diff --git a/packages/service/core/dataset/search/controller.ts b/packages/service/core/dataset/search/controller.ts
index 3554a452a..87350f25d 100644
--- a/packages/service/core/dataset/search/controller.ts
+++ b/packages/service/core/dataset/search/controller.ts
@@ -787,6 +787,7 @@ export const defaultSearchDatasetData = async ({
   ...props
 }: DefaultSearchDatasetDataProps): Promise<SearchDatasetDataResponse> => {
   const query = props.queries[0];
+  const histories = props.histories;
 
   const extensionModel = datasetSearchUsingExtensionQuery
     ? getLLMModel(datasetSearchExtensionModel)
@@ -796,7 +797,8 @@ export const defaultSearchDatasetData = async ({
     await datasetSearchQueryExtension({
       query,
       extensionModel,
-      extensionBg: datasetSearchExtensionBg
+      extensionBg: datasetSearchExtensionBg,
+      histories
     });
 
   const result = await searchDatasetData({