From 5a471caf1686354e35bb1920d311f08a5bc2652d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 2 Nov 2023 10:17:40 -0700 Subject: [PATCH] (test) add cohere embedding v3 --- litellm/tests/test_embedding.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index 10c2ce8b200..d533b7c42e6 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -69,6 +69,7 @@ def test_openai_azure_embedding(): def test_cohere_embedding(): try: + # litellm.set_verbose=True response = embedding( model="embed-english-v2.0", input=["good morning from litellm", "this is another item"] ) @@ -78,6 +79,19 @@ def test_cohere_embedding(): # test_cohere_embedding() +def test_cohere_embedding3(): + try: + litellm.set_verbose=True + response = embedding( + model="embed-english-v3.0", + input=["good morning from litellm", "this is another item"], + ) + print(f"response:", response) + except Exception as e: + pytest.fail(f"Error occurred: {e}") + +# test_cohere_embedding3() + def test_bedrock_embedding(): try: response = embedding( @@ -99,7 +113,7 @@ def test_hf_embedding(): print(f"response:", response) except Exception as e: pytest.fail(f"Error occurred: {e}") -test_hf_embedding() +# test_hf_embedding() # test async embeddings def test_aembedding():