refactor: 重构文档处理流程和任务管理
This commit is contained in:
@@ -48,7 +48,7 @@ func (c *document) List(ctx context.Context, req *dto.ListDocumentReq) (res *dto
|
||||
}
|
||||
|
||||
// Process 处理文件(向量化)
|
||||
func (c *document) Process(ctx context.Context, req *dto.ProcessDocumentReq) (res *dto.ProcessDocumentRes, err error) {
|
||||
res, err = service.Document.Process(ctx, req)
|
||||
func (c *document) Process(ctx context.Context, req *dto.ProcessDocumentReq) (res *beans.ResponseEmpty, err error) {
|
||||
err = service.Document.Process(ctx, req)
|
||||
return
|
||||
}
|
||||
|
||||
17
controller/rag_query.go
Normal file
17
controller/rag_query.go
Normal file
@@ -0,0 +1,17 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"context"
|
||||
"rag/model/dto"
|
||||
"rag/service"
|
||||
)
|
||||
|
||||
type ragQuery struct{}
|
||||
|
||||
var RAGQuery = new(ragQuery)
|
||||
|
||||
// Query 执行RAG查询
|
||||
func (c *ragQuery) Query(ctx context.Context, req *dto.RAGQueryReq) (res *dto.RAGQueryRes, err error) {
|
||||
res, err = service.RAGQuery.Query(ctx, req)
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user