scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

fix _validate_format for BedrockImageProcessor

Browse source
This commit is contained in:
Ishaan Jaff 2025-08-05 13:43:50 -07:00
parent 32501c85f5
commit 645ac875aa
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/litellm_core_utils/prompt_templates/factory.py
View file

@ -2479,7 +2479,9 @@ class BedrockImageProcessor:
)
if is_document:
potential_extensions = mimetypes.guess_all_extensions(mime_type)
potential_extensions = mimetypes.guess_all_extensions(
mime_type, strict=False
)
valid_extensions = [
ext[1:]
for ext in potential_extensions
@ -2950,7 +2952,10 @@ def process_empty_text_blocks(
]
modified_message = message.copy()
modified_message["content"] = modified_content_block
modified_message["content"] = cast(
Union[List[ChatCompletionTextObject], List[ChatCompletionThinkingBlock]],
modified_content_block,
)
return modified_message