From ec03903372433a20f8d0121af8ec5f5d363256b5 Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Thu, 21 Nov 2024 15:44:14 +0900
Subject: [PATCH] Add OAI prediction

---
 src/lang/en.ts                              | 1 +
 src/lib/Setting/Pages/PromptSettings.svelte | 2 ++
 src/ts/process/request.ts                   | 7 +++++++
 src/ts/storage/database.svelte.ts           | 2 ++
 4 files changed, 12 insertions(+)

diff --git a/src/lang/en.ts b/src/lang/en.ts
index 48262f64..50473fbf 100644
--- a/src/lang/en.ts
+++ b/src/lang/en.ts
@@ -796,4 +796,5 @@ export const languageEnglish = {
     preview: "Preview",
     recommended: "Recommended",
     newChat: "New Chat",
+    predictedOutput: "Predicted Output",
 }
\ No newline at end of file
diff --git a/src/lib/Setting/Pages/PromptSettings.svelte b/src/lib/Setting/Pages/PromptSettings.svelte
index 18ff35ea..d52cb0dd 100644
--- a/src/lib/Setting/Pages/PromptSettings.svelte
+++ b/src/lib/Setting/Pages/PromptSettings.svelte
@@ -145,6 +145,8 @@
     <TextAreaInput bind:value={DBState.db.customPromptTemplateToggle}/>
     <span class="text-textcolor mt-4">{language.defaultVariables} <Help key='defaultVariables' /></span>
     <TextAreaInput bind:value={DBState.db.templateDefaultVariables}/>
+    <span class="text-textcolor mt-4">{language.predictedOutput}</span>
+    <TextAreaInput bind:value={DBState.db.OAIPrediction}/>
     <span class="text-textcolor mt-4">{language.groupInnerFormat} <Help key='groupInnerFormat' /></span>
     <TextAreaInput placeholder={`<{{char}}\'s Message>\n{{slot}}\n</{{char}}\'s Message>`} bind:value={DBState.db.groupTemplate}/>
     {#if DBState.db.jsonSchemaEnabled}
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 4fa99cea..ae390628 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -535,6 +535,13 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                 }
             }
 
+            if(db.OAIPrediction){
+                body.prediction = {
+                    type: "content",
+                    content: db.OAIPrediction
+                }
+            }
+
             if(aiModel === 'openrouter'){
                 if(db.openrouterFallback){
                     body.route = "fallback"
diff --git a/src/ts/storage/database.svelte.ts b/src/ts/storage/database.svelte.ts
index 9f685a22..c9507f87 100644
--- a/src/ts/storage/database.svelte.ts
+++ b/src/ts/storage/database.svelte.ts
@@ -301,6 +301,7 @@ export function setDatabase(data:Database){
     data.sendWithEnter ??= true
     data.autoSuggestPrompt ??= defaultAutoSuggestPrompt
     data.autoSuggestPrefix ??= ""
+    data.OAIPrediction ??= ''
     data.autoSuggestClean ??= true
     data.imageCompression ??= true
     if(!data.formatingOrder.includes('personaPrompt')){
@@ -815,6 +816,7 @@ export interface Database{
     customGUI:string
     guiHTML:string
     logShare:boolean
+    OAIPrediction:string
 }
 
 export interface customscript{