From 859474dd44acb75f605b6509b1ed395ebbe87322 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 20 Dec 2025 14:00:42 +0530 Subject: [PATCH] fix patch --- .../image/test_bedrock_image_prepare_request.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/tests/test_litellm/llms/bedrock/image/test_bedrock_image_prepare_request.py b/tests/test_litellm/llms/bedrock/image/test_bedrock_image_prepare_request.py index 6c56ccc1ef7..5d4fd45271c 100644 --- a/tests/test_litellm/llms/bedrock/image/test_bedrock_image_prepare_request.py +++ b/tests/test_litellm/llms/bedrock/image/test_bedrock_image_prepare_request.py @@ -1,15 +1,17 @@ -from unittest.mock import patch, MagicMock +from unittest.mock import MagicMock, patch from litellm.llms.bedrock.image_generation.image_handler import BedrockImageGeneration + def test_bedrock_image_prepare_request_with_arn() -> None: + """Test that ARN model identifiers are correctly URL-encoded in the request endpoint.""" dummy_arn = "arn:aws:bedrock:us-east-1:123456789012:application-inference-profile/abcdefghi123" image_generation = BedrockImageGeneration() with ( - patch("litellm.llms.bedrock.image.image_handler.BedrockImageGeneration._get_boto_credentials_from_optional_params"), - patch("litellm.llms.bedrock.image.image_handler.BedrockImageGeneration.get_request_headers"), + patch("litellm.llms.bedrock.image_generation.image_handler.BedrockImageGeneration._get_boto_credentials_from_optional_params"), + patch("litellm.llms.bedrock.image_generation.image_handler.BedrockImageGeneration.get_request_headers"), ): request = image_generation._prepare_request( model="amazon.nova-canvas-v1:0", @@ -27,11 +29,12 @@ def test_bedrock_image_prepare_request_with_arn() -> None: def test_bedrock_image_prepare_request_without_arn() -> None: + """Test that regular model identifiers are used directly in the request endpoint.""" image_generation = BedrockImageGeneration() with ( - patch("litellm.llms.bedrock.image.image_handler.BedrockImageGeneration._get_boto_credentials_from_optional_params"), - patch("litellm.llms.bedrock.image.image_handler.BedrockImageGeneration.get_request_headers"), + patch("litellm.llms.bedrock.image_generation.image_handler.BedrockImageGeneration._get_boto_credentials_from_optional_params"), + patch("litellm.llms.bedrock.image_generation.image_handler.BedrockImageGeneration.get_request_headers"), ): request = image_generation._prepare_request( model="amazon.nova-canvas-v1:0",