youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-02 08:28:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix variable_pool

Browse Source
This commit is contained in:
jyong
2025-08-13 17:38:14 +08:00
parent 3797416fe0
commit 72ea3b4d01
3 changed files with 21 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
api/core/workflow/nodes/datasource/datasource_node.py
View File

@ -204,16 +204,7 @@ class DatasourceNode(BaseNode):
size=upload_file.size,
storage_key=upload_file.key,
)
variable_pool.add([self.node_id, "file"], [file_info])
for key, value in datasource_info.items():
# construct new key list
new_key_list = ["file", key]
self._append_variables_recursively(
variable_pool=variable_pool,
node_id=self.node_id,
variable_key_list=new_key_list,
variable_value=value,
)
variable_pool.add([self.node_id, "file"], file_info.to_dict())
yield RunCompletedEvent(
run_result=NodeRunResult(
status=WorkflowNodeExecutionStatus.SUCCEEDED,
@ -307,7 +298,7 @@ class DatasourceNode(BaseNode):
:param variable_value: variable value
:return:
"""
variable_pool.add([node_id] + variable_key_list, variable_value)
variable_pool.add([node_id] + [".".join(variable_key_list)], variable_value)
# if variable_value is a dict, then recursively append variables
if isinstance(variable_value, dict):
@ -533,16 +524,8 @@ class DatasourceNode(BaseNode):
mapping=mapping,
tenant_id=self.tenant_id,
)
variable_pool.add([self.node_id, "file"], [file])
for key, value in datasource_info.items():
# construct new key list
new_key_list = ["file", key]
self._append_variables_recursively(
variable_pool=variable_pool,
node_id=self.node_id,
variable_key_list=new_key_list,
variable_value=value,
)
if file:
variable_pool.add([self.node_id, "file"], file.to_dict())
yield RunCompletedEvent(
run_result=NodeRunResult(
status=WorkflowNodeExecutionStatus.SUCCEEDED,