Stable Diffusion: wait for API reconnect after switching to the image-prompt profile

8a47b3fdc48dc9404d14df231a7bfc71cd61f4a0

permissionBRICK <permissionBRICK@gmail.com>

public/scripts/extensions/stable-diffusion/index.js+8 -3 Viewed

     getCurrentChatId,
     getRequestHeaders,
     getUserAvatar,
 +    online_status,
     saveSettingsDebounced,
     substituteParams,
     substituteParamsExtended,

     resetScrollHeight,
     saveBase64AsFile,
     stringFormat,
 +    waitUntilCondition,
 } from '../../utils.js';
 import { getMessageTimeStamp, humanizedDateTime } from '../../RossAscends-mods.js';
 import { SECRET_KEYS, secret_state } from '../../secrets.js';

     const switchToProfile = async (profileId) => {
         const loaded = new Promise(resolve => eventSource.once(event_types.CONNECTION_PROFILE_LOADED, resolve));
 -        const timeout = new Promise(resolve => setTimeout(resolve, 10000));
         const index = Array.from(select.options).findIndex(o => o.value === profileId);
         select.selectedIndex = index >= 0 ? index : 0;
         select.dispatchEvent(new Event('change'));
         // Don'tWait hangfor the imageprofile's generationcommands to finish applying (don't hang forever if the event never fires).
         await Promise.race([loaded, timeoutdelay(10000)]);
 +        // Applying a profile reconnects the API, which is asynchronous. Generating before the
 +        // connection is re-established fails instantly, so wait for it to come back up
 +        // (mirrors the built-in /profile command). rejectOnTimeout:false -> proceed anyway after the timeout.
 +        await waitUntilCondition(() => online_status !== 'no_connection', 10000, 100, { rejectOnTimeout: false });
     };
     await switchToProfile(targetProfileId);