From ebd4f3d94713d536a0f5cca26f37d8430c83208e Mon Sep 17 00:00:00 2001 From: Alexsander Hamir Date: Tue, 10 Feb 2026 11:41:15 -0800 Subject: [PATCH] test: pass dummy api_key for AIML mocked test so validate_environment passes; add flaky(retries=0) --- tests/image_gen_tests/test_image_generation.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/image_gen_tests/test_image_generation.py b/tests/image_gen_tests/test_image_generation.py index 34ff080c507..3b4abeeb82f 100644 --- a/tests/image_gen_tests/test_image_generation.py +++ b/tests/image_gen_tests/test_image_generation.py @@ -183,6 +183,7 @@ class TestAimlImageGeneration(BaseImageGenTest): return {"model": "aiml/flux-pro/v1.1"} @pytest.mark.asyncio(scope="module") + @pytest.mark.flaky(retries=0) async def test_basic_image_generation(self): """Test basic image generation""" from unittest.mock import AsyncMock, patch @@ -213,8 +214,11 @@ class TestAimlImageGeneration(BaseImageGenTest): litellm.callbacks = [custom_logger] base_image_generation_call_args = self.get_base_image_generation_call_args() litellm.set_verbose = True + # Pass dummy api_key so validate_environment passes; HTTP is mocked response = await litellm.aimage_generation( - **base_image_generation_call_args, prompt="A image of a otter" + **base_image_generation_call_args, + prompt="A image of a otter", + api_key="test-key-mocked-no-credits-needed", ) print("FAL AI RESPONSE: ", response)