test(test_health_check.py): fix azure embedding health check test

ventz · Dec 28, 2023 · 5089753 · 5089753
1 parent ed5111f
commit 5089753
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litellm/tests/test_health_check.py b/litellm/tests/test_health_check.py
@@ -37,11 +37,11 @@ async def test_azure_embedding_health_check():
     response = await litellm.ahealth_check(
         model_params={
             "model": "azure/azure-embedding-model",
-            "messages": [{"role": "user", "content": "Hey, how's it going?"}],
             "api_key": os.getenv("AZURE_API_KEY"),
             "api_base": os.getenv("AZURE_API_BASE"),
             "api_version": os.getenv("AZURE_API_VERSION"),
         },
+        input=["test for litellm"],
         mode="embedding",
     )
     print(f"response: {response}")