Added /continue "await" arg

aad65c9273176a2bf3960bf655a23db1adf4ae7e

Wolfsblvt <wolfsblvt@gmail.com>

Parents: 79069b49b8c8003d4fb57b721e830fbe3e576dc9

public/scripts/slash-commands.js+37 -12 Viewed

         name: 'continue',
         callback: continueChatCallback,
         aliases: ['cont'],
 +        namedArgumentList: [
 +            new SlashCommandNamedArgument(
 +                'await',
 +                'Whether to await for the continued generation before continuing',
 +                [ARGUMENT_TYPE.BOOLEAN],
 +                false,
 +                false,
 +                'false',
 +            ),
 +        ],
         unnamedArgumentList: [
             new SlashCommandArgument(
                 'prompt', [ARGUMENT_TYPE.STRING], false,

             Continues the last message in the chat, with an optional additional prompt.
         </div>
         <div>
 +            If <code>await=true</code> named argument is passed, the command will await for the continued generation before continuing.
 +        </div>
 +        <div>
             <strong>Example:</strong>
             <ul>
                 <li>

     await reloadCurrentChat();
+}
 async function continueChatCallback(_args, prompt) {
 -    setTimeout(async () => {
 +    const shouldAwait = isTrueBoolean(args?.await);
 -        try {
 -            await waitUntilCondition(() => !is_send_press && !is_group_generating, 10000, 100);
 -        } catch {
 -            console.warn('Timeout waiting for generation unlock');
 -            toastr.warning('Cannot run /continue command while the reply is being generated.');
 -        }
 -        // Prevent infinite recursion
 +    const outerPromise = new Promise((resolve) => {
 -        $('#send_textarea').val('')[0].dispatchEvent(new Event('input', { bubbles: true }));
 +        setTimeout(async () => {
 -        $('#option_continue').trigger('click', { fromSlashCommand: true, additionalPrompt: prompt });
 +            try {
 -    }, 1);
 +                await waitUntilCondition(() => !is_send_press && !is_group_generating, 10000, 100);
 +            } catch {
 +                console.warn('Timeout waiting for generation unlock');
 +                toastr.warning('Cannot run /continue command while the reply is being generated.');
 +            }
++
 +            // Prevent infinite recursion
 +            $('#send_textarea').val('')[0].dispatchEvent(new Event('input', { bubbles: true }));
++
 +            const options = prompt?.trim() ? { quiet_prompt: prompt.trim(), quietToLoud: true } : {};
 +            await Generate('continue', options);
++
 +            resolve();
 +        }, 1);
 +    });
++
 +    if (shouldAwait) {
 +        await outerPromise;
 +    }
     return '';
+}