提示词内容生成

controllers/llm/llm_http/request.go

@@ -1,5 +1,7 @@
 package llm_http
 
+import "encoding/json"
+
 type LLMQuestionReq struct {
 	Question      string `description:"提问"`
 	KnowledgeBase string `description:"知识库"`
@@ -23,3 +25,9 @@ type GenerateContentReq struct {
 	WechatArticleId int    `json:"WechatArticleId" description:"公众号Id"`
 	Promote         string `json:"Promote" description:"提示词"`
 }
+type SaveContentReq struct {
+	WechatArticleId int             `json:"WechatArticleId" description:"公众号Id"`
+	Title           string          `json:"Title" description:"标题"`
+	Promote         json.RawMessage `json:"Promote" description:"提示词"`
+	AigcContent     json.RawMessage `json:"AigcContent" description:"生成内容"`
+}

controllers/llm/llm_http/response.go

@@ -15,3 +15,14 @@ type UserChatResp struct {
 type UserChatAddResp struct {
 	SendTime string
 }
+
+type AIGCResp struct {
+	Promote Content
+	Answer Content
+}
+
+type Content struct {
+	Role     string
+	Content  string
+	SendTime string
+}

controllers/llm/promote_controller.go

@@ -5,8 +5,10 @@ import (
 	"eta/eta_api/controllers"
 	"eta/eta_api/controllers/llm/llm_http"
 	"eta/eta_api/models"
+	"eta/eta_api/models/rag"
 	"eta/eta_api/services/llm/facade"
 	"eta/eta_api/utils"
+	"time"
 )
 
 type PromoteController struct {
@@ -95,6 +97,11 @@ func (pCtrl *PromoteController) GenerateContent() {
 		br.ErrMsg = "公众号文章编号非法"
 		return
 	}
+	userContent := llm_http.Content{
+		Content:  gcReq.Promote,
+		Role:     "user",
+		SendTime: time.Now().Format(utils.FormatDateTime),
+	}
 	res, err := facade.AIGCBaseOnPromote(facade.AIGC{
 		Promote:   gcReq.Promote,
 		ArticleId: gcReq.WechatArticleId,
@@ -104,8 +111,119 @@ func (pCtrl *PromoteController) GenerateContent() {
 		br.ErrMsg = "内容生成失败，Err:" + err.Error()
 		return
 	}
-	br.Data = res
+	aiContent := llm_http.Content{
+		Content:  res.Answer,
+		Role:     "assistant",
+		SendTime: time.Now().Format(utils.FormatDateTime),
+	}
+	br.Data = llm_http.AIGCResp{
+		Promote: userContent,
+		Answer:  aiContent,
+	}
 	br.Ret = 200
 	br.Success = true
 	br.Msg = "内容生成成功"
 }
+
+// SavePromoteContent @Title 保存问答内容
+// @Description 生成问答内容
+// @Success 101 {object} response.ListResp
+// @router /promote/save_content [post]
+func (pCtrl *PromoteController) SavePromoteContent() {
+	br := new(models.BaseResponse).Init()
+	defer func() {
+		pCtrl.Data["json"] = br
+		pCtrl.ServeJSON()
+	}()
+	var gcReq llm_http.SaveContentReq
+	err := json.Unmarshal(pCtrl.Ctx.Input.RequestBody, &gcReq)
+	if err != nil {
+		br.Msg = "参数解析异常！"
+		br.ErrMsg = "参数解析失败，Err:" + err.Error()
+		return
+	}
+	sysUser := pCtrl.SysUser
+	if sysUser == nil {
+		br.Msg = "请登录"
+		br.ErrMsg = "请登录，SysUser Is Empty"
+		br.Ret = 408
+		return
+	}
+	if gcReq.Promote == nil {
+		br.Msg = "提示词内容不能为空"
+		br.ErrMsg = "提示词不能为空"
+		return
+	}
+	if gcReq.AigcContent == nil {
+		br.Msg = "回答内容不能为空"
+		br.ErrMsg = "提示词不能为空"
+		return
+	}
+	if gcReq.WechatArticleId <= 0 {
+		br.Msg = "公众号文章编号非法"
+		br.ErrMsg = "公众号文章编号非法"
+		return
+	}
+	var userContent, assistantContent llm_http.Content
+	parseErr := json.Unmarshal([]byte(gcReq.AigcContent), &assistantContent)
+	if parseErr != nil {
+		br.Msg = "内容参数解析异常！"
+		br.ErrMsg = "内容参数解析异，err" + parseErr.Error()
+		return
+	}
+	parseErr = json.Unmarshal([]byte(gcReq.Promote), &userContent)
+	if parseErr != nil {
+		br.Msg = "内容参数解析异常！"
+		br.ErrMsg = "内容参数解析异，err" + parseErr.Error()
+		return
+	}
+	var titile string
+	if gcReq.Title != "" {
+		titile = gcReq.Title
+	} else {
+		titile = userContent.Content
+	}
+	var userSendTime, assistantSendTime time.Time
+	if userContent.SendTime != "" {
+		userSendTime, parseErr = time.ParseInLocation(utils.FormatDateTime, userContent.SendTime, time.Local)
+		if parseErr != nil {
+			br.Msg = "用户发送时间解析异常！"
+			br.ErrMsg = "用户发送时间解析异常，err" + parseErr.Error()
+			return
+		}
+	} else {
+		br.Msg = "用户发送时间不能为空！"
+		br.ErrMsg = "用户发送时间不能为空"
+		return
+	}
+	if assistantContent.SendTime != "" {
+		assistantSendTime, parseErr = time.ParseInLocation(utils.FormatDateTime, assistantContent.SendTime, time.Local)
+		if parseErr != nil {
+			br.Msg = "AI生成时间解析异常！"
+			br.ErrMsg = "AI生成时间解析异常，err" + parseErr.Error()
+			return
+		}
+	} else {
+		br.Msg = "AI生成时间不能为空！"
+		br.ErrMsg = "AI生成时间不能为空"
+		return
+	}
+	saveContentReq := rag.PromoteTrainRecord{
+		WechatArticleId: gcReq.WechatArticleId,
+		Title:           titile,
+		AigcContent:     assistantContent.Content,
+		AigcSendTime:    assistantSendTime,
+		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 = "保存内容成功"
+}

models/rag/promote_train_record.go

@@ -0,0 +1,28 @@
+package rag
+
+import (
+	"eta/eta_api/global"
+	"eta/eta_api/utils"
+	"time"
+)
+
+type PromoteTrainRecord struct {
+	Id              int       `gorm:"id;primaryKey"`
+	Title           string    `gorm:"title"`
+	WechatArticleId int       `gorm:"wechat_article_id"`
+	TemplatePromote string    `gorm:"template_promote"`
+	PromoteSendTime time.Time `gorm:"promote_send_time"`
+	AigcContent     string    `gorm:"aigc_content"`
+	AigcSendTime    time.Time `gorm:"aigc_send_time"`
+	IsDeleted       bool      `gorm:"is_deleted"`
+	CreatedTime     time.Time `gorm:"created_time"`
+	UpdateTime      time.Time `gorm:"update_time"`
+}
+
+func (p *PromoteTrainRecord) TableName() string {
+	return "promote_train_record"
+}
+
+func (p *PromoteTrainRecord) SaveContent() error {
+	return global.DbMap[utils.DbNameAI].Create(p).Error
+}

routers/commentsRouter.go

@@ -8413,6 +8413,15 @@ func init() {
             Filters: nil,
             Params: nil})
 
+    beego.GlobalControllerRouter["eta/eta_api/controllers/llm:PromoteController"] = append(beego.GlobalControllerRouter["eta/eta_api/controllers/llm:PromoteController"],
+        beego.ControllerComments{
+            Method: "SavePromoteContent",
+            Router: `/promote/save_content`,
+            AllowHTTPMethods: []string{"post"},
+            MethodParams: param.Make(),
+            Filters: nil,
+            Params: nil})
+
     beego.GlobalControllerRouter["eta/eta_api/controllers/llm:PromoteController"] = append(beego.GlobalControllerRouter["eta/eta_api/controllers/llm:PromoteController"],
         beego.ControllerComments{
             Method: "PromoteTrainRecordList",

services/llm/facade/llm_service.go

@@ -118,7 +118,16 @@ func AIGCBaseOnPromote(aigc AIGC) (resp bus_response.AIGCEtaResponse, err error)
 			err = fmt.Errorf("内容生成失败,code:%v,msg:%v", response.Ret, response.Msg)
 			return
 		} else {
-			dataStr := string(response.Data)
+			var dataStr string
+			// 按行分割输入
+			lines := strings.Split(string(response.Data), "\n")
+			// 遍历每一行，提取以 "data:" 开头的内容
+			for _, line := range lines {
+				if !strings.HasPrefix(line, ": ping") && strings.TrimSpace(line) != "" {
+					// 去掉 "data:" 前缀
+					dataStr += line
+				}
+			}
 			// 去除 "data: " 前缀
 			if strings.HasPrefix(dataStr, "data: ") {
 				dataStr = strings.TrimPrefix(dataStr, "data: ")
@@ -161,7 +170,17 @@ func AIGCBaseOnPromote(aigc AIGC) (resp bus_response.AIGCEtaResponse, err error)
 				utils.FileLog.Error("内容生成失败,code:%v,msg:%v", gcResp.Ret, gcResp.Msg)
 				err = fmt.Errorf("内容生成失败,err:%v", gcResp.Msg)
 			}
-			dataStr = string(gcResp.Data)
+
+			var gcStr string
+			// 按行分割输入
+			lines = strings.Split(string(gcResp.Data), "\n")
+			// 遍历每一行，提取以 "data:" 开头的内容
+			for _, line := range lines {
+				if !strings.HasPrefix(line, ": ping") && strings.TrimSpace(line) != "" {
+					// 去掉 "data:" 前缀
+					gcStr += line
+				}
+			}
 			// 去除 "data: " 前缀
 			if strings.HasPrefix(dataStr, "data: ") {
 				dataStr = strings.TrimPrefix(dataStr, "data: ")