feat: rag初始版

controller/document.go

@@ -0,0 +1,54 @@
+package controller
+
+import (
+	"context"
+
+	"rag/model/dto"
+	"rag/service"
+
+	"gitea.com/red-future/common/beans"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+type document struct{}
+
+var Document = new(document)
+
+// Create 创建文件
+func (c *document) Create(ctx context.Context, req *dto.CreateDocumentReq) (res *dto.CreateDocumentRes, err error) {
+	res, err = service.Document.Create(ctx, req)
+	return
+}
+
+// Update 更新文件
+func (c *document) Update(ctx context.Context, req *dto.UpdateDocumentReq) (res *beans.ResponseEmpty, err error) {
+	err = service.Document.Update(ctx, req)
+	return
+}
+
+// Delete 删除文件
+func (c *document) Delete(ctx context.Context, req *dto.DeleteDocumentReq) (res *beans.ResponseEmpty, err error) {
+	err = service.Document.Delete(ctx, req)
+	return
+}
+
+// Get 获取文件详情
+func (c *document) Get(ctx context.Context, req *dto.GetDocumentReq) (res *dto.DocumentVO, err error) {
+	res, err = service.Document.Get(ctx, req)
+	return
+}
+
+// List 文件列表
+func (c *document) List(ctx context.Context, req *dto.ListDocumentReq) (res *dto.ListDocumentRes, err error) {
+	if !g.IsEmpty(req.Page) {
+		req.Page = &beans.Page{PageNum: 1, PageSize: 20}
+	}
+	res, err = service.Document.List(ctx, req)
+	return
+}
+
+// Process 处理文件(向量化)
+func (c *document) Process(ctx context.Context, req *dto.ProcessDocumentReq) (res *dto.ProcessDocumentRes, err error) {
+	res, err = service.Document.Process(ctx, req)
+	return
+}