From d9e1267b24f24f8170201aeca3060d6e5e47c6b1 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 24 Nov 2023 18:28:38 -0800 Subject: [PATCH] (fix) embedding test --- litellm/tests/test_embedding.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index 6fd58120bd..934367707d 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -21,11 +21,11 @@ def test_openai_embedding(): ) litellm_response = dict(response) litellm_response_keys = set(litellm_response.keys()) - litellm_response_keys.discard('_responses_ms') - + litellm_response_keys.discard('_response_ms') + print(litellm_response_keys) print("LiteLLM Response\n") - print(litellm_response) + # print(litellm_response) # same request with OpenAI 1.0+ import openai @@ -36,6 +36,7 @@ def test_openai_embedding(): response = dict(response) openai_response_keys = set(response.keys()) + print(openai_response_keys) assert litellm_response_keys == openai_response_keys # ENSURE the Keys in litellm response is exactly what the openai package returns assert len(litellm_response["data"]) == 2 # expect two embedding responses from litellm_response since input had two print(openai_response_keys) @@ -185,4 +186,4 @@ def test_aembedding(): pass asyncio.run(embedding_call()) -test_aembedding() +# test_aembedding()