From bfbed2d93d7874a8e831575e7a21940ef46b2c1a Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 2 Jan 2024 15:41:51 +0530 Subject: [PATCH] (test) xinference embeddings --- litellm/tests/test_embedding.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index a1dfb6ab3..2a86f79d7 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -332,6 +332,17 @@ def test_voyage_embeddings(): # test_voyage_embeddings() +# def test_xinference_embeddings(): +# try: +# litellm.set_verbose = True +# response = litellm.embedding( +# model="xinference/bge-base-en", +# input=["good morning from litellm"], +# ) +# print(f"response: {response}") +# except Exception as e: +# pytest.fail(f"Error occurred: {e}") +# test_xinference_embeddings() # test_sagemaker_embeddings() # def local_proxy_embeddings():