2f7d694f · permissionBRICK/SillyTavern

public/scripts/sysprompt.js+16 -0 Viewed

 const $content = $('#sysprompt_content');
 const $contentBlock = $('#SystemPromptBlock');
+function migrateSystemPromptFromInstructMode() {
+    if ('system_prompt' in power_user.instruct) {
+        power_user.sysprompt.enabled = power_user.instruct.enabled;
+        power_user.sysprompt.content = String(power_user.instruct.system_prompt);
+        delete power_user.instruct.system_prompt;
+        if (system_prompts.some(x => x.name === power_user.instruct.preset)) {
+            power_user.sysprompt.name = power_user.instruct.preset;
+        }
+        saveSettingsDebounced();
+        toastr.info('System prompt settings have been moved from the Instruct Mode.', 'Migration notice', { timeOut: 5000 });
+    }
+}
 /**
  * Loads sysprompt settings from the given data object.
  * @param {object} data Settings data object.

         system_prompts = data.sysprompt;
+    }
+    migrateSystemPromptFromInstructMode();
     toggleSyspromptDisabledControls();
     for (const prompt of system_prompts) {

server.js+1 -0 Viewed

 userModule.initUserStorage(dataRoot)
     .then(userModule.ensurePublicDirectoriesExist)
     .then(userModule.migrateUserData)
+    .then(userModule.migrateSystemPrompts)
     .then(verifySecuritySettings)
     .then(preSetupTasks)
     .finally(startServer);

src/users.js+32 -0 Viewed

 const express = require('express');
 const mime = require('mime-types');
 const archiver = require('archiver');
+const writeFileAtomicSync = require('write-file-atomic').sync;
 const { USER_DIRECTORY_TEMPLATE, DEFAULT_USER, PUBLIC_DIRECTORIES, DEFAULT_AVATAR, SETTINGS_FILE } = require('./constants');
 const { getConfigValue, color, delay, setConfigValue, generateTimestamp } = require('./util');

     console.log(color.green('Migration completed!'));
+}
+async function migrateSystemPrompts() {
+    const directories = await getUserDirectoriesList();
+    for (const directory of directories) {
+        try {
+            const migrateMarker = path.join(directory.sysprompt, '.migrated');
+            if (fs.existsSync(migrateMarker)) {
+                continue;
+            }
+            const instucts = fs.readdirSync(directory.instruct);
+            for (const instruct of instucts) {
+                const instructPath = path.join(directory.instruct, instruct);
+                const syspromptPath = path.join(directory.sysprompt, instruct);
+                if (path.extname(instruct) === '.json' && !fs.existsSync(syspromptPath)) {
+                    const instructData = JSON.parse(fs.readFileSync(instructPath, 'utf8'));
+                    if ('system_prompt' in instructData && 'name' in instructData) {
+                        const syspromptData = { name: instructData.name, content: instructData.system_prompt };
+                        writeFileAtomicSync(syspromptPath, JSON.stringify(syspromptData, null, 4));
+                        delete instructData.system_prompt;
+                        writeFileAtomicSync(instructPath, JSON.stringify(instructData, null, 4));
+                        console.log(`Migrated system prompt ${instruct} for ${directory.root.split(path.sep).pop()}`);
+                    }
+                }
+            }
+            writeFileAtomicSync(migrateMarker, '');
+        } catch {
+            // Ignore errors
+        }
+    }
+}
 /**
  * Converts a user handle to a storage key.
  * @param {string} handle User handle

     requireLoginMiddleware,
     requireAdminMiddleware,
     migrateUserData,
+    migrateSystemPrompts,
     getPasswordSalt,
     getPasswordHash,
     getCsrfSecret,

Migrate sysprompts from instruct