Merge pull request #3538 from SillyTavern/reasoning-parse-on-edit Reasoning edit/parse updates

6df2dc11f79fbeff6bcc7a86ae7cfd4b022bde29

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

Signed

public/scripts/reasoning.js+33 -6 Viewed

 import {
     moment,
 } from '../lib.js';
 import { chat, closeMessageEditor, event_types, eventSource, main_api, messageFormatting, saveChatConditional, saveChatDebounced, saveSettingsDebounced, substituteParams, 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';

+}
 function setReasoningEventHandlers() {
 +    /**
 +     * Updates the reasoning block of a message from a value.
 +     * @param {object} message Message object
 +     * @param {string} value Reasoning value
 +     */
 +    function updateReasoningFromValue(message, value) {
 +        const reasoning = getRegexedString(value, regex_placement.REASONING, { isEdit: true });
 +        message.extra.reasoning = reasoning;
 +        message.extra.reasoning_type = message.extra.reasoning_type ? ReasoningType.Edited : ReasoningType.Manual;
 +    }
++
     $(document).on('click', '.mes_reasoning_details', function (e) {
         if (!e.target.closest('.mes_reasoning_actions') && !e.target.closest('.mes_reasoning_header')) {
             e.preventDefault();

+        }
         const textarea = messageBlock.find('.reasoning_edit_textarea');
 -        const reasoning = getRegexedString(String(textarea.val()), regex_placement.REASONING, { isEdit: true });
 +        updateReasoningFromValue(message, String(textarea.val()));
 -        message.extra.reasoning = reasoning;
 -        message.extra.reasoning_type = message.extra.reasoning_type ? ReasoningType.Edited : ReasoningType.Manual;
         await saveChatConditional();
         updateMessageBlock(messageId, message);
         textarea.remove();

         await copyText(reasoning);
         toastr.info(t`Copied!`, '', { timeOut: 2000 });
     });
++
 +    $(document).on('input', '.reasoning_edit_textarea', function () {
 +        if (!power_user.auto_save_msg_edits) {
 +            return;
 +        }
++
 +        const { message } = getMessageFromJquery(this);
 +        if (!message?.extra) {
 +            return;
 +        }
++
 +        updateReasoningFromValue(message, String($(this).val()));
 +        saveChatDebounced();
 +    });
+}
 /**

@@ -973,7 +996,7 @@ function parseReasoningFromString(str, { strict = true } = {}) {
973	996	}
974	997
975	998	function registerReasoningAppEvents() {
976	999	eventSource.makeFirst(event_types.MESSAGE_RECEIVED,const eventHandler = (/** @type {number} */ idx) => {
977	1000	if (!power_user.reasoning.auto_parse) {
978	1001	return;
979	1002	}

@@ -1029,7 +1052,11 @@ function registerReasoningAppEvents() {
1029	1052	updateMessageBlock(idx, message);
1030	1053	}
1031	1054	}
1032	1055	});
	1056	+
	1057	+ for (const event of [event_types.MESSAGE_RECEIVED, event_types.MESSAGE_UPDATED]) {
	1058	+ eventSource.on(event, eventHandler);
	1059	+ }
1033	1060	}
1034	1061
1035	1062	export function initReasoning() {