mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[feat] improve ooba template & ooba config

Browse Source
This commit is contained in:
kwaroran
2023-08-30 20:04:21 +09:00
parent 4ef79c9e3c
commit 1b39526474
8 changed files with 231 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/ts/process/stringlize.ts
View File

@@ -29,6 +29,9 @@ export function stringlizeChat(formated:OpenAIChat[], char:string, continued:boo
}
function appendWhitespace(prefix:string, seperator:string=" ") {
if(!prefix){
return ""
}
if(prefix && !"> \n".includes(prefix[prefix.length-1])){
prefix += seperator.includes("\n\n") ? "\n" : " "
}
@@ -37,18 +40,15 @@ function appendWhitespace(prefix:string, seperator:string=" ") {
export function stringlizeChatOba(formated:OpenAIChat[], characterName:string, suggesting:boolean, continued:boolean){
const db = get(DataBase)
let resultString:string[] = []
let { header, systemPrefix, userPrefix, assistantPrefix, seperator } = db.ooba.formating;
header = header ?? ""
let { systemPrefix, userPrefix, assistantPrefix, seperator } = db.ooba.formating;
systemPrefix = systemPrefix ?? ""
userPrefix = userPrefix ?? ""
assistantPrefix = assistantPrefix ?? ""
seperator = seperator ?? "\n\n"
seperator = seperator ?? "\n"
if(header) {
resultString.push(header)
}
for(const form of formated){
if(form.content === "[Start a new chat]"){
resultString.push("<START>")
continue
}
let prefix = ""
@@ -87,7 +87,8 @@ export function stringlizeChatOba(formated:OpenAIChat[], characterName:string, s
}
}
}
return resultString.join(seperator)
console.log(resultString)
return resultString.join(seperator).trim()
}
const userStrings = ["user", "human", "input", "inst", "instruction"]