phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(feat) add voyage ai embeddings

Browse source
This commit is contained in:
ishaan-jaff 2023-12-28 17:09:27 +05:30
parent cbcf406fd0
commit 95e6d2fbba
4 changed files with 34 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/tests/test_embedding.py
View file

@ -318,6 +318,21 @@ def test_mistral_embeddings():
# test_mistral_embeddings()
def test_voyage_embeddings():
try:
litellm.set_verbose = True
response = litellm.embedding(
model="voyage/voyage-01",
input=["good morning from litellm"],
)
print(f"response: {response}")
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_voyage_embeddings()
# test_sagemaker_embeddings()
# def local_proxy_embeddings():
# litellm.set_verbose=True