From fbcdbf3a825fa1ab24056cc58d4ebc1feb19bf8e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 24 Nov 2023 20:46:35 -0800 Subject: [PATCH] (fix) test embedding --- litellm/tests/test_embedding.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index 934367707d..4f79ed05a7 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -51,7 +51,8 @@ def test_openai_azure_embedding_simple(): input=["good morning from litellm"], ) print(response) - response_keys = dict(response).keys() + response_keys = set(dict(response).keys()) + response_keys.discard('_response_ms') assert set(["usage", "model", "object", "data"]) == set(response_keys) #assert litellm response has expected keys from OpenAI embedding response except Exception as e: