fix(tests): swap dall-e to gpt-image-1 after openai deprecation

DALL-E 2 and DALL-E 3 were removed from the OpenAI API on 2026-05-12,
causing e2e image-generation tests to fail with "model does not exist".
Swap all live-API DALL-E references in proxy-backed tests to gpt-image-1
and update the dall-e-2 alias in proxy_server_config.yaml to point at
openai/gpt-image-1 (preserves any historical dall-e-2 callers).
This commit is contained in:
Yuneng Jiang 2026-05-12 16:07:59 -07:00 committed by shivam
parent 6bffa3e998
commit 7066587f5c
No known key found for this signature in database
5 changed files with 9 additions and 9 deletions

View File

@ -38,9 +38,9 @@ model_list:
model_info:
mode: embedding
base_model: text-embedding-ada-002
- model_name: dall-e-2 # some tests use dall-e-2 which is now deprecated, alias to dall-e-3
- model_name: dall-e-2 # dall-e-2 and dall-e-3 were deprecated 2026-05-12; alias to gpt-image-1
litellm_params:
model: openai/dall-e-3
model: openai/gpt-image-1
- model_name: openai-dall-e-3
litellm_params:
model: dall-e-3

View File

@ -13,7 +13,7 @@ async def generate_key(
models=[
"gpt-4",
"text-embedding-ada-002",
"dall-e-2",
"gpt-image-1",
"fake-openai-endpoint",
"mistral-embed",
],

View File

@ -31,7 +31,7 @@ async def generate_key(session):
url = "http://0.0.0.0:4000/key/generate"
headers = {"Authorization": "Bearer sk-1234", "Content-Type": "application/json"}
data = {
"models": ["gpt-4", "text-embedding-ada-002", "dall-e-2"],
"models": ["gpt-4", "text-embedding-ada-002", "gpt-image-1"],
"duration": None,
}

View File

@ -62,7 +62,7 @@ async def generate_key(
i,
budget=None,
budget_duration=None,
models=["azure-models", "gpt-4", "dall-e-3"],
models=["azure-models", "gpt-4", "gpt-image-1"],
max_parallel_requests: Optional[int] = None,
user_id: Optional[str] = None,
team_id: Optional[str] = None,
@ -235,7 +235,7 @@ async def chat_completion(session, key, model="gpt-4"):
pass
async def image_generation(session, key, model="dall-e-3"):
async def image_generation(session, key, model="gpt-image-1"):
url = "http://0.0.0.0:4000/v1/images/generations"
headers = {
"Authorization": f"Bearer {key}",

View File

@ -23,7 +23,7 @@ async def generate_key(
models=[
"gpt-4",
"text-embedding-ada-002",
"dall-e-2",
"gpt-image-1",
"fake-openai-endpoint-2",
"mistral-embed",
],
@ -56,7 +56,7 @@ async def new_user(session):
url = "http://0.0.0.0:4000/user/new"
headers = {"Authorization": "Bearer sk-1234", "Content-Type": "application/json"}
data = {
"models": ["gpt-4", "text-embedding-ada-002", "dall-e-2"],
"models": ["gpt-4", "text-embedding-ada-002", "gpt-image-1"],
"duration": None,
}
@ -264,7 +264,7 @@ async def image_generation(session, key):
"Content-Type": "application/json",
}
data = {
"model": "dall-e-2",
"model": "gpt-image-1",
"prompt": "A cute baby sea otter",
}