From 9ffa576f26402a87abfdd5a341e70057067742e4 Mon Sep 17 00:00:00 2001
From: huminmin <mac@MacBook-Pro.local>
Date: 星期一, 17 八月 2026 16:36:41 +0800
Subject: [PATCH] 修改退货日期数据类型
---
src/api/ai/knowledge/segment/index.ts | 52 +++++++++++++++++++++++++++++++++++++++++-----------
1 files changed, 41 insertions(+), 11 deletions(-)
diff --git a/src/api/ai/knowledge/segment/index.ts b/src/api/ai/knowledge/segment/index.ts
index 572c2d5..258054e 100644
--- a/src/api/ai/knowledge/segment/index.ts
+++ b/src/api/ai/knowledge/segment/index.ts
@@ -1,4 +1,4 @@
-import type { PageParam, PageResult } from '..\..\..\..\packages\effects\request\src';
+import type { PageParam, PageResult } from '@vben/request';
import { requestClient } from '#/api/request';
@@ -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