feat: mypy for all type check (#10921)

This commit is contained in:
yihong
2024-12-24 18:38:51 +08:00
committed by GitHub
parent c91e8b1737
commit 56e15d09a9
584 changed files with 3975 additions and 2826 deletions

View File

@@ -64,7 +64,7 @@ def build_from_mapping(
if not build_func:
raise ValueError(f"Invalid file transfer method: {transfer_method}")
file = build_func(
file: File = build_func(
mapping=mapping,
tenant_id=tenant_id,
transfer_method=transfer_method,
@@ -72,7 +72,7 @@ def build_from_mapping(
if config and not _is_file_valid_with_config(
input_file_type=mapping.get("type", FileType.CUSTOM),
file_extension=file.extension,
file_extension=file.extension or "",
file_transfer_method=file.transfer_method,
config=config,
):
@@ -281,6 +281,7 @@ def _get_file_type_by_extension(extension: str) -> FileType | None:
return FileType.AUDIO
elif extension in DOCUMENT_EXTENSIONS:
return FileType.DOCUMENT
return None
def _get_file_type_by_mimetype(mime_type: str) -> FileType | None: