Browse Source

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

tags/0.11.0
-LAN- 1 year ago
parent
commit
2c4d8dbe9b
No account linked to committer's email address
1 changed files with 1 additions and 3 deletions
  1. 1
    3
      api/core/workflow/nodes/document_extractor/node.py

+ 1
- 3
api/core/workflow/nodes/document_extractor/node.py View File

@@ -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


Loading…
Cancel
Save