From e878f2b1efda56908e42d01df9cbaeabaff41320 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 25 Oct 2025 15:09:08 -0700 Subject: [PATCH] test_router_get_available_deployments --- tests/local_testing/test_least_busy_routing.py | 16 ++++++---------- 1 file changed, 6 insertions(+), 10 deletions(-) diff --git a/tests/local_testing/test_least_busy_routing.py b/tests/local_testing/test_least_busy_routing.py index 479b805aaa..2d7605ebb6 100644 --- a/tests/local_testing/test_least_busy_routing.py +++ b/tests/local_testing/test_least_busy_routing.py @@ -75,9 +75,8 @@ async def test_router_get_available_deployments(async_test): { "model_name": "azure-model", "litellm_params": { - "model": "azure/gpt-turbo", - "api_key": "os.environ/AZURE_FRANCE_API_KEY", - "api_base": "https://openai-france-1234.openai.azure.com", + "model": "openai/gpt-4.1-mini", + "api_key": "os.environ/OPENAI_API_KEY", "rpm": 1440, }, "model_info": {"id": 1}, @@ -85,9 +84,8 @@ async def test_router_get_available_deployments(async_test): { "model_name": "azure-model", "litellm_params": { - "model": "azure/gpt-35-turbo", - "api_key": "os.environ/AZURE_EUROPE_API_KEY", - "api_base": "https://my-endpoint-europe-berri-992.openai.azure.com", + "model": "openai/gpt-4.1-mini", + "api_key": "os.environ/OPENAI_API_KEY", "rpm": 6, }, "model_info": {"id": 2}, @@ -95,9 +93,8 @@ async def test_router_get_available_deployments(async_test): { "model_name": "azure-model", "litellm_params": { - "model": "azure/gpt-35-turbo", - "api_key": "os.environ/AZURE_CANADA_API_KEY", - "api_base": "https://my-endpoint-canada-berri992.openai.azure.com", + "model": "openai/gpt-4.1-mini", + "api_key": "os.environ/OPENAI_API_KEY", "rpm": 6, }, "model_info": {"id": 3}, @@ -113,7 +110,6 @@ async def test_router_get_available_deployments(async_test): router.leastbusy_logger.test_flag = True model_group = "azure-model" - deployment = "azure/gpt-4.1-mini" request_count_dict = {1: 10, 2: 54, 3: 100} cache_key = f"{model_group}_request_count" if async_test is True: