Remove reasoning from raw builder summaries
| @@ -28,6 +28,7 @@ import { ARGUMENT_TYPE, SlashCommandArgument, SlashCommandNamedArgument } from ' | ||
| 28 | 28 | import { MacrosParser } from '../../macros.js'; |
| 29 | 29 | import { countWebLlmTokens, generateWebLlmChatPrompt, getWebLlmContextSize, isWebLlmSupported } from '../shared.js'; |
| 30 | 30 | import { commonEnumProviders } from '../../slash-commands/SlashCommandCommonEnumsProvider.js'; |
| 31 | +import { removeReasoningFromString } from '../../reasoning.js'; | |
| 31 | 32 | export { MODULE_NAME }; |
| 32 | 33 | |
| 33 | 34 | const MODULE_NAME = '1_memory'; |
| @@ -504,7 +505,7 @@ async function summarizeCallback(args, text) { | ||
| 504 | 505 | case summary_sources.extras: |
| 505 | 506 | return await callExtrasSummarizeAPI(text); |
| 506 | 507 | case summary_sources.main: |
| 507 | 508 | return removeReasoningFromString(await generateRaw(text, '', false, false, prompt, extension_settings.memory.overrideResponseLength)); |
| 508 | 509 | case summary_sources.webllm: { |
| 509 | 510 | const messages = [{ role: 'system', content: prompt }, { role: 'user', content: text }].filter(m => m.content); |
| 510 | 511 | const params = extension_settings.memory.overrideResponseLength > 0 ? { max_tokens: extension_settings.memory.overrideResponseLength } : {}; |
| @@ -699,7 +700,8 @@ async function summarizeChatMain(context, force, skipWIAN) { | ||
| 699 | 700 | return null; |
| 700 | 701 | } |
| 701 | 702 | |
| 702 | 703 | summaryconst rawSummary = await generateRaw(rawPrompt, '', false, false, prompt, extension_settings.memory.overrideResponseLength); |
| 704 | + summary = removeReasoningFromString(rawSummary); | |
| 703 | 705 | index = lastUsedIndex; |
| 704 | 706 | } finally { |
| 705 | 707 | inApiCall = false; |
| @@ -36,6 +36,7 @@ import { slashCommandReturnHelper } from '../../slash-commands/SlashCommandRetur | ||
| 36 | 36 | import { callGenericPopup, POPUP_RESULT, POPUP_TYPE } from '../../popup.js'; |
| 37 | 37 | import { generateWebLlmChatPrompt, isWebLlmSupported } from '../shared.js'; |
| 38 | 38 | import { WebLlmVectorProvider } from './webllm.js'; |
| 39 | +import { removeReasoningFromString } from '../../reasoning.js'; | |
| 39 | 40 | |
| 40 | 41 | /** |
| 41 | 42 | * @typedef {object} HashedMessage |
| @@ -260,7 +261,7 @@ async function summarizeExtra(element) { | ||
| 260 | 261 | * @returns {Promise<boolean>} Sucess |
| 261 | 262 | */ |
| 262 | 263 | async function summarizeMain(element) { |
| 263 | 264 | element.text = removeReasoningFromString(await generateRaw(element.text, '', false, false, settings.summary_prompt)); |
| 264 | 265 | return true; |
| 265 | 266 | } |
| 266 | 267 | |