diff --git a/public/script.js b/public/script.js
index 4c8d65766..c18d047b1 100644
--- a/public/script.js
+++ b/public/script.js
@@ -2240,6 +2240,11 @@ function getMessageFromTemplate({
     return mes;
 }
 
+/**
+ * Re-renders a message block with updated content.
+ * @param {number} messageId Message ID
+ * @param {object} message Message object
+ */
 export function updateMessageBlock(messageId, message) {
     const messageElement = $(`#chat [mesid="${messageId}"]`);
     const text = message?.extra?.display_text ?? message.mes;
diff --git a/public/scripts/st-context.js b/public/scripts/st-context.js
index a02fd5492..1c7a46199 100644
--- a/public/scripts/st-context.js
+++ b/public/scripts/st-context.js
@@ -1,6 +1,7 @@
 import {
     activateSendButtons,
     addOneMessage,
+    appendMediaToMessage,
     callPopup,
     characters,
     chat,
@@ -41,6 +42,7 @@ import {
     substituteParamsExtended,
     this_chid,
     updateChatMetadata,
+    updateMessageBlock,
 } from '../script.js';
 import {
     extension_settings,
@@ -171,6 +173,8 @@ export function getContext() {
         getCharacters,
         uuidv4,
         humanizedDateTime,
+        updateMessageBlock,
+        appendMediaToMessage,
     };
 }