From 959eedde27ba13576abf2949f9ec7aeec774031b Mon Sep 17 00:00:00 2001
From: twwu <twwu@dify.ai>
Date: Tue, 24 Jun 2025 11:12:02 +0800
Subject: [PATCH] fix: update retrieval method cache on successful data
 retrieval in Step Two

---
 web/app/components/datasets/create/step-three/index.tsx | 4 ++--
 web/app/components/datasets/create/step-two/index.tsx   | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/web/app/components/datasets/create/step-three/index.tsx b/web/app/components/datasets/create/step-three/index.tsx
index cdaa003a23..3b546acb1d 100644
--- a/web/app/components/datasets/create/step-three/index.tsx
+++ b/web/app/components/datasets/create/step-three/index.tsx
@@ -52,8 +52,8 @@ const StepThree = ({ datasetId, datasetName, indexingType, creationCache, retrie
             datasetId={datasetId || creationCache?.dataset?.id || ''}
             batchId={creationCache?.batch || ''}
             documents={creationCache?.documents as FullDocumentDetail[]}
-            indexingType={indexingType || creationCache?.dataset?.indexing_technique}
-            retrievalMethod={retrievalMethod || creationCache?.dataset?.retrieval_model?.search_method}
+            indexingType={creationCache?.dataset?.indexing_technique || indexingType}
+            retrievalMethod={creationCache?.dataset?.retrieval_model?.search_method || retrievalMethod}
           />
         </div>
       </div>
diff --git a/web/app/components/datasets/create/step-two/index.tsx b/web/app/components/datasets/create/step-two/index.tsx
index acc891407f..9a76bc6834 100644
--- a/web/app/components/datasets/create/step-two/index.tsx
+++ b/web/app/components/datasets/create/step-two/index.tsx
@@ -575,6 +575,7 @@ const StepTwo = ({
         onSuccess(data) {
           updateIndexingTypeCache && updateIndexingTypeCache(indexType as string)
           updateResultCache && updateResultCache(data)
+          updateRetrievalMethodCache && updateRetrievalMethodCache(retrievalConfig.search_method as string)
         },
       })
     }