From 04ac35240732bbcd6cfbb929d1069bb2bf10b4d8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 11 May 2024 19:20:24 -0700 Subject: [PATCH] test fix - test_async_fallbacks_embeddings --- litellm/tests/test_router_fallbacks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_router_fallbacks.py b/litellm/tests/test_router_fallbacks.py index 0bce9894b7..c1035e3e00 100644 --- a/litellm/tests/test_router_fallbacks.py +++ b/litellm/tests/test_router_fallbacks.py @@ -269,7 +269,7 @@ def test_sync_fallbacks_embeddings(): response = router.embedding(**kwargs) print(f"customHandler.previous_models: {customHandler.previous_models}") time.sleep(0.05) # allow a delay as success_callbacks are on a separate thread - assert customHandler.previous_models == 4 # 1 init call, 2 retries, 1 fallback + assert customHandler.previous_models == 1 # 1 init call, 2 retries, 1 fallback router.reset() except litellm.Timeout as e: pass @@ -323,7 +323,7 @@ async def test_async_fallbacks_embeddings(): await asyncio.sleep( 0.05 ) # allow a delay as success_callbacks are on a separate thread - assert customHandler.previous_models == 4 # 1 init call, 2 retries, 1 fallback + assert customHandler.previous_models == 1 # 1 init call with a bad key router.reset() except litellm.Timeout as e: pass