feat: 完成模型库

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

@@ -0,0 +1,116 @@
+using Mapster;
+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.Shared.Enums;
+using Yi.Framework.AiHub.Domain.Shared.Extensions;
+using Yi.Framework.SqlSugarCore.Abstractions;
+
+namespace Yi.Framework.AiHub.Application.Services.Chat;
+
+/// <summary>
+/// 模型服务
+/// </summary>
+public class ModelService : ApplicationService, IModelService
+{
+    private readonly ISqlSugarRepository<AiModelEntity, Guid> _modelRepository;
+
+    public ModelService(ISqlSugarRepository<AiModelEntity, Guid> modelRepository)
+    {
+        _modelRepository = modelRepository;
+    }
+
+    /// <summary>
+    /// 获取模型库列表（公开接口，无需登录）
+    /// </summary>
+    public async Task<PagedResultDto<ModelLibraryDto>> GetListAsync(ModelLibraryGetListInput input)
+    {
+        RefAsync<int> total = 0;
+
+        // 查询所有未删除的模型，使用WhereIF动态添加筛选条件
+        var models = await _modelRepository._DbQueryable
+            .Where(x => !x.IsDeleted)
+            .WhereIF(!string.IsNullOrWhiteSpace(input.SearchKey), x =>
+                x.Name.Contains(input.SearchKey) || x.ModelId.Contains(input.SearchKey))
+            .WhereIF(!string.IsNullOrWhiteSpace(input.ProviderName), x =>
+                x.ProviderName == input.ProviderName)
+            .WhereIF(input.ModelType.HasValue, x =>
+                x.ModelType == input.ModelType.Value)
+            .WhereIF(input.ModelApiType.HasValue, x =>
+                x.ModelApiType == input.ModelApiType.Value)
+            .WhereIF(input.IsPremiumOnly == true, x =>
+                x.ModelType == ModelTypeEnum.PremiumChat)
+            .OrderBy(x => x.OrderNum)
+            .OrderBy(x => x.Name)
+            .ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
+
+        // 转换为DTO
+        var result = models.Select(model => new ModelLibraryDto
+        {
+            ModelId = model.ModelId,
+            Name = model.Name,
+            Description = model.Description,
+            ModelType = model.ModelType,
+            ModelTypeName = model.ModelType.GetDescription(),
+            ModelApiType = model.ModelApiType,
+            ModelApiTypeName = model.ModelApiType.GetDescription(),
+            MultiplierShow = model.MultiplierShow,
+            ProviderName = model.ProviderName,
+            IconUrl = model.IconUrl,
+            IsPremium = model.ModelType == ModelTypeEnum.PremiumChat
+        }).ToList();
+
+        return new PagedResultDto<ModelLibraryDto>(total, result);
+    }
+
+    /// <summary>
+    /// 获取供应商列表（公开接口，无需登录）
+    /// </summary>
+    public async Task<List<string>> GetProviderListAsync()
+    {
+        var providers = await _modelRepository._DbQueryable
+            .Where(x => !x.IsDeleted)
+            .Where(x => !string.IsNullOrEmpty(x.ProviderName))
+            .GroupBy(x => x.ProviderName)
+            .OrderBy(x => x.ProviderName)
+            .Select(x => x.ProviderName)
+            .ToListAsync();
+
+        return providers;
+    }
+
+    /// <summary>
+    /// 获取模型类型选项列表（公开接口，无需登录）
+    /// </summary>
+    public Task<List<ModelTypeOption>> GetModelTypeOptionsAsync()
+    {
+        var options = Enum.GetValues<ModelTypeEnum>()
+            .Select(e => new ModelTypeOption
+            {
+                Label = e.GetDescription(),
+                Value = (int)e
+            })
+            .ToList();
+
+        return Task.FromResult(options);
+    }
+
+    /// <summary>
+    /// 获取API类型选项列表（公开接口，无需登录）
+    /// </summary>
+    public Task<List<ModelApiTypeOption>> GetApiTypeOptionsAsync()
+    {
+        var options = Enum.GetValues<ModelApiTypeEnum>()
+            .Select(e => new ModelApiTypeOption
+            {
+                Label = e.GetDescription(),
+                Value = (int)e
+            })
+            .ToList();
+
+        return Task.FromResult(options);
+    }
+}