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

test_image_generation_openai

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-25 14:57:40 -07:00
parent 2c52791b83
commit 5cca4c8b4f
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/local_testing/test_custom_callback_input.py
View file

@ -930,10 +930,8 @@ def test_image_generation_openai():
response = litellm.image_generation(
prompt="A cute baby sea otter",
model="azure/dall-e-3-test",
api_version="2023-12-01-preview",
api_base=os.getenv("AZURE_SWEDEN_API_BASE"),
api_key=os.getenv("AZURE_SWEDEN_API_KEY"),
model="openai/dall-e-3",
api_key=os.getenv("OPENAI_API_KEY"),
)
print(f"response: {response}")