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

add more rule to parse filename when downloading (#3076)

Browse Source
This commit is contained in:
LawyZheng
2025-08-01 11:25:36 +08:00
committed by GitHub
parent 7844b8372a
commit 119ef71057
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -113,16 +113,15 @@ async def download_file(url: str, max_size_mb: int | None = None) -> str:
except Exception:
LOG.exception("Failed to retrieve the file extension from HTTP headers")
# parse the query params to get the file name
query_params = dict(parse_qsl(a.query))
if "download" in query_params:
file_name = query_params["download"]
if not file_name:
LOG.info("No file name retrieved from HTTP headers, using the file name from the URL")
file_name = os.path.basename(a.path)
# Check for download parameter in Supabase URLs
if "supabase.co" in a.netloc.lower():
query_params = dict(parse_qsl(a.query))
if "download" in query_params:
file_name = query_params["download"]
if not Path(file_name).suffix and file_suffix:
LOG.info("No file extension detected, adding the extension from HTTP headers")
file_name = file_name + file_suffix