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) =>