rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-03-19 13:30:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update prompt-converters.js

Browse Source
This commit is contained in:
DonMoralez 2023-12-15 20:58:03 +02:00
parent 04f3aa8b66
commit d061c830c2
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/endpoints/prompt-converters.js
View File

@ -9,7 +9,6 @@
* @returns {string} Prompt for Claude * @returns {string} Prompt for Claude
* @copyright Prompt Conversion script taken from RisuAI by kwaroran (GPLv3). * @copyright Prompt Conversion script taken from RisuAI by kwaroran (GPLv3).
*/ */
function convertClaudePrompt(messages, addAssistantPostfix, addAssistantPrefill, withSyspromptSupport, useSystemPrompt, addSysHumanMsg) { function convertClaudePrompt(messages, addAssistantPostfix, addAssistantPrefill, withSyspromptSupport, useSystemPrompt, addSysHumanMsg) {
console.log(JSON.stringify(messages, null, 2)); console.log(JSON.stringify(messages, null, 2));
@ -43,9 +42,9 @@ function convertClaudePrompt(messages, addAssistantPostfix, addAssistantPrefill,
} else { } else {
// Otherwise, use the default message format by setting the first message's role to 'user'(compatible with all claude models including 2.1.) // Otherwise, use the default message format by setting the first message's role to 'user'(compatible with all claude models including 2.1.)
messages[0].role = 'user'; messages[0].role = 'user';
// Also, fix messages order for default claude format when(messages > Context Size) by merging two messages with "\n\nHuman: " prefixes into one before the first Assistant's message. // Fix messages order for default message format when(messages > Context Size) by merging two messages with "\n\nHuman: " prefixes into one, before the first Assistant's message.
if (firstAssistantIndex > 0) { if (firstAssistantIndex > 0) {
messages[firstAssistantIndex - 1].role = firstAssistantIndex - 1 !== 0 && messages[firstAssistantIndex - 1].role === 'user' ? 'firstMsg' : messages[firstAssistantIndex - 1].role; messages[firstAssistantIndex - 1].role = firstAssistantIndex - 1 !== 0 && messages[firstAssistantIndex - 1].role === 'user' ? 'FixHumMsg' : messages[firstAssistantIndex - 1].role;
} }
} }
} }
@ -58,15 +57,14 @@ function convertClaudePrompt(messages, addAssistantPostfix, addAssistantPrefill,
v.content = `${v.name}: ${v.content}`; v.content = `${v.name}: ${v.content}`;
delete v.name; delete v.name;
} }
//let prefix = ''; // Set prefix according to the role.
let prefix = ''; let prefix = {
prefix = {
'assistant': '\n\nAssistant: ', 'assistant': '\n\nAssistant: ',
'user': '\n\nHuman: ', 'user': '\n\nHuman: ',
'system': i === 0 ? '' : v.name === 'example_assistant' ? '\n\nA: ' : v.name === 'example_user' ? '\n\nH: ' : '\n\n', 'system': i === 0 ? '' : v.name === 'example_assistant' ? '\n\nA: ' : v.name === 'example_user' ? '\n\nH: ' : '\n\n',
'firstMsg': '\n\nFirst message: ', 'FixHumMsg': '\n\nFirst message: ',
}[v.role] ?? '\n\n'; }[v.role] ?? '\n\n';
//}
return prefix + v.content; return prefix + v.content;
}).join(''); }).join('');