From 9c6eb957000e4642d645b856974406912e72f548 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Fri, 4 Jul 2025 18:49:37 +0800
Subject: [PATCH] r2

---
 api/core/file/file_manager.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/api/core/file/file_manager.py b/api/core/file/file_manager.py
index ada19ef8ce..ee58840c84 100644
--- a/api/core/file/file_manager.py
+++ b/api/core/file/file_manager.py
@@ -72,7 +72,9 @@ def to_prompt_message_content(
 
 
 def download(f: File, /):
-    if f.transfer_method in (FileTransferMethod.TOOL_FILE, FileTransferMethod.LOCAL_FILE):
+    if f.transfer_method in (FileTransferMethod.TOOL_FILE, 
+                             FileTransferMethod.LOCAL_FILE, 
+                             FileTransferMethod.DATASOURCE_FILE):
         return _download_file_content(f._storage_key)
     elif f.transfer_method == FileTransferMethod.REMOTE_URL:
         response = ssrf_proxy.get(f.remote_url, follow_redirects=True)