diff --git a/tests/image_gen_tests/test_image_generation.py b/tests/image_gen_tests/test_image_generation.py index 21f96095a7..885df94e19 100644 --- a/tests/image_gen_tests/test_image_generation.py +++ b/tests/image_gen_tests/test_image_generation.py @@ -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