add user message suffix

This commit is contained in:
Yazan Agha-Schrader 2024-05-29 10:28:07 +02:00
parent eac8d739a5
commit 5fa255edfb
2 changed files with 331 additions and 327 deletions

View file

@ -258,7 +258,10 @@
return;
}
transcriptUpdate([...session.value.transcript, ["{{user}}", msg]])
const suffix = session.value.userMsgSuffix || "";
const userMsg = msg + suffix;
transcriptUpdate([...session.value.transcript, ["{{user}}", userMsg]])
let prompt = template(session.value.template, {
message: msg,

View file

@ -317,7 +317,8 @@ template: `{{history}}
historyTemplate: `{{name}}
{{message}}`,
char: "<|assistant|>",
user: "<|user|>"
user: "<|user|>",
userMsgSuffix: "<|end|>"
},