Unverified Commit 19df70ef authored by Yeuoly's avatar Yeuoly

fix: node type

parent 975d0a16
......@@ -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'
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment