From e996bc2d7f3fed7504513bcb61db3d266d34abf2 Mon Sep 17 00:00:00 2001
From: ccnetcore <cc2016cc>
Date: Thu, 3 Jul 2025 22:44:52 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=AE=8C=E6=88=90token=E6=A8=A1?=
 =?UTF-8?q?=E5=9D=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Dtos/{OpenAiDto => OpenAi}/ChatCompletionsInput.cs    | 0
 .../Dtos/Token/TokenOutput.cs                             | 6 ++++++
 .../Services/OpenApiService.cs                            | 2 +-
 .../Services/TokenService.cs                              | 8 ++++++--
 4 files changed, 13 insertions(+), 3 deletions(-)
 rename Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/{OpenAiDto => OpenAi}/ChatCompletionsInput.cs (100%)
 create mode 100644 Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/Token/TokenOutput.cs

diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/OpenAiDto/ChatCompletionsInput.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/OpenAi/ChatCompletionsInput.cs
similarity index 100%
rename from Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/OpenAiDto/ChatCompletionsInput.cs
rename to Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/OpenAi/ChatCompletionsInput.cs
diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/Token/TokenOutput.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/Token/TokenOutput.cs
new file mode 100644
index 00000000..fc34730f
--- /dev/null
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/Token/TokenOutput.cs
@@ -0,0 +1,6 @@
+namespace Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
+
+public class TokenOutput
+{
+    public string? ApiKey { get; set; }
+}
\ No newline at end of file
diff --git 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
index 502924de..5444a3b8 100644
--- 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
@@ -42,7 +42,7 @@ public class OpenApiService : ApplicationService
     /// </summary>
     /// <param name="input"></param>
     /// <param name="cancellationToken"></param>
-    [HttpPost("/v1/chat/completions")]
+    [HttpPost("openApi/v1/chat/completions")]
     public async Task ChatCompletionsAsync(ChatCompletionsInput input, CancellationToken cancellationToken)
     {
         //前面都是校验，后面才是真正的调用
diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/TokenService.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/TokenService.cs
index de376e39..fb26b6d9 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/TokenService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/TokenService.cs
@@ -1,6 +1,7 @@
 using Microsoft.AspNetCore.Authorization;
 using Volo.Abp.Application.Services;
 using Volo.Abp.Users;
+using Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
 using Yi.Framework.AiHub.Domain.Entities.OpenApi;
 using Yi.Framework.AiHub.Domain.Managers;
 using Yi.Framework.SqlSugarCore.Abstractions;
@@ -28,9 +29,12 @@ public class TokenService : ApplicationService
     /// </summary>
     /// <returns></returns>
     [Authorize]
-    public async Task<string?> GetAsync()
+    public async Task<TokenOutput> GetAsync()
     {
-        return await _tokenManager.GetAsync(CurrentUser.GetId());
+        return new TokenOutput
+        {
+            ApiKey = await _tokenManager.GetAsync(CurrentUser.GetId())
+        };
     }
 
     /// <summary>