kobe6258 6 dagar sedan
förälder
incheckning
9a7592cb88
2 ändrade filer med 11 tillägg och 5 borttagningar
  1. 7 1
      controllers/llm/promote_controller.go
  2. 4 4
      models/rag/promote_train_record.go

+ 7 - 1
controllers/llm/promote_controller.go

@@ -8,6 +8,7 @@ import (
 	"eta/eta_api/models/rag"
 	"eta/eta_api/services/llm/facade"
 	"eta/eta_api/utils"
+	"fmt"
 	"time"
 )
 
@@ -64,7 +65,7 @@ func (pCtrl *PromoteController) GenerateContent() {
 		br.ErrMsg = "内容生成失败,Err:" + err.Error()
 		return
 	}
-	aiSendTime:=time.Now()
+	aiSendTime := time.Now()
 	aiContent := llm_http.Content{
 		Content:  res.Answer,
 		Role:     "assistant",
@@ -270,6 +271,11 @@ func (pCtrl *PromoteController) PromoteContentList() {
 		br.ErrMsg = "查询列表失败,err:" + err.Error()
 		return
 	}
+	total := len(list)
+	for _, item := range list {
+		item.Title = fmt.Sprintf("%d.%s", total, item.Title)
+		total--
+	}
 	br.Data = list
 	br.Ret = 200
 	br.Success = true

+ 4 - 4
models/rag/promote_train_record.go

@@ -19,8 +19,8 @@ type PromoteTrainRecord struct {
 	UpdateTime      time.Time `gorm:"update_time"`
 }
 
-func (p *PromoteTrainRecord) ToView() PromoteTrainRecordView {
-	return PromoteTrainRecordView{
+func (p *PromoteTrainRecord) ToView() *PromoteTrainRecordView {
+	return &PromoteTrainRecordView{
 		Id:              p.Id,
 		Title:           p.Title,
 		WechatArticleId: p.WechatArticleId,
@@ -52,10 +52,10 @@ func DeleteContent(id int) error {
 	return global.DbMap[utils.DbNameAI].Model(&PromoteTrainRecord{}).Where("id = ?", id).Update("is_deleted", true).Error
 }
 
-func GetRecordList(wechatArticleId int) (list []PromoteTrainRecordView, err error) {
+func GetRecordList(wechatArticleId int) (list []*PromoteTrainRecordView, err error) {
 	var ormList []PromoteTrainRecord
 	err = global.DbMap[utils.DbNameAI].Model(&PromoteTrainRecord{}).Where("wechat_article_id = ? and is_deleted=?", wechatArticleId, false).Order(`created_time DESC`).Find(&ormList).Error
-	if err!=nil{
+	if err != nil {
 		return
 	}
 	for _, item := range ormList {