From 8625280c32a66f4865367056a26dabe7346e1ef3 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 3 Jun 2024 23:32:19 -0700 Subject: [PATCH] test(test_image_generation.py): fix azure dall e test --- litellm/tests/test_image_generation.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_image_generation.py b/litellm/tests/test_image_generation.py index 4fabfd04f..fea014e67 100644 --- a/litellm/tests/test_image_generation.py +++ b/litellm/tests/test_image_generation.py @@ -115,7 +115,9 @@ async def test_async_image_generation_openai(): async def test_async_image_generation_azure(): try: response = await litellm.aimage_generation( - prompt="A cute baby sea otter", model="azure/dall-e-3-test" + prompt="A cute baby sea otter", + model="azure/dall-e-3-test", + api_version="2023-09-01-preview", ) print(f"response: {response}") except litellm.RateLimitError as e: