feat: 支持图片生成

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

@@ -3,9 +3,11 @@ using Microsoft.AspNetCore.Mvc;
 using Microsoft.Extensions.Logging;
 using Volo.Abp.Application.Services;
 using Yi.Framework.AiHub.Application.Contracts.Dtos.OpenAi;
+using Yi.Framework.AiHub.Application.Contracts.Dtos.OpenAi.Images;
 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.Enums;
 using Yi.Framework.SqlSugarCore.Abstractions;
 
 namespace Yi.Framework.AiHub.Application.Services;
@@ -17,16 +19,17 @@ public class OpenApiService : ApplicationService
     private readonly TokenManager _tokenManager;
     private readonly AiGateWayManager _aiGateWayManager;
     private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
-
+    private readonly AiBlacklistManager _aiBlacklistManager;
     public OpenApiService(IHttpContextAccessor httpContextAccessor, ILogger<OpenApiService> logger,
         TokenManager tokenManager, AiGateWayManager aiGateWayManager,
-        ISqlSugarRepository<AiModelEntity> aiModelRepository)
+        ISqlSugarRepository<AiModelEntity> aiModelRepository, AiBlacklistManager aiBlacklistManager)
     {
         _httpContextAccessor = httpContextAccessor;
         _logger = logger;
         _tokenManager = tokenManager;
         _aiGateWayManager = aiGateWayManager;
         _aiModelRepository = aiModelRepository;
+        _aiBlacklistManager = aiBlacklistManager;
     }
 
     /// <summary>
@@ -41,6 +44,7 @@ public class OpenApiService : ApplicationService
         //前面都是校验，后面才是真正的调用
         var httpContext = this._httpContextAccessor.HttpContext;
         var userId = await _tokenManager.GetUserIdAsync(GetTokenByHttpContext(httpContext));
+        await _aiBlacklistManager.VerifiyAiBlacklist(userId);
         //ai网关代理httpcontext
         if (input.Stream == true)
         {
@@ -55,6 +59,21 @@ public class OpenApiService : ApplicationService
         }
     }
 
+    /// <summary>
+    /// 图片生成
+    /// </summary>
+    /// <param name="input"></param>
+    /// <param name="cancellationToken"></param>
+    [HttpPost("openApi/v1/images/generations")]
+    public async Task ImagesGenerationsAsync([FromBody] ImageCreateRequest input, CancellationToken cancellationToken)
+    {
+        var httpContext = this._httpContextAccessor.HttpContext;
+        var userId = await _tokenManager.GetUserIdAsync(GetTokenByHttpContext(httpContext));
+        await _aiBlacklistManager.VerifiyAiBlacklist(userId);
+        await _aiGateWayManager.CreateImageForStatisticsAsync(httpContext, userId, null, input);
+    }
+
+
     /// <summary>
     /// 获取模型列表
     /// </summary>
@@ -63,6 +82,7 @@ public class OpenApiService : ApplicationService
     public async Task<ModelsListDto> ModelsAsync()
     {
         var data = await _aiModelRepository._DbQueryable
+            .Where(x => x.ModelType == ModelTypeEnum.Chat)
             .OrderByDescending(x => x.OrderNum)
             .Select(x => new ModelsDataDto
             {