From fed620b202f334b43e972d6d1855f6e58d4dec2d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 25 Nov 2023 11:10:02 -0800 Subject: [PATCH] (test) metadata test in embedding --- litellm/tests/test_embedding.py | 19 ++++++++++++++++--- 1 file changed, 16 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index 4f79ed05a7..348ee11668 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -16,8 +16,11 @@ litellm.set_verbose = False def test_openai_embedding(): try: + litellm.set_verbose=True response = embedding( - model="text-embedding-ada-002", input=["good morning from litellm", "this is another item"] + model="text-embedding-ada-002", + input=["good morning from litellm", "this is another item"], + metadata = {"anything": "good day"} ) litellm_response = dict(response) litellm_response_keys = set(litellm_response.keys()) @@ -42,7 +45,7 @@ def test_openai_embedding(): print(openai_response_keys) except Exception as e: pytest.fail(f"Error occurred: {e}") -test_openai_embedding() +# test_openai_embedding() def test_openai_azure_embedding_simple(): try: @@ -58,7 +61,7 @@ def test_openai_azure_embedding_simple(): except Exception as e: pytest.fail(f"Error occurred: {e}") -test_openai_azure_embedding_simple() +# test_openai_azure_embedding_simple() def test_openai_azure_embedding_timeouts(): @@ -188,3 +191,13 @@ def test_aembedding(): asyncio.run(embedding_call()) # test_aembedding() + +# def test_custom_openai_embedding(): +# litellm.set_verbose=True +# response = embedding( +# model="openai/custom_embedding", +# input=["good morning from litellm"], +# api_base="http://0.0.0.0:8000/" +# ) +# print(response) +# test_custom_openai_embedding()