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], truefalse, 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/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) {
             toastr.warning('You must specify a name andof textthe 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 || 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...`);
         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 ''askResult;
+}
 async function hideMessageCallback(_, arg) {