From 4a34dc3e235e8e3bc7a42dab443225f1fc61d75a Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Sat, 23 Sep 2023 16:18:20 +0900
Subject: [PATCH] [feat] add autofill option

---
 src/lib/Setting/Pages/AdvancedSettings.svelte | 3 +++
 src/ts/process/request.ts                     | 2 +-
 src/ts/storage/database.ts                    | 3 +++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/lib/Setting/Pages/AdvancedSettings.svelte b/src/lib/Setting/Pages/AdvancedSettings.svelte
index 2e3a3664..2a4bb2aa 100644
--- a/src/lib/Setting/Pages/AdvancedSettings.svelte
+++ b/src/lib/Setting/Pages/AdvancedSettings.svelte
@@ -53,6 +53,9 @@
 <div class="flex items-center mt-4">
     <Check bind:check={$DataBase.usePlainFetch} name="Force Plain Fetch"/>
 </div>
+<div class="flex items-center mt-4">
+    <Check bind:check={$DataBase.autofillRequestUrl} name="Autofill Request URL"/>
+</div>
 <button
     on:click={async () => {
         alertMd(getRequestLog())
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 8954bcbe..5414e0a1 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -201,7 +201,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                 replacerURL = replacerURL.replace("risu::", '')
             }
 
-            if(aiModel === 'reverse_proxy'){
+            if(aiModel === 'reverse_proxy' && db.autofillRequestUrl){
                 if(replacerURL.endsWith('v1')){
                     replacerURL += '/chat/completions'
                 }
diff --git a/src/ts/storage/database.ts b/src/ts/storage/database.ts
index 7d9aaa79..6a51637d 100644
--- a/src/ts/storage/database.ts
+++ b/src/ts/storage/database.ts
@@ -315,6 +315,7 @@ export function setDatabase(data:Database){
     data.NAIsettings.cfg_scale ??= 1
     data.NAIsettings.mirostat_tau ??= 0
     data.NAIsettings.mirostat_lr ??= 1
+    data.autofillRequestUrl ??= true
     changeLanguage(data.language)
     DataBase.set(data)
 }
@@ -484,6 +485,8 @@ export interface Database{
         freeApi:boolean
     }
     localStopStrings?:string[]
+    autofillRequestUrl:boolean
+    customProxyRequestModel:string
 }
 
 export interface customscript{