From c2ded79cb2bd752866b42ca8b0a9640da1be9e66 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Thu, 14 Mar 2024 11:58:56 +0800 Subject: [PATCH] fix: node type --- api/core/workflow/nodes/tool/tool_node.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/core/workflow/nodes/tool/tool_node.py b/api/core/workflow/nodes/tool/tool_node.py index c62e025e75..89c8389085 100644 --- a/api/core/workflow/nodes/tool/tool_node.py +++ b/api/core/workflow/nodes/tool/tool_node.py @@ -136,12 +136,12 @@ class ToolNode(BaseNode): @classmethod - def _extract_variable_selector_to_variable_mapping(cls, node_data: BaseNodeData) -> dict[str, list[str]]: + def _extract_variable_selector_to_variable_mapping(cls, node_data: ToolNodeData) -> dict[str, list[str]]: """ Extract variable selector to variable mapping """ return { k.variable: k.value_selector - for k in cast(ToolNodeData, node_data).tool_parameters + for k in node_data.tool_parameters if k.variable_type == 'selector' }