/stop slash command to stop generation

7de43d3a9cdc2e27826577a2c373363c8fad8625

Wolfsblvt <wolfsblvt@gmail.com>

public/script.js+23 -9 Viewed

 let dialogueResolve = null;
 let dialogueCloseStop = false;
 export let chat_metadata = {};
 +/** @type {StreamingProcessor} */
 export let streamingProcessor = null;
 let crop_data = undefined;
 let is_delete_mode = false;

 //novel settings
 export let novelai_settings;
 export let novelai_setting_names;
 +/** @type {AbortController} */
 let abortController;
 //css

+}
 /**
 + * Stops the generation and any streaming if it is currently running.
 + */
 +export function stopGeneration() {
 +    let stopped = false;
 +    if (streamingProcessor) {
 +        streamingProcessor.onStopStreaming();
 +        streamingProcessor = null;
 +        stopped = true;
 +    }
 +    if (abortController) {
 +        abortController.abort('Clicked stop button');
 +        hideStopButton();
 +        stopped = true;
 +    }
 +    eventSource.emit(event_types.GENERATION_STOPPED);
 +    return stopped;
 +}
++
 +/**
  * Injects extension prompts into chat messages.
  * @param {object[]} messages Array of chat messages
  * @param {boolean} isContinue Whether the generation is a continuation. If true, the extension prompts of depth 0 are injected at position 1.

     });
     $(document).on('click', '.mes_stop', function () {
 -        if (streamingProcessor) {
 +        stopGeneration();
 -            streamingProcessor.onStopStreaming();
 -            streamingProcessor = null;
 -        }
 -        if (abortController) {
 -            abortController.abort('Clicked stop button');
 -            hideStopButton();
 -        }
 -        eventSource.emit(event_types.GENERATION_STOPPED);
     });
     $(document).on('click', '#form_sheld .stscript_continue', function () {

     initCustomSelectedSamplers();
 });
++

public/scripts/slash-commands.js+19 -0 Viewed

     setCharacterName,
     setExtensionPrompt,
     setUserName,
 +    stopGeneration,
     substituteParams,
     system_avatar,
     system_message_types,

         helpString: 'Adds a swipe to the last chat message.',
     }));
     SlashCommandParser.addCommandObject(SlashCommand.fromProps({
 +        name: 'stop',
 +        callback: () => {
 +            const stopped = stopGeneration();
 +            return String(stopped);
 +        },
 +        returns: 'true/false, whether the generation was running and got stopped',
 +        helpString: `
 +            <div>
 +                Stops the generation and any streaming if it is currently running.
 +            </div>
 +            <div>
 +                Note: This command cannot be executed from the chat input, as sending any message or script from there is blocked during generation.
 +                But it can be executed via automations or QR scripts/buttons.
 +            </div>
 +        `,
 +        aliases: ['generate-stop'],
 +    }));
 +    SlashCommandParser.addCommandObject(SlashCommand.fromProps({
         name: 'abort',
         callback: abortCallback,
         namedArgumentList: [