scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

test_e2e_bedrock_embedding_image_twelvelabs_marengodocs fix

Browse source
This commit is contained in:
Ishaan Jaffer 2025-09-20 09:34:08 -07:00
parent 7236202e41
commit 39cf690d50
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/llm_translation/test_bedrock_embedding.py
View file

@ -78,13 +78,15 @@ def test_bedrock_embedding_models(model, input_type, embed_response):
pytest.fail(f"Error occurred: {e}")
@patch.dict(os.environ, {}, clear=True)
def test_e2e_bedrock_embedding():
"""
Test text embedding with TwelveLabs Marengo.
Validates that the transformation properly extracts embedding data from TwelveLabs response format.
"""
print("Testing text embedding...")
original_region_name = os.environ.get("AWS_REGION_NAME")
os.environ.pop("AWS_REGION_NAME")
litellm._turn_on_debug()
response = litellm.embedding(
model="bedrock/us.twelvelabs.marengo-embed-2-7-v1:0",
@ -115,15 +117,18 @@ def test_e2e_bedrock_embedding():
print(f"Text embedding successful! Vector size: {len(embedding_obj.embedding)}, Response: {response}")
# Restore original region name
if original_region_name:
os.environ["AWS_REGION_NAME"] = original_region_name
@patch.dict(os.environ, {}, clear=True)
def test_e2e_bedrock_embedding_image_twelvelabs_marengo():
"""
Test image embedding with TwelveLabs Marengo.
Validates that the transformation properly extracts embedding data from TwelveLabs response format for images.
"""
print("Testing image embedding...")
original_region_name = os.environ.get("AWS_REGION_NAME")
os.environ.pop("AWS_REGION_NAME")
litellm._turn_on_debug()
# Load duck.png and convert to base64
@ -165,3 +170,6 @@ def test_e2e_bedrock_embedding_image_twelvelabs_marengo():
print(f"Image embedding successful! Vector size: {len(embedding_obj.embedding)}, Response: {response}")
# Restore original region name
if original_region_name:
os.environ["AWS_REGION_NAME"] = original_region_name