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();
+}
-function continueChatCallback(_, prompt) {
+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 '';
+}