opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-28 20:17:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix file name

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