Merge branch 'staging' into request-proxy

051cdf554889269cb5f6a40b38b70dc28dd46560

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

Parents: f62e57d737a594a57bf3b734948c2c3375b765d8 81cd38d4f70bfbb31641618382afd6d0d8df99aa

public/scripts/slash-commands.js+26 -24 Viewed

     SlashCommandParser.addCommandObject(SlashCommand.fromProps({
         name: 'ask',
         callback: askCharacter,
+        returns: 'the generated text',
         namedArgumentList: [
             SlashCommandNamedArgument.fromProps({
                 name: 'name',

         ],
         unnamedArgumentList: [
             new SlashCommandArgument(
-                'prompt', [ARGUMENT_TYPE.STRING], true, false,
+                'prompt', [ARGUMENT_TYPE.STRING], false, false,
             ),
         ],
         helpString: 'Asks a specified character card a prompt. Character name must be provided in a named argument.',

     // Not supported in group chats
     // TODO: Maybe support group chats?
     if (selected_group) {
-        toastr.error('Cannot run this command in a group chat!');
+        toastr.error('Cannot run /ask command in a group chat!');
-        return '';
-    }
-    if (!text) {
-        console.warn('WARN: No text provided for /ask command');
-        toastr.warning('No text provided for /ask command');
         return '';
+    }
     let name = '';
-    let mesText = '';
     if (args?.name) {
         name = args.name.trim();
-        mesText = text.trim();
-        if (!name && !mesText) {
+        if (!name) {
-            toastr.warning('You must specify a name and text to ask.');
+            toastr.warning('You must specify a name of the character to ask.');
             return '';
+        }
+    }
-    mesText = getRegexedString(mesText, regex_placement.SLASH_COMMAND);
     const prevChId = this_chid;
     // Find the character
-    const chId = characters.findIndex((e) => e.name === name);
+    const chId = characters.findIndex((e) => e.name === name || e.avatar === name);
     if (!characters[chId] || chId === -1) {
         toastr.error('Character not found.');
         return '';
+    }
+    if (text) {
+        const mesText = getRegexedString(text.trim(), regex_placement.SLASH_COMMAND);
+        // Sending a message implicitly saves the chat, so this needs to be done before changing the character
+        // Otherwise, a corruption will occur
+        await sendMessageAsUser(mesText, '');
+    }
     // Override character and send a user message
     setCharacterId(String(chId));
-    // TODO: Maybe look up by filename instead of name
     const character = characters[chId];
     let force_avatar, original_avatar;

     setCharacterName(character.name);
-    await sendMessageAsUser(mesText, '');
     const restoreCharacter = () => {
+        if (String(this_chid) !== String(chId)) {
+            return;
+        }
         setCharacterId(prevChId);
         setCharacterName(characters[prevChId].name);

+        }
     };
+    let askResult = '';
     // Run generate and restore previous character
     try {
+        eventSource.once(event_types.MESSAGE_RECEIVED, restoreCharacter);
         toastr.info(`Asking ${character.name} something...`);
-        await Generate('ask_command');
+        askResult = await Generate('ask_command');
-        await saveChatConditional();
     } catch (error) {
+        restoreCharacter();
         console.error('Error running /ask command', error);
     } finally {
-        restoreCharacter();
+        if (String(this_chid) === String(prevChId)) {
+            await saveChatConditional();
-        if (this_chid !== prevChId) {
+        } else {
             toastr.error('It is strongly recommended to reload the page.', 'Something went wrong');
+        }
+    }
-    return '';
+    return askResult;
+}
 async function hideMessageCallback(_, arg) {