From 30bca8e39ba91c9a158223f580520e0fee407d63 Mon Sep 17 00:00:00 2001
From: M0cho <77959408+M0ch0@users.noreply.github.com>
Date: Fri, 15 Nov 2024 05:26:10 +0900
Subject: [PATCH] Supports GEMINI EXP

---
 public/index.html                               | 1 +
 public/scripts/extensions/caption/settings.html | 1 +
 public/scripts/openai.js                        | 3 +++
 src/endpoints/backends/chat-completions.js      | 2 +-
 src/prompt-converters.js                        | 1 +
 5 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/public/index.html b/public/index.html
index 1b125e3a0..7c130227b 100644
--- a/public/index.html
+++ b/public/index.html
@@ -2979,6 +2979,7 @@
                                         <option value="chat-bison-001">PaLM 2 Chat (Legacy)</option>
                                     </optgroup>
                                     <optgroup label="Subversions">
+                                        <option value="gemini-exp-1114">Gemini Experimental 2024-11-14</option>
                                         <option value="gemini-1.5-pro-exp-0801">Gemini 1.5 Pro Experiment 2024-08-01</option>
                                         <option value="gemini-1.5-pro-exp-0827">Gemini 1.5 Pro Experiment 2024-08-27</option>
                                         <option value="gemini-1.5-pro-latest">Gemini 1.5 Pro [latest]</option>
diff --git a/public/scripts/extensions/caption/settings.html b/public/scripts/extensions/caption/settings.html
index f0e0ca4eb..e9363eebe 100644
--- a/public/scripts/extensions/caption/settings.html
+++ b/public/scripts/extensions/caption/settings.html
@@ -58,6 +58,7 @@
                         <option data-type="google" value="gemini-1.5-flash-exp-0827">gemini-1.5-flash-exp-0827</option>
                         <option data-type="google" value="gemini-1.5-flash-8b-exp-0827">gemini-1.5-flash-8b-exp-0827</option>
                         <option data-type="google" value="gemini-1.5-flash-8b-exp-0924">gemini-1.5-flash-8b-exp-0924</option>
+                        <option data-type="google" value="gemini-exp-1114">gemini-exp-1114</option>
                         <option data-type="google" value="gemini-1.5-pro">gemini-1.5-pro</option>
                         <option data-type="google" value="gemini-1.5-pro-latest">gemini-1.5-pro-latest</option>
                         <option data-type="google" value="gemini-1.5-pro-001">gemini-1.5-pro-001</option>
diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index becc27fe6..a20378fd0 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -4056,6 +4056,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', max_2mil);
+        } else if (value.match('gemini-exp-1114')){
+            $('#openai_max_context').attr('max', max_32k);
         } else if (value.includes('gemini-1.5-pro')) {
             $('#openai_max_context').attr('max', max_2mil);
         } else if (value.includes('gemini-1.5-flash')) {
@@ -4742,6 +4744,7 @@ export function isImageInliningSupported() {
         'gemini-1.5-flash-8b',
         'gemini-1.5-flash-8b-exp-0827',
         'gemini-1.5-flash-8b-exp-0924',
+        'gemini-exp-1114',
         'gemini-1.0-pro-vision-latest',
         'gemini-1.5-pro',
         'gemini-1.5-pro-latest',
diff --git a/src/endpoints/backends/chat-completions.js b/src/endpoints/backends/chat-completions.js
index fe258cfc9..e614f8795 100644
--- a/src/endpoints/backends/chat-completions.js
+++ b/src/endpoints/backends/chat-completions.js
@@ -267,7 +267,7 @@ async function sendMakerSuiteRequest(request, response) {
             delete generationConfig.stopSequences;
         }
 
-        const should_use_system_prompt = (model.includes('gemini-1.5-flash') || model.includes('gemini-1.5-pro')) && request.body.use_makersuite_sysprompt;
+        const should_use_system_prompt = (model.includes('gemini-1.5-flash') || model.includes('gemini-1.5-pro') || model.match('gemini-exp-1114')) && request.body.use_makersuite_sysprompt;
         const prompt = convertGooglePrompt(request.body.messages, model, should_use_system_prompt, request.body.char_name, request.body.user_name);
         let body = {
             contents: prompt.contents,
diff --git a/src/prompt-converters.js b/src/prompt-converters.js
index 9c4502303..0513e7f4a 100644
--- a/src/prompt-converters.js
+++ b/src/prompt-converters.js
@@ -346,6 +346,7 @@ export function convertGooglePrompt(messages, model, useSysPrompt = false, charN
         'gemini-1.5-flash-8b',
         'gemini-1.5-flash-8b-exp-0827',
         'gemini-1.5-flash-8b-exp-0924',
+        'gemini-exp-1114',
         'gemini-1.5-pro',
         'gemini-1.5-pro-latest',
         'gemini-1.5-pro-001',