From d7922bb71de5fada79c5783a94c309883c4e85f8 Mon Sep 17 00:00:00 2001
From: ccnetcore <cc2016cc>
Date: Sat, 27 Sep 2025 17:40:31 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=20tokenUsage=20?=
 =?UTF-8?q?=E4=B8=BA=E7=A9=BA=E6=97=B6=E7=9A=84=E7=A9=BA=E5=BC=95=E7=94=A8?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Managers/UsageStatisticsManager.cs                        | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/UsageStatisticsManager.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/UsageStatisticsManager.cs
index c1ba7be0..682c4b96 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/UsageStatisticsManager.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/UsageStatisticsManager.cs
@@ -21,11 +21,11 @@ public class UsageStatisticsManager : DomainService
     public async Task SetUsageAsync(Guid? userId, string modelId, ThorUsageResponse? tokenUsage)
     {
         long inputTokenCount = tokenUsage?.PromptTokens
-                               ?? tokenUsage.InputTokens
+                               ?? tokenUsage?.InputTokens
                                ?? 0;
 
         long outputTokenCount = tokenUsage?.CompletionTokens
-                                ?? tokenUsage.OutputTokens
+                                ?? tokenUsage?.OutputTokens
                                 ?? 0;
 
         await using (await DistributedLock.AcquireLockAsync($"UsageStatistics:{userId?.ToString()}"))