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

(test) add bedrock/amazon.titan-embed-text-v1

Browse source
This commit is contained in:
ishaan-jaff 2023-11-01 13:55:27 -07:00
parent 2ad81bdd7b
commit a46d6a2dc9
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/tests/test_embedding.py
View file

@ -78,6 +78,16 @@ def test_cohere_embedding():
# test_cohere_embedding() # test_cohere_embedding()
def test_bedrock_embedding():
try:
response = embedding(
model="bedrock/amazon.titan-embed-text-v1", input=["good morning from litellm, attempting to embed data"]
)
print(f"response:", response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_bedrock_embedding()
# comment out hf tests - since hf endpoints are unstable # comment out hf tests - since hf endpoints are unstable
# def test_hf_embedding(): # def test_hf_embedding():
# try: # try: