Add slash commands and connection profiles support

0bc6a572c6ca35a863fe2688911dabfee5a5056d

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

Parents: 912bebeb019d05a6f69a8bf4599ff69a6f18b4ff

public/scripts/extensions/connection-manager/index.js+9 -0 Viewed

 const TC_COMMANDS = [
     ...COMMON_COMMANDS,
 +    'sysprompt',
 +    'sysprompt-state',
     'instruct',
     'context',
     'instruct-state',

     'preset': 'Settings Preset',
     'model': 'Model',
     'proxy': 'Proxy Preset',
 +    'sysprompt-state': 'Use System Prompt',
 +    'sysprompt': 'System Prompt Name',
     'instruct-state': 'Instruct Mode',
     'instruct': 'Instruct Template',
     'context': 'Context Template',

+    }
     if (cleanUp) {
 +        for (const command of commands) {
 +            if (command.endsWith('-state') && profile[command] === 'false') {
 +                delete profile[command.replace('-state', '')];
 +            }
 +        }
         for (const command of opposingCommands) {
             if (commands.includes(command)) {
                 continue;

public/scripts/sysprompt.js+130 -1 Viewed

 import { saveSettingsDebounced } from '../script.js';
 import { power_user } from './power-user.js';
 import { resetScrollHeightSlashCommand } from './utilsslash-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 { isTrueBoolean, resetScrollHeight } from './utils.js';
 export let system_prompts = [];

     $contentBlock.toggleClass('disabled', !power_user.sysprompt.enabled);
+}
 +function enableSystemPromptCallback() {
 +    power_user.sysprompt.enabled = true;
 +    $enabled.prop('checked', true);
 +    toggleSyspromptDisabledControls();
 +    saveSettingsDebounced();
 +    return '';
 +}
++
 +function disableSystemPromptCallback() {
 +    power_user.sysprompt.enabled = false;
 +    $enabled.prop('checked', false);
 +    toggleSyspromptDisabledControls();
 +    saveSettingsDebounced();
 +    return '';
 +}
++
 +function toggleSystemPromptCallback(_args, state) {
 +    if (!state || typeof state !== 'string') {
 +        return String(power_user.sysprompt.enabled);
 +    }
++
 +    const newState = isTrueBoolean(state);
 +    newState ? enableSystemPromptCallback() : disableSystemPromptCallback();
 +    return String(power_user.sysprompt.enabled);
 +}
++
 +function selectSystemPromptCallback(args, name) {
 +    if (!power_user.sysprompt.enabled && !isTrueBoolean(args.forceGet)) {
 +        return '';
 +    }
++
 +    if (!name) {
 +        return power_user.sysprompt.name;
 +    }
++
 +    const quiet = isTrueBoolean(args?.quiet);
 +    const instructNames = system_prompts.map(preset => preset.name);
 +    const fuse = new Fuse(instructNames);
 +    const result = fuse.search(name);
++
 +    if (result.length === 0) {
 +        !quiet && toastr.warning(`System prompt "${name}" not found`);
 +        return '';
 +    }
++
 +    const foundName = result[0].item;
 +    $select.val(foundName).trigger('input');
 +    !quiet && toastr.success(`System prompt "${foundName}" selected`);
 +    return foundName;
 +}
++
 jQuery(function () {
     $enabled.on('input', function () {
         power_user.sysprompt.enabled = !!$(this).prop('checked');

     });
     $select.on('input', async function () {
 +        if (!power_user.sysprompt.enabled) {
 +            $enabled.prop('checked', true).trigger('input');
 +        }
++
         const name = String($(this).val());
         const prompt = system_prompts.find(p => p.name === name);
         if (prompt) {

         power_user.sysprompt.content = String($(this).val());
         saveSettingsDebounced();
     });
++
 +    SlashCommandParser.addCommandObject(SlashCommand.fromProps({
 +        name: 'sysprompt',
 +        aliases: ['system-prompt'],
 +        callback: selectSystemPromptCallback,
 +        returns: 'current prompt name',
 +        namedArgumentList: [
 +            SlashCommandNamedArgument.fromProps({
 +                name: 'quiet',
 +                description: 'Suppress the toast message on prompt change',
 +                typeList: [ARGUMENT_TYPE.BOOLEAN],
 +                defaultValue: 'false',
 +                enumList: commonEnumProviders.boolean('trueFalse')(),
 +            }),
 +            SlashCommandNamedArgument.fromProps({
 +                name: 'forceGet',
 +                description: 'Force getting a name even if system prompt is disabled',
 +                typeList: [ARGUMENT_TYPE.BOOLEAN],
 +                defaultValue: 'false',
 +                enumList: commonEnumProviders.boolean('trueFalse')(),
 +            }),
 +        ],
 +        unnamedArgumentList: [
 +            SlashCommandArgument.fromProps({
 +                description: 'system prompt name',
 +                typeList: [ARGUMENT_TYPE.STRING],
 +                enumProvider: () => system_prompts.map(x => new SlashCommandEnumValue(x.name, null, enumTypes.enum, enumIcons.preset)),
 +            }),
 +        ],
 +        helpString: `
 +            <div>
 +                Selects a system prompt by name. Enables the use of system prompt if not already enabled.
 +                Gets the current system prompt if no name is provided and sysprompt is enabled or <code>forceGet=true</code> is passed.
 +            </div>
 +            <div>
 +                <strong>Example:</strong>
 +                <ul>
 +                    <li>
 +                        <pre><code class="language-stscript">/sysprompt </code></pre>
 +                    </li>
 +                </ul>
 +            </div>
 +        `,
 +    }));
 +    SlashCommandParser.addCommandObject(SlashCommand.fromProps({
 +        name: 'sysprompt-on',
 +        aliases: ['sysprompt-enable'],
 +        callback: enableSystemPromptCallback,
 +        helpString: 'Enables system prompt.',
 +    }));
 +    SlashCommandParser.addCommandObject(SlashCommand.fromProps({
 +        name: 'sysprompt-off',
 +        aliases: ['sysprompt-disable'],
 +        callback: disableSystemPromptCallback,
 +        helpString: 'Disables system prompt',
 +    }));
 +    SlashCommandParser.addCommandObject(SlashCommand.fromProps({
 +        name: 'sysprompt-state',
 +        aliases: ['sysprompt-toggle'],
 +        helpString: 'Gets the current system prompt state. If an argument is provided, it will set the system prompt state.',
 +        unnamedArgumentList: [
 +            SlashCommandArgument.fromProps({
 +                description: 'system prompt state',
 +                typeList: [ARGUMENT_TYPE.BOOLEAN],
 +                enumList: commonEnumProviders.boolean('trueFalse')(),
 +            }),
 +        ],
 +        callback: toggleSystemPromptCallback,
 +    }));
 });