rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add memory and authors note to chat completion

This commit is contained in:
maver 2023-06-15 21:03:06 +02:00
parent bf5404f6ce
commit 5270d261aa
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
public/scripts/openai.js
View File

@ -464,6 +464,22 @@ async function prepareOpenAIMessages({
addMessageToChatCompletion('system', bias, 'main');
}
// Add extension prompts
if (0 < extensionPrompts.length) {
const summary = extensionPrompts['1_memory'] ?? null;
if (summary) {
const summaryMessage = new Message('system', summary.content, 'authorsNote');
chatCompletion.insert(summaryMessage, 'main')
}
const authorsNote = extensionPrompts['2_floating_prompt'] ?? null;
if (authorsNote) {
const authorsNoteMessage = new Message('system', authorsNote.content, 'authorsNote');
if (extension_prompt_types.AFTER_SCENARIO) chatCompletion.insert(authorsNoteMessage, 'scenario')
else chatCompletion.insert(authorsNoteMessage, 'main')
}
}
// Chat History
chatCompletion.add(new MessageCollection('chatHistory'), prompts.index('chatHistory'));
const mainChat = selected_group ? '[Start a new group chat. Group members: ${names}]' : '[Start a new Chat]';