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

Prevent preset migration from running multiple times.

Browse Source
This commit is contained in:
maver
2023-08-08 20:37:48 +02:00
parent a999d6e171
commit 7fb896ce2a
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
public/scripts/PromptManager.js
View File

@ -8,8 +8,6 @@ import {power_user} from "./power-user.js";
const registerPromptManagerMigration = (saveSettingsCallback) => { const registerPromptManagerMigration = (saveSettingsCallback) => {
const migrate = (settings, preset) => { const migrate = (settings, preset) => {
// If any of the specified settings exist, run the migration // If any of the specified settings exist, run the migration
if (preset.main_prompt || preset.nsfw_prompt || preset.jailbreak_prompt) {
console.log('Running one-time configuration migration for prompt manager.')
if (settings.prompts === undefined || settings.prompts.length === 0) settings.prompts = chatCompletionDefaultPrompts.prompts; if (settings.prompts === undefined || settings.prompts.length === 0) settings.prompts = chatCompletionDefaultPrompts.prompts;
const findPrompt = (identifier) => settings.prompts.find(prompt => identifier === prompt.identifier); const findPrompt = (identifier) => settings.prompts.find(prompt => identifier === prompt.identifier);
@ -27,15 +25,21 @@ const registerPromptManagerMigration = (saveSettingsCallback) => {
findPrompt('jailbreak').content = preset.jailbreak_prompt findPrompt('jailbreak').content = preset.jailbreak_prompt
delete settings.jailbreak_prompt; delete settings.jailbreak_prompt;
} }
saveSettingsCallback();
}
}; };
const migrateSettings = (settings) => migrate(settings, settings); const migrateSettings = (settings) => {
if (settings.main_prompt || settings.nsfw_prompt || settings.jailbreak_prompt) {
console.log(`Migrating configuration`);
migrate(settings, settings);
}
}
const migratePreset = (event) => { const migratePreset = (event) => {
migrate(event.settings, event.preset); if (event.preset.main_prompt || event.preset.nsfw_prompt || event.preset.jailbreak_prompt) {
event.callback(event.presetName, event.settings, false); console.log(`Migrating preset ${event.presetName}`);
migrate(event.settings, event.preset);
event.callback(event.presetName, event.settings, false);
}
} }
eventSource.on(event_types.SETTINGS_LOADED_BEFORE, settings => migrateSettings(settings)); eventSource.on(event_types.SETTINGS_LOADED_BEFORE, settings => migrateSettings(settings));