From 020a05d8d31565fc6fe2b0f76645cdabe8374e1c Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期一, 03 八月 2026 17:56:21 +0800
Subject: [PATCH] refactor(srm): 优化招标项目功能实现
---
src/api/ai/knowledge/segment/index.ts | 50 ++++++++++++++++++++++++++++++++++++++++----------
1 files changed, 40 insertions(+), 10 deletions(-)
diff --git a/src/api/ai/knowledge/segment/index.ts b/src/api/ai/knowledge/segment/index.ts
index 577ce0b..258054e 100644
--- a/src/api/ai/knowledge/segment/index.ts
+++ b/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,
- });
+ );
}
--
Gitblit v1.9.3