phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) mistral-embed

Browse source
This commit is contained in:
ishaan-jaff 2023-12-28 16:42:36 +05:30
parent 78f0c0228b
commit 41d5dceaec
1 changed files with 15 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/tests/test_embedding.py
View file

@ -201,7 +201,7 @@ def test_bedrock_embedding_titan():
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
test_bedrock_embedding_titan() # test_bedrock_embedding_titan()
def test_bedrock_embedding_cohere(): def test_bedrock_embedding_cohere():
@ -304,6 +304,20 @@ def test_sagemaker_embeddings():
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
def test_mistral_embeddings():
try:
litellm.set_verbose = True
response = litellm.embedding(
model="mistral/mistral-embed",
input=["good morning from litellm"],
)
print(f"response: {response}")
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_mistral_embeddings()
# test_sagemaker_embeddings() # test_sagemaker_embeddings()
# def local_proxy_embeddings(): # def local_proxy_embeddings():
# litellm.set_verbose=True # litellm.set_verbose=True