From 24addd8a4fcf67d50b55fcbe552b6406999ddcd8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 21 Jun 2024 21:02:34 -0700 Subject: [PATCH] fix test test_router_completion_streaming --- litellm/tests/test_tpm_rpm_routing_v2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_tpm_rpm_routing_v2.py b/litellm/tests/test_tpm_rpm_routing_v2.py index e362eedb0ea..1f3de09104f 100644 --- a/litellm/tests/test_tpm_rpm_routing_v2.py +++ b/litellm/tests/test_tpm_rpm_routing_v2.py @@ -394,7 +394,7 @@ async def test_router_completion_streaming(): "model": "azure/gpt-turbo", "api_key": "os.environ/AZURE_FRANCE_API_KEY", "api_base": "https://openai-france-1234.openai.azure.com", - "rpm": 1440, + "tpm": 1440, "mock_response": "Hello world", }, "model_info": {"id": 1}, @@ -405,7 +405,7 @@ async def test_router_completion_streaming(): "model": "azure/gpt-35-turbo", "api_key": "os.environ/AZURE_EUROPE_API_KEY", "api_base": "https://my-endpoint-europe-berri-992.openai.azure.com", - "rpm": 6, + "tpm": 6, "mock_response": "Hello world", }, "model_info": {"id": 2},