test: update test to handle rate limit error

2025-05-10 16:15:48 -07:00 · 2025-05-10 16:15:48 -07:00 · 7934610f63
commit 7934610f63
parent c7b6f661b3
1 changed files with 2 additions and 0 deletions
--- 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