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

@ -1,458 +1,459 @@
// extended list // extended list
export const promptFormats = { export const promptFormats = {
"airoborosl2": { "airoborosl2": {
template: `{{prompt}} {{history}} {{char}}`, template: `{{prompt}} {{history}} {{char}}`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"alpaca": { "alpaca": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
### {{char}}:`, ### {{char}}:`,
historyTemplate: `### {{name}}: historyTemplate: `### {{name}}:
{{message}}`, {{message}}`,
char: "Response", char: "Response",
user: "Instruction" user: "Instruction"
}, },
"bakllava": { "bakllava": {
template: `{{history}}{{char}}:`, template: `{{history}}{{char}}:`,
historyTemplate: `{{name}}: {{message}} historyTemplate: `{{name}}: {{message}}
`, `,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"chatml": { "chatml": {
template: `<|im_start|>system template: `<|im_start|>system
{{prompt}}<|im_end|> {{prompt}}<|im_end|>
{{history}} {{history}}
<|im_start|>{{char}}`, <|im_start|>{{char}}`,
historyTemplate: `<|im_start|>{{user}} historyTemplate: `<|im_start|>{{user}}
{{message}}<|im_end|>`, {{message}}<|im_end|>`,
char: "assistant", char: "assistant",
user: "user" user: "user"
}, },
"codeCherryPop": { "codeCherryPop": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
### {{char}}:`, ### {{char}}:`,
historyTemplate: `### {{name}}: historyTemplate: `### {{name}}:
{{message}}`, {{message}}`,
char: "Response", char: "Response",
user: "Instruction" user: "Instruction"
}, },
"deepseekCoder": { "deepseekCoder": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
### {{char}}:`, ### {{char}}:`,
historyTemplate: `### {{name}}: historyTemplate: `### {{name}}:
{{message}}`, {{message}}`,
char: "Response", char: "Response",
user: "Instruction" user: "Instruction"
}, },
"dolphinMistral": { "dolphinMistral": {
template: `<|im_start|>system template: `<|im_start|>system
{{prompt}}<|im_end|> {{prompt}}<|im_end|>
{{history}} {{history}}
<|im_start|>{{char}}`, <|im_start|>{{char}}`,
historyTemplate: `<|im_start|>{{user}} historyTemplate: `<|im_start|>{{user}}
{{message}}<|im_end|>`, {{message}}<|im_end|>`,
char: "assistant", char: "assistant",
user: "user" user: "user"
}, },
"evolvedSeeker": { "evolvedSeeker": {
template: `<|im_start|>system template: `<|im_start|>system
{{prompt}}<|im_end|> {{prompt}}<|im_end|>
{{history}} {{history}}
<|im_start|>{{char}}`, <|im_start|>{{char}}`,
historyTemplate: `<|im_start|>{{user}} historyTemplate: `<|im_start|>{{user}}
{{message}}<|im_end|>`, {{message}}<|im_end|>`,
char: "assistant", char: "assistant",
user: "user" user: "user"
}, },
"goliath120b": { "goliath120b": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
{{char}}:`, {{char}}:`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"jordan": { "jordan": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
{{char}}:`, {{char}}:`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"llama2": { "llama2": {
template: `<s>[INST] <<SYS>> template: `<s>[INST] <<SYS>>
{{prompt}} {{prompt}}
<</SYS>> <</SYS>>
{{history}} [/INST] {{char}} </s><s>[INST] `, {{history}} [/INST] {{char}} </s><s>[INST] `,
historyTemplate: `{{name}}: {{message}} [/INST]`, historyTemplate: `{{name}}: {{message}} [/INST]`,
char: "llama", char: "llama",
user: "user" user: "user"
}, },
"llava": { "llava": {
template: `{{history}}{{char}}:`, template: `{{history}}{{char}}:`,
historyTemplate: `{{name}}: {{message}} historyTemplate: `{{name}}: {{message}}
`, `,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"leoHessianai": { "leoHessianai": {
template: `<|im_start|>system template: `<|im_start|>system
{{prompt}}<|im_end|> {{prompt}}<|im_end|>
{{history}} {{history}}
<|im_start|>{{char}}`, <|im_start|>{{char}}`,
historyTemplate: `<|im_start|>{{user}} historyTemplate: `<|im_start|>{{user}}
{{message}}<|im_end|>`, {{message}}<|im_end|>`,
char: "assistant", char: "assistant",
user: "user" user: "user"
}, },
"leoMistral": { "leoMistral": {
template: `{{prompt}} {{history}} {{char}}`, template: `{{prompt}} {{history}} {{char}}`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"marx": { "marx": {
template: `{{history}} template: `{{history}}
{{char}}:`, {{char}}:`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"med42": { "med42": {
template: `<|system|>: {{prompt}} template: `<|system|>: {{prompt}}
{{history}} {{history}}
{{char}}`, {{char}}`,
historyTemplate: `<|{{name}}|>:{{message}}`, historyTemplate: `<|{{name}}|>:{{message}}`,
char: "assistant", char: "assistant",
user: "prompter" user: "prompter"
}, },
"metaMath": { "metaMath": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
### {{char}}:`, ### {{char}}:`,
historyTemplate: `### {{name}}: historyTemplate: `### {{name}}:
{{message}}`, {{message}}`,
char: "Response", char: "Response",
user: "Instruction" user: "Instruction"
}, },
"mistralInstruct": { "mistralInstruct": {
template: `<s>[INST] ({{prompt}}) {{history}} {{char}}</s>`, template: `<s>[INST] ({{prompt}}) {{history}} {{char}}</s>`,
historyTemplate: `{{name}} {{message}}`, historyTemplate: `{{name}} {{message}}`,
char: "[/INST] Assistant:", char: "[/INST] Assistant:",
user: "[INST] User:" user: "[INST] User:"
}, },
"mistralOpenOrca": { "mistralOpenOrca": {
template: `<|im_start|>system template: `<|im_start|>system
{{prompt}}<|im_end|> {{prompt}}<|im_end|>
{{history}} {{history}}
<|im_start|>{{char}}`, <|im_start|>{{char}}`,
historyTemplate: `<|im_start|>{{user}} historyTemplate: `<|im_start|>{{user}}
{{message}}<|im_end|>`, {{message}}<|im_end|>`,
char: "assistant", char: "assistant",
user: "user" user: "user"
}, },
"mythomax": { "mythomax": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
### {{char}}:`, ### {{char}}:`,
historyTemplate: `### {{name}}: historyTemplate: `### {{name}}:
{{message}}`, {{message}}`,
char: "Response", char: "Response",
user: "Instruction" user: "Instruction"
}, },
"neuralchat": { "neuralchat": {
template: `### System: template: `### System:
{{prompt}} {{prompt}}
{{history}} {{history}}
### {{char}}:`, ### {{char}}:`,
historyTemplate: `### {{name}}: historyTemplate: `### {{name}}:
{{message}}`, {{message}}`,
char: "Assistant", char: "Assistant",
user: "User" user: "User"
}, },
"nousCapybara": { "nousCapybara": {
template: `{{history}} template: `{{history}}
{{char}}`, {{char}}`,
historyTemplate: ` historyTemplate: `
{{name}}: {{message}}`, {{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"nousHermes": { "nousHermes": {
template: `### Instruction: {{prompt}} template: `### Instruction: {{prompt}}
{{history}} {{history}}
### {{char}}:`, ### {{char}}:`,
historyTemplate: ` historyTemplate: `
### {{name}}: {{message}}`, ### {{name}}: {{message}}`,
char: "Response", char: "Response",
user: "Input" user: "Input"
}, },
"openChat": { "openChat": {
template: `{{history}}{{char}}`, template: `{{history}}{{char}}`,
historyTemplate: `GPT4 {{user}}: {{message}}<|end_of_turn|>`, historyTemplate: `GPT4 {{user}}: {{message}}<|end_of_turn|>`,
char: "Assistant", char: "Assistant",
user: "User" user: "User"
}, },
"openhermes2Mistral": { "openhermes2Mistral": {
template: `<|im_start|>system template: `<|im_start|>system
{{prompt}}<|im_end|> {{prompt}}<|im_end|>
{{history}} {{history}}
<|im_start|>{{char}}`, <|im_start|>{{char}}`,
historyTemplate: `<|im_start|>{{user}} historyTemplate: `<|im_start|>{{user}}
{{message}}<|im_end|>`, {{message}}<|im_end|>`,
char: "assistant", char: "assistant",
user: "user" user: "user"
}, },
"orcamini": { "orcamini": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
### {{char}}:`, ### {{char}}:`,
historyTemplate: `### {{name}}: historyTemplate: `### {{name}}:
{{message}}`, {{message}}`,
char: "Response", char: "Response",
user: "Instruction" user: "Instruction"
}, },
"phi3": { "phi3": {
template: `{{history}} template: `{{history}}
{{char}} {{char}}
`, `,
historyTemplate: `{{name}} historyTemplate: `{{name}}
{{message}}`, {{message}}`,
char: "<|assistant|>", char: "<|assistant|>",
user: "<|user|>" user: "<|user|>",
}, userMsgSuffix: "<|end|>"
},
"sauerkraut": { "sauerkraut": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
{{char}}:`, {{char}}:`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "Assistant", char: "Assistant",
user: "User" user: "User"
}, },
"samantha": { "samantha": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
{{char}}:`, {{char}}:`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"samanthaMistral": { "samanthaMistral": {
template: `<|im_start|>system template: `<|im_start|>system
{{prompt}}<|im_end|> {{prompt}}<|im_end|>
{{history}} {{history}}
<|im_start|>{{char}}`, <|im_start|>{{char}}`,
historyTemplate: `<|im_start|>{{user}} historyTemplate: `<|im_start|>{{user}}
{{message}}<|im_end|>`, {{message}}<|im_end|>`,
char: "assistant", char: "assistant",
user: "user" user: "user"
}, },
"scarlett": { "scarlett": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
{{char}}:`, {{char}}:`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"starlingLM": { "starlingLM": {
template: `{{history}}{{char}}`, template: `{{history}}{{char}}`,
historyTemplate: `GPT4 Correct {{user}}: {{message}}<|end_of_turn|>`, historyTemplate: `GPT4 Correct {{user}}: {{message}}<|end_of_turn|>`,
char: "Assistant", char: "Assistant",
user: "User" user: "User"
}, },
"starlingLMCode": { "starlingLMCode": {
template: `{{history}}{{char}}`, template: `{{history}}{{char}}`,
historyTemplate: `Code {{user}}: {{message}}<|end_of_turn|>`, historyTemplate: `Code {{user}}: {{message}}<|end_of_turn|>`,
char: "Assistant", char: "Assistant",
user: "User" user: "User"
}, },
"sydney": { "sydney": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
{{char}}`, {{char}}`,
historyTemplate: `### {{name}}: historyTemplate: `### {{name}}:
{{message}} {{message}}
`, `,
char: "Response", char: "Response",
user: "Instruction" user: "Instruction"
}, },
"synthia": { "synthia": {
template: `SYSTEM: {{prompt}} template: `SYSTEM: {{prompt}}
{{history}} {{history}}
{{char}}:`, {{char}}:`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"tess": { "tess": {
template: `SYSTEM: {{prompt}} template: `SYSTEM: {{prompt}}
{{history}} {{history}}
{{char}}:`, {{char}}:`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"vicuna": { "vicuna": {
template: `{{prompt}} template: `{{prompt}}
{{history}} {{history}}
{{char}}:`, {{char}}:`,
historyTemplate: ` historyTemplate: `
{{name}}: {{message}}`, {{name}}: {{message}}`,
char: "ASSISTANT", char: "ASSISTANT",
user: "USER" user: "USER"
}, },
"yi34b": { "yi34b": {
template: `{{history}} {{char}}`, template: `{{history}} {{char}}`,
historyTemplate: `{{name}}: {{message}}`, historyTemplate: `{{name}}: {{message}}`,
char: "Assistant", char: "Assistant",
user: "Human" user: "Human"
}, },
"zephyr": { "zephyr": {
template: `<|system|> template: `<|system|>
{{prompt}}</s> {{prompt}}</s>
{{history}} {{history}}
{{char}}`, {{char}}`,
historyTemplate: `<|{{name}}|> historyTemplate: `<|{{name}}|>
{{message}}</s>`, {{message}}</s>`,
char: "assistant", char: "assistant",
user: "user" user: "user"
} }
}; };