test_add_model_run_health

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