diff --git a/proxy_server_config.yaml b/proxy_server_config.yaml index 5d3d810926..d9838c852a 100644 --- a/proxy_server_config.yaml +++ b/proxy_server_config.yaml @@ -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 diff --git a/tests/otel_tests/test_otel.py b/tests/otel_tests/test_otel.py index a0f58dd5b8..9ded859eb9 100644 --- a/tests/otel_tests/test_otel.py +++ b/tests/otel_tests/test_otel.py @@ -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", ], diff --git a/tests/test_health.py b/tests/test_health.py index 15dc2330ff..cc551fd938 100644 --- a/tests/test_health.py +++ b/tests/test_health.py @@ -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, } diff --git a/tests/test_keys.py b/tests/test_keys.py index 6d4c24aa80..e6bda59c2c 100644 --- a/tests/test_keys.py +++ b/tests/test_keys.py @@ -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}", diff --git a/tests/test_openai_endpoints.py b/tests/test_openai_endpoints.py index 8a3f9361ba..e898b88a55 100644 --- a/tests/test_openai_endpoints.py +++ b/tests/test_openai_endpoints.py @@ -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", }