Merge branch 'ai-agent-backend' into ai-agent

# Conflicts: # Yi.Ai.Vue3/src/pages/console/index.vue # Yi.Ai.Vue3/src/routers/modules/staticRouter.ts
2026-01-02 19:45:55 +08:00
parent 560a76558a a1be2bebf7
commit 436b5b910c
34 changed files with 1860 additions and 137 deletions
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Activities/ActivationCodeService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Activities/ActivationCodeService.cs
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Activities/DailyTaskService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Activities/DailyTaskService.cs
@@ -7,6 +7,7 @@ using Volo.Abp.Users;
 using Yi.Framework.AiHub.Application.Contracts.Dtos.DailyTask;
 using Yi.Framework.AiHub.Domain.Entities;
 using Yi.Framework.AiHub.Domain.Entities.Chat;
+using Yi.Framework.AiHub.Domain.Entities.Model;
 using Yi.Framework.AiHub.Domain.Extensions;
 using Yi.Framework.AiHub.Domain.Managers;
 using Yi.Framework.AiHub.Domain.Shared.Consts;
@@ -25,6 +26,7 @@ public class DailyTaskService : ApplicationService
    private readonly ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;
    private readonly ILogger<DailyTaskService> _logger;
    private IDistributedLockProvider DistributedLock => LazyServiceProvider.LazyGetRequiredService<IDistributedLockProvider>();
+    private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
    // 任务配置
    private readonly Dictionary<int, (long RequiredTokens, long RewardTokens, string Name, string Description)>
        _taskConfigs = new()
@@ -37,12 +39,13 @@ public class DailyTaskService : ApplicationService
        ISqlSugarRepository<DailyTaskRewardRecordAggregateRoot> dailyTaskRepository,
        ISqlSugarRepository<MessageAggregateRoot> messageRepository,
        ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository,
-        ILogger<DailyTaskService> logger)
+        ILogger<DailyTaskService> logger, ISqlSugarRepository<AiModelEntity> aiModelRepository)
    {
        _dailyTaskRepository = dailyTaskRepository;
        _messageRepository = messageRepository;
        _premiumPackageRepository = premiumPackageRepository;
        _logger = logger;
+        _aiModelRepository = aiModelRepository;
    }

    /// <summary>
@@ -179,10 +182,16 @@ public class DailyTaskService : ApplicationService
        var tomorrow = today.AddDays(1);

        // 查询今日所有使用尊享包模型的消息（role=system 表示消耗）
+        // 先获取所有尊享模型的ModelId列表
+        var premiumModelIds = await _aiModelRepository._DbQueryable
+            .Where(x => x.IsPremium)
+            .Select(x => x.ModelId)
+            .ToListAsync();
+
        var totalTokens = await _messageRepository._DbQueryable
            .Where(x => x.UserId == userId)
            .Where(x => x.Role == "system") // system角色表示实际消耗
-            .Where(x => PremiumPackageConst.ModeIds.Contains(x.ModelId)) // 尊享包模型
+            .Where(x => premiumModelIds.Contains(x.ModelId)) // 尊享包模型
            .Where(x => x.CreationTime >= today && x.CreationTime < tomorrow)
            .SumAsync(x => x.TokenUsage.TotalTokenCount);

--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/ChannelService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/ChannelService.cs
@@ -0,0 +1,240 @@
+using Mapster;
+using Microsoft.AspNetCore.Authorization;
+using Microsoft.AspNetCore.Mvc;
+using SqlSugar;
+using Volo.Abp.Application.Dtos;
+using Volo.Abp.Application.Services;
+using Yi.Framework.AiHub.Application.Contracts.Dtos.Channel;
+using Yi.Framework.AiHub.Application.Contracts.IServices;
+using Yi.Framework.AiHub.Domain.Entities.Model;
+using Yi.Framework.SqlSugarCore.Abstractions;
+
+namespace Yi.Framework.AiHub.Application.Services;
+
+/// <summary>
+/// 渠道商管理服务实现
+/// </summary>
+[Authorize]
+public class ChannelService : ApplicationService, IChannelService
+{
+    private readonly ISqlSugarRepository<AiAppAggregateRoot, Guid> _appRepository;
+    private readonly ISqlSugarRepository<AiModelEntity, Guid> _modelRepository;
+
+    public ChannelService(
+        ISqlSugarRepository<AiAppAggregateRoot, Guid> appRepository,
+        ISqlSugarRepository<AiModelEntity, Guid> modelRepository)
+    {
+        _appRepository = appRepository;
+        _modelRepository = modelRepository;
+    }
+
+    #region AI应用管理
+
+    /// <summary>
+    /// 获取AI应用列表
+    /// </summary>
+    [HttpGet("channel/app")]
+    public async Task<PagedResultDto<AiAppDto>> GetAppListAsync(AiAppGetListInput input)
+    {
+        RefAsync<int> total = 0;
+
+        var entities = await _appRepository._DbQueryable
+            .WhereIF(!string.IsNullOrWhiteSpace(input.SearchKey), x => x.Name.Contains(input.SearchKey))
+            .OrderByDescending(x => x.OrderNum)
+            .OrderByDescending(x => x.CreationTime)
+            .ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
+
+        var output = entities.Adapt<List<AiAppDto>>();
+        return new PagedResultDto<AiAppDto>(total, output);
+    }
+
+    /// <summary>
+    /// 根据ID获取AI应用
+    /// </summary>
+    [HttpGet("channel/app/{id}")]
+    public async Task<AiAppDto> GetAppByIdAsync([FromRoute]Guid id)
+    {
+        var entity = await _appRepository.GetByIdAsync(id);
+        return entity.Adapt<AiAppDto>();
+    }
+
+    /// <summary>
+    /// 创建AI应用
+    /// </summary>
+    public async Task<AiAppDto> CreateAppAsync(AiAppCreateInput input)
+    {
+        var entity = new AiAppAggregateRoot
+        {
+            Name = input.Name,
+            Endpoint = input.Endpoint,
+            ExtraUrl = input.ExtraUrl,
+            ApiKey = input.ApiKey,
+            OrderNum = input.OrderNum
+        };
+
+        await _appRepository.InsertAsync(entity);
+        return entity.Adapt<AiAppDto>();
+    }
+
+    /// <summary>
+    /// 更新AI应用
+    /// </summary>
+    public async Task<AiAppDto> UpdateAppAsync(AiAppUpdateInput input)
+    {
+        var entity = await _appRepository.GetByIdAsync(input.Id);
+
+        entity.Name = input.Name;
+        entity.Endpoint = input.Endpoint;
+        entity.ExtraUrl = input.ExtraUrl;
+        entity.ApiKey = input.ApiKey;
+        entity.OrderNum = input.OrderNum;
+
+        await _appRepository.UpdateAsync(entity);
+        return entity.Adapt<AiAppDto>();
+    }
+
+    /// <summary>
+    /// 删除AI应用
+    /// </summary>
+    [HttpDelete("channel/app/{id}")]
+    public async Task DeleteAppAsync([FromRoute]Guid id)
+    {
+        // 检查是否有关联的模型
+        var hasModels = await _modelRepository._DbQueryable
+            .Where(x => x.AiAppId == id && !x.IsDeleted)
+            .AnyAsync();
+
+        if (hasModels)
+        {
+            throw new Volo.Abp.UserFriendlyException("该应用下存在模型,无法删除");
+        }
+
+        await _appRepository.DeleteAsync(id);
+    }
+
+    #endregion
+
+    #region AI模型管理
+
+    /// <summary>
+    /// 获取AI模型列表
+    /// </summary>
+    [HttpGet("channel/model")]
+    public async Task<PagedResultDto<AiModelDto>> GetModelListAsync(AiModelGetListInput input)
+    {
+        RefAsync<int> total = 0;
+
+        var query = _modelRepository._DbQueryable
+            .Where(x => !x.IsDeleted)
+            .WhereIF(!string.IsNullOrWhiteSpace(input.SearchKey), x =>
+                x.Name.Contains(input.SearchKey) || x.ModelId.Contains(input.SearchKey))
+            .WhereIF(input.AiAppId.HasValue, x => x.AiAppId == input.AiAppId.Value)
+            .WhereIF(input.IsPremiumOnly == true, x => x.IsPremium);
+
+        var entities = await query
+            .OrderBy(x => x.OrderNum)
+            .OrderByDescending(x => x.Id)
+            .ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
+
+        var output = entities.Adapt<List<AiModelDto>>();
+        return new PagedResultDto<AiModelDto>(total, output);
+    }
+
+    /// <summary>
+    /// 根据ID获取AI模型
+    /// </summary>
+    [HttpGet("channel/model/{id}")]
+    public async Task<AiModelDto> GetModelByIdAsync([FromRoute]Guid id)
+    {
+        var entity = await _modelRepository.GetByIdAsync(id);
+        return entity.Adapt<AiModelDto>();
+    }
+
+    /// <summary>
+    /// 创建AI模型
+    /// </summary>
+    public async Task<AiModelDto> CreateModelAsync(AiModelCreateInput input)
+    {
+        // 验证应用是否存在
+        var appExists = await _appRepository._DbQueryable
+            .Where(x => x.Id == input.AiAppId)
+            .AnyAsync();
+
+        if (!appExists)
+        {
+            throw new Volo.Abp.UserFriendlyException("指定的AI应用不存在");
+        }
+
+        var entity = new AiModelEntity
+        {
+            HandlerName = input.HandlerName,
+            ModelId = input.ModelId,
+            Name = input.Name,
+            Description = input.Description,
+            OrderNum = input.OrderNum,
+            AiAppId = input.AiAppId,
+            ExtraInfo = input.ExtraInfo,
+            ModelType = input.ModelType,
+            ModelApiType = input.ModelApiType,
+            Multiplier = input.Multiplier,
+            MultiplierShow = input.MultiplierShow,
+            ProviderName = input.ProviderName,
+            IconUrl = input.IconUrl,
+            IsPremium = input.IsPremium,
+            IsDeleted = false
+        };
+
+        await _modelRepository.InsertAsync(entity);
+        return entity.Adapt<AiModelDto>();
+    }
+
+    /// <summary>
+    /// 更新AI模型
+    /// </summary>
+    public async Task<AiModelDto> UpdateModelAsync(AiModelUpdateInput input)
+    {
+        var entity = await _modelRepository.GetByIdAsync(input.Id);
+
+        // 验证应用是否存在
+        if (entity.AiAppId != input.AiAppId)
+        {
+            var appExists = await _appRepository._DbQueryable
+                .Where(x => x.Id == input.AiAppId)
+                .AnyAsync();
+
+            if (!appExists)
+            {
+                throw new Volo.Abp.UserFriendlyException("指定的AI应用不存在");
+            }
+        }
+
+        entity.HandlerName = input.HandlerName;
+        entity.ModelId = input.ModelId;
+        entity.Name = input.Name;
+        entity.Description = input.Description;
+        entity.OrderNum = input.OrderNum;
+        entity.AiAppId = input.AiAppId;
+        entity.ExtraInfo = input.ExtraInfo;
+        entity.ModelType = input.ModelType;
+        entity.ModelApiType = input.ModelApiType;
+        entity.Multiplier = input.Multiplier;
+        entity.MultiplierShow = input.MultiplierShow;
+        entity.ProviderName = input.ProviderName;
+        entity.IconUrl = input.IconUrl;
+        entity.IsPremium = input.IsPremium;
+
+        await _modelRepository.UpdateAsync(entity);
+        return entity.Adapt<AiModelDto>();
+    }
+
+    /// <summary>
+    /// 删除AI模型(软删除)
+    /// </summary>
+    [HttpDelete("channel/model/{id}")]
+    public async Task DeleteModelAsync(Guid id)
+    {
+        await _modelRepository.DeleteByIdAsync(id);
+    }
+
+    #endregion
+}
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiChatService.cs
+++ b/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;
    }


@@ -102,7 +104,7 @@ public class AiChatService : ApplicationService
                ModelName = x.Name,
                ModelDescribe = x.Description,
                Remark = x.Description,
-                IsPremiumPackage = PremiumPackageConst.ModeIds.Contains(x.ModelId)
+                IsPremiumPackage = x.IsPremium
            }).ToListAsync();
        return output;
    }
@@ -137,19 +139,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>
@@ -185,7 +192,7 @@ public class AiChatService : ApplicationService

        //ai网关代理httpcontext
        await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
-            CurrentUser.Id, null, null, cancellationToken);
+            CurrentUser.Id, null, null, CancellationToken.None);
    }


@@ -212,7 +219,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);
@@ -230,7 +239,7 @@ public class AiChatService : ApplicationService
            input.ModelId,
            tokenValidation.UserId,
            input.Tools,
-            cancellationToken);
+            CancellationToken.None);
    }

    /// <summary>
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiImageService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiImageService.cs
@@ -29,6 +29,7 @@ public class AiImageService : ApplicationService
    private readonly IBackgroundJobManager _backgroundJobManager;
    private readonly AiBlacklistManager _aiBlacklistManager;
    private readonly PremiumPackageManager _premiumPackageManager;
+    private readonly ModelManager _modelManager;
    private readonly IGuidGenerator _guidGenerator;
    private readonly IWebHostEnvironment _webHostEnvironment;
    private readonly TokenManager _tokenManager;
@@ -37,6 +38,7 @@ public class AiImageService : ApplicationService
        IBackgroundJobManager backgroundJobManager,
        AiBlacklistManager aiBlacklistManager,
        PremiumPackageManager premiumPackageManager,
+        ModelManager modelManager,
        IGuidGenerator guidGenerator,
        IWebHostEnvironment webHostEnvironment, TokenManager tokenManager)
    {
@@ -44,6 +46,7 @@ public class AiImageService : ApplicationService
        _backgroundJobManager = backgroundJobManager;
        _aiBlacklistManager = aiBlacklistManager;
        _premiumPackageManager = premiumPackageManager;
+        _modelManager = modelManager;
        _guidGenerator = guidGenerator;
        _webHostEnvironment = webHostEnvironment;
        _tokenManager = tokenManager;
@@ -76,8 +79,9 @@ public class AiImageService : ApplicationService
            throw new UserFriendlyException("图片生成功能需要VIP用户才能使用，请购买VIP后重新登录重试");
        }

-        // 尊享包校验
-        if (PremiumPackageConst.ModeIds.Contains(input.ModelId))
+        // 尊享包校验 - 使用ModelManager统一判断
+        var isPremium = await _modelManager.IsPremiumModelAsync(input.ModelId);
+        if (isPremium)
        {
            var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(userId);
            if (availableTokens <= 0)
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/ModelService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/ModelService.cs
@@ -1,10 +1,12 @@
 using Mapster;
+using Microsoft.AspNetCore.Mvc;
 using SqlSugar;
 using Volo.Abp.Application.Dtos;
 using Volo.Abp.Application.Services;
 using Yi.Framework.AiHub.Application.Contracts.Dtos.Model;
 using Yi.Framework.AiHub.Application.Contracts.IServices;
 using Yi.Framework.AiHub.Domain.Entities.Model;
+using Yi.Framework.AiHub.Domain.Managers;
 using Yi.Framework.AiHub.Domain.Shared.Consts;
 using Yi.Framework.AiHub.Domain.Shared.Enums;
 using Yi.Framework.AiHub.Domain.Shared.Extensions;
@@ -18,10 +20,12 @@ namespace Yi.Framework.AiHub.Application.Services.Chat;
 public class ModelService : ApplicationService, IModelService
 {
    private readonly ISqlSugarRepository<AiModelEntity, Guid> _modelRepository;
+    private readonly ModelManager _modelManager;

-    public ModelService(ISqlSugarRepository<AiModelEntity, Guid> modelRepository)
+    public ModelService(ISqlSugarRepository<AiModelEntity, Guid> modelRepository, ModelManager modelManager)
    {
        _modelRepository = modelRepository;
+        _modelManager = modelManager;
    }

    /// <summary>
@@ -41,8 +45,7 @@ public class ModelService : ApplicationService, IModelService
                input.ModelTypes.Contains(x.ModelType))
            .WhereIF(input.ModelApiTypes is not null, x =>
                input.ModelApiTypes.Contains(x.ModelApiType))
-            .WhereIF(input.IsPremiumOnly == true, x =>
-                PremiumPackageConst.ModeIds.Contains(x.ModelId))
+            .WhereIF(input.IsPremiumOnly == true, x => x.IsPremium)
            .GroupBy(x => x.ModelId)
            .Select(x => x.ModelId)
            .ToPageListAsync(input.SkipCount, input.MaxResultCount, total));
@@ -61,7 +64,7 @@ public class ModelService : ApplicationService, IModelService
            MultiplierShow = x.First().MultiplierShow,
            ProviderName = x.First().ProviderName,
            IconUrl = x.First().IconUrl,
-            IsPremium = PremiumPackageConst.ModeIds.Contains(x.First().ModelId),
+            IsPremium = x.First().IsPremium,
            OrderNum = x.First().OrderNum
        }).ToList();
        
@@ -77,11 +80,11 @@ public class ModelService : ApplicationService, IModelService
            .Where(x => !x.IsDeleted)
            .Where(x => !string.IsNullOrEmpty(x.ProviderName))
            .GroupBy(x => x.ProviderName)
-            .OrderBy(x => x.ProviderName)
+            .OrderBy(x => x.OrderNum)
            .Select(x => x.ProviderName)
            .ToListAsync();

-        return providers;
+        return providers!;
    }

    /// <summary>
@@ -115,4 +118,13 @@ public class ModelService : ApplicationService, IModelService

        return Task.FromResult(options);
    }
+
+    /// <summary>
+    /// 清除尊享模型ID缓存
+    /// </summary>
+    [HttpPost("model/clear-premium-cache")]
+    public async Task ClearPremiumModelCacheAsync()
+    {
+        await _modelManager.ClearPremiumModelIdsCacheAsync();
+    }
 }
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/TokenService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/TokenService.cs
@@ -7,8 +7,10 @@ using Volo.Abp.Application.Services;
 using Volo.Abp.Users;
 using Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
 using Yi.Framework.AiHub.Domain.Entities;
+using Yi.Framework.AiHub.Domain.Entities.Model;
 using Yi.Framework.AiHub.Domain.Entities.OpenApi;
 using Yi.Framework.AiHub.Domain.Extensions;
+using Yi.Framework.AiHub.Domain.Managers;
 using Yi.Framework.AiHub.Domain.Shared.Consts;
 using Yi.Framework.Ddd.Application.Contracts;
 using Yi.Framework.SqlSugarCore.Abstractions;
@@ -23,13 +25,15 @@ public class TokenService : ApplicationService
 {
    private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
    private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
-
+    private readonly ModelManager _modelManager;
    public TokenService(
        ISqlSugarRepository<TokenAggregateRoot> tokenRepository,
-        ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository)
+        ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository,
+        ModelManager modelManager)
    {
        _tokenRepository = tokenRepository;
        _usageStatisticsRepository = usageStatisticsRepository;
+        _modelManager = modelManager;
    }

    /// <summary>
@@ -51,8 +55,8 @@ public class TokenService : ApplicationService
            return new PagedResultDto<TokenGetListOutputDto>();
        }

-        // 获取尊享包模型ID列表
-        var premiumModelIds = PremiumPackageConst.ModeIds;
+        // 通过ModelManager获取尊享包模型ID列表
+        var premiumModelIds = await _modelManager.GetPremiumModelIdsAsync();

        // 批量查询所有Token的尊享包已使用额度
        var tokenIds = tokens.Select(t => t.Id).ToList();
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/OpenApiService.cs
+++ b/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);
        }
    }

--- 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
@@ -9,8 +9,10 @@ using Yi.Framework.AiHub.Application.Contracts.Dtos.UsageStatistics;
 using Yi.Framework.AiHub.Application.Contracts.IServices;
 using Yi.Framework.AiHub.Domain.Entities;
 using Yi.Framework.AiHub.Domain.Entities.Chat;
+using Yi.Framework.AiHub.Domain.Entities.Model;
 using Yi.Framework.AiHub.Domain.Entities.OpenApi;
 using Yi.Framework.AiHub.Domain.Extensions;
+using Yi.Framework.AiHub.Domain.Managers;
 using Yi.Framework.AiHub.Domain.Shared.Consts;
 using Yi.Framework.Ddd.Application.Contracts;
 using Yi.Framework.SqlSugarCore.Abstractions;
@@ -27,17 +29,19 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
    private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _usageStatisticsRepository;
    private readonly ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;
    private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
-
+    private readonly ModelManager _modelManager;
    public UsageStatisticsService(
        ISqlSugarRepository<MessageAggregateRoot> messageRepository,
        ISqlSugarRepository<UsageStatisticsAggregateRoot> usageStatisticsRepository,
        ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository,
-        ISqlSugarRepository<TokenAggregateRoot> tokenRepository)
+        ISqlSugarRepository<TokenAggregateRoot> tokenRepository,
+        ModelManager modelManager)
    {
        _messageRepository = messageRepository;
        _usageStatisticsRepository = usageStatisticsRepository;
        _premiumPackageRepository = premiumPackageRepository;
        _tokenRepository = tokenRepository;
+        _modelManager = modelManager;
    }

    /// <summary>
@@ -181,7 +185,9 @@ public class UsageStatisticsService : ApplicationService, IUsageStatisticsServic
    public async Task<List<TokenPremiumUsageDto>> GetPremiumTokenUsageByTokenAsync()
    {
        var userId = CurrentUser.GetId();
-        var premiumModelIds = PremiumPackageConst.ModeIds;
+
+        // 通过ModelManager获取所有尊享模型的ModelId列表
+        var premiumModelIds = await _modelManager.GetPremiumModelIdsAsync();

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