Qun
6 months ago
1 changed files with 5 additions and 1 deletions
+ 5
- 1
api/core/tools/tool_file_manager.py
View File
| except httpx.TimeoutException: | except httpx.TimeoutException: | ||||
| raise ValueError(f"timeout when downloading file from {file_url}") | 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" | extension = guess_extension(mimetype) or ".bin" | ||||
| unique_name = uuid4().hex | unique_name = uuid4().hex | ||||
| filename = f"{unique_name}{extension}" | filename = f"{unique_name}{extension}" |
Loading…