refactor: 重构文档处理流程和任务管理

model/dto/document.go

@@ -84,12 +84,6 @@ type ProcessDocumentReq struct {
 	DatasetId int64 `json:"datasetId" v:"required#数据集ID不能为空"`
 }
 
-// ProcessDocumentRes 处理文件响应
-type ProcessDocumentRes struct {
-	ChunkCount int64 `json:"chunkCount"`
-	CostTime   int64 `json:"costTime"`
-}
-
 type ListDocumentChunkRPC struct {
 	List []*DocumentChunkRPC `json:"list"`
 }

model/dto/keyword.go

@@ -52,6 +52,7 @@ type ListKeywordReq struct {
 	DatasetId  int64       `json:"datasetId"`
 	DocumentId int64       `json:"documentId"`
 	Word       string      `json:"word"`
+	Words      []string    `json:"words"`
 	Keyword    string      `json:"keyword" dc:"关键词搜索"`
 }
 
@@ -62,9 +63,11 @@ type ListKeywordRes struct {
 }
 
 type KeywordVO struct {
-	Id        int64       `json:"id,string" dc:"id"`
-	Word      string      `json:"word" dc:"关键词名称"`
-	Weight    int16       `json:"weight" dc:"权重"`
-	CreatedAt *gtime.Time `json:"createdAt" dc:"创建时间"`
-	UpdatedAt *gtime.Time `json:"updatedAt" dc:"更新时间"`
+	Id         int64       `json:"id,string" dc:"id"`
+	Word       string      `json:"word" dc:"关键词名称"`
+	Weight     int16       `json:"weight" dc:"权重"`
+	DatasetId  int64       `json:"datasetId,string" dc:"数据集ID"`
+	DocumentId int64       `json:"documentId,string" dc:"文档ID"`
+	CreatedAt  *gtime.Time `json:"createdAt" dc:"创建时间"`
+	UpdatedAt  *gtime.Time `json:"updatedAt" dc:"更新时间"`
 }

model/dto/rag_query.go

@@ -0,0 +1,21 @@
+package dto
+
+import (
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+// RAGQueryReq RAG查询请求
+type RAGQueryReq struct {
+	g.Meta `path:"/ragQuery" method:"post" tags:"RAG查询" summary:"执行RAG查询" dc:"执行RAG查询"`
+
+	Content    string  `json:"content" v:"required#查询内容不能为空" dc:"用户问题"`
+	DatasetIds []int64 `json:"datasetIds" dc:"数据集ID"`
+	TopK       int     `json:"topK" d:"5" dc:"检索topK，默认5"`
+}
+
+// RAGQueryRes RAG查询响应
+type RAGQueryRes struct {
+	Answer    string   `json:"answer" dc:"生成的答案"`
+	DatasetId string   `json:"datasetId" dc:"使用的数据集ID"`
+	Sources   []string `json:"sources" dc:"参考来源"`
+}

model/dto/task.go

@@ -0,0 +1,65 @@
+package dto
+
+import (
+	"rag/common/task"
+)
+
+// WriteTaskProgressReq 写入任务进度请求
+type WriteTaskProgressReq struct {
+	TaskType task.TaskType   `json:"taskType" dc:"任务类型"`
+	Status   task.TaskStatus `json:"status" dc:"任务状态"`
+	TaskId   int64           `json:"taskId" dc:"任务ID"`
+	Remark   string          `json:"remark" dc:"备注"`
+}
+
+// CreateTaskReq 创建任务请求
+type CreateTaskReq struct {
+	TaskType task.TaskType   `json:"taskType" dc:"任务类型"`
+	Status   task.TaskStatus `json:"status" dc:"任务状态"`
+	TaskId   int64           `json:"taskId" dc:"任务ID"`
+	Remark   string          `json:"remark" dc:"备注"`
+}
+
+// UpdateTaskReq 更新任务请求
+type UpdateTaskReq struct {
+	Id     int64           `json:"id" dc:"任务ID"`
+	TaskId int64           `json:"taskId" dc:"任务ID"`
+	Status task.TaskStatus `json:"status" dc:"任务状态"`
+	Remark string          `json:"remark" dc:"备注"`
+}
+
+// DeleteTaskByTaskIdReq 删除任务请求
+type DeleteTaskByTaskIdReq struct {
+	TaskId int64 `json:"taskId" v:"required#任务id不能为空"`
+}
+
+// GetTaskReq 获取任务请求
+type GetTaskReq struct {
+	Id       int64         `json:"id" dc:"任务ID"`
+	TaskId   int64         `json:"taskId" dc:"任务ID"`
+	TaskType task.TaskType `json:"taskType" dc:"任务类型"`
+}
+
+// TaskVO 任务视图对象
+type TaskVO struct {
+	Id             int64             `json:"id" dc:"任务ID"`
+	TaskType       task.TaskType     `json:"taskType" dc:"任务类型"`
+	Status         task.TaskStatus   `json:"status" dc:"任务状态"`
+	Priority       task.TaskPriority `json:"priority" dc:"任务优先级"`
+	ParentTaskID   int64             `json:"parentTaskId" dc:"父任务ID"`
+	TotalItems     int64             `json:"totalItems" dc:"总项数"`
+	ProcessedItems int64             `json:"processedItems" dc:"已处理项数"`
+	Progress       float64           `json:"progress" dc:"进度百分比"`
+	StartTime      *int64            `json:"startTime" dc:"开始时间戳"`
+	EndTime        *int64            `json:"endTime" dc:"结束时间戳"`
+	Duration       int64             `json:"duration" dc:"耗时(毫秒)"`
+	SuccessCount   int64             `json:"successCount" dc:"成功数"`
+	FailCount      int64             `json:"failCount" dc:"失败数"`
+	Executor       string            `json:"executor" dc:"执行器"`
+	DocumentID     int64             `json:"documentId" dc:"文档ID"`
+	Remark         string            `json:"remark" dc:"备注"`
+	Creator        string            `json:"creator" dc:"创建人"`
+	CreatedAt      int64             `json:"createdAt" dc:"创建时间"`
+	Updater        string            `json:"updater" dc:"更新人"`
+	UpdatedAt      int64             `json:"updatedAt" dc:"更新时间"`
+}