Merge branch 'staging' of https://github.com/Cohee1207/SillyTavern into staging

aa6df2cfb4f63fed850cc5917458c25763dc2ed2

RossAscends <124905043+RossAscends@users.noreply.github.com>

public/script.js+41 -11 Viewed

         this.messageDom = null;
         this.messageTextDom = null;
         this.messageTimerDom = null;
+        this.messageTokenCounterDom = null;
+        /** @type {HTMLTextAreaElement} */
         this.sendTextarea = document.querySelector('#send_textarea');
         this.type = type;
         this.force_name2 = force_name2;

         this.messageLogprobs = [];
+    }
+    #checkDomElements(messageId) {
+        if (this.messageDom === null || this.messageTextDom === null) {
+            this.messageDom = document.querySelector(`#chat .mes[mesid="${messageId}"]`);
+            this.messageTextDom = this.messageDom?.querySelector('.mes_text');
+            this.messageTimerDom = this.messageDom?.querySelector('.mes_timer');
+            this.messageTokenCounterDom = this.messageDom?.querySelector('.tokenCounterDisplay');
+        }
+    }
     showMessageButtons(messageId) {
         if (messageId == -1) {
             return;

         else {
             await saveReply(this.type, text, true);
             messageId = chat.length - 1;
-            this.messageDom = document.querySelector(`#chat .mes[mesid="${messageId}"]`);
+            this.#checkDomElements(messageId);
-            this.messageTextDom = this.messageDom.querySelector('.mes_text');
-            this.messageTimerDom = this.messageDom.querySelector('.mes_timer');
             this.showMessageButtons(messageId);
+        }

             this.sendTextarea.dispatchEvent(new Event('input', { bubbles: true }));
+        }
         else {
-            let currentTime = new Date();
+            this.#checkDomElements(messageId);
+            const currentTime = new Date();
             // Don't waste time calculating token count for streaming
-            let currentTokenCount = isFinal && power_user.message_token_count_enabled ? getTokenCount(processedText, 0) : 0;
+            const currentTokenCount = isFinal && power_user.message_token_count_enabled ? getTokenCount(processedText, 0) : 0;
             const timePassed = formatGenerationTimer(this.timeStarted, currentTime, currentTokenCount);
             chat[messageId]['mes'] = processedText;
             chat[messageId]['gen_started'] = this.timeStarted;

@@ -2895,8 +2905,9 @@ class StreamingProcessor {
2895	}	2905	}
2896		2906
2897	chat[messageId]['extra']['token_count'] = currentTokenCount;	2907	chat[messageId]['extra']['token_count'] = currentTokenCount;
2898	const tokenCounter = $(`#chat .mes[mesid="${messageId}"] .tokenCounterDisplay`);	2908	if (this.messageTokenCounterDom instanceof HTMLElement) {
2899	tokenCounter.text(`${currentTokenCount}t`);	2909	this.messageTokenCounterDom.textContent = `${currentTokenCount}t`;
		2910	}
2900	}	2911	}
2901		2912
2902	if ((this.type == 'swipe' \|\| this.type === 'continue') && Array.isArray(chat[messageId]['swipes'])) {	2913	if ((this.type == 'swipe' \|\| this.type === 'continue') && Array.isArray(chat[messageId]['swipes'])) {

                 chat[messageId]['swipe_info'][chat[messageId]['swipe_id']] = { 'send_date': chat[messageId]['send_date'], 'gen_started': chat[messageId]['gen_started'], 'gen_finished': chat[messageId]['gen_finished'], 'extra': JSON.parse(JSON.stringify(chat[messageId]['extra'])) };
+            }
-            let formattedText = messageFormatting(
+            const formattedText = messageFormatting(
                 processedText,
                 chat[messageId].name,
                 chat[messageId].is_system,
                 chat[messageId].is_user,
                 messageId,
             );
+            if (this.messageTextDom instanceof HTMLElement) {
                 this.messageTextDom.innerHTML = formattedText;
+            }
+            if (this.messageTimerDom instanceof HTMLElement) {
                 this.messageTimerDom.textContent = timePassed.timerValue;
                 this.messageTimerDom.title = timePassed.timerTitle;
+            }
             this.setFirstSwipe(messageId);
+        }

+}
 /**
+ * Removes last message from the chat DOM.
+ * @returns {Promise<void>} Resolves when the message is removed.
+ */
+function removeLastMessage() {
+    return new Promise((resolve) => {
+        const lastMes = $('#chat').children('.mes').last();
+        if (lastMes.length === 0) {
+            return resolve();
+        }
+        lastMes.hide(animation_duration, function () {
+            $(this).remove();
+            resolve();
+        });
+    });
+}
+/**
  * Runs a generation using the current chat context.
  * @param {string} type Generation type
  * @param {GenerateOptions} options Generation options

@@ -3331,9 +3363,7 @@ export async function Generate(type, { automatic_trigger, force_name2, quiet_pro
3331	}	3363	}
3332	else if (type !== 'quiet' && type !== 'swipe' && !isImpersonate && !dryRun && chat.length) {	3364	else if (type !== 'quiet' && type !== 'swipe' && !isImpersonate && !dryRun && chat.length) {
3333	chat.length = chat.length - 1;	3365	chat.length = chat.length - 1;
3334	$('#chat').children().last().hide(250, function () {	3366	await removeLastMessage();
3335	$(this).remove();
3336	});
3337	await eventSource.emit(event_types.MESSAGE_DELETED, chat.length);	3367	await eventSource.emit(event_types.MESSAGE_DELETED, chat.length);
3338	}	3368	}
3339	}	3369	}

public/scripts/bookmarks.js+2 -2 Viewed

     saveChatConditional,
     saveItemizedPrompts,
 } from '../script.js';
-import { humanizedDateTime } from './RossAscends-mods.js';
+import { humanizedDateTime, getMessageTimeStamp } from './RossAscends-mods.js';
 import {
     getGroupPastChats,
     group_activation_strategy,

     if (groupChat.length === 0) {
         const newMessage = {
             ...system_messages[system_message_types.GROUP],
-            send_date: humanizedDateTime(),
+            send_date: getMessageTimeStamp(),
             extra: { type: system_message_types.GROUP },
         };
         groupChat.push(newMessage);