mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-05 10:24:03 +00:00
test_img_gen_on_router
This commit is contained in:
@@ -1154,16 +1154,7 @@ def test_img_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)
|
||||
response = router.image_generation(
|
||||
|
||||
Reference in New Issue
Block a user