From 3059f30672c5547d61f977b6c13b0308d23e100e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 22 Nov 2023 14:21:58 -0800 Subject: [PATCH] (test) verify azure response have expected keys --- litellm/tests/test_embedding.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index ce0a84573..40fea7dd7 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -39,21 +39,22 @@ def test_openai_embedding(): print(openai_response_keys) except Exception as e: pytest.fail(f"Error occurred: {e}") -test_openai_embedding() +# test_openai_embedding() def test_openai_azure_embedding_simple(): try: - response = embedding( model="azure/azure-embedding-model", input=["good morning from litellm"], ) print(response) + response_keys = dict(response).keys() + assert set(["usage", "model", "object", "data"]) == set(response_keys) #assert litellm response has expected keys from OpenAI embedding response except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_openai_azure_embedding_simple() +test_openai_azure_embedding_simple() def test_openai_azure_embedding(): try: