From a876baf0a9cc46f03e8e67af4dfebb2398309683 Mon Sep 17 00:00:00 2001
From: majian <46617237@qq.com>
Date: Thu, 25 Jul 2024 11:15:54 +0800
Subject: [PATCH] Resolve variable type parameter error (#6646)

---
 api/core/workflow/nodes/tool/tool_node.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/api/core/workflow/nodes/tool/tool_node.py b/api/core/workflow/nodes/tool/tool_node.py
index 1bd126f842..238477117d 100644
--- a/api/core/workflow/nodes/tool/tool_node.py
+++ b/api/core/workflow/nodes/tool/tool_node.py
@@ -125,11 +125,15 @@ class ToolNode(BaseNode):
                 ]
             else:
                 tool_input = node_data.tool_parameters[parameter_name]
-                segment_group = parser.convert_template(
-                    template=str(tool_input.value),
-                    variable_pool=variable_pool,
-                )
-                result[parameter_name] = segment_group.log if for_log else segment_group.text
+                if tool_input.type == 'variable':
+                    parameter_value = variable_pool.get(tool_input.value).value
+                else:
+                    segment_group = parser.convert_template(
+                        template=str(tool_input.value),
+                        variable_pool=variable_pool,
+                    )
+                    parameter_value = segment_group.log if for_log else segment_group.text
+                result[parameter_name] = parameter_value
 
         return result