src/api/ai/knowledge/segment/index.ts
@@ -14,6 +14,27 @@
    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;
  }
}
@@ -61,21 +82,30 @@
/** 切片内容 */
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,
  });
  );
}