phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_amazing_vertex_completion.py): fix test to handle api instability

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-31 08:33:10 -08:00
parent c43b6e6f41
commit dd47c9d854
2 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/utils.py
View file

@ -2248,7 +2248,12 @@ def client(original_function):
)
if num_retries:
if isinstance(e, openai.APIError) or isinstance(e, openai.Timeout):
if (
isinstance(e, openai.APIError)
or isinstance(e, openai.Timeout)
or isinstance(openai.APIConnectionError)
):
print_verbose(f"RETRY TRIGGERED!")
kwargs["num_retries"] = num_retries
return litellm.completion_with_retries(*args, **kwargs)
elif (