From 5089753925b6bd035cd9d4f3985ff32e7aa3968a Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 28 Dec 2023 10:48:31 +0530
Subject: [PATCH] test(test_health_check.py): fix azure embedding health check
 test

---
 litellm/tests/test_health_check.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litellm/tests/test_health_check.py b/litellm/tests/test_health_check.py
index 9a59ca315451..3996c8d09867 100644
--- 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}")