From 54547f0d7c65b1d944f329dce437b441f032023a Mon Sep 17 00:00:00 2001
From: chenchun <chenchun@hytch.com>
Date: Tue, 25 Nov 2025 10:18:45 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E7=BC=A9=E6=94=BE=20ThorChatCompletions?=
 =?UTF-8?q?Response.Usage.TotalTokens=20=E6=8C=89=20multiplier?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

当 Usage.TotalTokens 不为 null 时，按 multiplier 进行四舍五入缩放；与 PromptTokens/CompletionTokens 的缩放逻辑保持一致，修复 TotalTokens 未被缩放的问题。
---
 .../Dtos/OpenAi/ThorChatCompletionsResponse.cs              | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain.Shared/Dtos/OpenAi/ThorChatCompletionsResponse.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain.Shared/Dtos/OpenAi/ThorChatCompletionsResponse.cs
index f842c610..2afc32b9 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain.Shared/Dtos/OpenAi/ThorChatCompletionsResponse.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain.Shared/Dtos/OpenAi/ThorChatCompletionsResponse.cs
@@ -73,6 +73,12 @@ public record ThorChatCompletionsResponse
                 (int)Math.Round((this.Usage.CompletionTokens ?? 0) * multiplier);
             this.Usage.PromptTokens =
                 (int)Math.Round((this.Usage.PromptTokens ?? 0) * multiplier);
+
+            if (this.Usage.TotalTokens is not null)
+            {
+                this.Usage.TotalTokens =
+                    (int)Math.Round((this.Usage.TotalTokens ?? 0) * multiplier);
+            }
         }
     }
 }
\ No newline at end of file