浏览代码

Merge branch 'rag/4.0' into debug

Roc 6 天之前
父节点
当前提交
cd25d1c509
共有 2 个文件被更改,包括 7 次插入1 次删除
  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()
 		br.ErrMsg = "设置失败,Err:" + err.Error()
 		return
 		return
 	}
 	}
+	// 新增/编辑ES数据
+	go services.AddOrEditEsRagQuestion(item.QuestionId)
 
 
 	br.Ret = 200
 	br.Ret = 200
 	br.Success = true
 	br.Success = true
@@ -554,6 +556,8 @@ func (c *QuestionController) UnSetDefault() {
 		br.ErrMsg = "取消设置失败,Err:" + err.Error()
 		br.ErrMsg = "取消设置失败,Err:" + err.Error()
 		return
 		return
 	}
 	}
+	// 新增/编辑ES数据
+	go services.AddOrEditEsRagQuestion(item.QuestionId)
 
 
 	// todo 对应的提示词生成的摘要库和向量库内容也取消,同时需要加锁,不允许重复操作
 	// todo 对应的提示词生成的摘要库和向量库内容也取消,同时需要加锁,不允许重复操作
 
 
@@ -631,6 +635,8 @@ func (c *QuestionController) GenerateAbstract() {
 		br.ErrMsg = "取消设置失败,Err:" + err.Error()
 		br.ErrMsg = "取消设置失败,Err:" + err.Error()
 		return
 		return
 	}
 	}
+	// 新增/编辑ES数据
+	go services.AddOrEditEsRagQuestion(item.QuestionId)
 
 
 	// 添加任务
 	// 添加任务
 	services.AddGenerateAbstractTask(item, c.SysUser)
 	services.AddGenerateAbstractTask(item, c.SysUser)

+ 1 - 1
models/rag/question.go

@@ -169,7 +169,7 @@ func (m *Question) GetTitlePageListByCondition(condition string, pars []interfac
 		return
 		return
 	}
 	}
 	if total > 0 {
 	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
 	return