From a9b64950ca683caff588719cbd63e1144e0a60a7 Mon Sep 17 00:00:00 2001 From: MaxRobinsonTheGreat Date: Tue, 14 May 2024 21:34:14 -0500 Subject: [PATCH] init prompt as system --- src/utils/text.js | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/utils/text.js b/src/utils/text.js index c075d50..b0a273a 100644 --- a/src/utils/text.js +++ b/src/utils/text.js @@ -14,11 +14,9 @@ export function stringifyTurns(turns) { } export function toSinglePrompt(turns, system=null, stop_seq='***', model_nickname='assistant') { - let messages = turns; - if (system) messages.unshift({role: 'system', content: system}); - let prompt = ""; - let role = ""; - messages.forEach((message) => { + let prompt = system ? `${system}${stop_seq}` : ''; + let role = ''; + turns.forEach((message) => { role = message.role; if (role === 'assistant') role = model_nickname; prompt += `${role}: ${message.content}${stop_seq}`;