Improve reasoning block inline editing

a14e3b0657ecc385b950a96d9c86e40023d3017f

Wolfsblvt <wolfsblvt@gmail.com>

2 files changed, +9 -4Showing whitespace changes
public/script.js+4 -2
@@ -7193,8 +7193,10 @@ export function setGenerationParamsFromPreset(preset) {
71937193// Common code for message editor done and auto-save
71947194function updateMessage(div) {
71957195 const mesBlock = div.closest('.mes_block');
71967196 let text = mesBlock.find('.edit_textarea').val();
7197- const mes = chat[this_edit_mes_id];
7197+ ?? mesBlock.find('.mes_text').text();
7198+ const mesElement = div.closest('.mes');
7199+ const mes = chat[mesElement.attr('mesid')];
71987200
71997201 let regexPlacement;
72007202 if (mes.is_user) {
public/scripts/reasoning.js+5 -2
@@ -288,6 +288,8 @@ function setReasoningEventHandlers() {
288288 await saveChatConditional();
289289 updateMessageBlock(messageId, message);
290290 textarea.remove();
291+
292+ messageBlock.find('.mes_edit_done:visible').trigger('click');
291293 });
292294
293295 $(document).on('click', '.mes_reasoning_edit_cancel', function (e) {
@@ -297,6 +299,8 @@ function setReasoningEventHandlers() {
297299 const { messageBlock } = getMessageFromJquery(this);
298300 const textarea = messageBlock.find('.reasoning_edit_textarea');
299301 textarea.remove();
302+
303+ messageBlock.find('.mes_reasoning_edit_cancel:visible').trigger('click');
300304 });
301305
302306 $(document).on('click', '.mes_edit_add_reasoning', async function () {
@@ -311,9 +315,8 @@ function setReasoningEventHandlers() {
311315 }
312316
313317 message.extra.reasoning = PromptReasoning.REASONING_PLACEHOLDER;
314- await saveChatConditional();
315- closeMessageEditor();
316318 updateMessageBlock(messageId, message);
319+ await saveChatConditional();
317320 });
318321
319322 $(document).on('click', '.mes_reasoning_delete', async function (e) {