From 01055762ca4ae2cf666017fe683ef40ead56c50a Mon Sep 17 00:00:00 2001
From: lq1405 <2769838458@qq.com>
Date: Sun, 16 Feb 2025 16:00:04 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E4=B8=80=E4=B8=AAopenAI?=
 =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E7=9A=84=E8=BD=AC=E5=8F=91=E6=8E=A5=E5=8F=A3?=
 =?UTF-8?q?=EF=BC=8C=E7=9B=B4=E6=8E=A5=E8=BD=AC=E5=8F=91=E6=95=B4=E4=B8=AA?=
 =?UTF-8?q?=E8=AF=B7=E6=B1=82=E4=BD=93?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 LMS.Repository/Forward/SimpleTransferModel.cs | 19 ++++++
 LMS.service/Controllers/ForwardController.cs  | 30 +++++++++-
 LMS.service/Service/ForwardWordService.cs     | 59 +++++++++++++++++++
 3 files changed, 107 insertions(+), 1 deletion(-)
 create mode 100644 LMS.Repository/Forward/SimpleTransferModel.cs

diff --git a/LMS.Repository/Forward/SimpleTransferModel.cs b/LMS.Repository/Forward/SimpleTransferModel.cs
new file mode 100644
index 0000000..4aaa430
--- /dev/null
+++ b/LMS.Repository/Forward/SimpleTransferModel.cs
@@ -0,0 +1,19 @@
+namespace LMS.Repository.Forward;
+
+public class SimpleTransferModel
+{
+    /// <summary>
+    /// GPT的完整地址
+    /// </summary>
+    public string url { get; set; }
+
+    /// <summary>
+    /// 对应的API站的使用的APIkey
+    /// </summary>
+    public string APIKey { get; set; }
+
+    /// <summary>
+    /// 实际的GPT请求的数据字符串，直接序列化再传递就行
+    /// </summary>
+    public string dataString { get; set; }
+}
diff --git a/LMS.service/Controllers/ForwardController.cs b/LMS.service/Controllers/ForwardController.cs
index 4f97bbe..805d646 100644
--- a/LMS.service/Controllers/ForwardController.cs
+++ b/LMS.service/Controllers/ForwardController.cs
@@ -11,11 +11,13 @@ namespace LMS.service.Controllers;
 [Route("lms/[controller]/[action]")]
 [ApiController]
 
-// ceshi
 public class ForwardController(ForwardWordService forwardWordService) : ControllerBase
 {
     private readonly ForwardWordService _forwardWordService = forwardWordService;
 
+
+    #region 非流转发接口，需要系统数据
+
     /// <summary>
     /// 转发OpenAi格式的请求
     /// </summary>
@@ -31,6 +33,10 @@ public class ForwardController(ForwardWordService forwardWordService) : Controll
         return await _forwardWordService.ForwardWord(request);
     }
 
+    #endregion
+
+
+    #region 流式转发接口，需要系统数据
     /// <summary>
     /// 流式转发
     /// </summary>
@@ -72,4 +78,26 @@ public class ForwardController(ForwardWordService forwardWordService) : Controll
             return propertyName.ToLower();
         }
     }
+
+    #endregion
+
+
+    #region 普通转发接口，直接转发，不要系统数据
+
+    /// <summary>
+    /// 转发所有的OpenAI格式的AI请求
+    /// </summary>
+    /// <param name="request"></param>
+    /// <returns></returns>
+    [HttpPost]
+    public async Task<ActionResult<APIResponseModel<object>>> SimpleTransfer([FromBody] SimpleTransferModel request)
+    {
+        if (!ModelState.IsValid)
+        {
+            return APIResponseModel<object>.CreateErrorResponseModel(ResponseCode.ParameterError);
+        }
+        return await _forwardWordService.SimpleTransfer(request);
+    }
+
+    #endregion
 }
diff --git a/LMS.service/Service/ForwardWordService.cs b/LMS.service/Service/ForwardWordService.cs
index b67a376..674fdcf 100644
--- a/LMS.service/Service/ForwardWordService.cs
+++ b/LMS.service/Service/ForwardWordService.cs
@@ -18,6 +18,8 @@ namespace LMS.service.Service;
 public class ForwardWordService(ApplicationDbContext context)
 {
     private readonly ApplicationDbContext _context = context;
+
+    #region 非流转发接口，需要系统数据
     /// <summary>
     /// 转发OpenAi格式的请求 非流
     /// </summary>
@@ -94,6 +96,15 @@ public class ForwardWordService(ApplicationDbContext context)
         }
     }
 
+    #endregion
+
+    #region 流式转发接口，需要系统数据
+    /// <summary>
+    /// 流式转发
+    /// </summary>
+    /// <param name="request"></param>
+    /// <returns></returns>
+    /// <exception cref="Exception"></exception>
     public async IAsyncEnumerable<string> ForwardWordStream(ForwardModel request)
     {
 
@@ -144,4 +155,52 @@ public class ForwardWordService(ApplicationDbContext context)
         }
 
     }
+
+    #endregion
+
+    #region 简单的AI转发，OpenAI格式
+    /// <summary>
+    /// 简单的转发接口
+    /// </summary>
+    /// <param name="request"></param>
+    /// <returns></returns>
+    /// <exception cref="NotImplementedException"></exception>
+    public async Task<ActionResult<APIResponseModel<object>>> SimpleTransfer(SimpleTransferModel request)
+    {
+        try
+        {
+            // 开始拼接请求体
+            using HttpClient client = new();
+            client.DefaultRequestHeaders.Add("Authorization", "Bearer " + request.APIKey);
+
+            // 判断请求的url是不是满足条件
+            if (string.IsNullOrEmpty(request.url))
+            {
+                throw new Exception("请求的url为空");
+            }
+            if (!request.url.StartsWith("https://ark.cn-beijing.volces.com") && !request.url.StartsWith("https://api.moonshot.cn") && !request.url.StartsWith("https://laitool.net") && !request.url.StartsWith("https://api.laitool.cc") && !request.url.StartsWith("https://laitool.cc"))
+            {
+                throw new Exception("请求的url不合法");
+            }
+            client.Timeout = Timeout.InfiniteTimeSpan;
+            var response = await client.PostAsync(request.url, new StringContent(request.dataString, Encoding.UTF8, "application/json"));
+
+            // 判断返回的状态码
+            if (response.StatusCode != HttpStatusCode.OK)
+            {
+                // 读取响应体
+                string responseContent = await response.Content.ReadAsStringAsync();
+                return APIResponseModel<object>.CreateErrorResponseModel(ResponseCode.ForwardWordFail, responseContent, "请求失败");
+            }
+            var content = await response.Content.ReadAsStringAsync();
+            // 序列化一下
+            return APIResponseModel<object>.CreateSuccessResponseModel(content);
+        }
+        catch (Exception e)
+        {
+            return APIResponseModel<object>.CreateErrorResponseModel(ResponseCode.SystemError, e.Message);
+        }
+    }
+
+    #endregion
 }