Parse reasoning in multi-swipe swipes

0ea64050ff1e5a49d3a140ff3122bc59872e6574

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

public/script.js+7 -9 Viewed

 import { initBulkEdit } from './scripts/bulk-edit.js';
 import { deriveTemplatesFromChatTemplate } from './scripts/chat-templates.js';
 import { getContext } from './scripts/st-context.js';
 import { extractReasoningFromData, initReasoning, parseReasoningInSwipes, PromptReasoning, ReasoningHandler, removeReasoningFromString, updateReasoningUI } from './scripts/reasoning.js';
 import { accountStorage } from './scripts/util/AccountStorage.js';
 // API OBJECT FOR EXTERNAL WIRING

         if (Array.isArray(this.swipes) && this.swipes.length > 0) {
             const message = chat[messageId];
             const swipeInfoExtra = structuredClone(message.extra ?? {});
             delete swipeInfoExtra.token_count;
             delete swipeInfoExtra.reasoning;
             delete swipeInfoExtra.reasoning_duration;

                 gen_finished: message.gen_finished,
                 extra: swipeInfoExtra,
             };
 -            const swipeInfoArray = [];
 +            const swipeInfoArray = Array(this.swipes.length).fill().map(() => structuredClone(swipeInfo));
 -            swipeInfoArray.length = this.swipes.length;
 +            parseReasoningInSwipes(this.swipes, swipeInfoArray, message.extra?.reasoning_duration);
 -            swipeInfoArray.fill(swipeInfo);
             chat[messageId].swipes.push(...this.swipes);
             chat[messageId].swipe_info.push(...swipeInfoArray);
+        }

+    }
     if (Array.isArray(swipes) && swipes.length > 0) {
         const swipeInfoExtra = structuredClone(item.extra ?? {});
         delete swipeInfoExtra.token_count;
         delete swipeInfoExtra.reasoning;
         delete swipeInfoExtra.reasoning_duration;

             gen_finished: item.gen_finished,
             extra: swipeInfoExtra,
         };
 -        const swipeInfoArray = [];
 +        const swipeInfoArray = Array(swipes.length).fill().map(() => structuredClone(swipeInfo));
 -        swipeInfoArray.length = swipes.length;
 +        parseReasoningInSwipes(swipes, swipeInfoArray, item.extra?.reasoning_duration);
 -        swipeInfoArray.fill(swipeInfo, 0, swipes.length);
         item.swipes.push(...swipes);
         item.swipe_info.push(...swipeInfoArray);
+    }

public/scripts/reasoning.js+26 -0 Viewed

+    }
+}
 +/**
 + * Parse reasoning in an array of swipe strings if auto-parsing is enabled.
 + * @param {string[]} swipes Array of swipe strings
 + * @param {{extra: {reasoning: string, reasoning_duration: number}}[]} swipeInfoArray Array of swipe info objects
 + * @param {number?} duration Duration of the reasoning
 + */
 +export function parseReasoningInSwipes(swipes, swipeInfoArray, duration) {
 +    if (!power_user.reasoning.auto_parse) {
 +        return;
 +    }
++
 +    // Something ain't right, don't parse
 +    if (!Array.isArray(swipes) || !Array.isArray(swipeInfoArray) || swipes.length !== swipeInfoArray.length) {
 +        return;
 +    }
++
 +    for (let index = 0; index < swipes.length; index++) {
 +        const parsedReasoning = parseReasoningFromString(swipes[index]);
 +        if (parsedReasoning) {
 +            swipes[index] = parsedReasoning.content;
 +            swipeInfoArray[index].extra.reasoning = parsedReasoning.reasoning;
 +            swipeInfoArray[index].extra.reasoning_duration = duration;
 +        }
 +    }
 +}
++
 function registerReasoningAppEvents() {
     const eventHandler = (/** @type {number} */ idx) => {
         if (!power_user.reasoning.auto_parse) {