Disable newline wrapping of in-chat extension prompts

This commit is contained in:
Cohee
2024-03-25 13:11:28 +02:00
parent 7520de445d
commit 0218eb1a18
2 changed files with 9 additions and 7 deletions

View File

@ -671,12 +671,13 @@ function populationInjectionPrompts(prompts, messages) {
const roles = ['system', 'user', 'assistant'];
const roleMessages = [];
const separator = '\n';
const wrap = false;
for (const role of roles) {
// Get prompts for current role
const rolePrompts = depthPrompts.filter(prompt => prompt.role === role).map(x => x.content).join(separator);
// Get extension prompt
const extensionPrompt = getExtensionPrompt(extension_prompt_types.IN_CHAT, i, separator, roleTypes[role]);
const extensionPrompt = getExtensionPrompt(extension_prompt_types.IN_CHAT, i, separator, roleTypes[role], wrap);
const jointPrompt = [rolePrompts, extensionPrompt].filter(x => x).map(x => x.trim()).join(separator);