scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test_aimg_gen_on_router

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-25 15:11:21 -07:00
parent 20d8345a7c
commit ab0fc0a30d
1 changed files with 1 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/local_testing/test_router.py
View file

@ -1119,16 +1119,7 @@ async def test_aimg_gen_on_router():
"litellm_params": {
"model": "dall-e-3",
},
},
{
"model_name": "dall-e-3",
"litellm_params": {
"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"),
},
},
}
]
router = Router(model_list=model_list, num_retries=3)
response = await router.aimage_generation(