Add OAI prediction

This commit is contained in:
kwaroran
2024-11-21 15:44:14 +09:00
parent e895008814
commit ec03903372
4 changed files with 12 additions and 0 deletions

View File

@@ -796,4 +796,5 @@ export const languageEnglish = {
preview: "Preview", preview: "Preview",
recommended: "Recommended", recommended: "Recommended",
newChat: "New Chat", newChat: "New Chat",
predictedOutput: "Predicted Output",
} }

View File

@@ -145,6 +145,8 @@
<TextAreaInput bind:value={DBState.db.customPromptTemplateToggle}/> <TextAreaInput bind:value={DBState.db.customPromptTemplateToggle}/>
<span class="text-textcolor mt-4">{language.defaultVariables} <Help key='defaultVariables' /></span> <span class="text-textcolor mt-4">{language.defaultVariables} <Help key='defaultVariables' /></span>
<TextAreaInput bind:value={DBState.db.templateDefaultVariables}/> <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> <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}/> <TextAreaInput placeholder={`<{{char}}\'s Message>\n{{slot}}\n</{{char}}\'s Message>`} bind:value={DBState.db.groupTemplate}/>
{#if DBState.db.jsonSchemaEnabled} {#if DBState.db.jsonSchemaEnabled}

View File

@@ -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(aiModel === 'openrouter'){
if(db.openrouterFallback){ if(db.openrouterFallback){
body.route = "fallback" body.route = "fallback"

View File

@@ -301,6 +301,7 @@ export function setDatabase(data:Database){
data.sendWithEnter ??= true data.sendWithEnter ??= true
data.autoSuggestPrompt ??= defaultAutoSuggestPrompt data.autoSuggestPrompt ??= defaultAutoSuggestPrompt
data.autoSuggestPrefix ??= "" data.autoSuggestPrefix ??= ""
data.OAIPrediction ??= ''
data.autoSuggestClean ??= true data.autoSuggestClean ??= true
data.imageCompression ??= true data.imageCompression ??= true
if(!data.formatingOrder.includes('personaPrompt')){ if(!data.formatingOrder.includes('personaPrompt')){
@@ -815,6 +816,7 @@ export interface Database{
customGUI:string customGUI:string
guiHTML:string guiHTML:string
logShare:boolean logShare:boolean
OAIPrediction:string
} }
export interface customscript{ export interface customscript{