fix azure gpt-4o test

This commit is contained in:
Ishaan Jaff 2024-09-19 16:20:43 -07:00
parent b022247168
commit b54bbf510e

View file

@ -935,6 +935,9 @@ def test_completion_base64(model):
print(f"\nResponse: {resp}")
prompt_tokens = resp.usage.prompt_tokens
except litellm.ServiceUnavailableError as e:
print("got service unavailable error: ", e)
pass
except Exception as e:
if "500 Internal error encountered.'" in str(e):
pass