feat: 支持尊享包渠道

2025-12-31 00:02:25 +08:00
parent e4621d9049
commit 33d28a8cb0
28 changed files with 1666 additions and 34 deletions
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/UsageStatisticsService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/UsageStatisticsService.cs
@@ -181,7 +181,12 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
    public async Task<List<TokenPremiumUsageDto>> GetPremiumTokenUsageByTokenAsync()
    {
        var userId = CurrentUser.GetId();
-        var premiumModelIds = PremiumPackageConst.ModeIds;
+
+        // 先获取所有尊享模型的ModelId列表
+        var premiumModelIds = await _aiModelRepository._DbQueryable
+            .Where(x => x.IsPremium)
+            .Select(x => x.ModelId)
+            .ToListAsync();

        // 从UsageStatistics表获取尊享模型的token消耗统计（按TokenId聚合）
        var tokenUsages = await _usageStatisticsRepository._DbQueryable