From c52861906b049d1b1cec09fa5a72550bb128d637 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 27 Nov 2023 22:03:02 -0800 Subject: [PATCH] (test) router cooldowns --- litellm/tests/test_acooldowns_router.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_acooldowns_router.py b/litellm/tests/test_acooldowns_router.py index f261058faf..afab5944c8 100644 --- a/litellm/tests/test_acooldowns_router.py +++ b/litellm/tests/test_acooldowns_router.py @@ -138,7 +138,7 @@ def test_cooldown_same_model_name(): redis_port=int(os.getenv("REDIS_PORT")), routing_strategy="simple-shuffle", set_verbose=True, - num_retries=1 + num_retries=3 ) # type: ignore response = router.completion( @@ -146,7 +146,7 @@ def test_cooldown_same_model_name(): messages=[ { "role": "user", - "content": "hello this request will fail" + "content": "hello this request will pass" } ] ) @@ -162,4 +162,4 @@ def test_cooldown_same_model_name(): except Exception as e: pytest.fail(f"Got unexpected exception on router! - {e}") -# test_cooldown_same_model_name() +test_cooldown_same_model_name()