xwl2023/Yi.Framework
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修复 tokenUsage 为空时的空引用问题

Browse Source
This commit is contained in:
ccnetcore
2025-09-27 17:40:31 +08:00
parent fa3ac91ba4
commit d7922bb71d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/UsageStatisticsManager.cs
View File

@@ -21,11 +21,11 @@ public class UsageStatisticsManager : DomainService
public async Task SetUsageAsync(Guid? userId, string modelId, ThorUsageResponse? tokenUsage) public async Task SetUsageAsync(Guid? userId, string modelId, ThorUsageResponse? tokenUsage)
{ {
long inputTokenCount = tokenUsage?.PromptTokens long inputTokenCount = tokenUsage?.PromptTokens
?? tokenUsage.InputTokens ?? tokenUsage?.InputTokens
?? 0; ?? 0;
long outputTokenCount = tokenUsage?.CompletionTokens long outputTokenCount = tokenUsage?.CompletionTokens
?? tokenUsage.OutputTokens ?? tokenUsage?.OutputTokens
?? 0; ?? 0;
await using (await DistributedLock.AcquireLockAsync($"UsageStatistics:{userId?.ToString()}")) await using (await DistributedLock.AcquireLockAsync($"UsageStatistics:{userId?.ToString()}"))