From 63a5a2d8c2a5d62d397124f290920522a9c0f2c7 Mon Sep 17 00:00:00 2001 From: Moe Khalil Date: Wed, 28 Oct 2026 02:18:28 +0000 Subject: [PATCH] test(images): assert the public invalid retry error mapping Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/unit/images/test_main.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/unit/images/test_main.py b/tests/unit/images/test_main.py index 50f3b2f36eb..192c861e258 100644 --- a/tests/unit/images/test_main.py +++ b/tests/unit/images/test_main.py @@ -160,7 +160,7 @@ async def test_image_generation_rejects_non_integer_retries_before_sending(is_as base_url="https://image.test/v1", http_client=httpx.AsyncClient(transport=transport), ) as client: - with pytest.raises(litellm.UnprocessableEntityError) as raised: + with pytest.raises(litellm.BadRequestError) as raised: await litellm.aimage_generation( model="openai/gpt-image-1-mini", prompt="a red circle", @@ -175,7 +175,7 @@ async def test_image_generation_rejects_non_integer_retries_before_sending(is_as base_url="https://image.test/v1", http_client=httpx.Client(transport=transport), ) as client: - with pytest.raises(litellm.UnprocessableEntityError) as raised: + with pytest.raises(litellm.BadRequestError) as raised: litellm.image_generation( model="openai/gpt-image-1-mini", prompt="a red circle", @@ -184,7 +184,8 @@ async def test_image_generation_rejects_non_integer_retries_before_sending(is_as client=client, max_retries="invalid", ) - assert raised.value.status_code == 422 + assert raised.value.status_code == 400 + assert "max retries must be an int" in raised.value.message assert requests.qsize() == 0