Migrate system prompts from imported instruct template

b0c537d014f3707e12bd32e830cf2556152aebd5

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

public/scripts/preset-manager.js+6 -2 Viewed

 import { enumIcons } from './slash-commands/SlashCommandCommonEnumsProvider.js';
 import { SlashCommandEnumValue, enumTypes } from './slash-commands/SlashCommandEnumValue.js';
 import { SlashCommandParser } from './slash-commands/SlashCommandParser.js';
-import { system_prompts } from './sysprompt.js';
+import { checkForSystemPromptInInstructTemplate, system_prompts } from './sysprompt.js';
 import {
     textgenerationwebui_preset_names,
     textgenerationwebui_presets,

  * @param {string} apiId API id
  * @returns {PresetManager} Preset manager
  */
-function getPresetManager(apiId = '') {
+export function getPresetManager(apiId = '') {
     if (!apiId) {
         apiId = main_api == 'koboldhorde' ? 'kobold' : main_api;
+    }

+    }
     async savePreset(name, settings) {
+        if (this.apiId === 'instruct') {
+            await checkForSystemPromptInInstructTemplate(name, settings);
+        }
         const preset = settings ?? this.getPresetSettings(name);
         const response = await fetch('/api/presets/save', {

public/scripts/sysprompt.js+30 -5 Viewed

 import { saveSettingsDebounced } from '../script.js';
+import { callGenericPopup, POPUP_TYPE } from './popup.js';
 import { power_user } from './power-user.js';
+import { getPresetManager } from './preset-manager.js';
 import { SlashCommand } from './slash-commands/SlashCommand.js';
 import { ARGUMENT_TYPE, SlashCommandArgument, SlashCommandNamedArgument } from './slash-commands/SlashCommandArgument.js';
 import { commonEnumProviders, enumIcons } from './slash-commands/SlashCommandCommonEnumsProvider.js';
 import { enumTypes, SlashCommandEnumValue } from './slash-commands/SlashCommandEnumValue.js';
 import { SlashCommandParser } from './slash-commands/SlashCommandParser.js';
+import { renderTemplateAsync } from './templates.js';
 import { isTrueBoolean, resetScrollHeight } from './utils.js';
 export let system_prompts = [];

+    }
     migrateSystemPromptFromInstructMode();
-    toggleSyspromptDisabledControls();
+    toggleSystemPromptDisabledControls();
     for (const prompt of system_prompts) {
         $('<option>').val(prompt.name).text(prompt.name).appendTo($select);

+    }
+}
-function toggleSyspromptDisabledControls() {
+/**
+ * Checks if the instruct template has a system prompt and prompts the user to save it as a system prompt.
+ * @param {string} name Name of the instruct template
+ * @param {object} template Instruct template object
+ */
+export async function checkForSystemPromptInInstructTemplate(name, template) {
+    if ('system_prompt' in template && name && template.system_prompt && !system_prompts.some(x => x.name === name)) {
+        const html = await renderTemplateAsync('migrateInstructPrompt', { prompt: template.system_prompt });
+        const confirm = await callGenericPopup(html, POPUP_TYPE.CONFIRM);
+        if (confirm) {
+            const prompt = { name: name, content: template.system_prompt };
+            const presetManager = getPresetManager('sysprompt');
+            await presetManager.savePreset(prompt.name, prompt);
+            toastr.success(`System prompt "${prompt.name}" has been saved.`);
+        } else {
+            toastr.info('System prompt has been discarded.');
+        }
+        delete template.system_prompt;
+    }
+}
+function toggleSystemPromptDisabledControls() {
     $enabled.parent().find('i').toggleClass('toggleEnabled', !!power_user.sysprompt.enabled);
     $contentBlock.toggleClass('disabled', !power_user.sysprompt.enabled);
+}

 function enableSystemPromptCallback() {
     power_user.sysprompt.enabled = true;
     $enabled.prop('checked', true);
-    toggleSyspromptDisabledControls();
+    toggleSystemPromptDisabledControls();
     saveSettingsDebounced();
     return '';
+}

 function disableSystemPromptCallback() {
     power_user.sysprompt.enabled = false;
     $enabled.prop('checked', false);
-    toggleSyspromptDisabledControls();
+    toggleSystemPromptDisabledControls();
     saveSettingsDebounced();
     return '';
+}

 jQuery(function () {
     $enabled.on('input', function () {
         power_user.sysprompt.enabled = !!$(this).prop('checked');
-        toggleSyspromptDisabledControls();
+        toggleSystemPromptDisabledControls();
         saveSettingsDebounced();
     });

public/scripts/templates/migrateInstructPrompt.html+7 -0 Viewed

+<h3>
+    This instruct template also contains a system prompt.
+</h3>
+<div class="marginBot10">
+    Would you like to migrate the system prompt from the template?
+</div>
+<textarea class="wide100p textarea_compact" rows="8">{{prompt}}</textarea>