feat: 兼容了用量使用显示

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
             };
         }

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/UsageStatisticsAggregateRoot.cs

@@ -37,22 +37,22 @@ public class UsageStatisticsAggregateRoot : FullAuditedAggregateRoot<Guid>
     /// <summary>
     /// 使用输出token总数
     /// </summary>
-    public int UsageOutputTokenCount { get; set; }
+    public long UsageOutputTokenCount { get; set; }
 
     /// <summary>
     /// 使用输入总数
     /// </summary>
-    public int UsageInputTokenCount { get; set; }
+    public long UsageInputTokenCount { get; set; }
 
     /// <summary>
     /// 总token使用数量
     /// </summary>
-    public int TotalTokenCount { get; set; }
+    public long TotalTokenCount { get; set; }
 
     /// <summary>
     /// 新增一次聊天统计
     /// </summary>
-    public void AddOnceChat(int inputTokenCount, int outputTokenCount)
+    public void AddOnceChat(long inputTokenCount, long outputTokenCount)
     {
         UsageTotalNumber += 1;
         UsageOutputTokenCount += outputTokenCount;

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/ValueObjects/TokenUsageValueObject.cs

@@ -2,9 +2,9 @@
 
 public class TokenUsageValueObject
 {
-    public int OutputTokenCount { get; set; }
+    public long OutputTokenCount { get; set; }
 
-    public int InputTokenCount { get; set; }
+    public long InputTokenCount { get; set; }
 
     public long TotalTokenCount { get; set; }
 }