scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 114

(test) add cohere embedding v3

Browse source
This commit is contained in:
ishaan-jaff 2023-11-02 10:17:40 -07:00
parent 03860984eb
commit 5a471caf16
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/tests/test_embedding.py
View file

@ -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():