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, saveSettingsDebounced, substituteParams, updateMessageBlock } from '../script.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	function registerReasoningAppEvents() {	998	function registerReasoningAppEvents() {
976	eventSource.makeFirst(event_types.MESSAGE_RECEIVED, (/** @type {number} */ idx) => {	999	const eventHandler = (/** @type {number} */ idx) => {
977	if (!power_user.reasoning.auto_parse) {	1000	if (!power_user.reasoning.auto_parse) {
978	return;	1001	return;
979	}	1002	}

@@ -1029,7 +1052,11 @@ function registerReasoningAppEvents() {
1029	updateMessageBlock(idx, message);	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	export function initReasoning() {	1062	export function initReasoning() {