🎨 优化模型配置 && 新增apikey功能 && 完善通用接口

server/api/v1/app/ai_preset.go

@@ -1,6 +1,9 @@
 package app
 
 import (
+	"encoding/json"
+	"strings"
+
 	"git.echol.cn/loser/ai_proxy/server/global"
 	"git.echol.cn/loser/ai_proxy/server/model/app"
 	"git.echol.cn/loser/ai_proxy/server/model/common/request"
@@ -150,7 +153,7 @@ func (a *AiPresetApi) GetAiPresetList(c *gin.Context) {
 	}
 }
 
-// ImportAiPreset 导入预设
+// ImportAiPreset 导入预设(JSON粘贴)
 // @Tags AiPreset
 // @Summary 导入预设(支持SillyTavern格式)
 // @Security ApiKeyAuth
@@ -160,17 +163,75 @@ func (a *AiPresetApi) GetAiPresetList(c *gin.Context) {
 // @Success 200 {object} response.Response{msg=string} "导入成功"
 // @Router /aiPreset/importAiPreset [post]
 func (a *AiPresetApi) ImportAiPreset(c *gin.Context) {
-	var preset app.AiPreset
-	err := c.ShouldBindJSON(&preset)
+	var rawData map[string]interface{}
+	err := c.ShouldBindJSON(&rawData)
 	if err != nil {
 		response.FailWithMessage(err.Error(), c)
 		return
 	}
+
+	preset, err := aiPresetService.ParseImportedPreset(rawData)
+	if err != nil {
+		response.FailWithMessage("解析预设失败:"+err.Error(), c)
+		return
+	}
 	preset.UserID = utils.GetUserID(c)
 
-	if err := aiPresetService.CreateAiPreset(&preset); err != nil {
+	if err := aiPresetService.CreateAiPreset(preset); err != nil {
 		global.GVA_LOG.Error("导入失败!", zap.Error(err))
-		response.FailWithMessage("导入失败", c)
+		response.FailWithMessage("导入失败:"+err.Error(), c)
+	} else {
+		response.OkWithMessage("导入成功", c)
+	}
+}
+
+// ImportAiPresetFile 导入预设文件
+// @Tags AiPreset
+// @Summary 通过文件导入预设(支持SillyTavern格式)
+// @Security ApiKeyAuth
+// @accept multipart/form-data
+// @Produce application/json
+// @Param file formData file true "预设JSON文件"
+// @Success 200 {object} response.Response{msg=string} "导入成功"
+// @Router /aiPreset/importAiPresetFile [post]
+func (a *AiPresetApi) ImportAiPresetFile(c *gin.Context) {
+	file, err := c.FormFile("file")
+	if err != nil {
+		response.FailWithMessage("文件上传失败:"+err.Error(), c)
+		return
+	}
+
+	// 检查文件类型
+	if file.Header.Get("Content-Type") != "application/json" && !strings.HasSuffix(file.Filename, ".json") {
+		response.FailWithMessage("只支持JSON文件", c)
+		return
+	}
+
+	// 读取文件内容
+	fileContent, err := file.Open()
+	if err != nil {
+		response.FailWithMessage("文件读取失败:"+err.Error(), c)
+		return
+	}
+	defer fileContent.Close()
+
+	// 解析JSON
+	var rawData map[string]interface{}
+	if err := json.NewDecoder(fileContent).Decode(&rawData); err != nil {
+		response.FailWithMessage("JSON解析失败:"+err.Error(), c)
+		return
+	}
+
+	preset, err := aiPresetService.ParseImportedPreset(rawData)
+	if err != nil {
+		response.FailWithMessage("解析预设失败:"+err.Error(), c)
+		return
+	}
+	preset.UserID = utils.GetUserID(c)
+
+	if err := aiPresetService.CreateAiPreset(preset); err != nil {
+		global.GVA_LOG.Error("导入失败!", zap.Error(err))
+		response.FailWithMessage("导入失败:"+err.Error(), c)
 	} else {
 		response.OkWithMessage("导入成功", c)
 	}