hongze 1 yıl önce
ebeveyn
işleme
93a900ffc1
2 değiştirilmiş dosya ile 19 ekleme ve 7 silme
  1. 11 3
      controllers/ai/ai_file.go
  2. 8 4
      services/aiser/ai.go

+ 11 - 3
controllers/ai/ai_file.go

@@ -334,9 +334,17 @@ func (this *AiFileController) FileRetrieve() {
 			return
 		}
 
-		assistantId = fileRetrieveResp.Data.AssistantId
-		threadId = fileRetrieveResp.Data.ThreadId
-		answer = fileRetrieveResp.Data.Answer
+		if fileRetrieveResp != nil {
+			if fileRetrieveResp.Ret == 200 {
+				assistantId = fileRetrieveResp.Data.AssistantId
+				threadId = fileRetrieveResp.Data.ThreadId
+				answer = fileRetrieveResp.Data.Answer
+			} else {
+				br.Msg = "获取数据失败!"
+				br.ErrMsg = fileRetrieveResp.Msg
+				return
+			}
+		}
 	}
 
 	if req.AiChatTopicId <= 0 { //新增

+ 8 - 4
services/aiser/ai.go

@@ -157,6 +157,9 @@ func FileRetrieve(assistantId, threadId string, historyChatList []aimod.HistoryC
 	defer response.Body.Close()
 
 	body, err := ioutil.ReadAll(response.Body)
+	if err != nil {
+		return
+	}
 
 	utils.FileLog.Info("result:" + string(body))
 
@@ -192,12 +195,13 @@ type OpenAiFile struct {
 }
 
 type OpenAiFileRetrieveResp struct {
-	Ret  int
-	Data *OpenAiFileRetrieve
-	Msg  string
+	Ret    int
+	Data   *FileRetrieveResp
+	Msg    string
+	ErrMsg string
 }
 
-type OpenAiFileRetrieve struct {
+type FileRetrieveResp struct {
 	AssistantId string `description:"助手ID"`
 	ThreadId    string `description:"进程id"`
 	Answer      string