From 2415e7d0a6bbb95f9418e82e2df544d7d5b6ece3 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 12 Nov 2024 16:18:49 -0800
Subject: [PATCH] TestAzureOpenAIDalle3

---
 tests/image_gen_tests/test_image_generation.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/tests/image_gen_tests/test_image_generation.py b/tests/image_gen_tests/test_image_generation.py
index 4e02b493c..411aab475 100644
--- a/tests/image_gen_tests/test_image_generation.py
+++ b/tests/image_gen_tests/test_image_generation.py
@@ -23,6 +23,10 @@ import litellm
 import json
 import tempfile
 from base_image_generation_test import BaseImageGenTest
+import logging
+from litellm._logging import verbose_logger
+
+verbose_logger.setLevel(logging.DEBUG)
 
 
 def get_vertex_ai_creds_json() -> dict:
@@ -114,10 +118,15 @@ class TestOpenAIDalle3(BaseImageGenTest):
 
 class TestAzureOpenAIDalle3(BaseImageGenTest):
     def get_base_image_generation_call_args(self) -> dict:
+        litellm.set_verbose = True
         return {
             "model": "azure/dall-e-3-test",
             "api_version": "2023-09-01-preview",
-            "base_model": "dall-e-3",
+            "metadata": {
+                "model_info": {
+                    "base_model": "dall-e-3",
+                }
+            },
         }