This commit is contained in:
Ishaan Jaff
2025-09-08 18:46:20 -07:00
parent 2cc85936ed
commit 87f116bded
@@ -198,17 +198,6 @@ class TestAzureOpenAIDalle3(BaseImageGenTest):
}
class TestAzureFoundryFlux(BaseImageGenTest):
def get_base_image_generation_call_args(self) -> dict:
litellm.set_verbose = True
return {
"model": "azure_ai/FLUX.1-Kontext-pro",
"api_base": os.getenv("AZURE_FLUX_API_BASE"),
"api_key": os.getenv("AZURE_GPT5_API_KEY"),
"n": 1,
"quality": "standard",
}
@pytest.mark.flaky(retries=3, delay=1)
def test_image_generation_azure_dall_e_3():