diff --git a/api/core/workflow/nodes/datasource/datasource_node.py b/api/core/workflow/nodes/datasource/datasource_node.py index 37b4974ee3..b206fe479c 100644 --- a/api/core/workflow/nodes/datasource/datasource_node.py +++ b/api/core/workflow/nodes/datasource/datasource_node.py @@ -202,6 +202,7 @@ class DatasourceNode(Node): related_id=upload_file.id, size=upload_file.size, storage_key=upload_file.key, + url=upload_file.source_url, ) variable_pool.add([self._node_id, "file"], file_info) # variable_pool.add([self.node_id, "file"], file_info.to_dict()) diff --git a/api/factories/file_factory.py b/api/factories/file_factory.py index 5b13901dd3..198fec7cd3 100644 --- a/api/factories/file_factory.py +++ b/api/factories/file_factory.py @@ -345,7 +345,7 @@ def _build_from_datasource_file( ) return File( - id=mapping.get("id"), + id=mapping.get("datasource_file_id"), tenant_id=tenant_id, filename=datasource_file.name, type=file_type, @@ -356,6 +356,7 @@ def _build_from_datasource_file( mime_type=datasource_file.mime_type, size=datasource_file.size, storage_key=datasource_file.key, + url=datasource_file.source_url, )