scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

fix patch

Browse source
This commit is contained in:
Ishaan Jaffer 2025-12-20 14:00:42 +05:30
parent 13fa573d61
commit 859474dd44
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/test_litellm/llms/bedrock/image/test_bedrock_image_prepare_request.py
View file

@ -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",