From c3c0727366ffb6ac562c9636e3322005240dff0e Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 6 Mar 2024 18:53:30 -0800 Subject: [PATCH] fix(factory.py): retry failed get request --- litellm/llms/prompt_templates/factory.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/litellm/llms/prompt_templates/factory.py b/litellm/llms/prompt_templates/factory.py index e776bee50..616833a2e 100644 --- a/litellm/llms/prompt_templates/factory.py +++ b/litellm/llms/prompt_templates/factory.py @@ -485,7 +485,12 @@ def convert_url_to_base64(url): import requests import base64 - response = requests.get(url) + for _ in range(3): + try: + response = requests.get(url) + break + except: + pass if response.status_code == 200: image_bytes = response.content 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, } except Exception as e: + if "Error: Unable to fetch image from URL" in str(e): + raise e 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'] """ )