phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

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

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-28 10:48:31 +05:30
parent ed5111ff2c
commit 5089753925
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_health_check.py
View file

@ -37,11 +37,11 @@ async def test_azure_embedding_health_check():
response = await litellm.ahealth_check( response = await litellm.ahealth_check(
model_params={ model_params={
"model": "azure/azure-embedding-model", "model": "azure/azure-embedding-model",
"messages": [{"role": "user", "content": "Hey, how's it going?"}],
"api_key": os.getenv("AZURE_API_KEY"), "api_key": os.getenv("AZURE_API_KEY"),
"api_base": os.getenv("AZURE_API_BASE"), "api_base": os.getenv("AZURE_API_BASE"),
"api_version": os.getenv("AZURE_API_VERSION"), "api_version": os.getenv("AZURE_API_VERSION"),
}, },
input=["test for litellm"],
mode="embedding", mode="embedding",
) )
print(f"response: {response}") print(f"response: {response}")