Explorar el Código

Resolve variable type parameter error (#6646)

majian hace 9 meses
padre
commit
a876baf0a9
Se han modificado 1 ficheros con 9 adiciones y 5 borrados
  1. 9 5
      api/core/workflow/nodes/tool/tool_node.py

+ 9 - 5
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