Remove reasoning from raw builder summaries

7659dfb85c77f098f37fac50e667c3c6b9b87073

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

public/scripts/extensions/memory/index.js+4 -2 Viewed

 import { MacrosParser } from '../../macros.js';
 import { countWebLlmTokens, generateWebLlmChatPrompt, getWebLlmContextSize, isWebLlmSupported } from '../shared.js';
 import { commonEnumProviders } from '../../slash-commands/SlashCommandCommonEnumsProvider.js';
+import { removeReasoningFromString } from '../../reasoning.js';
 export { MODULE_NAME };
 const MODULE_NAME = '1_memory';

             case summary_sources.extras:
                 return await callExtrasSummarizeAPI(text);
             case summary_sources.main:
-                return await generateRaw(text, '', false, false, prompt, extension_settings.memory.overrideResponseLength);
+                return removeReasoningFromString(await generateRaw(text, '', false, false, prompt, extension_settings.memory.overrideResponseLength));
             case summary_sources.webllm: {
                 const messages = [{ role: 'system', content: prompt }, { role: 'user', content: text }].filter(m => m.content);
                 const params = extension_settings.memory.overrideResponseLength > 0 ? { max_tokens: extension_settings.memory.overrideResponseLength } : {};

                 return null;
+            }
-            summary = await generateRaw(rawPrompt, '', false, false, prompt, extension_settings.memory.overrideResponseLength);
+            const rawSummary = await generateRaw(rawPrompt, '', false, false, prompt, extension_settings.memory.overrideResponseLength);
+            summary = removeReasoningFromString(rawSummary);
             index = lastUsedIndex;
         } finally {
             inApiCall = false;

public/scripts/extensions/vectors/index.js+2 -1 Viewed

 import { callGenericPopup, POPUP_RESULT, POPUP_TYPE } from '../../popup.js';
 import { generateWebLlmChatPrompt, isWebLlmSupported } from '../shared.js';
 import { WebLlmVectorProvider } from './webllm.js';
+import { removeReasoningFromString } from '../../reasoning.js';
 /**
  * @typedef {object} HashedMessage

  * @returns {Promise<boolean>} Sucess
  */
 async function summarizeMain(element) {
-    element.text = await generateRaw(element.text, '', false, false, settings.summary_prompt);
+    element.text = removeReasoningFromString(await generateRaw(element.text, '', false, false, settings.summary_prompt));
     return true;
+}