From 785c024324d75609e5f9e68715ec5f63918c30d4 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 24 Nov 2023 18:14:06 -0800 Subject: [PATCH] (test) fix embedding --- litellm/tests/test_embedding.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index e33ebe79f4..6fd58120bd 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -21,6 +21,8 @@ def test_openai_embedding(): ) litellm_response = dict(response) litellm_response_keys = set(litellm_response.keys()) + litellm_response_keys.discard('_responses_ms') + print(litellm_response_keys) print("LiteLLM Response\n") print(litellm_response)