From 145ef794b61bd3b16891aa9599355dd7e6df7452 Mon Sep 17 00:00:00 2001
From: AkaraChen <akarachen@outlook.com>
Date: Thu, 9 Jan 2025 16:08:20 +0800
Subject: [PATCH] fix: switch version not work

---
 .../nodes/_base/components/agent-strategy-selector.tsx      | 6 +++---
 web/app/components/workflow/nodes/agent/use-config.ts       | 5 +++++
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/web/app/components/workflow/nodes/_base/components/agent-strategy-selector.tsx b/web/app/components/workflow/nodes/_base/components/agent-strategy-selector.tsx
index 221ce479d0..a96654c4d9 100644
--- a/web/app/components/workflow/nodes/_base/components/agent-strategy-selector.tsx
+++ b/web/app/components/workflow/nodes/_base/components/agent-strategy-selector.tsx
@@ -99,7 +99,7 @@ export const AgentStrategySelector = memo((props: AgentStrategySelectorProps) =>
     if (!list) return []
     return list.filter(tool => tool.name.toLowerCase().includes(query.toLowerCase()))
   }, [query, list])
-  const { strategyStatus } = useStrategyInfo(
+  const { strategyStatus, refetch: refetchStrategyInfo } = useStrategyInfo(
     value?.agent_strategy_provider_name,
     value?.agent_strategy_name,
   )
@@ -178,13 +178,13 @@ export const AgentStrategySelector = memo((props: AgentStrategySelectorProps) =>
               : <RiArrowDownSLine className='size-4 text-text-tertiary' />
           }
           {showSwitchVersion && <SwitchPluginVersion
-            uniqueIdentifier={'langgenius/openai:12'}
+            uniqueIdentifier={value.plugin_unique_identifier}
             tooltip={<ToolTipContent
               title={t('workflow.nodes.agent.unsupportedStrategy')}>
               {t('workflow.nodes.agent.strategyNotFoundDescAndSwitchVersion')}
             </ToolTipContent>}
             onChange={() => {
-              // TODO: refresh all strategies
+              refetchStrategyInfo()
             }}
           />}
         </div>
diff --git a/web/app/components/workflow/nodes/agent/use-config.ts b/web/app/components/workflow/nodes/agent/use-config.ts
index 64a666d82f..88c793723f 100644
--- a/web/app/components/workflow/nodes/agent/use-config.ts
+++ b/web/app/components/workflow/nodes/agent/use-config.ts
@@ -49,10 +49,15 @@ export const useStrategyInfo = (
       isExistInPlugin: strategyExist,
     }
   }, [strategy, marketplace, strategyProvider.isError, strategyProvider.isLoading])
+  const refetch = useCallback(() => {
+    strategyProvider.refetch()
+    marketplace.refetch()
+  }, [marketplace, strategyProvider])
   return {
     strategyProvider,
     strategy,
     strategyStatus,
+    refetch,
   }
 }