From 0552b1695a56a9befb1028f243c682890d62a22b Mon Sep 17 00:00:00 2001
From: Wang Qi <wangq8@outlook.com>
Date: Fri, 8 May 2026 20:20:09 +0800
Subject: [PATCH] Fix UI search multiple datasets (#14689)

### What problem does this PR solve?

Fix UI search multiple datasets

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 web/src/services/knowledge-service.ts | 11 ++++++++---
 web/src/utils/api.ts                  |  3 +--
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/web/src/services/knowledge-service.ts b/web/src/services/knowledge-service.ts
index ae5cc889ea..58b5f468d3 100644
--- a/web/src/services/knowledge-service.ts
+++ b/web/src/services/knowledge-service.ts
@@ -147,14 +147,19 @@ const getAvailableParam = (available?: number) => {
 
 const chunkService = {
   retrievalTest: async (params: Record<string, any>) => {
-    const datasetId = getDatasetId(params);
+    const datasetId = params.dataset_id || params.kb_id || params.knowledge_id;
     if (!datasetId) {
       throw new Error(
         'dataset_id (or kb_id/knowledge_id) is required for retrievalTest',
       );
     }
-    return request.post(api.retrievalTest(datasetId), {
-      data: params,
+    const datasetIds = Array.isArray(datasetId) ? datasetId : [datasetId];
+    const rest = { ...params };
+    delete rest.dataset_id;
+    delete rest.kb_id;
+    delete rest.knowledge_id;
+    return request.post(api.retrievalTest, {
+      data: { ...rest, dataset_ids: datasetIds },
     });
   },
   chunkList: async (params: Record<string, any>) => {
diff --git a/web/src/utils/api.ts b/web/src/utils/api.ts
index b74d169611..7c6307bc42 100644
--- a/web/src/utils/api.ts
+++ b/web/src/utils/api.ts
@@ -109,8 +109,7 @@ export default {
     `${restAPIv1}/datasets/${datasetId}/documents/${documentId}/chunks`,
   chunkDetail: (datasetId: string, documentId: string, chunkId: string) =>
     `${restAPIv1}/datasets/${datasetId}/documents/${documentId}/chunks/${chunkId}`,
-  retrievalTest: (datasetId: string) =>
-    `${restAPIv1}/datasets/${datasetId}/search`,
+  retrievalTest: `${restAPIv1}/datasets/search`,
 
   // document
   getDocumentList: (datasetId: string) =>