Yoink reasoning placeholder from being saved

d60609716b542669e6ab83c220dd5ab3b4716d82

Wolfsblvt <wolfsblvt@gmail.com>

public/scripts/reasoning.js+16 -10 Viewed

  * Keeps track of the number of reasoning additions.
  */
 export class PromptReasoning {
-    static REASONING_PLACEHOLDER = '\u200B';
+    static REASONING_UI_PLACEHOLDER = '\u200B';
-    static REASONING_PLACEHOLDER_REGEX = new RegExp(`${PromptReasoning.REASONING_PLACEHOLDER}$`);
     constructor() {
         this.counter = 0;

@@ -433,8 +432,8 @@ export class PromptReasoning {
433	return content;	432	return content;
434	}	433	}
435		434
436	// No reasoning provided or a placeholder	435	// No reasoning provided
437	if (!reasoning \|\| reasoning === PromptReasoning.REASONING_PLACEHOLDER) {	436	if (!reasoning) {
438	return content;	437	return content;
439	}	438	}
440		439

             const messageId = !isNaN(parseInt(value.toString())) ? parseInt(value.toString()) : chat.length - 1;
             const message = chat[messageId];
             const reasoning = String(message?.extra?.reasoning ?? '');
-            return reasoning.replace(PromptReasoning.REASONING_PLACEHOLDER_REGEX, '');
+            return reasoning;
         },
     }));

         const textarea = document.createElement('textarea');
         const reasoningBlock = messageBlock.find('.mes_reasoning');
         textarea.classList.add('reasoning_edit_textarea');
-        textarea.value = reasoning.replace(PromptReasoning.REASONING_PLACEHOLDER_REGEX, '');
+        textarea.value = reasoning;
         $(textarea).insertBefore(reasoningBlock);
         if (!CSS.supports('field-sizing', 'content')) {

         const textarea = messageBlock.find('.reasoning_edit_textarea');
         textarea.remove();
+        // Make sure we remove the fake placeholder from the reasoning string
+        const text = messageBlock.find('.mes_reasoning').text();
+        if (text === PromptReasoning.REASONING_UI_PLACEHOLDER) {
+            messageBlock.find('.mes_reasoning').text('');
+        }
         messageBlock.find('.mes_reasoning_edit_cancel:visible').trigger('click');
     });
     $(document).on('click', '.mes_edit_add_reasoning', async function () {
-        const { message, messageId, messageBlock } = getMessageFromJquery(this);
+        const { message, messageBlock } = getMessageFromJquery(this);
         if (!message?.extra) {
             return;
+        }

             return;
+        }
-        message.extra.reasoning = PromptReasoning.REASONING_PLACEHOLDER;
+        // To be able to edit, we need to "fake" content being there inside the reasoning string
-        updateMessageBlock(messageId, message, { rerenderMessage: false });
+        messageBlock.find('.mes_reasoning').text(PromptReasoning.REASONING_UI_PLACEHOLDER);
         messageBlock.find('.mes_reasoning_edit').trigger('click');
         await saveChatConditional();
     });

     $(document).on('pointerup', '.mes_reasoning_copy', async function () {
         const { message } = getMessageFromJquery(this);
-        const reasoning = String(message?.extra?.reasoning ?? '').replace(PromptReasoning.REASONING_PLACEHOLDER_REGEX, '');
+        const reasoning = String(message?.extra?.reasoning ?? '');
         if (!reasoning) {
             return;