43 lines
1.0 KiB
Go
43 lines
1.0 KiB
Go
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
|
|
}
|