Parcourir la source

enhance guessing mimetype of tool file (#17640)

tags/1.3.0
Qun il y a 6 mois
Parent
révision
0e136b42a2
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 5 ajouts et 1 suppressions
  1. 5
    1
      api/core/tools/tool_file_manager.py

+ 5
- 1
api/core/tools/tool_file_manager.py Voir le fichier

@@ -108,7 +108,11 @@ class ToolFileManager:
except httpx.TimeoutException:
raise ValueError(f"timeout when downloading file from {file_url}")

mimetype = guess_type(file_url)[0] or "application/octet-stream"
mimetype = (
guess_type(file_url)[0]
or response.headers.get("Content-Type", "").split(";")[0].strip()
or "application/octet-stream"
)
extension = guess_extension(mimetype) or ".bin"
unique_name = uuid4().hex
filename = f"{unique_name}{extension}"

Chargement…
Annuler
Enregistrer