test: update test to handle rate limit error

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