feat(document_extractor): support tool file in document extractor (#10217)

1 year ago · 2c4d8dbe9b
parent 9305ad2102
commit 2c4d8dbe9b
1 changed files with 1 additions and 3 deletions
--- a/api/core/workflow/nodes/document_extractor/node.py
+++ b/api/core/workflow/nodes/document_extractor/node.py
@ -198,10 +198,8 @@ def _download_file_content(file: File) -> bytes:
            response = ssrf_proxy.get(file.remote_url)
            response.raise_for_status()
            return response.content
-        elif file.transfer_method == FileTransferMethod.LOCAL_FILE:
-            return file_manager.download(file)
        else:
-            raise ValueError(f"Unsupported transfer method: {file.transfer_method}")
+            return file_manager.download(file)
    except Exception as e:
        raise FileDownloadError(f"Error downloading file: {str(e)}") from e