From a3ad188b073bef59fb77258ff91a341cc808073c Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Fri, 13 Sep 2024 08:14:30 +0900
Subject: [PATCH] Add Gpt4o1 support

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

diff --git a/src/lib/UI/ModelList.svelte b/src/lib/UI/ModelList.svelte
index 0bbab798..6d9c31c8 100644
--- a/src/lib/UI/ModelList.svelte
+++ b/src/lib/UI/ModelList.svelte
@@ -56,6 +56,8 @@
                     <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4o-2024-05-13')}}>GPT-4o 20240513</button>
                     <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4o-2024-08-06')}}>GPT-4o 20240806</button>
                     <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4o-chatgpt')}}>GPT-4o ChatGPT</button>
+                    <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4o1-preview')}}>o1 Preview</button>
+                    <button class="p-2 hover:text-green-500" on:click={() => {changeModel('gpt4o1-mini')}}>o1 Mini</button>
                 {/if}
             </Arcodion>
             <Arcodion name="Anthropic Claude">
diff --git a/src/ts/model/names.ts b/src/ts/model/names.ts
index 41b19540..c0d1f72c 100644
--- a/src/ts/model/names.ts
+++ b/src/ts/model/names.ts
@@ -105,6 +105,10 @@ export function getModelName(name:string){
             return 'GPT-4o ChatGPT'
         case 'gpt4om':
             return 'GPT-4o Mini'
+        case 'gpt4o1-preview':
+            return 'o1 Preview'
+        case 'gpt4o1-mini':
+            return 'o1 Mini'
         case 'gpt4om-2024-07-18':
             return 'GPT-4o Mini (2024-07-18)'
         case 'gemini-1.5-pro-latest':
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 0ada4266..7c890d97 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -236,6 +236,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
         case 'gpt4om-2024-07-18':
         case 'gpt4o-2024-08-06':
         case 'gpt4o-chatgpt':
+        case 'gpt4o1-preview':
+        case 'gpt4o1-mini':
         case 'reverse_proxy':{
             let formatedChat:OpenAIChatExtra[] = []
             for(let i=0;i<formated.length;i++){
@@ -301,6 +303,15 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                 })
             }
 
+            if(aiModel.startsWith('gpt4o1')){
+                for(let i=0;i<formatedChat.length;i++){
+                    if(formatedChat[i].role === 'system'){
+                        formatedChat[i].content = `<system>${formatedChat[i].content}</system>`
+                        formatedChat[i].role = 'user'
+                    }
+                }
+            }
+
             for(let i=0;i<biasString.length;i++){
                 const bia = biasString[i]
                 if(bia[1] === -101){
@@ -483,6 +494,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                     : requestModel === 'gpt4om-2024-07-18' ? 'gpt-4o-mini-2024-07-18'
                     : requestModel === 'gpt4o-2024-08-06' ? 'gpt-4o-2024-08-06'
                     : requestModel === 'gpt4o-chatgpt' ? 'chatgpt-4o-latest'
+                    : requestModel === 'gpt4o1-preview' ? 'o1-preview'
+                    : requestModel === 'gpt4o1-mini' ? 'o1-mini'
                     : (!requestModel) ? 'gpt-3.5-turbo'
                     : requestModel,
                 messages: formatedChat,