scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

test: pass dummy api_key for AIML mocked test so validate_environment passes; add flaky(retries=0)

Browse source
This commit is contained in:
Alexsander Hamir 2026-02-10 11:41:15 -08:00
parent ca7c74e358
commit ebd4f3d947
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/image_gen_tests/test_image_generation.py
View file

@ -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)