feat: 支持多租户多模型对话及文档去重优化

service/task.go

@@ -2,6 +2,8 @@ package service
 
 import (
 	"context"
+	"rag/consts/document"
+	"rag/consts/public"
 	"rag/consts/task"
 	"rag/dao"
 	"rag/model/dto"
@@ -37,7 +39,7 @@ func (s *taskService) WriteTaskProgress(ctx context.Context, req *dto.WriteTaskP
 			TaskType: req.TaskType,
 			Status:   req.Status,
 		})
-		completed = IsAllSubTasksCompleted(taskVO)
+		completed = IsAllSubTasks(taskVO, task.TaskStatusCompleted)
 	}
 
 	// 1. 查询是否已存在该文档的该类型任务
@@ -49,7 +51,7 @@ func (s *taskService) WriteTaskProgress(ctx context.Context, req *dto.WriteTaskP
 		g.Log().Errorf(ctx, "查询任务失败: %v", err)
 		return err
 	}
-	err = gfdb.DB(ctx).Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
+	err = gfdb.DB(ctx, public.DbNameKnowledge).Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
 		// 2. 如果不存在，则创建新任务
 		if g.IsEmpty(existTask) {
 			createReq := &dto.CreateTaskReq{
@@ -80,17 +82,36 @@ func (s *taskService) WriteTaskProgress(ctx context.Context, req *dto.WriteTaskP
 				Status: task.TaskStatusCompleted,
 				Remark: "文档解析完成",
 			})
+			if err != nil {
+				g.Log().Errorf(ctx, "更新任务失败: %v", err)
+				return err
+			}
+			_, err = dao.Document.Update(ctx, &dto.UpdateDocumentReq{
+				Id:           req.TaskId,
+				VectorStatus: document.VectorStatusCompleted.Code(),
+			})
+			if err != nil {
+				return err
+			}
+		} else {
+			if task.TaskStatusFailed == req.Status {
+				_, err = dao.Document.Update(ctx, &dto.UpdateDocumentReq{
+					Id:           req.TaskId,
+					VectorStatus: document.VectorStatusFailed.Code(),
+				})
+				if err != nil {
+					return err
+				}
+			}
 		}
-
 		return nil
 	})
 
 	return
 }
 
-// IsAllSubTasksCompleted 判断三个子任务是否全部完成
-// 参数：传入当前文档的所有子任务列表
-func IsAllSubTasksCompleted(subTasks []*dto.TaskVO) bool {
+// IsAllSubTasks 判断三个子任务
+func IsAllSubTasks(subTasks []*dto.TaskVO, taskStatus task.TaskStatus) bool {
 	// 必须包含 3 种任务类型
 	hasKeywords := false
 	hasVector := false
@@ -98,7 +119,7 @@ func IsAllSubTasksCompleted(subTasks []*dto.TaskVO) bool {
 
 	for _, t := range subTasks {
 		// 子任务必须是【已完成】状态才计数
-		if t.Status == task.TaskStatusCompleted {
+		if t.Status == taskStatus {
 			switch t.TaskType {
 			case task.TaskTypeExtractKeywords:
 				hasKeywords = true
@@ -113,3 +134,15 @@ func IsAllSubTasksCompleted(subTasks []*dto.TaskVO) bool {
 	// 三个任务全部完成 → 返回true
 	return hasKeywords && hasVector && hasFullText
 }
+
+func (s *taskService) Get(ctx context.Context, req *dto.GetTaskReq) (res *dto.ListTaskRes, err error) {
+	list, total, err := dao.Task.Get(ctx, req)
+	if err != nil {
+		return
+	}
+	res = &dto.ListTaskRes{
+		Total: total,
+	}
+	err = gconv.Struct(list, &res.List)
+	return
+}