From 57a20100e49ccb46e9473171951972a72b24ca0d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 19 Jan 2024 13:58:08 -0800 Subject: [PATCH] (test) usage based routing --- 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 ca4aac3e6..fa3c175d7 100644 --- a/litellm/tests/test_router_fallbacks.py +++ b/litellm/tests/test_router_fallbacks.py @@ -710,7 +710,7 @@ def test_usage_based_routing_fallbacks(): # Constants for TPM and RPM allocation AZURE_FAST_TPM = 3 - AZURE_BASIC_TPM = 2 + AZURE_BASIC_TPM = 4 OPENAI_TPM = 2000 ANTHROPIC_TPM = 100000 @@ -776,7 +776,7 @@ def test_usage_based_routing_fallbacks(): ) messages = [ - {"content": "Tell me a joke.", "modality": "voice"}, + {"content": "Tell me a joke.", "role": "user"}, ] response = router.completion(