feat: Thor搭建

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

@@ -12,6 +12,7 @@ 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.OpenAi;
 using Yi.Framework.AiHub.Domain.Entities;
 using Yi.Framework.AiHub.Domain.Entities.Model;
 using Yi.Framework.AiHub.Domain.Extensions;
@@ -92,7 +93,8 @@ public class AiChatService : ApplicationService
     /// </summary>
     /// <param name="input"></param>
     /// <param name="cancellationToken"></param>
-    public async Task PostSendAsync(SendMessageInput input, CancellationToken cancellationToken)
+    public async Task PostSendAsync([FromBody] ThorChatCompletionsRequest input, [FromQuery] Guid sessionId,
+        CancellationToken cancellationToken)
     {
         //除了免费模型，其他的模型都要校验
         if (!input.Model.Contains("DeepSeek-R1"))
@@ -111,22 +113,8 @@ public class AiChatService : ApplicationService
                 throw new UserFriendlyException("未登录用户，只能使用未加速的DeepSeek-R1，请登录后重试");
             }
         }
-
-        var history = new List<ChatMessage>();
-        foreach (var aiChatContextDto in input.Messages)
-        {
-            if (aiChatContextDto.Role == "assistant")
-            {
-                history.Add(ChatMessage.CreateAssistantMessage(aiChatContextDto.Content));
-            }
-            else if (aiChatContextDto.Role == "user")
-            {
-                history.Add(ChatMessage.CreateUserMessage(aiChatContextDto.Content));
-            }
-        }
-
         //ai网关代理httpcontext
-        await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input.Model, history,
-            CurrentUser.Id, input.SessionId, cancellationToken);
+        await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
+            CurrentUser.Id, sessionId, cancellationToken);
     }
 }