feat: 完成openapi改造

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

@@ -1,5 +1,6 @@
 using System.Collections.Concurrent;
 using System.Text;
+using Microsoft.AspNetCore.Authorization;
 using Microsoft.AspNetCore.Http;
 using Microsoft.AspNetCore.Mvc;
 using Microsoft.Extensions.DependencyInjection;
@@ -8,10 +9,13 @@ using Newtonsoft.Json;
 using Newtonsoft.Json.Serialization;
 using OpenAI.Chat;
 using Volo.Abp.Application.Services;
+using Volo.Abp.Users;
 using Yi.Framework.AiHub.Application.Contracts.Dtos;
 using Yi.Framework.AiHub.Application.Contracts.Dtos.OpenAiDto;
+using Yi.Framework.AiHub.Domain.Entities.OpenApi;
 using Yi.Framework.AiHub.Domain.Managers;
 using Yi.Framework.AiHub.Domain.Shared.Dtos;
+using Yi.Framework.SqlSugarCore.Abstractions;
 
 namespace Yi.Framework.AiHub.Application.Services;
 
@@ -19,18 +23,34 @@ public class OpenApiService : ApplicationService
 {
     private readonly IHttpContextAccessor _httpContextAccessor;
     private readonly ILogger<OpenApiService> _logger;
+    private readonly TokenManager _tokenManager;
+    private readonly AiMessageManager _aiMessageManager;
+    private readonly UsageStatisticsManager _usageStatisticsManager;
 
-    public OpenApiService(IHttpContextAccessor httpContextAccessor, ILogger<OpenApiService> logger)
+    public OpenApiService(IHttpContextAccessor httpContextAccessor, ILogger<OpenApiService> logger,
+        TokenManager tokenManager, AiMessageManager aiMessageManager, UsageStatisticsManager usageStatisticsManager)
     {
         _httpContextAccessor = httpContextAccessor;
         _logger = logger;
+        _tokenManager = tokenManager;
+        _aiMessageManager = aiMessageManager;
+        _usageStatisticsManager = usageStatisticsManager;
     }
 
+    /// <summary>
+    /// 对话
+    /// </summary>
+    /// <param name="input"></param>
+    /// <param name="cancellationToken"></param>
     [HttpPost("/v1/chat/completions")]
     public async Task ChatCompletionsAsync(ChatCompletionsInput input, CancellationToken cancellationToken)
     {
         //前面都是校验，后面才是真正的调用
         var httpContext = this._httpContextAccessor.HttpContext;
+
+        var userId = await _tokenManager.GetUserIdAsync(GetTokenByHttpContext(httpContext));
+
+
         var response = httpContext.Response;
         // 设置响应头，声明是 SSE 流
         response.ContentType = "text/event-stream";
@@ -125,6 +145,26 @@ public class OpenApiService : ApplicationService
         isComplete = true;
 
         await outputTask;
+        
+        await _aiMessageManager.CreateUserMessageAsync(userId, null,
+            new MessageInputDto
+            {
+                Content = input.Messages.LastOrDefault()
+                    .Content,
+                ModelId = input.Model,
+                TokenUsage = tokenUsage,
+            });
+
+        await _aiMessageManager.CreateSystemMessageAsync(userId, null,
+            new MessageInputDto
+            {
+                Content = backupSystemContent.ToString(),
+                ModelId = input.Model,
+                TokenUsage = tokenUsage
+            });
+
+        await _usageStatisticsManager.SetUsageAsync(userId, input.Model, tokenUsage.InputTokenCount,
+            tokenUsage.OutputTokenCount);
     }
 
     private SendMessageOutputDto MapToMessage(string modelId, string content)
@@ -204,4 +244,18 @@ public class OpenApiService : ApplicationService
 
         return output;
     }
+
+    private string? GetTokenByHttpContext(HttpContext httpContext)
+    {
+        // 获取Authorization头
+        string authHeader = httpContext.Request.Headers["Authorization"];
+
+        // 检查是否有Bearer token
+        if (authHeader != null && authHeader.StartsWith("Bearer "))
+        {
+            return authHeader.Substring("Bearer ".Length).Trim();
+        }
+
+        return null;
+    }
 }

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

@@ -0,0 +1,50 @@
+using Microsoft.AspNetCore.Authorization;
+using Volo.Abp.Application.Services;
+using Volo.Abp.Users;
+using Yi.Framework.AiHub.Domain.Entities.OpenApi;
+using Yi.Framework.AiHub.Domain.Managers;
+using Yi.Framework.SqlSugarCore.Abstractions;
+
+namespace Yi.Framework.AiHub.Application.Services;
+
+public class TokenService : ApplicationService
+{
+    private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
+    private readonly TokenManager _tokenManager;
+
+    /// <summary>
+    /// 构造函数
+    /// </summary>
+    /// <param name="tokenRepository"></param>
+    /// <param name="tokenManager"></param>
+    public TokenService(ISqlSugarRepository<TokenAggregateRoot> tokenRepository, TokenManager tokenManager)
+    {
+        _tokenRepository = tokenRepository;
+        _tokenManager = tokenManager;
+    }
+
+    /// <summary>
+    /// 获取token
+    /// </summary>
+    /// <returns></returns>
+    [Authorize]
+    public async Task<string?> GetAsync()
+    {
+        return await _tokenManager.GetAsync(CurrentUser.GetId());
+    }
+
+    /// <summary>
+    /// 创建token
+    /// </summary>
+    /// <exception cref="UserFriendlyException"></exception>
+    [Authorize]
+    public async Task CreateAsync()
+    {
+        if (!CurrentUser.Roles.Contains("YiXinAi-Vip") && CurrentUser.UserName != "cc")
+        {
+            throw new UserFriendlyException("充值成为Vip，畅想第三方token服务");
+        }
+
+        await _tokenManager.CreateAsync(CurrentUser.GetId());
+    }
+}