111 lines
3.6 KiB
TypeScript
111 lines
3.6 KiB
TypeScript
import { GET, POST, PUT, DELETE } from '../request';
|
|
import type {
|
|
DatasetItemType,
|
|
FileInfo,
|
|
KbFileItemType,
|
|
KbItemType,
|
|
KbListItemType,
|
|
KbPathItemType
|
|
} from '@/types/plugin';
|
|
import { TrainingModeEnum } from '@/constants/plugin';
|
|
import {
|
|
Props as PushDataProps,
|
|
Response as PushDateResponse
|
|
} from '@/pages/api/openapi/kb/pushData';
|
|
import {
|
|
Props as SearchTestProps,
|
|
Response as SearchTestResponse
|
|
} from '@/pages/api/openapi/kb/searchTest';
|
|
import { Props as UpdateDataProps } from '@/pages/api/openapi/kb/updateData';
|
|
import type { KbUpdateParams, CreateKbParams, GetKbDataListProps } from '../request/kb';
|
|
import { QuoteItemType } from '@/types/chat';
|
|
import { KbTypeEnum } from '@/constants/kb';
|
|
|
|
/* knowledge base */
|
|
export const getKbList = (data: { parentId?: string; type?: `${KbTypeEnum}` }) =>
|
|
GET<KbListItemType[]>(`/plugins/kb/list`, data);
|
|
export const getAllDataset = () => GET<KbListItemType[]>(`/plugins/kb/allDataset`);
|
|
|
|
export const getKbPaths = (parentId?: string) =>
|
|
GET<KbPathItemType[]>('/plugins/kb/paths', { parentId });
|
|
|
|
export const getKbById = (id: string) => GET<KbItemType>(`/plugins/kb/detail?id=${id}`);
|
|
|
|
export const postCreateKb = (data: CreateKbParams) => POST<string>(`/plugins/kb/create`, data);
|
|
|
|
export const putKbById = (data: KbUpdateParams) => PUT(`/plugins/kb/update`, data);
|
|
|
|
export const delKbById = (id: string) => DELETE(`/plugins/kb/delete?id=${id}`);
|
|
|
|
/* kb file */
|
|
export const getKbFiles = (data: { kbId: string; searchText: string }) =>
|
|
GET<KbFileItemType[]>(`/plugins/kb/file/list`, data);
|
|
export const deleteKbFileById = (params: { fileId: string; kbId: string }) =>
|
|
DELETE(`/plugins/kb/file/delFileByFileId`, params);
|
|
export const getFileInfoById = (fileId: string) =>
|
|
GET<FileInfo>(`/plugins/kb/file/getFileInfo`, { fileId });
|
|
export const delEmptyFiles = (kbId: string) =>
|
|
DELETE(`/plugins/kb/file/deleteEmptyFiles`, { kbId });
|
|
|
|
/* kb data */
|
|
export const getKbDataList = (data: GetKbDataListProps) =>
|
|
POST(`/plugins/kb/data/getDataList`, data);
|
|
|
|
/**
|
|
* 获取导出数据（不分页）
|
|
*/
|
|
export const getExportDataList = (data: { kbId: string; fileId: string }) =>
|
|
GET<[string, string, string][]>(`/plugins/kb/data/exportModelData`, data, {
|
|
timeout: 600000
|
|
});
|
|
|
|
/**
|
|
* 获取模型正在拆分数据的数量
|
|
*/
|
|
export const getTrainingData = (data: { kbId: string; init: boolean }) =>
|
|
POST<{
|
|
qaListLen: number;
|
|
vectorListLen: number;
|
|
}>(`/plugins/kb/data/getTrainingData`, data);
|
|
|
|
/* get length of system training queue */
|
|
export const getTrainingQueueLen = () => GET<number>(`/plugins/kb/data/getQueueLen`);
|
|
|
|
export const getKbDataItemById = (dataId: string) =>
|
|
GET<QuoteItemType>(`/plugins/kb/data/getDataById`, { dataId });
|
|
|
|
/**
|
|
* 直接push数据
|
|
*/
|
|
export const postKbDataFromList = (data: PushDataProps) =>
|
|
POST<PushDateResponse>(`/openapi/kb/pushData`, data);
|
|
|
|
/**
|
|
* insert one data to dataset
|
|
*/
|
|
export const insertData2Kb = (data: { kbId: string; data: DatasetItemType }) =>
|
|
POST<string>(`/plugins/kb/data/insertData`, data);
|
|
|
|
/**
|
|
* 更新一条数据
|
|
*/
|
|
export const putKbDataById = (data: UpdateDataProps) => PUT('/openapi/kb/updateData', data);
|
|
/**
|
|
* 删除一条知识库数据
|
|
*/
|
|
export const delOneKbDataByDataId = (dataId: string) =>
|
|
DELETE(`/openapi/kb/delDataById?dataId=${dataId}`);
|
|
|
|
/**
|
|
* 拆分数据
|
|
*/
|
|
export const postSplitData = (data: {
|
|
kbId: string;
|
|
chunks: string[];
|
|
prompt: string;
|
|
mode: `${TrainingModeEnum}`;
|
|
}) => POST(`/openapi/text/pushData`, data);
|
|
|
|
export const searchText = (data: SearchTestProps) =>
|
|
POST<SearchTestResponse>(`/openapi/kb/searchTest`, data);
|