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-03 21:26:28 +08:00
parent 46731dc8a1
commit e3092837e4
1 changed files with 9 additions and 4 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,7 +197,9 @@ class DatasourceNode(Node):
storage_key=upload_file.key,
url=upload_file.source_url,
)
variable_pool.add([self._node_id, "file"], file_info)
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.to_dict())
yield StreamCompletedEvent(
node_run_result=NodeRunResult(
@ -205,7 +207,7 @@ class DatasourceNode(Node):
inputs=parameters_for_log,
metadata={WorkflowNodeExecutionMetadataKey.DATASOURCE_INFO: datasource_info},
outputs={
"file": datasource_info,
"file": file_info_dict,
"datasource_type": datasource_type,
},
)
@ -501,6 +503,7 @@ 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)
@ -527,14 +530,16 @@ class DatasourceNode(Node):
tenant_id=self.tenant_id,
)
if file:
variable_pool.add([self._node_id, "file"], file)
file_info_dict = file.to_dict()
file_info_dict["name"] = file.filename
variable_pool.add([self._node_id, "file"], file_info_dict)
yield StreamCompletedEvent(
node_run_result=NodeRunResult(
status=WorkflowNodeExecutionStatus.SUCCEEDED,
inputs=parameters_for_log,
metadata={WorkflowNodeExecutionMetadataKey.DATASOURCE_INFO: datasource_info},
outputs={
"file": file,
"file": file_info_dict,
"datasource_type": datasource_type,
},
)