phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Fix file type handling of uppercase extensions

Browse source
This commit is contained in:
nick-rackauckas 2024-06-13 15:00:16 -07:00
parent de79eab37d
commit 25ce49770a
1 changed files with 7 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/types/files.py
View file

@ -151,23 +151,16 @@ Util Functions
""" """
def get_file_mime_type_from_extension(extension: str) -> str:
for file_type, extensions in FILE_EXTENSIONS.items():
if extension in extensions:
return FILE_MIME_TYPES[file_type]
raise ValueError(f"Unknown mime type for extension: {extension}")
def get_file_extension_from_mime_type(mime_type: str) -> str: def get_file_extension_from_mime_type(mime_type: str) -> str:
for file_type, mime in FILE_MIME_TYPES.items(): for file_type, mime in FILE_MIME_TYPES.items():
if mime == mime_type: if mime.lower() == mime_type.lower():
return FILE_EXTENSIONS[file_type][0] return FILE_EXTENSIONS[file_type][0]
raise ValueError(f"Unknown extension for mime type: {mime_type}") raise ValueError(f"Unknown extension for mime type: {mime_type}")
def get_file_type_from_extension(extension: str) -> FileType: def get_file_type_from_extension(extension: str) -> FileType:
for file_type, extensions in FILE_EXTENSIONS.items(): for file_type, extensions in FILE_EXTENSIONS.items():
if extension in extensions: if extension.lower() in extensions:
return file_type return file_type
raise ValueError(f"Unknown file type for extension: {extension}") raise ValueError(f"Unknown file type for extension: {extension}")
@ -181,6 +174,11 @@ def get_file_mime_type_for_file_type(file_type: FileType) -> str:
return FILE_MIME_TYPES[file_type] return FILE_MIME_TYPES[file_type]
def get_file_mime_type_from_extension(extension: str) -> str:
file_type = get_file_type_from_extension(extension)
return get_file_mime_type_for_file_type(file_type)
""" """
FileType Type Groupings (Videos, Images, etc) FileType Type Groupings (Videos, Images, etc)
""" """