From 20d84ddef1ed0f2079a3457bc5af791b198749ce Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 2 Apr 2025 21:03:59 -0700 Subject: [PATCH] get_base_image_generation_call_args --- tests/image_gen_tests/test_image_generation.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/image_gen_tests/test_image_generation.py b/tests/image_gen_tests/test_image_generation.py index 63274dfedf3..bee9d1f7d50 100644 --- a/tests/image_gen_tests/test_image_generation.py +++ b/tests/image_gen_tests/test_image_generation.py @@ -168,6 +168,8 @@ class TestAzureOpenAIDalle3(BaseImageGenTest): return { "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"), "metadata": { "model_info": { "base_model": "azure/dall-e-3",