From 8a1023fa2da0c470866685fcc4422fddec8cb8b0 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 2 Apr 2025 21:33:17 -0700
Subject: [PATCH] test image gen fix in build and test

---
 proxy_server_config.yaml | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/proxy_server_config.yaml b/proxy_server_config.yaml
index 16a93dbed1..364a218c33 100644
--- a/proxy_server_config.yaml
+++ b/proxy_server_config.yaml
@@ -46,12 +46,9 @@ model_list:
     model_info:
       mode: embedding
       base_model: text-embedding-ada-002
-  - model_name: dall-e-2
+  - model_name: dall-e-2 # some tests use dall-e-2 which is now deprecated, alias to dall-e-3
     litellm_params:
-      model: azure/
-      api_version: 2023-06-01-preview
-      api_base: https://openai-gpt-4-test-v-1.openai.azure.com/
-      api_key: os.environ/AZURE_API_KEY
+      model: openai/dall-e-3
   - model_name: openai-dall-e-3
     litellm_params:
       model: dall-e-3