新增get转发接口

新增Laitool 设置中的绘图设置

LMS.Common/Dictionary/AllOptions.cs

@@ -1,4 +1,7 @@
-namespace LMS.Common.Dictionary;
+
+using LMS.Common.Enum;
+
+namespace LMS.Common.Dictionary;
 
 public class AllOptions
 {
@@ -7,6 +10,7 @@ public class AllOptions
         { "all", [] },
         { "tts", ["EdgeTTsRoles"] },
         { "software", ["LaitoolHomePage", "LaitoolNotice", "LaitoolUpdateContent","LaitoolVersion"]},
-        { "trial" , ["LaiToolTrialDays"] }
+        { "trial" , ["LaiToolTrialDays"] },
+        { "image", [OptionKeyName.LaitoolFluxApiModelList] }
     };
 }

LMS.Common/Enum/OptionTypeEnum.cs

@@ -6,3 +6,11 @@ public enum OptionTypeEnum
     JSON = 2,
     Number = 3,
 }
+
+public static class OptionKeyName
+{
+    /// <summary>
+    /// LaiTool Flux API 模型列表的Option Key
+    /// </summary>
+    public const string LaitoolFluxApiModelList = "LaitoolFluxApiModelList";
+}

LMS.Repository/Forward/SimpleTransferModel.cs

@@ -17,3 +17,16 @@ public class SimpleTransferModel
     /// </summary>
     public string dataString { get; set; }
 }
+
+public class GetTransferModel
+{
+    /// <summary>
+    /// GPT的完整地址
+    /// </summary>
+    public string url { get; set; }
+
+    /// <summary>
+    /// 对应的API站的使用的APIkey
+    /// </summary>
+    public string? APIKey { get; set; }
+}

LMS.service/Configuration/InitConfiguration/DatabaseConfiguration.cs

@@ -1,12 +1,10 @@
-using LMS.DAO;
+using LMS.Common.Enum;
+using LMS.DAO;
 using LMS.Repository.DB;
 using LMS.Repository.Models.DB;
 using LMS.Tools.Extensions;
 using Microsoft.AspNetCore.Identity;
 using Microsoft.EntityFrameworkCore;
-using LMS.Common.Enum;
-using System.Threading;
-using static LMS.Common.Enums.ResponseCodeEnum;
 
 namespace LMS.service.Configuration.InitConfiguration;
 
@@ -102,7 +100,8 @@ public class DatabaseConfiguration(IServiceProvider serviceProvider) : IHostedSe
             new Options { Key = "LaitoolUpdateContent", Value = string.Empty, Type = OptionTypeEnum.String },
             new Options { Key = "LaitoolNotice", Value = string.Empty, Type = OptionTypeEnum.String },
             new Options { Key = "LaitoolVersion", Value = string.Empty, Type = OptionTypeEnum.String },
-            new Options { Key = "LaiToolTrialDays", Value = "2" , Type = OptionTypeEnum.Number}
+            new Options { Key = "LaiToolTrialDays", Value = "2" , Type = OptionTypeEnum.Number},
+            new Options { Key = OptionKeyName.LaitoolFluxApiModelList, Value = "{}" , Type = OptionTypeEnum.JSON }
         ];
 
         // 遍历所有的配置项，如果没有则添加

LMS.service/Controllers/ForwardController.cs

@@ -82,7 +82,7 @@ public class ForwardController(ForwardWordService forwardWordService) : Controll
     #endregion
 
 
-    #region 普通转发接口，直接转发，不要系统数据
+    #region Post 直接转发接口 
 
     /// <summary>
     /// 转发所有的OpenAI格式的AI请求
@@ -100,4 +100,18 @@ public class ForwardController(ForwardWordService forwardWordService) : Controll
     }
 
     #endregion
+
+    #region Get直接转发接口
+
+    [HttpPost]
+    public async Task<ActionResult<APIResponseModel<object>>> GetTransfer([FromBody] GetTransferModel getTransferModel)
+    {
+        if (!ModelState.IsValid)
+        {
+            return APIResponseModel<object>.CreateErrorResponseModel(ResponseCode.ParameterError);
+        }
+        return await _forwardWordService.GetTransfer(getTransferModel);
+
+    }
+    #endregion
 }

LMS.service/Controllers/OptionsController.cs

@@ -61,12 +61,5 @@ namespace LMS.service.Controllers
 
         #endregion
 
-        //[HttpPost]
-        //[Authorize]
-        //public async Task<ActionResult<APIResponseModel<Options>>> GetOptions([FromBody] OptionsRequestModel request)
-        //{
-        //    return await _optionsService.GetOptions(request);
-        //}
-
     }
 }

LMS.service/Service/ForwardWordService.cs

@@ -158,7 +158,7 @@ public class ForwardWordService(ApplicationDbContext context)
 
     #endregion
 
-    #region 简单的AI转发，OpenAI格式
+    #region Post 转发接口
     /// <summary>
     /// 简单的转发接口
     /// </summary>
@@ -171,20 +171,88 @@ public class ForwardWordService(ApplicationDbContext context)
         {
             // 开始拼接请求体
             using HttpClient client = new();
+            if (request.url.StartsWith("https://api.laitool.net"))
+            {
+                //client.DefaultRequestHeaders.Add("Authorization", "Bearer " + request.APIKey);
+                client.DefaultRequestHeaders.Add("mj-api-secret", request.APIKey);
+            }
+            else
+            {
                 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"))
+            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.net")
+                && !request.url.StartsWith("https://api.laitool.cc")
+                && !request.url.StartsWith("https://laitool.cc"))
             {
-                throw new Exception("请求的url不合法");
+                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)
+            {
+                if (response.StatusCode == HttpStatusCode.Unauthorized)
+                {
+                    return APIResponseModel<object>.CreateErrorResponseModel(ResponseCode.ForwardWordFail, "responseCode: 401", "请求失败");
+                }
+                // 读取响应体
+                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 
+
+    #region Get直接转发接口
+    internal async Task<ActionResult<APIResponseModel<object>>> GetTransfer(GetTransferModel getTransferModel)
+    {
+        try
+        {
+            // 开始拼接请求体
+            using HttpClient client = new();
+
+            if (getTransferModel.url.StartsWith("https://api.laitool.net"))
+            {
+                client.DefaultRequestHeaders.Add("mj-api-secret", getTransferModel.APIKey);
+            }
+            else
+            {
+                client.DefaultRequestHeaders.Add("Authorization", "Bearer " + getTransferModel.APIKey);
+            }
+
+            // 判断请求的url是不是满足条件
+            if (string.IsNullOrEmpty(getTransferModel.url))
+            {
+                throw new Exception("请求的url为空");
+            }
+            if (!getTransferModel.url.StartsWith("https://ark.cn-beijing.volces.com")
+                && !getTransferModel.url.StartsWith("https://api.moonshot.cn")
+                && !getTransferModel.url.StartsWith("https://laitool.net")
+                && !getTransferModel.url.StartsWith("https://api.laitool.net")
+                && !getTransferModel.url.StartsWith("https://api.laitool.cc")
+                && !getTransferModel.url.StartsWith("https://laitool.cc"))
+            {
+                throw new Exception("请求的url不支持转发");
+            }
+            client.Timeout = Timeout.InfiniteTimeSpan;
+            var response = await client.GetAsync(getTransferModel.url);
             // 判断返回的状态码
             if (response.StatusCode != HttpStatusCode.OK)
             {
@@ -201,6 +269,5 @@ public class ForwardWordService(ApplicationDbContext context)
             return APIResponseModel<object>.CreateErrorResponseModel(ResponseCode.SystemError, e.Message);
         }
     }
-
     #endregion
 }