From 2c50ea94c8120863265c3bf78dbdad01f4cf2477 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 23 Nov 2023 11:15:50 -0800 Subject: [PATCH] (test) embedding: azurea - dd response_ms --- litellm/tests/test_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index d24b06bfbe..a3026e44cc 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -50,7 +50,7 @@ def test_openai_azure_embedding_simple(): ) 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 + assert set(["usage", "model", "object", "data", "_response_ms"]) == set(response_keys) #assert litellm response has expected keys from OpenAI embedding response except Exception as e: pytest.fail(f"Error occurred: {e}")