From 17370dc50fab5586b25c7a26f09545cadf39d6ff Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 26 Jan 2024 10:37:01 -0800 Subject: [PATCH] (test) dimension param - openai --- litellm/tests/test_embedding.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index 42ac6f7f9d..a005a6ad16 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -64,7 +64,9 @@ def test_openai_embedding_3(): model="text-embedding-3-small", input=["good morning from litellm", "this is another item"], metadata={"anything": "good day"}, + dimensions=5, ) + print(f"response:", response) litellm_response = dict(response) litellm_response_keys = set(litellm_response.keys()) litellm_response_keys.discard("_response_ms") @@ -80,6 +82,7 @@ def test_openai_embedding_3(): response = client.embeddings.create( model="text-embedding-3-small", input=["good morning from litellm", "this is another item"], + dimensions=5, ) response = dict(response)