vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix download file name bug (#3163)

Browse Source
This commit is contained in:
LawyZheng
2025-08-12 00:38:31 +08:00
committed by GitHub
parent d46d79e9e3
commit e8df79347b
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
skyvern/forge/sdk/api/files.py
View File

@@ -64,6 +64,11 @@ def extract_google_drive_file_id(url: str) -> str | None:
return None
def is_valid_mime_type(file_path: str) -> bool:
mime_type, _ = mimetypes.guess_type(file_path)
return mime_type is not None
async def download_file(url: str, max_size_mb: int | None = None) -> str:
try:
# Check if URL is a Google Drive link
@@ -122,7 +127,7 @@ async def download_file(url: str, max_size_mb: int | None = None) -> str:
LOG.info("No file name retrieved from HTTP headers, using the file name from the URL")
file_name = os.path.basename(a.path)
if not Path(file_name).suffix and file_suffix:
if not is_valid_mime_type(file_name) and file_suffix:
LOG.info("No file extension detected, adding the extension from HTTP headers")
file_name = file_name + file_suffix