fix(tests): drop dall-e-only test classes; route live image tests via gpt-image-1

Second wave of failures from the 2026-05-12 DALL-E shutdown:
- tests/image_gen_tests/test_image_edits.py::TestOpenAIImageEditDallE2
  and tests/image_gen_tests/test_image_generation.py::TestOpenAIDalle3
  are explicitly named for the deprecated models and can't pass; remove.
  gpt-image-1 coverage already exists in sibling classes.
- tests/local_testing/test_router.py image gen tests use dall-e-3 only
  as a routing example; swap to gpt-image-1.
- tests/local_testing/test_custom_callback_input.py image_generation
  success/failure paths swapped to gpt-image-1.
This commit is contained in:
Yuneng Jiang
2026-05-12 16:16:59 -07:00
parent 8c8621ece3
commit 945b10ded4
4 changed files with 8 additions and 27 deletions
-14
View File
@@ -133,20 +133,6 @@ class TestOpenAIImageEditGPTImage1(BaseLLMImageEditTest):
}
class TestOpenAIImageEditDallE2(BaseLLMImageEditTest):
"""
Concrete implementation of BaseLLMImageEditTest for OpenAI DALL-E-2 image edits.
DALL-E-2 only supports a single image (not an array).
"""
def get_base_image_edit_call_args(self) -> dict:
"""Return base call args for OpenAI DALL-E-2 image edit (single image only)"""
return {
"model": "dall-e-2",
"image": SINGLE_TEST_IMAGE,
}
class TestAzureAIFlux2ImageEdit(BaseLLMImageEditTest):
"""
Concrete implementation of BaseLLMImageEditTest for Azure AI FLUX 2 image edits.
@@ -163,11 +163,6 @@ class TestBedrockNovaCanvasColorGuidedGeneration(BaseImageGenTest):
}
class TestOpenAIDalle3(BaseImageGenTest):
def get_base_image_generation_call_args(self) -> dict:
return {"model": "dall-e-3"}
class TestOpenAIGPTImage1(BaseImageGenTest):
def get_base_image_generation_call_args(self) -> dict:
return {"model": "gpt-image-1"}
@@ -930,7 +930,7 @@ def test_image_generation_openai():
response = litellm.image_generation(
prompt="A cute baby sea otter",
model="openai/dall-e-3",
model="openai/gpt-image-1",
api_key=os.getenv("OPENAI_API_KEY"),
)
@@ -948,7 +948,7 @@ def test_image_generation_openai():
try:
response = litellm.image_generation(
prompt="A cute baby sea otter",
model="dall-e-2",
model="gpt-image-1",
api_key="my-bad-api-key",
)
except Exception:
+6 -6
View File
@@ -995,15 +995,15 @@ async def test_aimg_gen_on_router():
try:
model_list = [
{
"model_name": "dall-e-3",
"model_name": "gpt-image-1",
"litellm_params": {
"model": "dall-e-3",
"model": "gpt-image-1",
},
}
]
router = Router(model_list=model_list, num_retries=3)
response = await router.aimage_generation(
model="dall-e-3", prompt="A cute baby sea otter"
model="gpt-image-1", prompt="A cute baby sea otter"
)
print(response)
assert len(response.data) > 0
@@ -1030,15 +1030,15 @@ def test_img_gen_on_router():
try:
model_list = [
{
"model_name": "dall-e-3",
"model_name": "gpt-image-1",
"litellm_params": {
"model": "dall-e-3",
"model": "gpt-image-1",
},
}
]
router = Router(model_list=model_list)
response = router.image_generation(
model="dall-e-3", prompt="A cute baby sea otter"
model="gpt-image-1", prompt="A cute baby sea otter"
)
print(response)
assert len(response.data) > 0