From 55572ce3cd511fbbeb824d3952e05c6ed7e3bb12 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 27 Sep 2025 10:59:25 -0700 Subject: [PATCH] test_add_model_run_health --- tests/test_models.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/tests/test_models.py b/tests/test_models.py index b1a5b84b3b..67e77dcaaf 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -307,10 +307,8 @@ async def add_model_for_health_checking(session, model_id="123"): data = { "model_name": f"azure-model-health-check-{model_id}", "litellm_params": { - "model": "azure/chatgpt-v-3", - "api_key": os.getenv("AZURE_API_KEY"), - "api_base": "https://openai-gpt-4-test-v-1.openai.azure.com/", - "api_version": "2023-05-15", + "model": "gpt-4.1-nano", + "api_key": os.getenv("OPENAI_API_KEY"), }, "model_info": {"id": model_id}, } @@ -432,7 +430,7 @@ async def test_add_model_run_health(): assert _health_info["healthy_count"] == 1 assert ( - _healthy_endpooint["model"] == "azure/chatgpt-v-3" + _healthy_endpooint["model"] == "gpt-4.1-nano" ) # this is the model that got added # assert httpx client is is unchanges