import type { PageParam, PageResult } from '@vben/request';
|
|
import { requestClient } from '#/api/request';
|
|
export namespace AiKnowledgeDocumentApi {
|
export interface KnowledgeDocument {
|
id: number; // 编号
|
knowledgeId: number; // 知识库编号
|
name: string; // 文档名称
|
url: string; // 文档 URL
|
contentLength: number; // 字符数
|
tokens: number; // token 数
|
segmentMaxTokens: number; // 分片最大 token 数
|
retrievalCount: number; // 召回次数
|
status: number; // 是否启用
|
}
|
|
export interface CreateDocumentParams {
|
knowledgeId: number;
|
name: string;
|
url: string;
|
}
|
|
export interface CreateDocumentListParams {
|
knowledgeId: number;
|
segmentMaxTokens: number;
|
list: Array<{ name: string; url: string }>;
|
}
|
|
export interface UpdateDocumentParams {
|
id: number;
|
name?: string;
|
url?: string;
|
segmentMaxTokens?: number;
|
}
|
|
export interface UpdateDocumentStatusParams {
|
id: number;
|
status: number;
|
}
|
|
export interface DocumentProgress {
|
documentId: number;
|
count: number;
|
embeddingCount: number;
|
}
|
}
|
|
/** 查询知识库文档分页 */
|
export function getKnowledgeDocumentPage(params: PageParam) {
|
return requestClient.get<
|
PageResult<AiKnowledgeDocumentApi.KnowledgeDocument>
|
>('/ai/knowledge/document/page', { params });
|
}
|
|
/** 查询知识库文档详情 */
|
export function getKnowledgeDocument(id: number) {
|
return requestClient.get<AiKnowledgeDocumentApi.KnowledgeDocument>(
|
`/ai/knowledge/document/get?id=${id}`,
|
);
|
}
|
|
/** 新增知识库文档（单个） */
|
export function createKnowledge(data: AiKnowledgeDocumentApi.CreateDocumentParams) {
|
return requestClient.post('/ai/knowledge/document/create', data);
|
}
|
|
/** 新增知识库文档（多个） */
|
export function createKnowledgeDocumentList(
|
data: AiKnowledgeDocumentApi.CreateDocumentListParams,
|
) {
|
return requestClient.post('/ai/knowledge/document/create-list', data);
|
}
|
|
/** 修改知识库文档 */
|
export function updateKnowledgeDocument(
|
data: AiKnowledgeDocumentApi.UpdateDocumentParams,
|
) {
|
return requestClient.put('/ai/knowledge/document/update', data);
|
}
|
|
/** 修改知识库文档状态 */
|
export function updateKnowledgeDocumentStatus(
|
data: AiKnowledgeDocumentApi.UpdateDocumentStatusParams,
|
) {
|
return requestClient.put('/ai/knowledge/document/update-status', data);
|
}
|
|
/** 删除知识库文档 */
|
export function deleteKnowledgeDocument(id: number) {
|
return requestClient.delete(`/ai/knowledge/document/delete?id=${id}`);
|
}
|
|
/** 处理文档（分段+向量化） */
|
export function processDocument(id: number) {
|
return requestClient.post(`/ai/knowledge/document/process/${id}`);
|
}
|
|
/** 查询文档处理进度 */
|
export function getDocumentProgress(documentIds: number[]) {
|
return requestClient.get<AiKnowledgeDocumentApi.DocumentProgress[]>(
|
'/ai/knowledge/document/progress',
|
{ params: { documentIds: documentIds.join(',') } },
|
);
|
}
|
