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