diff --git a/tests/local_testing/test_health_check.py b/tests/litellm_utils_tests/test_health_check.py similarity index 99% rename from tests/local_testing/test_health_check.py rename to tests/litellm_utils_tests/test_health_check.py index f14bb617daa..6cff1f08d9b 100644 --- a/tests/local_testing/test_health_check.py +++ b/tests/litellm_utils_tests/test_health_check.py @@ -20,7 +20,7 @@ import litellm async def test_azure_health_check(): response = await litellm.ahealth_check( model_params={ - "model": "azure/chatgpt-v-3", + "model": "azure/gpt-4.1-nano", "messages": [{"role": "user", "content": "Hey, how's it going?"}], "api_key": os.getenv("AZURE_API_KEY"), "api_base": os.getenv("AZURE_API_BASE"), @@ -51,7 +51,7 @@ async def test_text_completion_health_check(): async def test_azure_embedding_health_check(): response = await litellm.ahealth_check( model_params={ - "model": "azure/azure-embedding-model", + "model": "azure/text-embedding-ada-002", "api_key": os.getenv("AZURE_API_KEY"), "api_base": os.getenv("AZURE_API_BASE"), "api_version": os.getenv("AZURE_API_VERSION"),