From e797f301d79b0ebe234bcc68d2a187d0d5b23d23 Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Fri, 9 Feb 2024 01:14:39 +0900
Subject: [PATCH] Add new Gemini models to ModelList and requestChatDataMain

---
 src/lib/UI/ModelList.svelte | 10 ++++++++++
 src/ts/process/request.ts   |  4 +++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/lib/UI/ModelList.svelte b/src/lib/UI/ModelList.svelte
index 074e06a1..961ecd55 100644
--- a/src/lib/UI/ModelList.svelte
+++ b/src/lib/UI/ModelList.svelte
@@ -82,6 +82,14 @@
                 return "Gemini Pro"
             case 'horde:::auto':
                 return 'Horde Auto Model'
+            case 'gpt4_0125':
+                return 'GPT-4 Turbo 0125'
+            case 'gpt35_0125':
+                return 'GPT-3.5 Turbo 0125'
+            case 'gemini-ultra':
+                return 'Gemini Ultra'
+            case 'gemini-ultra-vision':
+                return 'Gemini Ultra Vision'
             default:
                 if(name.startsWith("horde:::")){
                     const split = name.split(":::")
@@ -183,6 +191,8 @@
                 <Arcodion name="Google Gemini">
                     <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('gemini-pro')}}>Gemini Pro</button>
                     <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('gemini-pro-vision')}}>Gemini Pro Vision</button>
+                    <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('gemini-ultra')}}>Gemini Ultra</button>
+                    <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('gemini-ultra-vision')}}>Gemini Ultra Vision</button>
                 </Arcodion>
                 <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('kobold')}}>Kobold</button>
             {/if}
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index adb47879..746128c5 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -1064,7 +1064,9 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
             }
         }
         case 'gemini-pro':
-        case 'gemini-pro-vision':{
+        case 'gemini-pro-vision':
+        case 'gemini-ultra':
+        case 'gemini-ultra-vision':{
             interface GeminiPart{
                 text?:string
                 "inlineData"?: {