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

Copy migrated preset into loaded presets

Browse Source 
Since this only is populated once during page load
This commit is contained in:
maver
2023-08-10 19:56:18 +02:00
parent 7fb896ce2a
commit b1254fa2ab
3 changed files with 19 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
public/scripts/PromptManager.js
View File

@ -5,45 +5,32 @@ import {power_user} from "./power-user.js";
/**
* Register migrations for the prompt manager when settings are loaded or an Open AI preset is loaded.
*/
const registerPromptManagerMigration = (saveSettingsCallback) => {
const migrate = (settings, preset) => {
// If any of the specified settings exist, run the migration
const registerPromptManagerMigration = () => {
const migrate = (settings) => {
if (settings.main_prompt || settings.nsfw_prompt || settings.jailbreak_prompt) {
console.log('Running prompt manager configuration migration');
if (settings.prompts === undefined || settings.prompts.length === 0) settings.prompts = chatCompletionDefaultPrompts.prompts;
const findPrompt = (identifier) => settings.prompts.find(prompt => identifier === prompt.identifier);
if (preset.main_prompt) {
findPrompt('main').content = preset.main_prompt
if (settings.main_prompt) {
findPrompt('main').content = settings.main_prompt
delete settings.main_prompt;
}
if (preset.nsfw_prompt) {
findPrompt('nsfw').content = preset.nsfw_prompt
if (settings.nsfw_prompt) {
findPrompt('nsfw').content = settings.nsfw_prompt
delete settings.nsfw_prompt;
}
if (preset.jailbreak_prompt) {
findPrompt('jailbreak').content = preset.jailbreak_prompt
if (settings.jailbreak_prompt) {
findPrompt('jailbreak').content = settings.jailbreak_prompt
delete settings.jailbreak_prompt;
}
}
};
const migrateSettings = (settings) => {
if (settings.main_prompt || settings.nsfw_prompt || settings.jailbreak_prompt) {
console.log(`Migrating configuration`);
migrate(settings, settings);
}
}
const migratePreset = (event) => {
if (event.preset.main_prompt || event.preset.nsfw_prompt || event.preset.jailbreak_prompt) {
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.OAI_PRESET_CHANGED, settings => migratePreset(settings));
eventSource.on(event_types.SETTINGS_LOADED_BEFORE, settings => migrate(settings));
eventSource.on(event_types.OAI_PRESET_CHANGED, event => migrate(event.preset));
}
/**