fix(utils.py): fix exception mapping for azure internal server error

This commit is contained in:
Krrish Dholakia 2024-06-06 17:08:05 -07:00
parent ea4334f760
commit 6f3468aee4
6 changed files with 6 additions and 3 deletions

View file

@ -76,7 +76,7 @@ def test_image_generation_azure_dall_e_3():
)
print(f"response: {response}")
assert len(response.data) > 0
except litellm.RateLimitError as e:
except litellm.InternalServerError as e:
pass
except litellm.ContentPolicyViolationError:
pass # OpenAI randomly raises these errors - skip when they occur