r2

10 months ago · 9c6eb95700
parent a8b676ade0
commit 9c6eb95700
1 changed files with 3 additions and 1 deletions
--- 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)