scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 87

test: update test to handle rate limit error

Browse source
This commit is contained in:
Krrish Dholakia 2025-05-10 16:15:48 -07:00
parent c7b6f661b3
commit 7934610f63
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/image_gen_tests/test_image_generation.py
View file

@ -207,6 +207,8 @@ def test_image_generation_azure_dall_e_3():
pass # OpenAI randomly raises these errors - skip when they occur
except litellm.InternalServerError:
pass
except litellm.RateLimitError as e:
pass
except Exception as e:
if "Your task failed as a result of our safety system." in str(e):
pass