[feat] new nai formating/kayra

This commit is contained in:
kwaroran
2023-07-29 08:57:28 +09:00
parent a370b9c033
commit 538d664baf
7 changed files with 465 additions and 21 deletions

View File

@@ -1,6 +1,6 @@
import { DataBase, setPreset, type botPreset, setDatabase } from "src/ts/storage/database";
import { get } from "svelte/store";
import { prebuiltPresets } from "./templates";
import { prebuiltNAIpresets, prebuiltPresets } from "./templates";
import { alertConfirm, alertSelect } from "src/ts/alert";
import { language } from "src/lang";
@@ -20,6 +20,17 @@ export async function setRecommended(model: string, ask:'ask'|'force') {
const pr:botPreset = prebuiltPresets.OAI
setDatabase(setPreset(db, pr))
}
else if(db.aiModel.startsWith('novelai')){
const pr:botPreset = prebuiltPresets.OAI
pr.NAISettings = prebuiltNAIpresets
pr.temperature = 1.05
pr.maxContext = 8000
pr.maxResponse = 300
pr.mainPrompt = "***\n[ Style: chat ]"
pr.jailbreak = ''
pr.globalNote = ''
setDatabase(setPreset(db, pr))
}
else if(db.aiModel === 'textgen_webui'){
const sel = parseInt(await alertSelect(["Llama, Alpaca", "Koala", "Vicuna", "WizardLM", "Others"]))
let pr = prebuiltPresets.ooba
@@ -60,5 +71,5 @@ export async function setRecommended(model: string, ask:'ask'|'force') {
}
export function recommendedPresetExist(model:string){
return model.startsWith('gpt') || model === 'openrouter' || model === 'reverse_proxy' || model === 'textgen_webui'
return model.startsWith('gpt') || model === 'openrouter' || model === 'reverse_proxy' || model === 'textgen_webui' || model.startsWith('novelai')
}