jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: mypy for all type check (#10921)

Browse source
This commit is contained in:
yihong 2024-12-24 18:38:51 +08:00 • committed by GitHub
commit 56e15d09a9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
584 changed files with 3975 additions and 2826 deletions
Download patch file Download diff file Expand all files Collapse all files

5
api/factories/file_factory.py
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: