Skip to content

Commit 7934610

Browse files
test: update test to handle rate limit error
1 parent c7b6f66 commit 7934610

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/image_gen_tests/test_image_generation.py

+2
Original file line numberDiff line numberDiff line change
@@ -207,6 +207,8 @@ def test_image_generation_azure_dall_e_3():
207207
pass # OpenAI randomly raises these errors - skip when they occur
208208
except litellm.InternalServerError:
209209
pass
210+
except litellm.RateLimitError as e:
211+
pass
210212
except Exception as e:
211213
if "Your task failed as a result of our safety system." in str(e):
212214
pass

0 commit comments

Comments
 (0)