diff --git a/ui/src/api/dataset.ts b/ui/src/api/dataset.ts index ef5f09fb1..caee05921 100644 --- a/ui/src/api/dataset.ts +++ b/ui/src/api/dataset.ts @@ -32,9 +32,49 @@ const delDateset: (dataset_id: String) => Promise> = (dataset_id return del(`${prefix}/${dataset_id}`) } +/** + * 创建数据集 + * @param 参数 + * { + "name": "string", + "desc": "string", + "documents": [ + { + "name": "string", + "paragraphs": [ + { + "content": "string", + "title": "string", + "is_active": true, + "problem_list": [ + { + "id": "string", + "content": "string" + } + ] + } + ] + } + ] +} + */ +const postDateset: (data: any) => Promise> = (data) => { + return post(`${prefix}`, data) +} + +/** + * 分段预览(上传文档) + * @param 参数 file:file,limit:number,patterns:array,with_filter:boolean + */ +const postSplitDocument: (data: any) => Promise> = (data) => { + console.log(data) + return post(`${prefix}/document/split`, data) +} export default { getDateset, getAllDateset, - delDateset + delDateset, + postDateset, + postSplitDocument } diff --git a/ui/src/api/type/dataset.ts b/ui/src/api/type/dataset.ts index f8247e1ac..becec96d7 100644 --- a/ui/src/api/type/dataset.ts +++ b/ui/src/api/type/dataset.ts @@ -4,4 +4,10 @@ interface datasetListRequest { search_text: string } -export type { datasetListRequest } +interface datasetData { + name: String + desc: String + documents?: Array +} + +export type { datasetListRequest, datasetData } diff --git a/ui/src/components/card-add/index.vue b/ui/src/components/card-add/index.vue index c6f7673be..678322484 100644 --- a/ui/src/components/card-add/index.vue +++ b/ui/src/components/card-add/index.vue @@ -1,5 +1,5 @@