diff --git a/litellm/tests/test_custom_callback_router.py b/litellm/tests/test_custom_callback_router.py index 544af78ee6..80fc096e78 100644 --- a/litellm/tests/test_custom_callback_router.py +++ b/litellm/tests/test_custom_callback_router.py @@ -692,7 +692,7 @@ async def test_rate_limit_error_callback(): messages=[{"role": "user", "content": "Hey, how's it going?"}], litellm_logging_obj=litellm_logging_obj, ) - except litellm.RateLimitError: + except (litellm.RateLimitError, ValueError): pass await asyncio.sleep(3)