| | |
| | | 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 splitContent(url: string, segmentMaxTokens: number) { |
| | | return requestClient.get('/ai/knowledge/segment/split', { |
| | | params: { url, segmentMaxTokens }, |
| | | }); |
| | | return requestClient.get<AiKnowledgeSegmentApi.SplitResult[]>( |
| | | '/ai/knowledge/segment/split', |
| | | { |
| | | params: { url, segmentMaxTokens }, |
| | | }, |
| | | ); |
| | | } |
| | | |
| | | /** 获取文档处理列表 */ |
| | | export function getKnowledgeSegmentProcessList(documentIds: number[]) { |
| | | return requestClient.get('/ai/knowledge/segment/get-process-list', { |
| | | params: { documentIds: documentIds.join(',') }, |
| | | }); |
| | | return requestClient.get<AiKnowledgeSegmentApi.ProcessInfo[]>( |
| | | '/ai/knowledge/segment/get-process-list', |
| | | { |
| | | params: { documentIds: documentIds.join(',') }, |
| | | }, |
| | | ); |
| | | } |
| | | |
| | | /** 搜索知识库分段 */ |
| | | export function searchKnowledgeSegment(params: any) { |
| | | return requestClient.get('/ai/knowledge/segment/search', { |
| | | /** 搜索知识库分段(RAG 向量检索) */ |
| | | export function searchKnowledgeSegment( |
| | | params: AiKnowledgeSegmentApi.SearchParams, |
| | | ) { |
| | | return requestClient.post<AiKnowledgeSegmentApi.KnowledgeSegment[]>( |
| | | '/ai/knowledge/segment/search', |
| | | params, |
| | | }); |
| | | ); |
| | | } |