Browse Source

add:添加前端到后端的模型映射

zqbao 1 year ago
parent
commit
84790287f3
2 changed files with 5 additions and 9 deletions
  1. 2 2
      controllers/ai/ai.go
  2. 3 7
      controllers/ai/ai_file.go

+ 2 - 2
controllers/ai/ai.go

@@ -132,7 +132,7 @@ func (this *AiController) List() {
 		chatItem.Ask = req.Ask
 		chatItem.AskUuid = utils.MD5(req.Ask)
 		chatItem.Answer = answer
-		chatItem.Model = req.Model
+		chatItem.Model = EnabledModelsForMap[req.Model]
 		chatItem.SysUserId = this.SysUser.AdminId
 		chatItem.SysUserRealName = this.SysUser.RealName
 		chatItem.CreateTime = time.Now()
@@ -150,7 +150,7 @@ func (this *AiController) List() {
 		chatItem.Ask = req.Ask
 		chatItem.AskUuid = utils.MD5(req.Ask)
 		chatItem.Answer = answer
-		chatItem.Model = req.Model
+		chatItem.Model = EnabledModelsForMap[req.Model]
 		chatItem.SysUserId = this.SysUser.AdminId
 		chatItem.SysUserRealName = this.SysUser.RealName
 		chatItem.CreateTime = time.Now()

+ 3 - 7
controllers/ai/ai_file.go

@@ -418,11 +418,7 @@ func (this *AiFileController) FileRetrieve() {
 			return
 		}
 	}
-	if req.Model == "Kimi" {
-		resp.Model = "Kimi"
-	} else {
-		resp.Model = "GPT-4 Turbo"
-	}
+	resp.Model = aimod.ModelViewMap[req.Model]
 	resp.AiChatTopicId = req.AiChatTopicId
 	resp.Ask = req.Ask
 	resp.Answer = answer
@@ -434,6 +430,6 @@ func (this *AiFileController) FileRetrieve() {
 }
 
 var EnabledModelsForMap = map[string]string{
-	"Kimi": "moonshot-v1-8k",
-	"GPT":  "gpt-4-1106-preview",
+	"Kimi":        "moonshot-v1-32k",
+	"GPT-4 Turbo": "gpt-4-1106-preview",
 }