Jelajahi Sumber

Merge branch 'rag/4.0' into debug

Roc 6 hari lalu
induk
melakukan
cd25d1c509
2 mengubah file dengan 7 tambahan dan 1 penghapusan
  1. 6 0
      controllers/llm/question.go
  2. 1 1
      models/rag/question.go

+ 6 - 0
controllers/llm/question.go

@@ -484,6 +484,8 @@ func (c *QuestionController) SetDefault() {
 		br.ErrMsg = "设置失败,Err:" + err.Error()
 		return
 	}
+	// 新增/编辑ES数据
+	go services.AddOrEditEsRagQuestion(item.QuestionId)
 
 	br.Ret = 200
 	br.Success = true
@@ -554,6 +556,8 @@ func (c *QuestionController) UnSetDefault() {
 		br.ErrMsg = "取消设置失败,Err:" + err.Error()
 		return
 	}
+	// 新增/编辑ES数据
+	go services.AddOrEditEsRagQuestion(item.QuestionId)
 
 	// todo 对应的提示词生成的摘要库和向量库内容也取消,同时需要加锁,不允许重复操作
 
@@ -631,6 +635,8 @@ func (c *QuestionController) GenerateAbstract() {
 		br.ErrMsg = "取消设置失败,Err:" + err.Error()
 		return
 	}
+	// 新增/编辑ES数据
+	go services.AddOrEditEsRagQuestion(item.QuestionId)
 
 	// 添加任务
 	services.AddGenerateAbstractTask(item, c.SysUser)

+ 1 - 1
models/rag/question.go

@@ -169,7 +169,7 @@ func (m *Question) GetTitlePageListByCondition(condition string, pars []interfac
 		return
 	}
 	if total > 0 {
-		items, err = m.GetListByCondition(`question_id,question_title,sort`, condition, pars, startSize, pageSize)
+		items, err = m.GetListByCondition(`question_id,question_title,sort,version,generate_status,is_default`, condition, pars, startSize, pageSize)
 	}
 
 	return