scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

fix(bedrock): keep the provider status code on unprocessable image errors (#43416)

Browse source 
Co-authored-by: Krrish Dholakia <krrishdholakia@berri.ai>
Co-authored-by: dbalintx <dbalintx@amazon.com>
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
devin-ai-integration[bot] 2026-09-27 05:10:24 +00:00 • committed by GitHub
parent c1f761eba5
commit b831e9b4ac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 37 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/litellm_core_utils/exception_mapping_utils.py
View file

@ -954,7 +954,7 @@ def _map_bedrock_exception(
llm_provider="bedrock",
response=getattr(original_exception, "response", None),
)
elif "Could not process image" in error_str:
elif "Could not process image" in error_str and getattr(original_exception, "status_code", 500) == 500:
raise litellm.InternalServerError(
message=f"BedrockException - {error_str}",
model=model,

37
tests/unit/litellm_core_utils/test_exception_mapping_utils.py
View file

@ -1280,7 +1280,7 @@ def test_bedrock_500_preserves_provider_response_headers():
"bedrock",
400,
'{"message":"Could not process image"}',
litellm.InternalServerError,
litellm.BadRequestError,
),
],
)
@ -1313,6 +1313,41 @@ def test_bedrock_classified_errors_preserve_provider_response_headers(
assert exc_info.value.response.headers["x-amzn-requestid"] == "req-classified"
@pytest.mark.parametrize(
"status_code, expected_exception",
[
(400, litellm.BadRequestError),
(503, litellm.ServiceUnavailableError),
(500, litellm.InternalServerError),
],
)
def test_bedrock_unprocessable_image_keeps_provider_status_code(status_code, expected_exception):
"""An unprocessable image maps to the status Bedrock sent, so the 400 it returns stays a client error."""
provider_message = '{"message":"The model returned the following errors: Could not process image"}'
provider_response = httpx.Response(
status_code=status_code,
text=provider_message,
request=httpx.Request("POST", "https://bedrock-runtime.us-east-1.amazonaws.com/"),
)
original_exception = BedrockError(
status_code=status_code,
message=provider_message,
headers=provider_response.headers,
response=provider_response,
)
with pytest.raises(expected_exception) as exc_info:
exception_type(
model="anthropic.claude-haiku-4-5-20251001-v1:0",
original_exception=original_exception,
custom_llm_provider="bedrock",
completion_kwargs={},
extra_kwargs={},
)
assert exc_info.value.status_code == status_code
@pytest.mark.parametrize(
"status_code, provider_message",
[