import type { PageParam, PageResult } from '@vben/request';
|
|
import { requestClient } from '#/api/request';
|
|
export namespace AiKnowledgeSegmentApi {
|
export interface KnowledgeSegment {
|
id: number; // 编号
|
documentId: number; // 文档编号
|
knowledgeId: number; // 知识库编号
|
vectorId: string; // 向量库编号
|
content: string; // 切片内容
|
contentLength: number; // 切片内容长度
|
tokens: number; // token 数量
|
retrievalCount: number; // 召回次数
|
status: number; // 文档状态
|
createTime: number; // 创建时间
|
score?: number; // 相似度分数（0~1），仅搜索接口返回
|
documentName?: string; // 来源文档名称，仅搜索接口返回
|
}
|
|
export interface SearchParams {
|
knowledgeId: number;
|
content: string;
|
topK?: number;
|
similarityThreshold?: number;
|
}
|
|
export interface SplitResult {
|
content: string;
|
contentLength: number;
|
tokens: number;
|
}
|
|
export interface ProcessInfo {
|
documentId: number;
|
count: number;
|
embeddingCount: number;
|
}
|
}
|
|
/** 查询知识库分段分页 */
|
export function getKnowledgeSegmentPage(params: PageParam) {
|
return requestClient.get<PageResult<AiKnowledgeSegmentApi.KnowledgeSegment>>(
|
'/ai/knowledge/segment/page',
|
{ params },
|
);
|
}
|
|
/** 查询知识库分段详情 */
|
export function getKnowledgeSegment(id: number) {
|
return requestClient.get<AiKnowledgeSegmentApi.KnowledgeSegment>(
|
`/ai/knowledge/segment/get?id=${id}`,
|
);
|
}
|
|
/** 新增知识库分段 */
|
export function createKnowledgeSegment(
|
data: AiKnowledgeSegmentApi.KnowledgeSegment,
|
) {
|
return requestClient.post('/ai/knowledge/segment/create', data);
|
}
|
|
/** 修改知识库分段 */
|
export function updateKnowledgeSegment(
|
data: AiKnowledgeSegmentApi.KnowledgeSegment,
|
) {
|
return requestClient.put('/ai/knowledge/segment/update', data);
|
}
|
|
/** 修改知识库分段状态 */
|
export function updateKnowledgeSegmentStatus(id: number, status: number) {
|
return requestClient.put('/ai/knowledge/segment/update-status', {
|
id,
|
status,
|
});
|
}
|
|
/** 删除知识库分段 */
|
export function deleteKnowledgeSegment(id: number) {
|
return requestClient.delete(`/ai/knowledge/segment/delete?id=${id}`);
|
}
|
|
/** 切片内容 */
|
export function splitContent(url: string, segmentMaxTokens: number) {
|
return requestClient.get<AiKnowledgeSegmentApi.SplitResult[]>(
|
'/ai/knowledge/segment/split',
|
{
|
params: { url, segmentMaxTokens },
|
},
|
);
|
}
|
|
/** 获取文档处理列表 */
|
export function getKnowledgeSegmentProcessList(documentIds: number[]) {
|
return requestClient.get<AiKnowledgeSegmentApi.ProcessInfo[]>(
|
'/ai/knowledge/segment/get-process-list',
|
{
|
params: { documentIds: documentIds.join(',') },
|
},
|
);
|
}
|
|
/** 搜索知识库分段（RAG 向量检索） */
|
export function searchKnowledgeSegment(
|
params: AiKnowledgeSegmentApi.SearchParams,
|
) {
|
return requestClient.post<AiKnowledgeSegmentApi.KnowledgeSegment[]>(
|
'/ai/knowledge/segment/search',
|
params,
|
);
|
}
|
