scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-05 07:37:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 233

test(images): assert the public invalid retry error mapping

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Moe Khalil 2026-10-28 02:18:28 +00:00
parent a6342918dd
commit 63a5a2d8c2
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/unit/images/test_main.py
View file

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