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

fix(factory.py): retry failed get request

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-06 18:53:30 -08:00
parent 726dad5756
commit c3c0727366
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/prompt_templates/factory.py
View file

@ -485,7 +485,12 @@ def convert_url_to_base64(url):
import requests import requests
import base64 import base64
for _ in range(3):
try:
response = requests.get(url) response = requests.get(url)
break
except:
pass
if response.status_code == 200: if response.status_code == 200:
image_bytes = response.content image_bytes = response.content
base64_image = base64.b64encode(image_bytes).decode("utf-8") base64_image = base64.b64encode(image_bytes).decode("utf-8")
@ -536,6 +541,8 @@ def convert_to_anthropic_image_obj(openai_image_url: str):
"data": base64_data, "data": base64_data,
} }
except Exception as e: except Exception as e:
if "Error: Unable to fetch image from URL" in str(e):
raise e
raise Exception( raise Exception(
"""Image url not in expected format. Example Expected input - "image_url": "data:image/jpeg;base64,{base64_image}". Supported formats - ['image/jpeg', 'image/png', 'image/gif', 'image/webp'] """ """Image url not in expected format. Example Expected input - "image_url": "data:image/jpeg;base64,{base64_image}". Supported formats - ['image/jpeg', 'image/png', 'image/gif', 'image/webp'] """
) )