phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 10:14:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) embedding test

Browse source
This commit is contained in:
ishaan-jaff 2023-11-24 18:28:38 -08:00
parent 785c024324
commit d9e1267b24
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_embedding.py
View file

@ -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()