Merge branch 'feature/deepseek_rag_2.0' into debug

controllers/llm/promote_controller.go

@@ -49,10 +49,11 @@ func (pCtrl *PromoteController) GenerateContent() {
 		br.ErrMsg = "公众号文章编号非法"
 		return
 	}
+	userSendTime := time.Now()
 	userContent := llm_http.Content{
 		Content:  gcReq.Promote,
 		Role:     "user",
-		SendTime: time.Now().Format(utils.FormatDateTime),
+		SendTime: userSendTime.Format(utils.FormatDateTime),
 	}
 	res, err := facade.AIGCBaseOnPromote(facade.AIGC{
 		Promote:   gcReq.Promote,
@@ -63,15 +64,31 @@ func (pCtrl *PromoteController) GenerateContent() {
 		br.ErrMsg = "内容生成失败，Err:" + err.Error()
 		return
 	}
+	aiSendTime:=time.Now()
 	aiContent := llm_http.Content{
 		Content:  res.Answer,
 		Role:     "assistant",
-		SendTime: time.Now().Format(utils.FormatDateTime),
+		SendTime: aiSendTime.Format(utils.FormatDateTime),
 	}
 	br.Data = llm_http.AIGCResp{
 		Promote: userContent,
 		Answer:  aiContent,
 	}
+	saveContentReq := rag.PromoteTrainRecord{
+		WechatArticleId: gcReq.WechatArticleId,
+		Title:           userContent.Content,
+		AigcContent:     res.Answer,
+		AigcSendTime:    aiSendTime,
+		TemplatePromote: userContent.Content,
+		PromoteSendTime: userSendTime,
+		CreatedTime:     time.Now(),
+	}
+	err = saveContentReq.SaveContent()
+	if err != nil {
+		br.Msg = "保存内容失败"
+		br.ErrMsg = "保存内容失败，err:" + err.Error()
+		return
+	}
 	br.Ret = 200
 	br.Success = true
 	br.Msg = "内容生成成功"