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

Update existing prompts during migration

Browse Source 
Instead of adding new prompts for main, jailbreak and nsfw
This commit is contained in:
maver
2023-07-31 15:18:44 +02:00
parent 222fd3429d
commit af81915ff3
2 changed files with 10 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
public/script.js
View File

@ -540,7 +540,7 @@ this
}; };
// Register configuration migrations // Register configuration migrations
registerPromptManagerMigration(); registerPromptManagerMigration(saveSettings);
$(document).ajaxError(function myErrorHandler(_, xhr) { $(document).ajaxError(function myErrorHandler(_, xhr) {
if (xhr.status == 403) { if (xhr.status == 403) {

35
public/scripts/PromptManager.js
View File

@ -5,48 +5,31 @@ import {power_user} from "./power-user.js";
/** /**
* Register migrations for the prompt manager when settings are loaded or an Open AI preset is loaded. * Register migrations for the prompt manager when settings are loaded or an Open AI preset is loaded.
*/ */
const registerPromptManagerMigration = () => { const registerPromptManagerMigration = (saveSettingsCallback) => {
const migrate = (settings) => { const migrate = (settings) => {
// If any of the specified settings exist, run the migration // If any of the specified settings exist, run the migration
if (settings.main_prompt || settings.nsfw_prompt || settings.jailbreak_prompt) { if (settings.main_prompt || settings.nsfw_prompt || settings.jailbreak_prompt) {
console.log('Running one-time configuration migration for prompt manager.') console.log('Running one-time configuration migration for prompt manager.')
if (settings.prompts === undefined) settings.prompts = []; if (settings.prompts === undefined || settings.prompts.length === 0) settings.prompts = chatCompletionDefaultPrompts.prompts;
const findPrompt = (identifier) => settings.prompts.find(prompt => identifier === prompt.identifier);
if (settings.main_prompt) { if (settings.main_prompt) {
settings.prompts.push({ findPrompt('main').content = settings.main_prompt
identifier: null, // Will be assigned by prompt manager during sanitization
name: 'My Main Prompt',
role: 'system',
content: settings.main_prompt,
system_prompt: false,
enabled: false,
});
delete settings.main_prompt; delete settings.main_prompt;
} }
if (settings.nsfw_prompt) { if (settings.nsfw_prompt) {
settings.prompts.push({ findPrompt('nsfw').content = settings.nsfw_prompt
identifier: null,
name: 'My NSFW Prompt',
role: 'system',
content: settings.nsfw_prompt,
system_prompt: false,
enabled: false,
});
delete settings.nsfw_prompt; delete settings.nsfw_prompt;
} }
if (settings.jailbreak_prompt) { if (settings.jailbreak_prompt) {
settings.prompts.push({ findPrompt('jailbreak').content = settings.jailbreak_prompt
identifier: null,
name: 'My Jailbreak',
role: 'system',
content: settings.jailbreak_prompt,
system_prompt: false,
enabled: false,
});
delete settings.jailbreak_prompt; delete settings.jailbreak_prompt;
} }
saveSettingsCallback();
} }
}; };