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

v0

Browse source
This commit is contained in:
ishaan-jaff 2024-01-11 22:56:18 +05:30
parent d5a1458b8a
commit a876748bf5
2 changed files with 4 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_embedding.py
View file

@ -189,10 +189,7 @@ def test_bedrock_embedding_titan():
litellm.set_verbose = True
response = embedding(
model="amazon.titan-embed-text-v1",
input=[
"good morning from litellm, attempting to embed data",
"lets test a second string for good measure",
],
input="good morning from litellm, attempting to embed data",
)
print(f"response:", response)
assert isinstance(
@ -206,7 +203,7 @@ def test_bedrock_embedding_titan():
pytest.fail(f"Error occurred: {e}")
# test_bedrock_embedding_titan()
test_bedrock_embedding_titan()
def test_bedrock_embedding_cohere():