feat: kb crud

src/api/model.ts

@@ -1,5 +1,5 @@
 import { GET, POST, DELETE, PUT } from './request';
-import type { ModelSchema, ModelDataSchema } from '@/types/mongoSchema';
+import type { ModelSchema } from '@/types/mongoSchema';
 import type { ModelUpdateParams, ShareModelItem } from '@/types/model';
 import { RequestPaging } from '../types/index';
 import { Obj2Query } from '@/utils/tools';
@@ -31,72 +31,6 @@ export const getModelById = (id: string) => GET<ModelSchema>(`/model/detail?mode
 export const putModelById = (id: string, data: ModelUpdateParams) =>
   PUT(`/model/update?modelId=${id}`, data);
 
-/* 模型 data */
-type GetModelDataListProps = RequestPaging & {
-  modelId: string;
-  searchText: string;
-};
-/**
- * 获取模型的知识库数据
- */
-export const getModelDataList = (props: GetModelDataListProps) =>
-  GET(`/model/data/getModelData?${Obj2Query(props)}`);
-
-/**
- * 获取导出数据（不分页）
- */
-export const getExportDataList = (modelId: string) =>
-  GET<[string, string][]>(`/model/data/exportModelData?modelId=${modelId}`);
-
-/**
- * 获取模型正在拆分数据的数量
- */
-export const getModelSplitDataListLen = (modelId: string) =>
-  GET<{
-    splitDataQueue: number;
-    embeddingQueue: number;
-  }>(`/model/data/getTrainingData?modelId=${modelId}`);
-
-/**
- * 获取 web 页面内容
- */
-export const getWebContent = (url: string) => POST<string>(`/model/data/fetchingUrlData`, { url });
-
-/**
- * 手动输入数据
- */
-export const postModelDataInput = (data: {
-  modelId: string;
-  data: { a: ModelDataSchema['a']; q: ModelDataSchema['q'] }[];
-}) => POST<number>(`/model/data/pushModelDataInput`, data);
-
-/**
- * 拆分数据
- */
-export const postModelDataSplitData = (data: {
-  modelId: string;
-  chunks: string[];
-  prompt: string;
-  mode: 'qa' | 'subsection';
-}) => POST(`/model/data/splitData`, data);
-
-/**
- * json导入数据
- */
-export const postModelDataCsvData = (modelId: string, data: string[][]) =>
-  POST<number>(`/model/data/pushModelDataCsv`, { modelId, data: data });
-
-/**
- * 更新模型数据
- */
-export const putModelDataById = (data: { dataId: string; a: string; q?: string }) =>
-  PUT('/model/data/putModelData', data);
-/**
- * 删除一条模型数据
- */
-export const delOneModelData = (dataId: string) =>
-  DELETE(`/model/data/delModelDataById?dataId=${dataId}`);
-
 /* 共享市场 */
 /**
  * 获取共享市场模型

src/api/plugins/kb.ts

@@ -0,0 +1,73 @@
+import { GET, POST, PUT, DELETE } from '../request';
+import type { KbItemType } from '@/types/plugin';
+import { RequestPaging } from '@/types/index';
+import { SplitTextTypEnum } from '@/constants/plugin';
+import { KbDataItemType } from '@/types/plugin';
+
+export type KbUpdateParams = { id: string; name: string; tags: string; avatar: string };
+
+/* knowledge base */
+export const getKbList = () => GET<KbItemType[]>(`/plugins/kb/list`);
+
+export const postCreateKb = (data: { name: string }) => 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 data */
+type GetKbDataListProps = RequestPaging & {
+  kbId: string;
+  searchText: string;
+};
+export const getKbDataList = (data: GetKbDataListProps) =>
+  POST(`/plugins/kb/data/getDataList`, data);
+
+/**
+ * 获取导出数据（不分页）
+ */
+export const getExportDataList = (kbId: string) =>
+  GET<[string, string][]>(`/plugins/kb/data/exportModelData?kbId=${kbId}`);
+
+/**
+ * 获取模型正在拆分数据的数量
+ */
+export const getTrainingData = (kbId: string) =>
+  GET<{
+    splitDataQueue: number;
+    embeddingQueue: number;
+  }>(`/plugins/kb/data/getTrainingData?kbId=${kbId}`);
+
+/**
+ * 获取 web 页面内容
+ */
+export const getWebContent = (url: string) => POST<string>(`/model/data/fetchingUrlData`, { url });
+
+/**
+ * 直接push数据
+ */
+export const postKbDataFromList = (data: {
+  kbId: string;
+  data: { a: KbDataItemType['a']; q: KbDataItemType['q'] }[];
+}) => POST(`/openapi/kb/pushData`, data);
+
+/**
+ * 更新一条数据
+ */
+export const putKbDataById = (data: { dataId: string; a: string; q?: string }) =>
+  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: `${SplitTextTypEnum}`;
+}) => POST(`/openapi/text/splitText`, data);