working fix for logprob reroll with autoparsed reasoning (#3998) * working fix for logprob reroll with autoparsed reasoning * fix prefix being added all the time * Code clean-up --------- Co-authored-by: Cohee <18619528+Cohee1207@users.noreply.github.com>

8d7648ccf7c4294239b9501cfb1cd1cf9c6871a8

RossAscends <124905043+RossAscends@users.noreply.github.com>

Signed

Parents: c3376da3939e9c55555515a17f0e50642e978655

public/scripts/logprobs.js+55 -3 Viewed

     getGeneratingApi,
     is_send_press,
     isStreamingEnabled,
+    substituteParamsExtended,
 } from '../script.js';
 import { debounce, delay, getStringHash } from './utils.js';
 import { decodeTextTokens, getTokenizerBestMatch } from './tokenizers.js';

 function createSwipe(messageId, prompt) {
     // need to call `cleanUpMessage` on our new prompt, because we were working
     // with raw model output and our new prompt is missing trimming/macro replacements
-    const cleanedPrompt = cleanUpMessage({
+    let cleanedPrompt = cleanUpMessage({
         getMessage: prompt,
         isImpersonate: false,
         isContinue: false,

     });
     const msg = chat[messageId];
+    const reasoningPrefix = substituteParamsExtended(power_user.reasoning.prefix);
+    const reasoningSuffix = substituteParamsExtended(power_user.reasoning.suffix);
+    const isReasoningAutoParsed = power_user.reasoning.auto_parse;
+    const msgHasParsedReasoning = msg.extra?.reasoning?.length > 0;
+    let shouldRerollReasoning = false;
+    //if we have pre-existing reasoning and are currently autoparsing
+    if (isReasoningAutoParsed && msgHasParsedReasoning) {
+        console.debug('saw autoparse on with reasoning in message');
+        //but the reroll prompt does not include the end of reasoning
+        if (cleanedPrompt.includes(reasoningPrefix) && !cleanedPrompt.includes(reasoningSuffix)) {
+            //we need to send the results to the reasoning block
+            //this will involve the ReasoningHandler from reasoning.js
+            console.debug('..with start tag but no end tag... reroll reasoning');
+            shouldRerollReasoning = true;
+        }
+        let hasReasoningPrefix = cleanedPrompt.includes(reasoningPrefix);
+        let hasReasoningSuffix = cleanedPrompt.includes(reasoningSuffix);
+        //..with both the start and end think tags
+        //OR
+        //..with only the end think tag (implying prefilled think start)
+        if (hasReasoningPrefix && hasReasoningSuffix) {
+            //we need to send the results to the response block without reasoning attached
+            console.debug('...incl. end tag...rerolling response');
+            const endOfThink = cleanedPrompt.indexOf(reasoningSuffix) + reasoningSuffix.length;
+            cleanedPrompt = cleanedPrompt.substring(endOfThink);
+        }
+        //if cleanedprompt includes the think prefix, but no suffix..
+        if (hasReasoningPrefix && !hasReasoningSuffix) {
+            console.debug('..no end tag...rerolling reasoning, so removing prefix');
+            cleanedPrompt = cleanedPrompt.replace(reasoningPrefix, '');
+        }
+    }
+    console.debug('cleanedPrompt: ', cleanedPrompt);
     const newSwipeInfo = {
         send_date: msg.send_date,
         gen_started: msg.gen_started,

     msg.swipe_info = msg.swipe_info || [];
     // Add our new swipe, then make sure the active swipe is the one just before
-    // it. The call to `swipe_right` will switch to it immediately.
+    // it. The call to `swipe_right` in addGeneration() will switch to it immediately.
-    msg.swipes.push(cleanedPrompt);
+    //if we determined that we need to reroll from reasoning
+    if (shouldRerollReasoning) {
+        //cleaned prompt goes into reasoning
+        newSwipeInfo.extra.reasoning = cleanedPrompt;
+        //mes_text becomes empty, causing the reasoning handler to parse the reasoning first
+        msg.swipes.push('');
+    } else {
+        //otherwise just add the cleaned prompt to the message and continue
+        msg.swipes.push(cleanedPrompt);
+    }
     msg.swipe_info.push(newSwipeInfo);
     msg.swipe_id = Math.max(0, msg.swipes.length - 2);
+}