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

fix: 缩放 ThorChatCompletionsResponse.Usage.TotalTokens 按 multiplier

Browse Source 
当 Usage.TotalTokens 不为 null 时，按 multiplier 进行四舍五入缩放；与 PromptTokens/CompletionTokens 的缩放逻辑保持一致，修复 TotalTokens 未被缩放的问题。
This commit is contained in:
chenchun
2025-11-25 10:18:45 +08:00
parent afe9c8bcae
commit 54547f0d7c
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain.Shared/Dtos/OpenAi/ThorChatCompletionsResponse.cs
View File

@@ -73,6 +73,12 @@ public record ThorChatCompletionsResponse
(int)Math.Round((this.Usage.CompletionTokens ?? 0) * multiplier); (int)Math.Round((this.Usage.CompletionTokens ?? 0) * multiplier);
this.Usage.PromptTokens = this.Usage.PromptTokens =
(int)Math.Round((this.Usage.PromptTokens ?? 0) * multiplier); (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);
}
} }
} }
} }