From 5cca4c8b4fb36b71ee345158fa964f179e0fefaf Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 25 Oct 2025 14:57:40 -0700 Subject: [PATCH] test_image_generation_openai --- tests/local_testing/test_custom_callback_input.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/local_testing/test_custom_callback_input.py b/tests/local_testing/test_custom_callback_input.py index e711344f654..7f4da72e04f 100644 --- a/tests/local_testing/test_custom_callback_input.py +++ b/tests/local_testing/test_custom_callback_input.py @@ -930,10 +930,8 @@ def test_image_generation_openai(): response = litellm.image_generation( prompt="A cute baby sea otter", - model="azure/dall-e-3-test", - api_version="2023-12-01-preview", - api_base=os.getenv("AZURE_SWEDEN_API_BASE"), - api_key=os.getenv("AZURE_SWEDEN_API_KEY"), + model="openai/dall-e-3", + api_key=os.getenv("OPENAI_API_KEY"), ) print(f"response: {response}")