From 9d73189133d20aa94e12ed3512243945caa54d77 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Sat, 25 Jan 2025 00:38:13 +0200
Subject: [PATCH] Add updateMessageBlock and appendMediaToMessage to getContext

---
 public/script.js             | 5 +++++
 public/scripts/st-context.js | 4 ++++
 2 files changed, 9 insertions(+)

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,
     };
 }