feat: 兼容了用量使用显示

2025-07-22 10:40:23 +08:00
parent b6d670c240
commit a0eb234539
8 changed files with 97 additions and 75 deletions
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Chat/MessageAggregateRoot.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Chat/MessageAggregateRoot.cs
@@ -29,10 +29,18 @@ public class MessageAggregateRoot : FullAuditedAggregateRoot<Guid>
        ModelId = modelId;
        if (tokenUsage is not null)
        {
+            long inputTokenCount = tokenUsage.PromptTokens 
+                                   ?? tokenUsage.InputTokens 
+                                   ?? 0;
+
+            long outputTokenCount = tokenUsage.CompletionTokens 
+                                    ?? tokenUsage.OutputTokens 
+                                    ?? 0;
+
            this.TokenUsage = new TokenUsageValueObject
            {
-                OutputTokenCount = tokenUsage.OutputTokens ?? 0,
-                InputTokenCount = tokenUsage.InputTokens ?? 0,
+                OutputTokenCount = outputTokenCount,
+                InputTokenCount = inputTokenCount,
                TotalTokenCount = tokenUsage.TotalTokens ?? 0
            };
        }