增加知识库模块

This commit is contained in:
WUSIJIAN
2026-02-02 14:04:37 +08:00
parent 0dc3eceace
commit 1df6d24be5
8 changed files with 2135 additions and 0 deletions

View 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,
});
}

View 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 },
});
}