opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: node type

This commit is contained in:
Yeuoly 2024-03-14 11:58:56 +08:00 committed by takatost
parent fb6e5bf4d5
commit c2ded79cb2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/workflow/nodes/tool/tool_node.py
View File

@ -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'
}