fix test_aimage_generation_bedrock_with_optional_params

This commit is contained in:
Ishaan Jaff 2024-11-12 16:58:07 -08:00
parent 501ced09ba
commit 84603763cc

View file

@ -181,6 +181,7 @@ def test_image_generation_azure_dall_e_3():
@pytest.mark.asyncio
async def test_aimage_generation_bedrock_with_optional_params():
try:
litellm.in_memory_llm_clients_cache = {}
response = await litellm.aimage_generation(
prompt="A cute baby sea otter",
model="bedrock/stability.stable-diffusion-xl-v1",