From 899b5710d18353166bcbbc3c5e0c456c267ecfa5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Mon, 30 Mar 2026 16:14:58 -0700 Subject: [PATCH] TestAzureAIFlux2ImageEdit --- tests/image_gen_tests/test_image_edits.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/image_gen_tests/test_image_edits.py b/tests/image_gen_tests/test_image_edits.py index 8fe2e25a99..3504065a10 100644 --- a/tests/image_gen_tests/test_image_edits.py +++ b/tests/image_gen_tests/test_image_edits.py @@ -158,7 +158,7 @@ class TestAzureAIFlux2ImageEdit(BaseLLMImageEditTest): return { "model": "azure_ai/flux.2-pro", "image": SINGLE_TEST_IMAGE, - "api_base": "https://litellm-ci-cd-prod.services.ai.azure.com", + "api_base": os.getenv("AZURE_AI_API_BASE"), "api_key": os.getenv("AZURE_AI_API_KEY"), "api_version": "preview", }