增加知识库模块
This commit is contained in:
77
src/api/knowledge/dataset/index.ts
Normal file
77
src/api/knowledge/dataset/index.ts
Normal file
@@ -0,0 +1,77 @@
|
||||
import { newService } from '/@/utils/request';
|
||||
|
||||
// 数据集查询参数
|
||||
export interface DatasetQueryParams {
|
||||
keyword?: string;
|
||||
status?: string;
|
||||
pageNum: number;
|
||||
pageSize: number;
|
||||
}
|
||||
|
||||
// 数据集信息
|
||||
export interface DatasetInfo {
|
||||
id?: string;
|
||||
name: string;
|
||||
description?: string;
|
||||
type: string; // text, qa, table
|
||||
documentCount?: number;
|
||||
charCount?: number;
|
||||
status: string; // enable, disable
|
||||
embeddingModel?: string;
|
||||
createdAt?: string;
|
||||
updatedAt?: string;
|
||||
}
|
||||
|
||||
// 获取数据集列表
|
||||
export function listDatasets(params: DatasetQueryParams) {
|
||||
return newService({
|
||||
url: '/knowledge/dataset/list',
|
||||
method: 'get',
|
||||
params,
|
||||
});
|
||||
}
|
||||
|
||||
// 获取数据集详情
|
||||
export function getDataset(id: string) {
|
||||
return newService({
|
||||
url: '/knowledge/dataset/detail',
|
||||
method: 'get',
|
||||
params: { id },
|
||||
});
|
||||
}
|
||||
|
||||
// 创建数据集
|
||||
export function createDataset(data: DatasetInfo) {
|
||||
return newService({
|
||||
url: '/knowledge/dataset/create',
|
||||
method: 'post',
|
||||
data,
|
||||
});
|
||||
}
|
||||
|
||||
// 更新数据集
|
||||
export function updateDataset(data: DatasetInfo) {
|
||||
return newService({
|
||||
url: '/knowledge/dataset/update',
|
||||
method: 'put',
|
||||
data,
|
||||
});
|
||||
}
|
||||
|
||||
// 删除数据集
|
||||
export function deleteDataset(id: string) {
|
||||
return newService({
|
||||
url: '/knowledge/dataset/delete',
|
||||
method: 'delete',
|
||||
params: { id },
|
||||
});
|
||||
}
|
||||
|
||||
// 更新数据集状态
|
||||
export function updateDatasetStatus(data: { id: string; status: string }) {
|
||||
return newService({
|
||||
url: '/knowledge/dataset/updateStatus',
|
||||
method: 'put',
|
||||
data,
|
||||
});
|
||||
}
|
||||
134
src/api/knowledge/document/index.ts
Normal file
134
src/api/knowledge/document/index.ts
Normal file
@@ -0,0 +1,134 @@
|
||||
import { newService } from '/@/utils/request';
|
||||
|
||||
// 文档查询参数
|
||||
export interface DocumentQueryParams {
|
||||
keyword?: string;
|
||||
datasetId?: string;
|
||||
status?: string;
|
||||
fileType?: string;
|
||||
pageNum: number;
|
||||
pageSize: number;
|
||||
}
|
||||
|
||||
// 文档信息
|
||||
export interface DocumentInfo {
|
||||
id?: string;
|
||||
name: string;
|
||||
datasetId: string;
|
||||
datasetName?: string;
|
||||
fileType: string; // pdf, docx, txt, md, html
|
||||
fileSize?: number;
|
||||
filePath?: string;
|
||||
charCount?: number;
|
||||
chunkCount?: number;
|
||||
status: string; // pending, processing, completed, failed
|
||||
indexStatus?: string; // not_indexed, indexing, indexed, failed
|
||||
errorMessage?: string;
|
||||
createdAt?: string;
|
||||
updatedAt?: string;
|
||||
}
|
||||
|
||||
// 文档分段信息
|
||||
export interface DocumentChunk {
|
||||
id: string;
|
||||
documentId: string;
|
||||
content: string;
|
||||
chunkIndex: number;
|
||||
charCount: number;
|
||||
tokenCount?: number;
|
||||
embedding?: number[];
|
||||
createdAt?: string;
|
||||
}
|
||||
|
||||
// 获取文档列表
|
||||
export function listDocuments(params: DocumentQueryParams) {
|
||||
return newService({
|
||||
url: '/knowledge/document/list',
|
||||
method: 'get',
|
||||
params,
|
||||
});
|
||||
}
|
||||
|
||||
// 获取文档详情
|
||||
export function getDocument(id: string) {
|
||||
return newService({
|
||||
url: '/knowledge/document/detail',
|
||||
method: 'get',
|
||||
params: { id },
|
||||
});
|
||||
}
|
||||
|
||||
// 上传文档
|
||||
export function uploadDocument(data: FormData) {
|
||||
return newService({
|
||||
url: '/knowledge/document/upload',
|
||||
method: 'post',
|
||||
data,
|
||||
headers: {
|
||||
'Content-Type': 'multipart/form-data',
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
// 删除文档
|
||||
export function deleteDocument(id: string) {
|
||||
return newService({
|
||||
url: '/knowledge/document/delete',
|
||||
method: 'delete',
|
||||
params: { id },
|
||||
});
|
||||
}
|
||||
|
||||
// 批量删除文档
|
||||
export function batchDeleteDocuments(ids: string[]) {
|
||||
return newService({
|
||||
url: '/knowledge/document/batchDelete',
|
||||
method: 'delete',
|
||||
data: { ids },
|
||||
});
|
||||
}
|
||||
|
||||
// 重新处理文档
|
||||
export function reprocessDocument(id: string) {
|
||||
return newService({
|
||||
url: '/knowledge/document/reprocess',
|
||||
method: 'post',
|
||||
params: { id },
|
||||
});
|
||||
}
|
||||
|
||||
// 获取文档分段列表
|
||||
export function listDocumentChunks(params: { documentId: string; pageNum: number; pageSize: number }) {
|
||||
return newService({
|
||||
url: '/knowledge/document/chunks',
|
||||
method: 'get',
|
||||
params,
|
||||
});
|
||||
}
|
||||
|
||||
// 更新文档分段
|
||||
export function updateDocumentChunk(data: { id: string; content: string }) {
|
||||
return newService({
|
||||
url: '/knowledge/document/chunk/update',
|
||||
method: 'put',
|
||||
data,
|
||||
});
|
||||
}
|
||||
|
||||
// 删除文档分段
|
||||
export function deleteDocumentChunk(id: string) {
|
||||
return newService({
|
||||
url: '/knowledge/document/chunk/delete',
|
||||
method: 'delete',
|
||||
params: { id },
|
||||
});
|
||||
}
|
||||
|
||||
// 预览文档内容
|
||||
export function previewDocument(id: string) {
|
||||
return newService({
|
||||
url: '/knowledge/document/preview',
|
||||
method: 'get',
|
||||
params: { id },
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user