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

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

@@ -27,25 +27,27 @@ public class OpenApiService : ApplicationService
     private readonly ILogger<OpenApiService> _logger;
     private readonly TokenManager _tokenManager;
     private readonly AiGateWayManager _aiGateWayManager;
-    private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
+    private readonly ModelManager _modelManager;
     private readonly AiBlacklistManager _aiBlacklistManager;
     private readonly IAccountService _accountService;
     private readonly PremiumPackageManager _premiumPackageManager;
     private readonly ISqlSugarRepository<ImageStoreTaskAggregateRoot> _imageStoreRepository;
+    private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
     public OpenApiService(IHttpContextAccessor httpContextAccessor, ILogger<OpenApiService> logger,
         TokenManager tokenManager, AiGateWayManager aiGateWayManager,
-        ISqlSugarRepository<AiModelEntity> aiModelRepository, AiBlacklistManager aiBlacklistManager,
-        IAccountService accountService, PremiumPackageManager premiumPackageManager, ISqlSugarRepository<ImageStoreTaskAggregateRoot> imageStoreRepository)
+        ModelManager modelManager, AiBlacklistManager aiBlacklistManager,
+        IAccountService accountService, PremiumPackageManager premiumPackageManager, ISqlSugarRepository<ImageStoreTaskAggregateRoot> imageStoreRepository, ISqlSugarRepository<AiModelEntity> aiModelRepository)
     {
         _httpContextAccessor = httpContextAccessor;
         _logger = logger;
         _tokenManager = tokenManager;
         _aiGateWayManager = aiGateWayManager;
-        _aiModelRepository = aiModelRepository;
+        _modelManager = modelManager;
         _aiBlacklistManager = aiBlacklistManager;
         _accountService = accountService;
         _premiumPackageManager = premiumPackageManager;
         _imageStoreRepository = imageStoreRepository;
+        _aiModelRepository = aiModelRepository;
     }
 
     /// <summary>
@@ -65,7 +67,9 @@ public class OpenApiService : ApplicationService
         await _aiBlacklistManager.VerifiyAiBlacklist(userId);
 
         //如果是尊享包服务，需要校验是是否尊享包足够
-        if (PremiumPackageConst.ModeIds.Contains(input.Model))
+        var isPremium = await _modelManager.IsPremiumModelAsync(input.Model);
+
+        if (isPremium)
         {
             // 检查尊享token包用量
             var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(userId);
@@ -79,13 +83,13 @@ public class OpenApiService : ApplicationService
         if (input.Stream == true)
         {
             await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
-                userId, null, tokenId, cancellationToken);
+                userId, null, tokenId,CancellationToken.None);
         }
         else
         {
             await _aiGateWayManager.CompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input, userId,
                 null, tokenId,
-                cancellationToken);
+                CancellationToken.None);
         }
     }
 
@@ -193,14 +197,14 @@ public class OpenApiService : ApplicationService
         {
             await _aiGateWayManager.AnthropicCompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext,
                 input,
-                userId, null, tokenId, cancellationToken);
+                userId, null, tokenId, CancellationToken.None);
         }
         else
         {
             await _aiGateWayManager.AnthropicCompleteChatForStatisticsAsync(_httpContextAccessor.HttpContext, input,
                 userId,
                 null, tokenId,
-                cancellationToken);
+                CancellationToken.None);
         }
     }
 
@@ -245,14 +249,14 @@ public class OpenApiService : ApplicationService
         {
             await _aiGateWayManager.OpenAiResponsesStreamForStatisticsAsync(_httpContextAccessor.HttpContext,
                 input,
-                userId, null, tokenId, cancellationToken);
+                userId, null, tokenId, CancellationToken.None);
         }
         else
         {
             await _aiGateWayManager.OpenAiResponsesAsyncForStatisticsAsync(_httpContextAccessor.HttpContext, input,
                 userId,
                 null, tokenId,
-                cancellationToken);
+                CancellationToken.None);
         }
     }
 
@@ -304,7 +308,7 @@ public class OpenApiService : ApplicationService
                 modelId, input,
                 userId,
                 null, tokenId,
-                cancellationToken);
+                CancellationToken.None);
         }
         else
         {
@@ -312,7 +316,7 @@ public class OpenApiService : ApplicationService
                 modelId, input,
                 userId,
                 null, tokenId,
-                cancellationToken);
+                CancellationToken.None);
         }
     }