test_img_gen_on_router

This commit is contained in:
Ishaan Jaffer
2025-10-25 15:11:52 -07:00
parent ab0fc0a30d
commit 679374fe79
+1 -10
View File
@@ -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(