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

(test) dimension param - openai

Browse source
This commit is contained in:
ishaan-jaff 2024-01-26 10:37:01 -08:00
parent c8da57710f
commit 17370dc50f
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_embedding.py
View file

@ -64,7 +64,9 @@ def test_openai_embedding_3():
model="text-embedding-3-small", model="text-embedding-3-small",
input=["good morning from litellm", "this is another item"], input=["good morning from litellm", "this is another item"],
metadata={"anything": "good day"}, metadata={"anything": "good day"},
dimensions=5,
) )
print(f"response:", response)
litellm_response = dict(response) litellm_response = dict(response)
litellm_response_keys = set(litellm_response.keys()) litellm_response_keys = set(litellm_response.keys())
litellm_response_keys.discard("_response_ms") litellm_response_keys.discard("_response_ms")
@ -80,6 +82,7 @@ def test_openai_embedding_3():
response = client.embeddings.create( response = client.embeddings.create(
model="text-embedding-3-small", model="text-embedding-3-small",
input=["good morning from litellm", "this is another item"], input=["good morning from litellm", "this is another item"],
dimensions=5,
) )
response = dict(response) response = dict(response)