TestAzureAIFlux2ImageEdit

This commit is contained in:
Ishaan Jaffer 2026-03-30 16:14:58 -07:00
parent 4c00a14ce0
commit 899b5710d1

View File

@ -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",
}