Migrate sysprompts from instruct

2f7d694f5429fd95e342bd0825e96eee1cc66170

Cohee <18619528+Cohee1207@users.noreply.github.com>

Parents: 73ee8697495dcfdffaef9f409d4562114866415b

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,