From dc65a386f279bd471d0963c0c95ec230afafc2ca Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Sat, 23 Sep 2023 16:24:57 +0900
Subject: [PATCH] [feat] custom proxy requst model

---
 src/lib/Setting/Pages/BotSettings.svelte | 8 +++++++-
 src/ts/process/request.ts                | 6 +++++-
 src/ts/storage/database.ts               | 1 +
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/lib/Setting/Pages/BotSettings.svelte b/src/lib/Setting/Pages/BotSettings.svelte
index 7d73fdb3..eabe2e37 100644
--- a/src/lib/Setting/Pages/BotSettings.svelte
+++ b/src/lib/Setting/Pages/BotSettings.svelte
@@ -135,7 +135,7 @@
     <span class="text-textcolor mt-4"> {language.proxyAPIKey}</span>
     <TextInput marginBottom={false} size={"sm"} placeholder="leave it blank if it hasn't password" bind:value={$DataBase.proxyKey} />
     <span class="text-textcolor mt-4"> {language.proxyRequestModel}</span>
-    <SelectInput className="mt-2 mb-4" bind:value={$DataBase.proxyRequestModel}>
+    <SelectInput className="mt-2" bind:value={$DataBase.proxyRequestModel}>
         <OptionInput value="">None</OptionInput>
         <OptionInput value="gpt35">GPT 3.5</OptionInput>
         <OptionInput value="gpt35_16k">GPT 3.5 16k</OptionInput>
@@ -151,7 +151,13 @@
         <OptionInput value="claude-v1.2">claude-v1.2</OptionInput>
         <OptionInput value="claude-instant-v1.1">claude-instant-v1.1</OptionInput>
         <OptionInput value="claude-instant-v1.1-100k">claude-instant-v1.1-100k</OptionInput>
+        <OptionInput value="custom">Custom</OptionInput>
     </SelectInput>
+    {#if $DataBase.proxyRequestModel === 'custom'}
+        <TextInput marginBottom={true} size={"sm"} bind:value={$DataBase.customProxyRequestModel} placeholder="Name" />
+    {:else}
+        <div class="mb-4"></div>
+    {/if}
 {/if}
 {#if $DataBase.aiModel === 'openrouter' || $DataBase.subModel === 'openrouter'}
     <span class="text-textcolor mt-4">Openrouter Key</span>
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 5414e0a1..7fd933bd 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -168,7 +168,11 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
 
 
             const oaiFunctionCall = oaiFunctions ? (arg.useEmotion ? {"name": "set_emotion"} : "auto") : undefined
-            const requestModel = (aiModel === 'reverse_proxy' || aiModel === 'openrouter') ? db.proxyRequestModel : aiModel
+            let requestModel = (aiModel === 'reverse_proxy' || aiModel === 'openrouter') ? db.proxyRequestModel : aiModel
+
+            if(aiModel === 'reverse_proxy' && db.proxyRequestModel === 'custom'){
+                requestModel = db.customProxyRequestModel
+            }
             const body = ({
                 model: aiModel === 'openrouter' ? db.openrouterRequestModel :
                     requestModel ===  'gpt35' ? 'gpt-3.5-turbo'
diff --git a/src/ts/storage/database.ts b/src/ts/storage/database.ts
index 6a51637d..d08d1f04 100644
--- a/src/ts/storage/database.ts
+++ b/src/ts/storage/database.ts
@@ -316,6 +316,7 @@ export function setDatabase(data:Database){
     data.NAIsettings.mirostat_tau ??= 0
     data.NAIsettings.mirostat_lr ??= 1
     data.autofillRequestUrl ??= true
+    data.customProxyRequestModel ??= ''
     changeLanguage(data.language)
     DataBase.set(data)
 }