From d134b66bd89e79353407d7490cf77813f5305e89 Mon Sep 17 00:00:00 2001
From: Henry Heng <henryheng@flowiseai.com>
Date: Wed, 4 Jun 2025 17:54:36 +0100
Subject: [PATCH] Bugfix/tool node selected tool (#4579)

* fix tool node selected tool

* fix tool node config

* update node to display tool node icons
---
 packages/components/nodes/agentflow/Tool/Tool.ts     | 5 ++++-
 packages/ui/src/views/agentflowsv2/AgentFlowNode.jsx | 7 +++++--
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/packages/components/nodes/agentflow/Tool/Tool.ts b/packages/components/nodes/agentflow/Tool/Tool.ts
index c37b1e01a..d8365b11c 100644
--- a/packages/components/nodes/agentflow/Tool/Tool.ts
+++ b/packages/components/nodes/agentflow/Tool/Tool.ts
@@ -187,7 +187,10 @@ class Tool_Agentflow implements INode {
 
     async run(nodeData: INodeData, input: string, options: ICommonObject): Promise<any> {
         const selectedTool = (nodeData.inputs?.selectedTool as string) || (nodeData.inputs?.toolAgentflowSelectedTool as string)
-        const selectedToolConfig = nodeData.inputs?.selectedToolConfig as ICommonObject
+        const selectedToolConfig =
+            (nodeData?.inputs?.selectedToolConfig as ICommonObject) ||
+            (nodeData?.inputs?.toolAgentflowSelectedToolConfig as ICommonObject) ||
+            {}
 
         const toolInputArgs = nodeData.inputs?.toolInputArgs as IToolInputArgs[]
         const _toolUpdateState = nodeData.inputs?.toolUpdateState
diff --git a/packages/ui/src/views/agentflowsv2/AgentFlowNode.jsx b/packages/ui/src/views/agentflowsv2/AgentFlowNode.jsx
index d52639d42..a5d3fbbdd 100644
--- a/packages/ui/src/views/agentflowsv2/AgentFlowNode.jsx
+++ b/packages/ui/src/views/agentflowsv2/AgentFlowNode.jsx
@@ -401,8 +401,11 @@ const AgentFlowNode = ({ data }) => {
                                     { tools: data.inputs?.llmTools, toolProperty: 'llmSelectedTool' },
                                     { tools: data.inputs?.agentTools, toolProperty: 'agentSelectedTool' },
                                     {
-                                        tools: data.inputs?.selectedTool ? [{ selectedTool: data.inputs?.selectedTool }] : [],
-                                        toolProperty: 'selectedTool'
+                                        tools:
+                                            data.inputs?.selectedTool ?? data.inputs?.toolAgentflowSelectedTool
+                                                ? [{ selectedTool: data.inputs?.selectedTool ?? data.inputs?.toolAgentflowSelectedTool }]
+                                                : [],
+                                        toolProperty: ['selectedTool', 'toolAgentflowSelectedTool']
                                     },
                                     { tools: data.inputs?.agentKnowledgeVSEmbeddings, toolProperty: ['vectorStore', 'embeddingModel'] }
                                 ]