From 0b4e10e068270566a65837bdb5e97c1314f83de4 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 22 Nov 2023 14:09:45 -0800 Subject: [PATCH] test(test_embedding.py): fix the embedding test --- litellm/tests/test_embedding.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index deff80632..ce0a84573 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -19,6 +19,7 @@ def test_openai_embedding(): model="text-embedding-ada-002", input=["good morning from litellm", "this is another item"] ) litellm_response = dict(response) + litellm_response.pop("_response_ms") litellm_response_keys = set(litellm_response.keys()) print(litellm_response_keys) print("LiteLLM Response\n")