From 2d47acad6ca3a6a510dd55a0b471825398227c67 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 3 Sep 2024 06:40:52 -0700 Subject: [PATCH] test: fix test --- litellm/tests/test_custom_callback_router.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_custom_callback_router.py b/litellm/tests/test_custom_callback_router.py index 544af78ee..80fc096e7 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)