From 7a41c0952995faabb2f7cdc305258eea58656da8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 4 Oct 2025 11:27:29 -0700 Subject: [PATCH] test bedrock embedding tests marengo --- .../bedrock/embed/twelvelabs_marengo_transformation.py | 7 +++---- tests/llm_translation/test_bedrock_embedding.py | 3 ++- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/litellm/llms/bedrock/embed/twelvelabs_marengo_transformation.py b/litellm/llms/bedrock/embed/twelvelabs_marengo_transformation.py index 0d25440cd7..ecbe027899 100644 --- a/litellm/llms/bedrock/embed/twelvelabs_marengo_transformation.py +++ b/litellm/llms/bedrock/embed/twelvelabs_marengo_transformation.py @@ -89,10 +89,8 @@ class TwelveLabsMarengoEmbeddingConfig: - Audio inputs (async-invoke only) - S3 URLs for all media types (async-invoke only) """ - if inference_params.get("inputType"): - input_type = inference_params["inputType"] - else: - raise ValueError("input_type is required") + # Get input_type or default to "text" + input_type = inference_params.get("inputType") or inference_params.get("input_type") or "text" # Validate that async-invoke is used for video/audio if input_type in ["video", "audio"] and not async_invoke_route: @@ -136,6 +134,7 @@ class TwelveLabsMarengoEmbeddingConfig: for k, v in inference_params.items(): if k not in [ "inputType", + "input_type", # Exclude both camelCase and snake_case "inputText", "mediaSource", "bucketOwner", # Don't include bucketOwner in the request diff --git a/tests/llm_translation/test_bedrock_embedding.py b/tests/llm_translation/test_bedrock_embedding.py index 0a4e24a82f..88918674aa 100644 --- a/tests/llm_translation/test_bedrock_embedding.py +++ b/tests/llm_translation/test_bedrock_embedding.py @@ -140,7 +140,8 @@ def test_e2e_bedrock_embedding_image_twelvelabs_marengo(): response = litellm.embedding( model="bedrock/us.twelvelabs.marengo-embed-2-7-v1:0", input=[duck_img_base64], - aws_region_name="us-east-1" + aws_region_name="us-east-1", + input_type="image" ) # Validate response structure