From 14ef1d95ce420041af3c0377a8f11a5299b407e1 Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Tue, 14 May 2024 02:35:00 +0900
Subject: [PATCH] feat: Add GPT-4o mode

---
 src/lib/UI/ModelList.svelte | 6 ++++--
 src/ts/model/names.ts       | 4 ++++
 src/ts/process/request.ts   | 4 ++++
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/lib/UI/ModelList.svelte b/src/lib/UI/ModelList.svelte
index ab0152e3..2f7bf335 100644
--- a/src/lib/UI/ModelList.svelte
+++ b/src/lib/UI/ModelList.svelte
@@ -35,7 +35,10 @@
                 <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt35')}}>GPT-3.5 Turbo</button>
                 <button class="p-2 hover:text-green-500" on:click={() => {changeModel('instructgpt35')}}>GPT-3.5 Instruct</button>
                 <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4_turbo')}}>GPT-4 Turbo</button>
+                <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4o')}}>GPT-4o</button>
                 {#if showUnrec}
+                    <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4')}}>GPT-4 (Old)</button>
+                    <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4_32k')}}>GPT-4 32K (Old)</button>
                     <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt35_16k')}}>GPT-3.5 Turbo 16K</button>
                     <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4_0314')}}>GPT-4 0314</button>
                     <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4_0613')}}>GPT-4 0613</button>
@@ -49,8 +52,7 @@
                     <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4_0125')}}>GPT-4 Turbo 0125</button>
                     <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gptvi4_1106')}}>GPT-4 Turbo 1106 Vision</button>
                     <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4_turbo_20240409')}}>GPT-4 Turbo 20240409</button>
-                    <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4')}}>GPT-4 (Old)</button>
-                    <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4_32k')}}>GPT-4 32K (Old)</button>
+                    <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4o-2024-05-13')}}>GPT-4o 20240513</button>
                 {/if}
             </Arcodion>
             <Arcodion name="Anthropic Claude">
diff --git a/src/ts/model/names.ts b/src/ts/model/names.ts
index a87a3aca..f26d6844 100644
--- a/src/ts/model/names.ts
+++ b/src/ts/model/names.ts
@@ -93,6 +93,10 @@ export function getModelName(name:string){
             return 'GPT-4 Turbo'
         case 'gpt4_turbo_20240409':
             return 'GPT-4 Turbo (20240409)'
+        case 'gpt4o':
+            return 'GPT-4o'
+        case 'gpt4o-2024-05-13':
+            return 'GPT-4o (2024-05-13)'
         case 'gemini-1.5-pro-latest':
             return 'Gemini 1.5 Pro'
         case 'ollama-hosted':
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 7edd32f0..63261528 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -172,6 +172,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
         case 'mistral-large-latest':
         case 'gpt4_turbo_20240409':
         case 'gpt4_turbo':
+        case 'gpt4o':
+        case 'gpt4o-2024-05-13':
         case 'reverse_proxy':{
             let formatedChat:OpenAIChatExtra[] = []
             for(let i=0;i<formated.length;i++){
@@ -417,6 +419,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                     : requestModel === 'gpt4_0314' ? 'gpt-4-0314'
                     : requestModel === 'gpt4_turbo_20240409' ? 'gpt-4-turbo-2024-04-09'
                     : requestModel === 'gpt4_turbo' ? 'gpt-4-turbo'
+                    : requestModel === 'gpt4o' ? 'gpt-4o'
+                    : requestModel === 'gpt4o-2024-05-13' ? 'gpt-4o-2024-05-13'
                     : (!requestModel) ? 'gpt-3.5-turbo'
                     : requestModel,
                 messages: formatedChat,