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

[fix] ooba formating

Browse Source
This commit is contained in:
kwaroran
2023-08-24 17:57:25 +09:00
parent d8a2df7c91
commit 327bc93456
1 changed files with 27 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -38,9 +38,11 @@ export function stringlizeChatOba(formated:OpenAIChat[], characterName:string, s
const db = get(DataBase)
let resultString:string[] = []
let { header, systemPrefix, userPrefix, assistantPrefix, seperator } = db.ooba.formating;
if(!seperator){
seperator = "\n\n"
}
header = header ?? ""
systemPrefix = systemPrefix ?? ""
userPrefix = userPrefix ?? ""
assistantPrefix = assistantPrefix ?? ""
seperator = seperator ?? "\n\n"
if(header) {
resultString.push(header)
@@ -50,25 +52,39 @@ export function stringlizeChatOba(formated:OpenAIChat[], characterName:string, s
continue
}
let prefix = ""
let name = ""
let name = form.name
if(form.role === 'user'){
prefix = appendWhitespace(suggesting ? assistantPrefix : userPrefix, seperator)
name = `${db.username}: `
name ??= `${db.username}: `
}
else if(form.role === 'assistant'){
prefix = appendWhitespace(suggesting ? userPrefix : assistantPrefix, seperator)
name = `${characterName}: `
name ??= `${characterName}: `
}
else if(form.role === 'system'){
prefix = appendWhitespace(systemPrefix, seperator)
}
resultString.push(prefix + name + form.content)
if(db.ooba.formating.useName){
resultString.push(prefix + name + form.content)
}
else{
resultString.push(prefix + form.content)
}
}
if(!continued){
if (suggesting){
resultString.push(appendWhitespace(assistantPrefix, seperator) + `${db.username}:\n` + db.autoSuggestPrefix)
} else {
resultString.push(assistantPrefix + `${characterName}:`)
if(db.ooba.formating.useName){
if (suggesting){
resultString.push(appendWhitespace(assistantPrefix, seperator) + `${db.username}:\n` + db.autoSuggestPrefix)
} else {
resultString.push(assistantPrefix + `${characterName}:`)
}
}
else{
if (suggesting){
resultString.push(appendWhitespace(assistantPrefix, seperator) + `\n` + db.autoSuggestPrefix)
} else {
resultString.push(assistantPrefix)
}
}
}
return resultString.join(seperator)