Fix : Add a process to fetch the mime type from the file name for signed url in remote_url (#10872)

1 year ago · 464cc26ccf
parent d18754afdd
commit 464cc26ccf
1 changed files with 1 additions and 0 deletions
--- a/api/factories/file_factory.py
+++ b/api/factories/file_factory.py
@ -169,6 +169,7 @@ def _get_remote_file_info(url: str):
    mime_type = mimetypes.guess_type(url)[0] or ""
    file_size = -1
    filename = url.split("/")[-1].split("?")[0] or "unknown_file"
+    mime_type = mime_type or mimetypes.guess_type(filename)[0]

    resp = ssrf_proxy.head(url, follow_redirects=True)
    if resp.status_code == httpx.codes.OK: