38 lines
943 B
Go
38 lines
943 B
Go
package dto
|
|
|
|
import (
|
|
"rag-local/kb/model/entity"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
type ListDatasetReq struct {
|
|
g.Meta `path:"/list" method:"get" tags:"数据集" summary:"数据集列表"`
|
|
}
|
|
|
|
type ListDatasetRes struct {
|
|
List []*entity.Dataset `json:"list"`
|
|
}
|
|
|
|
type SaveDatasetReq struct {
|
|
g.Meta `path:"/save" method:"post" tags:"数据集" summary:"保存数据集"`
|
|
Id int64 `json:"id"`
|
|
Name string `v:"required" json:"name"`
|
|
Description string `json:"description"`
|
|
EmbeddingCfgId int64 `json:"embedding_cfg_id"`
|
|
ChunkSize int `json:"chunk_size"`
|
|
ChunkOverlap int `json:"chunk_overlap"`
|
|
ChunkStrategy string `json:"chunk_strategy"`
|
|
}
|
|
|
|
type SaveDatasetRes struct {
|
|
Id int64 `json:"id"`
|
|
}
|
|
|
|
type DeleteDatasetReq struct {
|
|
g.Meta `path:"/delete" method:"post" tags:"数据集" summary:"删除数据集"`
|
|
Id int64 `v:"required" json:"id"`
|
|
}
|
|
|
|
type DeleteDatasetRes struct{}
|