phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_image_generation.py): fix azure dall e test

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-03 23:32:19 -07:00
parent 8ebae96008
commit 8625280c32
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_image_generation.py
View file

@ -115,7 +115,9 @@ async def test_async_image_generation_openai():
async def test_async_image_generation_azure(): async def test_async_image_generation_azure():
try: try:
response = await litellm.aimage_generation( 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}") print(f"response: {response}")
except litellm.RateLimitError as e: except litellm.RateLimitError as e: