init prompt as system

This commit is contained in:
MaxRobinsonTheGreat 2024-05-14 21:34:14 -05:00
parent dc520a9ea1
commit a9b64950ca

View file

@ -14,11 +14,9 @@ export function stringifyTurns(turns) {
} }
export function toSinglePrompt(turns, system=null, stop_seq='***', model_nickname='assistant') { export function toSinglePrompt(turns, system=null, stop_seq='***', model_nickname='assistant') {
let messages = turns; let prompt = system ? `${system}${stop_seq}` : '';
if (system) messages.unshift({role: 'system', content: system}); let role = '';
let prompt = ""; turns.forEach((message) => {
let role = "";
messages.forEach((message) => {
role = message.role; role = message.role;
if (role === 'assistant') role = model_nickname; if (role === 'assistant') role = model_nickname;
prompt += `${role}: ${message.content}${stop_seq}`; prompt += `${role}: ${message.content}${stop_seq}`;