b_dataset_file.go 1.3 KB

1234567891011121314151617181920212223242526272829303132
  1. package bll
  2. import (
  3. "context"
  4. "yx-dataset-server/app/schema"
  5. )
  6. // IDatasetFile DatasetFile业务逻辑接口
  7. type IDatasetFile interface {
  8. // 查询数据
  9. Query(ctx context.Context, params schema.DatasetFileQueryParam, opts ...schema.DatasetFileQueryOptions) (*schema.DatasetFileQueryResult, error)
  10. // 查询指定数据
  11. Get(ctx context.Context, recordID string, opts ...schema.DatasetFileQueryOptions) (*schema.DatasetFile, error)
  12. // 创建数据
  13. Create(ctx context.Context, item schema.DatasetFile) error
  14. // CreateV2 创建数据(V2版本,直接接收本地文件并上传到RagFlow)
  15. CreateV2(ctx context.Context, param schema.UpdateFileParam) error
  16. // BatchCreate 批量创建数据
  17. BatchCreate(ctx context.Context, files schema.DatasetFiles) error
  18. // 更新数据
  19. Update(ctx context.Context, recordID string, item schema.DatasetFile) error
  20. // 删除数据
  21. Delete(ctx context.Context, recordID string) error
  22. // BatchDelete 批量删除
  23. BatchDelete(ctx context.Context, fileIDs []string) error
  24. // 更新状态
  25. UpdateStatus(ctx context.Context, recordID string, status int) error
  26. // UpdateEnabled 更新启用状态
  27. UpdateEnabled(ctx context.Context, recordID string, status bool) error
  28. // BatchCreate2 批量上传多个文件到指定知识库
  29. BatchCreateV2(ctx context.Context, datasetId string, files []schema.UpdateFileParam) error
  30. }