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

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:"参考来源"`
+}