test_azure_img_gen_health_check

tests/litellm_utils_tests/test_health_check.py

@@ -85,12 +85,12 @@ async def test_openai_img_gen_health_check():
 
 
 async def test_azure_img_gen_health_check():
+    litellm._turn_on_debug()
     response = await litellm.ahealth_check(
         model_params={
             "model": "azure/dall-e-3",
             "api_base": os.getenv("AZURE_API_BASE"),
-            "api_key": os.getenv("AZURE_API_KEY"),
-            "api_version": "2023-06-01-preview",
+            "api_key": os.getenv("AZURE_API_KEY")
         },
         mode="image_generation",
         prompt="cute baby sea otter",
@@ -100,8 +100,6 @@ async def test_azure_img_gen_health_check():
     return response
 
 
-# asyncio.run(test_azure_img_gen_health_check())
-
 
 @pytest.mark.skip(reason="AWS Suspended Account")
 @pytest.mark.asyncio