From c9f90a2bd34737a2ca2e32a5a9a0a95d8393b331 Mon Sep 17 00:00:00 2001
From: InterestingDarkness <computeridiot999@gmail.com>
Date: Wed, 28 May 2025 09:26:06 +0800
Subject: [PATCH] Update Gemini model options in Vertex AI section

- Removed outdated model options from the Gemini 2.0 optgroup in index.html.
- Added 'gemini-2.0-flash-preview-image-generation' to the image generation models in chat-completions.js for improved functionality.
---
 public/index.html                          | 34 +---------------------
 src/endpoints/backends/chat-completions.js |  1 +
 2 files changed, 2 insertions(+), 33 deletions(-)

diff --git a/public/index.html b/public/index.html
index 3835913a8..74f88ef3e 100644
--- a/public/index.html
+++ b/public/index.html
@@ -3328,43 +3328,11 @@
                                         <option value="gemini-2.5-flash-preview-04-17">gemini-2.5-flash-preview-04-17</option>
                                     </optgroup>
                                     <optgroup id="vertexai_full_gemini_20" label="Gemini 2.0" data-mode="full" style="display: none;">
-                                        <option value="gemini-2.0-pro-exp-02-05">gemini-2.0-pro-exp-02-05 → 2.5-pro-exp-03-25</option>
-                                        <option value="gemini-2.0-pro-exp">gemini-2.0-pro-exp → 2.5-pro-exp-03-25</option>
-                                        <option value="gemini-exp-1206">gemini-exp-1206 → 2.5-pro-exp-03-25</option>
                                         <option value="gemini-2.0-flash-001">gemini-2.0-flash-001</option>
-                                        <option value="gemini-2.0-flash-exp-image-generation">gemini-2.0-flash-exp-image-generation</option>
                                         <option value="gemini-2.0-flash-exp">gemini-2.0-flash-exp</option>
+                                        <option value="gemini-2.0-flash-preview-image-generation">gemini-2.0-flash-preview-image-generation</option>
                                         <option value="gemini-2.0-flash">gemini-2.0-flash</option>
-                                        <option value="gemini-2.0-flash-thinking-exp-01-21">gemini-2.0-flash-thinking-exp-01-21 → 2.5-flash-preview-04-17</option>
-                                        <option value="gemini-2.0-flash-thinking-exp-1219">gemini-2.0-flash-thinking-exp-1219 → 2.5-flash-preview-04-17</option>
-                                        <option value="gemini-2.0-flash-thinking-exp">gemini-2.0-flash-thinking-exp → 2.5-flash-preview-04-17</option>
                                         <option value="gemini-2.0-flash-lite-001">gemini-2.0-flash-lite-001</option>
-                                        <option value="gemini-2.0-flash-lite-preview-02-05">gemini-2.0-flash-lite-preview-02-05</option>
-                                        <option value="gemini-2.0-flash-lite-preview">gemini-2.0-flash-lite-preview</option>
-                                    </optgroup>
-                                    <optgroup id="vertexai_full_gemini_15" label="Gemini 1.5" data-mode="full" style="display: none;">
-                                        <option value="gemini-1.5-pro-latest">gemini-1.5-pro-latest</option>
-                                        <option value="gemini-1.5-pro-002">gemini-1.5-pro-002</option>
-                                        <option value="gemini-1.5-pro-001">gemini-1.5-pro-001</option>
-                                        <option value="gemini-1.5-pro">gemini-1.5-pro</option>
-                                        <option value="gemini-1.5-flash-latest">gemini-1.5-flash-latest</option>
-                                        <option value="gemini-1.5-flash-002">gemini-1.5-flash-002</option>
-                                        <option value="gemini-1.5-flash-001">gemini-1.5-flash-001</option>
-                                        <option value="gemini-1.5-flash">gemini-1.5-flash</option>
-                                        <option value="gemini-1.5-flash-8b-001">gemini-1.5-flash-8b-001</option>
-                                        <option value="gemini-1.5-flash-8b-exp-0924">gemini-1.5-flash-8b-exp-0924</option>
-                                        <option value="gemini-1.5-flash-8b-exp-0827">gemini-1.5-flash-8b-exp-0827</option>
-                                        <option value="gemini-1.5-flash-8b">gemini-1.5-flash-8b</option>
-                                    </optgroup>
-                                    <optgroup id="vertexai_full_gemma" label="Gemma" data-mode="full" style="display: none;">
-                                        <option value="gemma-3-27b-it">gemma-3-27b-it</option>
-                                        <option value="gemma-3-12b-it">gemma-3-12b-it</option>
-                                        <option value="gemma-3-4b-it">gemma-3-4b-it</option>
-                                        <option value="gemma-3-1b-it">gemma-3-1b-it</option>
-                                    </optgroup>
-                                    <optgroup id="vertexai_full_learnlm" label="LearnLM" data-mode="full" style="display: none;">
-                                        <option value="learnlm-2.0-flash-experimental">learnlm-2.0-flash-experimental</option>
-                                        <option value="learnlm-1.5-pro-experimental">learnlm-1.5-pro-experimental</option>
                                     </optgroup>
                                 </select>
                             </div>
diff --git a/src/endpoints/backends/chat-completions.js b/src/endpoints/backends/chat-completions.js
index ec7a6d6c7..e7b2979f8 100644
--- a/src/endpoints/backends/chat-completions.js
+++ b/src/endpoints/backends/chat-completions.js
@@ -408,6 +408,7 @@ async function sendMakerSuiteRequest(request, response) {
         const imageGenerationModels = [
             'gemini-2.0-flash-exp',
             'gemini-2.0-flash-exp-image-generation',
+            'gemini-2.0-flash-preview-image-generation',
         ];
 
         // These models do not support setting the threshold to OFF at all.