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; 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, { let prompt = template(session.value.template, {
message: msg, message: msg,

View file

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