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

fix file type detect bug (#3172)

Browse Source
This commit is contained in:
LawyZheng
2025-08-13 11:24:20 +08:00
committed by GitHub
parent 5132501cc6
commit 2a4b26f8e6
2 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
skyvern/forge/sdk/workflow/models/block.py
View File

@@ -2450,12 +2450,14 @@ class FileParserBlock(Block):
def _detect_file_type_from_url(self, file_url: str) -> FileType:
"""Detect file type based on file extension in the URL."""
url_lower = file_url.lower()
if url_lower.endswith((".xlsx", ".xls", ".xlsm")):
url_parsed = urlparse(file_url)
# TODO: use filetype.guess(file_path) to make the detection more robust
suffix = Path(url_parsed.path).suffix.lower()
if suffix in (".xlsx", ".xls", ".xlsm"):
return FileType.EXCEL
elif url_lower.endswith(".pdf"):
elif suffix == ".pdf":
return FileType.PDF
elif url_lower.endswith(".tsv"):
elif suffix == ".tsv":
return FileType.CSV # TSV files are handled by the CSV parser
else:
return FileType.CSV # Default to CSV for .csv and any other extensions