mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-05-13 21:30:21 +02:00
20f2b9b2bb
* add support for image search * quick fix up * k * k * k * k * nit * quick fix for connector tests
47 lines
1.0 KiB
Python
47 lines
1.0 KiB
Python
"""
|
|
Centralized file type validation utilities.
|
|
"""
|
|
# Standard image MIME types supported by most vision LLMs
|
|
IMAGE_MIME_TYPES = [
|
|
"image/png",
|
|
"image/jpeg",
|
|
"image/jpg",
|
|
"image/webp",
|
|
]
|
|
|
|
# Image types that should be excluded from processing
|
|
EXCLUDED_IMAGE_TYPES = [
|
|
"image/bmp",
|
|
"image/tiff",
|
|
"image/gif",
|
|
"image/svg+xml",
|
|
]
|
|
|
|
|
|
def is_valid_image_type(mime_type: str) -> bool:
|
|
"""
|
|
Check if mime_type is a valid image type.
|
|
|
|
Args:
|
|
mime_type: The MIME type to check
|
|
|
|
Returns:
|
|
True if the MIME type is a valid image type, False otherwise
|
|
"""
|
|
if not mime_type:
|
|
return False
|
|
return mime_type.startswith("image/") and mime_type not in EXCLUDED_IMAGE_TYPES
|
|
|
|
|
|
def is_supported_by_vision_llm(mime_type: str) -> bool:
|
|
"""
|
|
Check if this image type can be processed by vision LLMs.
|
|
|
|
Args:
|
|
mime_type: The MIME type to check
|
|
|
|
Returns:
|
|
True if the MIME type is supported by vision LLMs, False otherwise
|
|
"""
|
|
return mime_type in IMAGE_MIME_TYPES
|