diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index b9e9d196ea..e11183938c 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -487,8 +487,13 @@ async def test_acompletion_gemini_stream(): raise Exception("Empty response received") except litellm.APIError as e: pass + except litellm.RateLimitError as e: + pass except Exception as e: - pytest.fail(f"Error occurred: {e}") + if "429 Resource has been exhausted" in str(e): + pass + else: + pytest.fail(f"Error occurred: {e}") # asyncio.run(test_acompletion_gemini_stream())