141 lines
3.1 KiB
TypeScript
141 lines
3.1 KiB
TypeScript
import request, { fileUploadRequest, fileDownloadRequest } from '@/utils/request'
|
|
|
|
export const createDataset = (data: any) => {
|
|
return request({
|
|
url: `/brichat-service/datasetManage/create`,
|
|
method: 'post',
|
|
data
|
|
})
|
|
}
|
|
export const updateDataset = (data: any) => {
|
|
return request({
|
|
url: `/brichat-service/datasetManage/update`,
|
|
method: 'post',
|
|
data
|
|
})
|
|
}
|
|
|
|
|
|
export const deleteDataset = (id: string) => {
|
|
return request({
|
|
url: `/brichat-service/datasetManage/${id}`,
|
|
method: 'delete',
|
|
})
|
|
}
|
|
|
|
export const getUsersAndDatasets = () => {
|
|
return request({
|
|
url: '/brichat-service/datasetManage/usersLinkDataset',
|
|
method: 'get',
|
|
})
|
|
}
|
|
|
|
export const getCurrUserDatasets = (userId: number, pageNo: number = 1, pageSize: number = 9) => {
|
|
return request({
|
|
url: '/brichat-service/datasetManage/currUserDatasets',
|
|
method: 'get',
|
|
params: {
|
|
userId,
|
|
pageNo,
|
|
pageSize
|
|
}
|
|
})
|
|
}
|
|
|
|
|
|
export const getAllDatasets = () => {
|
|
return request({
|
|
url: '/brichat-service/datasetManage/allDataset',
|
|
method: 'get',
|
|
})
|
|
}
|
|
|
|
interface UserBindDatasetDto {
|
|
userId: number;
|
|
datasetIds: string[];
|
|
}
|
|
|
|
export const bindDatasets = (data: UserBindDatasetDto) => {
|
|
return request({
|
|
url: '/brichat-service/datasetManage/bindDatasets',
|
|
method: 'post',
|
|
data
|
|
})
|
|
}
|
|
|
|
interface UnbindDatasetParams {
|
|
userId: number;
|
|
datasetId: string;
|
|
}
|
|
|
|
export const unbindDataset = (data: UnbindDatasetParams) => {
|
|
return request({
|
|
url: '/brichat-service/datasetManage/unbindDataset',
|
|
method: 'post',
|
|
params: data
|
|
})
|
|
}
|
|
|
|
|
|
export const getDatasetDocPage = (data: any) => {
|
|
return request({
|
|
url: '/brichat-service/datasetManage/document/page',
|
|
method: 'post',
|
|
data
|
|
})
|
|
}
|
|
|
|
export const deleteDocument = (params: any) => {
|
|
return request({
|
|
url: '/brichat-service/datasetManage/document/delete',
|
|
method: 'delete',
|
|
params
|
|
})
|
|
}
|
|
|
|
interface RenameParams {
|
|
documentId: string
|
|
newName: string
|
|
}
|
|
export const renameDocument = (data: RenameParams) => {
|
|
return request({
|
|
url: '/brichat-service/datasetManage/document/rename',
|
|
method: 'post',
|
|
data
|
|
})
|
|
}
|
|
|
|
export const uploadDocument = (data: FormData) => {
|
|
return fileUploadRequest({
|
|
url: '/brichat-service/datasetManage/document/upload',
|
|
method: 'post',
|
|
data
|
|
})
|
|
}
|
|
|
|
/**
|
|
* 下载文档文件
|
|
* @param documentId 文档ID
|
|
* @param datasetId 数据集ID
|
|
*/
|
|
export const downloadDocument = (documentId: string, datasetId: string) => {
|
|
return fileDownloadRequest({
|
|
url: '/brichat-service/datasetManage/document/download',
|
|
method: 'get',
|
|
params: {
|
|
documentId,
|
|
datasetId
|
|
}
|
|
})
|
|
}
|
|
|
|
export const previewDocumentUrl = (documentId: string, datasetId: string) => {
|
|
return request({
|
|
url: '/brichat-service/datasetManage/document/previewUrl',
|
|
method: 'get',
|
|
params: {
|
|
documentId,
|
|
datasetId
|
|
}
|
|
})
|
|
} |