From 6fe1b0f0e286c968d057ab39797e42fdca12ae4e Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Thu, 21 Mar 2024 19:25:54 +0200
Subject: [PATCH] #1933 Gemini 1.5 Pro

---
 public/index.html        | 1 +
 public/scripts/openai.js | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/public/index.html b/public/index.html
index 24cd3db85..7c82d0b1b 100644
--- a/public/index.html
+++ b/public/index.html
@@ -2572,6 +2572,7 @@
                             <div>
                                 <h4 data-i18n="Google Model">Google Model</h4>
                                 <select id="model_google_select">
+                                    <option value="gemini-1.5-pro">Gemini 1.5 Pro</option>
                                     <option value="gemini-pro">Gemini Pro</option>
                                     <option value="gemini-pro-vision">Gemini Pro Vision</option>
                                     <option value="text-bison-001">Bison Text</option>
diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index 20ed56f00..118a4f45a 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -115,6 +115,7 @@ const max_16k = 16383;
 const max_32k = 32767;
 const max_128k = 128 * 1000;
 const max_200k = 200 * 1000;
+const max_1mil = 1000 * 1000;
 const scale_max = 8191;
 const claude_max = 9000; // We have a proper tokenizer, so theoretically could be larger (up to 9k)
 const claude_100k_max = 99000;
@@ -3434,6 +3435,8 @@ async function onModelChange() {
     if (oai_settings.chat_completion_source == chat_completion_sources.MAKERSUITE) {
         if (oai_settings.max_context_unlocked) {
             $('#openai_max_context').attr('max', unlocked_max);
+        } else if (value === 'gemini-1.5-pro')  {
+            $('#openai_max_context').attr('max', max_1mil);
         } else if (value === 'gemini-pro') {
             $('#openai_max_context').attr('max', max_32k);
         } else if (value === 'gemini-pro-vision') {