ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(factory.py): handle escape characters in image url

Browse Source 
prevents mimetype errors in vertex ai calls
This commit is contained in:
Krrish Dholakia 2024-07-18 17:36:58 -07:00
parent b9d4d67357
commit 30d6c1d171
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
litellm/llms/prompt_templates/factory.py
View File

@ -709,6 +709,7 @@ def convert_to_anthropic_image_obj(openai_image_url: str) -> GenericImageParsing
openai_image_url = convert_url_to_base64(url=openai_image_url)
# Extract the media type and base64 data
media_type, base64_data = openai_image_url.split("data:")[1].split(";base64,")
media_type = media_type.replace("\\/", "/")
return GenericImageParsingChunk(
type="base64",

1
litellm/tests/test_prompt_factory.py
View File

@ -212,6 +212,7 @@ def test_convert_url_to_img():
[
("data:image/jpeg;base64,1234", "image/jpeg"),
("data:application/pdf;base64,1234", "application/pdf"),
("data:image\/jpeg;base64,1234", "image/jpeg"),
],
)
def test_base64_image_input(url, expected_media_type):