feat: rag初始版

controller/dataset.go

@@ -0,0 +1,48 @@
+package controller
+
+import (
+	"context"
+
+	"rag/model/dto"
+	"rag/service"
+
+	"gitea.com/red-future/common/beans"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+type dataset struct{}
+
+var Dataset = new(dataset)
+
+// Create 创建数据集
+func (c *dataset) Create(ctx context.Context, req *dto.CreateDatasetReq) (res *dto.CreateDatasetRes, err error) {
+	res, err = service.Dataset.Create(ctx, req)
+	return
+}
+
+// Update 更新数据集
+func (c *dataset) Update(ctx context.Context, req *dto.UpdateDatasetReq) (res *beans.ResponseEmpty, err error) {
+	err = service.Dataset.Update(ctx, req)
+	return
+}
+
+// Delete 删除数据集
+func (c *dataset) Delete(ctx context.Context, req *dto.DeleteDatasetReq) (res *beans.ResponseEmpty, err error) {
+	err = service.Dataset.Delete(ctx, req)
+	return
+}
+
+// List 数据集列表
+func (c *dataset) List(ctx context.Context, req *dto.ListDatasetReq) (res *dto.ListDatasetRes, err error) {
+	if !g.IsEmpty(req.Page) {
+		req.Page = &beans.Page{PageNum: 1, PageSize: 20}
+	}
+	res, err = service.Dataset.List(ctx, req)
+	return
+}
+
+// Search 搜索
+//func (c *dataset) Search(ctx context.Context, req *dto.SearchReq) (res *dto.SearchRes, err error) {
+//	res, err = service.Dataset.Search(ctx, req)
+//	return
+//}