test: fix test

2024-09-03 06:40:52 -07:00 · 2024-09-03 06:40:52 -07:00 · 2d47acad6c
commit 2d47acad6c
parent 9df0588c2c
1 changed files with 1 additions and 1 deletions
--- 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)