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

fix bedrock embedding test

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-20 20:05:22 -07:00
parent b901a65572
commit 48c365976f
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_embedding.py
View file

@ -279,6 +279,7 @@ def test_bedrock_embedding_titan():
response = embedding( response = embedding(
model="bedrock/amazon.titan-embed-text-v1", model="bedrock/amazon.titan-embed-text-v1",
input=f"good morning from litellm, attempting to embed data {current_time}", # input should always be a string in this test input=f"good morning from litellm, attempting to embed data {current_time}", # input should always be a string in this test
aws_region_name="us-west-2",
) )
print(f"response:", response) print(f"response:", response)
assert isinstance( assert isinstance(