feat: 完成用量统计功能模块

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

@@ -9,6 +9,16 @@ namespace Yi.Framework.AiHub.Domain.Entities;
 [SugarTable("Ai_UsageStatistics")]
 public class UsageStatisticsAggregateRoot : FullAuditedAggregateRoot<Guid>
 {
+    public UsageStatisticsAggregateRoot()
+    {
+    }
+
+    public UsageStatisticsAggregateRoot(Guid userId, string modelId)
+    {
+        UserId = userId;
+        ModelId = modelId;
+    }
+
     /// <summary>
     /// 用户id
     /// </summary>
@@ -19,18 +29,34 @@ public class UsageStatisticsAggregateRoot : FullAuditedAggregateRoot<Guid>
     /// </summary>
     public string ModelId { get; set; }
 
-    /// <summary>
-    /// 输入使用token使用
-    /// </summary>
-    public decimal InputTokens { get; set; }
-
-    /// <summary>
-    /// 输出使用token使用
-    /// </summary>
-    public decimal OutputTokens { get; set; }
-    
     /// <summary>
     /// 对话次数
     /// </summary>
-    public int Number { get; set; }
+    public int UsageTotalNumber { get; set; }
+
+    /// <summary>
+    /// 使用输出token总数
+    /// </summary>
+    public int UsageOutputTokenCount { get; set; }
+
+    /// <summary>
+    /// 使用输入总数
+    /// </summary>
+    public int UsageInputTokenCount { get; set; }
+
+    /// <summary>
+    /// 总token使用数量
+    /// </summary>
+    public int TotalTokenCount { get; set; }
+
+    /// <summary>
+    /// 新增一次聊天统计
+    /// </summary>
+    public void AddOnceChat(int inputTokenCount, int outputTokenCount)
+    {
+        UsageTotalNumber += 1;
+        UsageOutputTokenCount += outputTokenCount;
+        UsageInputTokenCount += inputTokenCount;
+        TotalTokenCount += (outputTokenCount + inputTokenCount);
+    }
 }