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

fix file name

This commit is contained in:
jyong 2025-09-04 10:36:44 +08:00
parent 5114569017
commit 647af6fc69
1 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -197,9 +197,7 @@ class DatasourceNode(Node):
storage_key=upload_file.key,
url=upload_file.source_url,
)
file_info_dict = file_info.to_dict()
file_info_dict["name"] = file_info.filename
variable_pool.add([self._node_id, "file"], file_info_dict)
variable_pool.add([self._node_id, "file"], file_info)
# variable_pool.add([self.node_id, "file"], file_info.to_dict())
yield StreamCompletedEvent(
node_run_result=NodeRunResult(
@ -207,7 +205,7 @@ class DatasourceNode(Node):
inputs=parameters_for_log,
metadata={WorkflowNodeExecutionMetadataKey.DATASOURCE_INFO: datasource_info},
outputs={
"file": file_info_dict,
"file": file_info,
"datasource_type": datasource_type,
},
)
@ -503,7 +501,6 @@ class DatasourceNode(Node):
conversation_id=None,
)
file = None
file_info_dict = None
for message in message_stream:
if message.type == DatasourceMessage.MessageType.BINARY_LINK:
assert isinstance(message.message, DatasourceMessage.TextMessage)
@ -530,16 +527,14 @@ class DatasourceNode(Node):
tenant_id=self.tenant_id,
)
if file:
file_info_dict = file.to_dict()
file_info_dict["name"] = file.filename
variable_pool.add([self._node_id, "file"], file_info_dict)
variable_pool.add([self._node_id, "file"], file)
yield StreamCompletedEvent(
node_run_result=NodeRunResult(
status=WorkflowNodeExecutionStatus.SUCCEEDED,
inputs=parameters_for_log,
metadata={WorkflowNodeExecutionMetadataKey.DATASOURCE_INFO: datasource_info},
outputs={
"file": file_info_dict,
"file": file,
"datasource_type": datasource_type,
},
)