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)