mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 03:34:10 +00:00
fix bedrock embedding test
This commit is contained in:
parent
e43604ef91
commit
9ddd9d628a
1 changed files with 1 additions and 0 deletions
|
@ -279,6 +279,7 @@ def test_bedrock_embedding_titan():
|
|||
response = embedding(
|
||||
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
|
||||
aws_region_name="us-west-2",
|
||||
)
|
||||
print(f"response:", response)
|
||||
assert isinstance(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue