tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-20 10:24:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

test e2e bedrock embedding

Browse Source
This commit is contained in:
Ishaan Jaffer
2025-09-20 11:35:31 -07:00
parent 3c2c14ffe4
commit aa992fb86f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/llm_translation/test_bedrock_embedding.py
+3 -3
View File
@@ -85,13 +85,13 @@ def test_e2e_bedrock_embedding():
"""
print("Testing text embedding...")
original_region_name = os.environ.get("AWS_REGION_NAME")
os.environ.pop("AWS_REGION_NAME")
os.environ["AWS_REGION_NAME"] = "us-east-1"
litellm._turn_on_debug()
response = litellm.embedding(
model="bedrock/us.twelvelabs.marengo-embed-2-7-v1:0",
input=["Hello world from LiteLLM with TwelveLabs Marengo!"],
aws_region_name="us-east-1"
)
# Validate response structure
@@ -128,7 +128,7 @@ def test_e2e_bedrock_embedding_image_twelvelabs_marengo():
"""
print("Testing image embedding...")
original_region_name = os.environ.get("AWS_REGION_NAME")
os.environ.pop("AWS_REGION_NAME")
os.environ["AWS_REGION_NAME"] = "us-east-1"
litellm._turn_on_debug()
# Load duck.png and convert to base64