rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add support for authors note and summary extensions

Browse Source
This commit is contained in:
maver
2023-06-11 17:17:38 +02:00
parent a66041f503
commit 77e183f5dd
2 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
public/scripts/openai.js
View File

@ -289,8 +289,6 @@ function setupOpenAIPromptManager(openAiSettings) {
promptManager.init(configuration, openAiSettings);
promptManager.render();
}
function generateOpenAIPromptCache() {
@ -366,7 +364,7 @@ function formatWorldInfo(value) {
return stringFormat(oai_settings.wi_format, value);
}
async function prepareOpenAIMessages({ systemPrompt, name2, charDescription, charPersonality, Scenario, worldInfoBefore, worldInfoAfter, extensionPrompt, bias, type, quietPrompt, jailbreakPrompt, cyclePrompt } = {}) {
async function prepareOpenAIMessages({ name2, charDescription, charPersonality, Scenario, worldInfoBefore, worldInfoAfter, bias, type, quietPrompt, extensionPrompts, cyclePrompt } = {}) {
const chatCompletion = promptManager.getChatCompletion();
// Prepare messages
@ -412,7 +410,16 @@ async function prepareOpenAIMessages({ systemPrompt, name2, charDescription, cha
}
// Handle extension prompt
if (extensionPrompt) chatCompletion.insertAfter('worldInfoAfter', 'extensionPrompt', chatCompletion.makeSystemMessage(substituteParams(extensionPrompt)));
if (0 < extensionPrompts.length) {
const summary = extensionPrompts['1_memory'];
if (summary) chatCompletion.insertAfter('scenario', 'extensionSummary', chatCompletion.makeSystemMessage(substituteParams(summary)));
const authorsNote = extensionPrompts['2_floating_prompt'];
if (authorsNote && (extension_prompt_types.AFTER_SCENARIO === authorsNote.position))
chatCompletion.insertAfter('scenario', 'extensionAuthorsNote', chatCompletion.makeSystemMessage(substituteParams(authorsNote)));
else
chatCompletion.insertAfter('chatHistory', 'extensionAuthorsNote', chatCompletion.makeSystemMessage(substituteParams(authorsNote)));
}
// Handle bias settings
if (bias && bias.trim().length) chatCompletion.add(biasMessage);