Merge pull request #842 from iceljc/features/add-global-stats

add global stats

Author: iceljc

src/Infrastructure/BotSharp.Abstraction/Infrastructures/Enums/CacheType.cs

@@ -1,14 +1,7 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
+namespace BotSharp.Abstraction.Infrastructures.Enums;
 
-namespace BotSharp.Abstraction.Infrastructures.Enums
+public enum CacheType
 {
-    public enum CacheType
-    {
-        MemoryCache,
-        RedisCache
-    }
+    MemoryCache = 1,
+    RedisCache = 2
 }

src/Infrastructure/BotSharp.Abstraction/Infrastructures/SharpCacheSettings.cs

@@ -3,6 +3,6 @@ namespace BotSharp.Abstraction.Infrastructures;
 public class SharpCacheSettings
 {
     public bool Enabled { get; set; } = true;
-    public CacheType CacheType { get; set; } = Enums.CacheType.MemoryCache;
+    public CacheType CacheType { get; set; } = CacheType.MemoryCache;
     public string Prefix { get; set; } = "cache";    
 }

src/Infrastructure/BotSharp.Abstraction/Repositories/IBotSharpRepository.cs

@@ -3,6 +3,7 @@
 using BotSharp.Abstraction.Repositories.Filters;
 using BotSharp.Abstraction.Roles.Models;
 using BotSharp.Abstraction.Shared;
+using BotSharp.Abstraction.Statistics.Models;
 using BotSharp.Abstraction.Tasks.Models;
 using BotSharp.Abstraction.Translation.Models;
 using BotSharp.Abstraction.Users.Enums;
@@ -119,7 +120,9 @@ public interface IBotSharpRepository : IHaveServiceProvider
     #endregion
 
     #region Statistics
-    void IncrementConversationCount();
+    BotSharpStats? GetGlobalStats(string category, string group, DateTime recordTime) => throw new NotImplementedException();
+    bool SaveGlobalStats(BotSharpStats body) => throw new NotImplementedException();
+
     #endregion
 
     #region Translation

src/Infrastructure/BotSharp.Abstraction/Statistics/Enums/StatCategory.cs

@@ -0,0 +1,7 @@
+namespace BotSharp.Abstraction.Statistics.Enums;
+
+public static class StatCategory
+{
+    public static string LlmCost = "llm-cost";
+    public static string AgentCall = "agent-call";
+}

src/Infrastructure/BotSharp.Abstraction/Statistics/Model/Statistics.cs

@@ -0,0 +1,34 @@
+namespace BotSharp.Abstraction.Statistics.Models;
+
+public class BotSharpStats
+{
+    [JsonPropertyName("category")]
+    public string Category { get; set; } = null!;
+
+    [JsonPropertyName("group")]
+    public string Group { get; set; } = null!;
+
+    [JsonPropertyName("data")]
+    public IDictionary<string, object> Data { get; set; } = new Dictionary<string, object>();
+
+    private DateTime innerRecordTime;
+
+    [JsonPropertyName("record_time")]
+    public DateTime RecordTime
+    {
+        get
+        {
+            return innerRecordTime;
+        }
+        set
+        {
+            var date = new DateTime(value.Year, value.Month, value.Day, value.Hour, 0, 0);
+            innerRecordTime = DateTime.SpecifyKind(date, DateTimeKind.Utc);
+        }
+    }
+
+    public override string ToString()
+    {
+        return $"{Category}-{Group}: {Data?.Count ?? 0} ({RecordTime})";
+    }
+}

src/Infrastructure/BotSharp.Abstraction/Statistics/Models/BotSharpStats.cs

@@ -0,0 +1,9 @@
+using BotSharp.Abstraction.Statistics.Models;
+
+namespace BotSharp.Abstraction.Statistics.Services;
+
+public interface IBotSharpStatService
+{
+    bool UpdateLlmCost(BotSharpStats stats);
+    bool UpdateAgentCall(BotSharpStats stats);
+}

src/Infrastructure/BotSharp.Abstraction/Statistics/Services/IBotSharpStatService.cs

@@ -1,11 +1,6 @@
-using System;
-using System.Collections.Generic;
-using System.Text;
+namespace BotSharp.Abstraction.Statistics.Settings;
 
-namespace BotSharp.Abstraction.Statistics.Settings
+public class StatisticsSettings
 {
-    public class StatisticsSettings
-    {
-        public string DataDir { get; set; }
-    }
+    public bool Enabled { get; set; }
 }

src/Infrastructure/BotSharp.Abstraction/Statistics/Settings/StatisticsSettings.cs

@@ -11,4 +11,9 @@ public static long Max(long a, long b, long c)
     {
         return Math.Max(Math.Max(a, b), c);
     }
+
+    public static decimal Round(decimal value, MidpointRounding rouding = MidpointRounding.AwayFromZero)
+    {
+        return Math.Round(value, rouding);
+    }
 }

src/Infrastructure/BotSharp.Abstraction/Utilities/MathExt.cs

@@ -32,6 +32,7 @@ public void RegisterDI(IServiceCollection services, IConfiguration config)
         services.AddScoped<ILlmProviderService, LlmProviderService>();
         services.AddScoped<IAgentService, AgentService>();
         services.AddScoped<IAgentHook, BasicAgentHook>();
+        services.AddScoped<IBotSharpStatService, BotSharpStatService>();
 
         services.AddScoped(provider =>
         {