From aa992fb86ff28712290e10e4415aef23d7f1d021 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 20 Sep 2025 11:35:31 -0700 Subject: [PATCH] test e2e bedrock embedding --- tests/llm_translation/test_bedrock_embedding.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/llm_translation/test_bedrock_embedding.py b/tests/llm_translation/test_bedrock_embedding.py index 4adb38e867..15a615b8cc 100644 --- a/tests/llm_translation/test_bedrock_embedding.py +++ b/tests/llm_translation/test_bedrock_embedding.py @@ -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