feat: 完成渠道商管理+尊享模型替换+v1前缀兼容

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiChatService.cs

@@ -40,35 +40,37 @@ namespace Yi.Framework.AiHub.Application.Services;
 public class AiChatService : ApplicationService
 {
     private readonly IHttpContextAccessor _httpContextAccessor;
-    private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
     private readonly AiBlacklistManager _aiBlacklistManager;
     private readonly ILogger<AiChatService> _logger;
     private readonly AiGateWayManager _aiGateWayManager;
+    private readonly ModelManager _modelManager;
     private readonly PremiumPackageManager _premiumPackageManager;
     private readonly ChatManager _chatManager;
     private readonly TokenManager _tokenManager;
     private readonly IAccountService _accountService;
     private readonly ISqlSugarRepository<AgentStoreAggregateRoot> _agentStoreRepository;
+    private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
 
     public AiChatService(IHttpContextAccessor httpContextAccessor,
         AiBlacklistManager aiBlacklistManager,
-        ISqlSugarRepository<AiModelEntity> aiModelRepository,
         ILogger<AiChatService> logger,
         AiGateWayManager aiGateWayManager,
+        ModelManager modelManager,
         PremiumPackageManager premiumPackageManager,
         ChatManager chatManager, TokenManager tokenManager, IAccountService accountService,
-        ISqlSugarRepository<AgentStoreAggregateRoot> agentStoreRepository)
+        ISqlSugarRepository<AgentStoreAggregateRoot> agentStoreRepository, ISqlSugarRepository<AiModelEntity> aiModelRepository)
     {
         _httpContextAccessor = httpContextAccessor;
         _aiBlacklistManager = aiBlacklistManager;
-        _aiModelRepository = aiModelRepository;
         _logger = logger;
         _aiGateWayManager = aiGateWayManager;
+        _modelManager = modelManager;
         _premiumPackageManager = premiumPackageManager;
         _chatManager = chatManager;
         _tokenManager = tokenManager;
         _accountService = accountService;
         _agentStoreRepository = agentStoreRepository;
+        _aiModelRepository = aiModelRepository;
     }
 
 
@@ -109,7 +111,7 @@ public class AiChatService : ApplicationService
                 ApiHost = null,
                 ApiKey = null,
                 Remark = x.Description,
-                IsPremiumPackage = PremiumPackageConst.ModeIds.Contains(x.ModelId)
+                IsPremiumPackage = x.IsPremium
             }).ToListAsync();
         return output;
     }
@@ -144,19 +146,24 @@ public class AiChatService : ApplicationService
         }
 
         //如果是尊享包服务，需要校验是是否尊享包足够
-        if (CurrentUser.IsAuthenticated && PremiumPackageConst.ModeIds.Contains(input.Model))
+        if (CurrentUser.IsAuthenticated)
         {
-            // 检查尊享token包用量
-            var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(CurrentUser.GetId());
-            if (availableTokens <= 0)
+            var isPremium = await _modelManager.IsPremiumModelAsync(input.Model);
+
+            if (isPremium)
             {
-                throw new UserFriendlyException("尊享token包用量不足，请先购买尊享token包");
+                // 检查尊享token包用量
+                var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(CurrentUser.GetId());
+                if (availableTokens <= 0)
+                {
+                    throw new UserFriendlyException("尊享token包用量不足，请先购买尊享token包");
+                }
             }
         }
 
         //ai网关代理httpcontext
         await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
-            CurrentUser.Id, sessionId, null, cancellationToken);
+            CurrentUser.Id, sessionId, null, CancellationToken.None);
     }
 
     /// <summary>
@@ -192,7 +199,7 @@ public class AiChatService : ApplicationService
 
         //ai网关代理httpcontext
         await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
-            CurrentUser.Id, null, null, cancellationToken);
+            CurrentUser.Id, null, null, CancellationToken.None);
     }
 
 
@@ -219,7 +226,9 @@ public class AiChatService : ApplicationService
         }
 
         //如果是尊享包服务，需要校验是是否尊享包足够
-        if (PremiumPackageConst.ModeIds.Contains(input.ModelId))
+        var isPremium = await _modelManager.IsPremiumModelAsync(input.ModelId);
+
+        if (isPremium)
         {
             // 检查尊享token包用量
             var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(tokenValidation.UserId);
@@ -237,7 +246,7 @@ public class AiChatService : ApplicationService
             input.ModelId,
             tokenValidation.UserId,
             input.Tools,
-            cancellationToken);
+            CancellationToken.None);
     }
 
     /// <summary>