From 859db3ab2d6283fe8c24e7b0a3124fc9f9550e25 Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Tue, 7 Nov 2023 05:21:56 +0900
Subject: [PATCH] [feat] oai seed

---
 src/lang/en.ts                           | 1 +
 src/lib/Setting/Pages/BotSettings.svelte | 5 +++++
 src/ts/process/request.ts                | 9 +++++++--
 src/ts/storage/database.ts               | 2 ++
 4 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/lang/en.ts b/src/lang/en.ts
index 2ce2e9d3..93d44064 100644
--- a/src/lang/en.ts
+++ b/src/lang/en.ts
@@ -437,4 +437,5 @@ export const languageEnglish = {
     customStopWords: "Custom Stop Words",
     defaultPrompt: "Default Prompt",
     additionalText: 'Additional Description',
+    seed: "Seed",
 }
\ No newline at end of file
diff --git a/src/lib/Setting/Pages/BotSettings.svelte b/src/lib/Setting/Pages/BotSettings.svelte
index eabe2e37..cc09927e 100644
--- a/src/lib/Setting/Pages/BotSettings.svelte
+++ b/src/lib/Setting/Pages/BotSettings.svelte
@@ -267,6 +267,11 @@
     <span class="text-textcolor">{language.maxResponseSize}</span>
     <NumberInput min={0} max={2048} marginBottom={true} bind:value={$DataBase.maxResponse}/>
 
+{#if $DataBase.aiModel.startsWith('gpt') || $DataBase.aiModel === 'reverse_proxy' || $DataBase.aiModel === 'openrouter'}
+<span class="text-textcolor">{language.seed}</span>
+
+<NumberInput bind:value={$DataBase.generationSeed} marginBottom={true}/>
+{/if}
 <span class="text-textcolor">{language.temperature} <Help key="tempature"/></span>
 
 {#if $DataBase.aiModel.startsWith("novelai")}
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index a69c54a3..ae7ca0ff 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -183,7 +183,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
 
 
             db.cipherChat = false
-            const body = ({
+            let body = ({
                 model: aiModel === 'openrouter' ? db.openrouterRequestModel :
                     requestModel ===  'gpt35' ? 'gpt-3.5-turbo'
                     : requestModel ===  'gpt35_0613' ? 'gpt-3.5-turbo-0613'
@@ -205,9 +205,14 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                 presence_penalty: arg.PresensePenalty || (db.PresensePenalty / 100),
                 frequency_penalty: arg.frequencyPenalty || (db.frequencyPenalty / 100),
                 logit_bias: bias,
-                stream: false
+                stream: false,
+                seed: db.generationSeed
             })
 
+            if(body.seed === -1){
+                delete body.seed
+            }
+
             let replacerURL = aiModel === 'openrouter' ? "https://openrouter.ai/api/v1/chat/completions" :
                 (aiModel === 'reverse_proxy') ? (db.forceReplaceUrl) : ('https://api.openai.com/v1/chat/completions')
 
diff --git a/src/ts/storage/database.ts b/src/ts/storage/database.ts
index 455c3397..956d057a 100644
--- a/src/ts/storage/database.ts
+++ b/src/ts/storage/database.ts
@@ -317,6 +317,7 @@ export function setDatabase(data:Database){
     data.NAIsettings.mirostat_lr ??= 1
     data.autofillRequestUrl ??= true
     data.customProxyRequestModel ??= ''
+    data.generationSeed ??= -1
     changeLanguage(data.language)
     DataBase.set(data)
 }
@@ -489,6 +490,7 @@ export interface Database{
     localStopStrings?:string[]
     autofillRequestUrl:boolean
     customProxyRequestModel:string
+    generationSeed:number
 }
 
 export interface customscript{