diff --git a/public/scripts/extensions/caption/index.js b/public/scripts/extensions/caption/index.js index b2e8c7664..7194abc49 100644 --- a/public/scripts/extensions/caption/index.js +++ b/public/scripts/extensions/caption/index.js @@ -169,7 +169,11 @@ async function sendCaptionedMessage(caption, image) { }, }; context.chat.push(message); + const messageId = context.chat.length - 1; + await eventSource.emit(event_types.MESSAGE_SENT, messageId); context.addOneMessage(message); + await eventSource.emit(event_types.USER_MESSAGE_RENDERED, messageId); + await context.saveChat(); } /** diff --git a/public/scripts/extensions/stable-diffusion/index.js b/public/scripts/extensions/stable-diffusion/index.js index 4d2c35722..2ea64a631 100644 --- a/public/scripts/extensions/stable-diffusion/index.js +++ b/public/scripts/extensions/stable-diffusion/index.js @@ -3351,8 +3351,11 @@ async function sendMessage(prompt, image, generationType, additionalNegativePref }, }; context.chat.push(message); + const messageId = context.chat.length - 1; + await eventSource.emit(event_types.MESSAGE_RECEIVED, messageId); context.addOneMessage(message); - context.saveChat(); + await eventSource.emit(event_types.CHARACTER_MESSAGE_RENDERED, messageId); + await context.saveChat(); } /**