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

test(test_embedding.py): fix the embedding test

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-22 14:09:45 -08:00
parent 448ec0a571
commit 0b4e10e068
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_embedding.py
View file

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