Convert system to user prompt for airoboros

This commit is contained in:
David Corbitt
2023-08-17 03:10:55 -07:00
parent 687f3dd85f
commit edac8da4a8

View File

@@ -156,7 +156,18 @@ export const templateAiroborosPrompt = (messages: OpenpipeChatInput["messages"])
}
}
let prompt = `${combinedSystemMessage}\n${conversationMessages.join(splitter)}`;
let systemMessage = "";
if (combinedSystemMessage) {
// If there is no user message, add a user tag to the system message
if (conversationMessages.find((message) => message.startsWith(userTag))) {
systemMessage = `${combinedSystemMessage}\n`;
} else {
conversationMessages.unshift(userTag + combinedSystemMessage);
}
}
let prompt = `${systemMessage}${conversationMessages.join(splitter)}`;
// Ensure that the prompt ends with an assistant message
const lastUserIndex = prompt.lastIndexOf(userTag);
@@ -166,5 +177,7 @@ export const templateAiroborosPrompt = (messages: OpenpipeChatInput["messages"])
prompt += splitter + assistantTag;
}
console.log("prompt is", prompt);
return prompt;
};