From 912ca6255ccb385abfa33ecececfe8cbbd108e34 Mon Sep 17 00:00:00 2001 From: "devin-ai-integration[bot]" <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 1 Jul 2026 22:57:29 -0700 Subject: [PATCH] test(bedrock): switch image gen live test off EOL Titan to Nova Canvas (#31937) Co-authored-by: mateo Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/image_gen_tests/test_bedrock_image_gen_unit_tests.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/image_gen_tests/test_bedrock_image_gen_unit_tests.py b/tests/image_gen_tests/test_bedrock_image_gen_unit_tests.py index 36ae9e1df67..6925bb2abc5 100644 --- a/tests/image_gen_tests/test_bedrock_image_gen_unit_tests.py +++ b/tests/image_gen_tests/test_bedrock_image_gen_unit_tests.py @@ -527,12 +527,11 @@ def test_backward_compatibility_regular_nova_model(): assert result["imageGenerationConfig"]["cfg_scale"] == 7 -def test_amazon_titan_image_gen(): - """Test Amazon Titan image generation with cost tracking.""" +def test_amazon_nova_canvas_image_gen(): + """Test Amazon Nova Canvas image generation with cost tracking.""" from litellm import image_generation - # Use v2 as v1 has reached end of life - model_id = "bedrock/amazon.titan-image-generator-v2:0" + model_id = "bedrock/amazon.nova-canvas-v1:0" response = litellm.image_generation( model=model_id,