From 5cf795f2b85e1d954862c75dde115db72b8819a3 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Fri, 29 Aug 2025 14:48:51 +0800
Subject: [PATCH] fix preview run

---
 api/core/workflow/nodes/datasource/datasource_node.py | 1 +
 api/factories/file_factory.py                         | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

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,
     )