Merge pull request #3561 from SillyTavern/fix-syncing-swipe-data Fix syncing swipe data on reasoning parse & utility function to sync swipe data

7b55d91d35e9de0b74624abca8899c1851337b4d

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed

Parents: 5955327f7b244ed55ee954eb93a7ad42d51e82fd 8b135f9ca3c7d305662a844afc0effbcc28916a3

public/script.js+53 -11 Viewed

     return { type, getMessage };
+}
 -export function syncCurrentSwipeInfoExtras() {
 +/**
 + * Syncs the current message and all its data into the swipe data at the given message ID (or the last message if no ID is given).
 + *
 + * If the swipe data is invalid in some way, this function will exit out without doing anything.
 + * @param {number?} [messageId=null] - The ID of the message to sync with the swipe data. If no ID is given, the last message is used.
 + * @returns {boolean} Whether the message was successfully synced
 + */
 +export function syncMesToSwipe(messageId = null) {
     if (!chat.length) {
         return false;
+    }
 -    const currentMessage = chat[chat.length - 1];
++
 -    if (currentMessage && Array.isArray(currentMessage.swipe_info) && typeof currentMessage.swipe_id === 'number') {
 +    const targetMessageId = messageId ?? chat.length - 1;
 -        const swipeInfo = currentMessage.swipe_info[currentMessage.swipe_id];
 +    if (chat.length > targetMessageId || targetMessageId < 0) {
 -        if (swipeInfo && typeof swipeInfo === 'object') {
 +        console.warn(`[syncMesToSwipe] Invalid message ID: ${messageId}`);
 -            swipeInfo.extra = structuredClone(currentMessage.extra);
 +        return false;
+            }
++
 +    const targetMessage = chat[targetMessageId];
++
 +    // No swipe data there yet, exit out
 +    if (typeof targetMessage.swipe_id !== 'number') {
 +        return false;
+    }
 +    // If swipes structure is invalid, exit out (for now?)
 +    if (!Array.isArray(targetMessage.swipe_info) || !Array.isArray(targetMessage.swipes)) {
 +        return false;
 +    }
 +    // If the swipe is not present yet, exit out (will likely be copied later)
 +    if (!targetMessage.swipes[targetMessage.swipe_id] || !targetMessage.swipe_info[targetMessage.swipe_id]) {
 +        return false;
 +    }
++
 +    const targetSwipeInfo = targetMessage.swipe_info[targetMessage.swipe_id];
 +    if (typeof targetSwipeInfo !== 'object') {
 +        return false;
 +    }
++
 +    targetMessage.swipes[targetMessage.swipe_id] = targetMessage.mes;
++
 +    targetSwipeInfo.send_date = targetMessage.send_date;
 +    targetSwipeInfo.gen_started = targetMessage.gen_started;
 +    targetSwipeInfo.gen_finished = targetMessage.gen_finished;
 +    targetSwipeInfo.extra = structuredClone(targetMessage.extra);
++
 +    return true;
+}
 function saveImageToMessage(img, mes) {

     if (chatSaveTimeout) {
         console.debug('Clearing chat save timeout');
         clearTimeout(chatSaveTimeout);
 +        chatSaveTimeout = null;
+    }
     chatSaveTimeout = setTimeout(async () => {

         console.debug('Chat save timeout triggered');
         await saveChatConditional();
         console.debug('Chat saved');
     }, 1000DEFAULT_SAVE_EDIT_TIMEOUT);
+}
 export async function saveChat(chatName, withMetadata, mesId) {

+    }
     try {
 +        if (chatSaveTimeout) {
 +            console.debug('Debounced chat save canceled');
 +            clearTimeout(chatSaveTimeout);
 +            chatSaveTimeout = null;
 +        }
++
         isChatSaving = true;
         if (selected_group) {

+    }
     // Make sure ad-hoc changes to extras are saved before swiping away
     syncCurrentSwipeInfoExtrassyncMesToSwipe();
     const swipe_duration = 120;
     const swipe_range = '700px';

+    }
     // Make sure ad-hoc changes to extras are saved before swiping away
     syncCurrentSwipeInfoExtrassyncMesToSwipe();
     const swipe_duration = 200;
     const swipe_range = 700;

public/scripts/reasoning.js+5 -2 Viewed

 import {
     moment,
 } from '../lib.js';
 import { chat, closeMessageEditor, event_types, eventSource, main_api, messageFormatting, saveChatConditional, saveChatDebounced, saveSettingsDebounced, substituteParams, syncMesToSwipe, updateMessageBlock } from '../script.js';
 import { getRegexedString, regex_placement } from './extensions/regex/engine.js';
 import { getCurrentLocale, t, translate } from './i18n.js';
 import { MacrosParser } from './macros.js';

             message.mes = parsedReasoning.content;
+        }
 -        // Find if a message already exists in DOM and must be updated
         if (contentUpdated) {
 +            syncMesToSwipe();
 +            saveChatDebounced();
++
 +            // Find if a message already exists in DOM and must be updated
             const messageRendered = document.querySelector(`.mes[mesid="${idx}"]`) !== null;
             if (messageRendered) {
                 console.debug('[Reasoning] Updating message block', idx);

public/scripts/slash-commands.js+2 -2 Viewed

     showMoreMessages,
     stopGeneration,
     substituteParams,
     syncCurrentSwipeInfoExtrassyncMesToSwipe,
     system_avatar,
     system_message_types,
     this_chid,

     if (isTrueBoolean(args.switch)) {
         // Make sure ad-hoc changes to extras are saved before swiping away
         syncCurrentSwipeInfoExtrassyncMesToSwipe();
         lastMessage.swipe_id = newSwipeId;
         lastMessage.mes = lastMessage.swipes[newSwipeId];
         lastMessage.extra = structuredClone(lastMessage.swipe_info?.[newSwipeId]?.extra ?? lastMessage.extra ?? {});