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

fix: tool variable selectors

This commit is contained in:
Yeuoly 2024-04-01 15:42:00 +08:00
parent 072967a1d3
commit 86a32517e5
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -7,7 +7,6 @@ from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool_engine import ToolEngine
from core.tools.tool_manager import ToolManager
from core.tools.utils.message_transformer import ToolFileMessageTransformer
from core.workflow.entities.base_node_data_entities import BaseNodeData
from core.workflow.entities.node_entities import NodeRunMetadataKey, NodeRunResult, NodeType
from core.workflow.entities.variable_pool import VariablePool
from core.workflow.nodes.base_node import BaseNode
@ -178,10 +177,21 @@ 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
:param node_data: node data
:return:
"""
return {}
result = {}
for parameter in node_data.tool_parameters:
if parameter.value_type == 'variable':
if isinstance(parameter.variable_value, str):
parser = VariableTemplateParser(parameter.variable_value)
variable_selectors = parser.extract_variable_selectors()
for selector in variable_selectors:
result[selector.variable] = selector.value_selector
elif isinstance(parameter.variable_value, list):
result[parameter.parameter_name] = parameter.variable_value
return result