slvccans/ST-Amily2-Chat-Optimisation
1
0
Fork 0
mirror of https://github.com/Wx-2025/ST-Amily2-Chat-Optimisation.git synced 2026-06-06 18:35:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update summarizer.js

Browse Source
This commit is contained in:
Wx-2025
2025-10-23 22:16:04 +08:00
committed by GitHub
parent c9994d0850
commit 17bc1c2ac4
1 changed files with 95 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

97
core/summarizer.js
View File

@@ -239,7 +239,100 @@ export async function processPlotOptimization(currentUserMessage, contextMessage
systemPrompt = systemPrompt.replace(regex, value);
}
const worldbookContent = await getPlotOptimizedWorldbookContent(context, settings);
let worldbookContent = await getPlotOptimizedWorldbookContent(context, settings);
// --- EJS 預處理（劇情優化專用）---
try {
if (settings.plotOpt_ejsEnabled !== false && globalThis.EjsTemplate?.evalTemplate && globalThis.EjsTemplate?.prepareContext) {
const safeUser = (userMessageContent ?? '').toString();
const safeWorld = (worldbookContent ?? '').toString();
const hasEjsUser = /<%[=_\-]?/.test(safeUser);
const hasEjsWorld = /<%[=_\-]?/.test(safeWorld);
const openTagRegex = /<%[=_\-]?/g;
const closeTagRegex = /[-_]?%>/g;
const openUser = (safeUser.match(openTagRegex) || []).length;
const closeUser = (safeUser.match(closeTagRegex) || []).length;
const openWorld = (safeWorld.match(openTagRegex) || []).length;
const closeWorld = (safeWorld.match(closeTagRegex) || []).length;
const balancedUser = hasEjsUser && openUser === closeUser && openUser > 0;
const balancedWorld = hasEjsWorld && openWorld === closeWorld && openWorld > 0;
if (hasEjsUser || hasEjsWorld) {
const env = await globalThis.EjsTemplate.prepareContext({ runType: 'plot_optimization', isDryRun: false });
try {
if (balancedUser) {
const compiledUser = await globalThis.EjsTemplate.evalTemplate(safeUser, env, { _with: true });
if (typeof compiledUser === 'string' && compiledUser.length > 0) {
currentUserMessage.mes = compiledUser;
}
} else if (hasEjsUser) {
console.warn('[ST-Amily2-Chat-Optimisation][PlotOpt] 检测到未闭合的 EJS 标签（用户输入），已跳过预处理。');
}
} catch (errUser) {
console.error('[ST-Amily2-Chat-Optimisation][PlotOpt] EJS 預處理-用户输入失败：', errUser);
toastr.error('EJS 预处理用户输入失败，已中止。', 'Amily2号');
return null;
}
try {
if (balancedWorld) {
const compiledWorld = await globalThis.EjsTemplate.evalTemplate(safeWorld, env, { _with: true });
if (typeof compiledWorld === 'string' && compiledWorld.length > 0) {
worldbookContent = compiledWorld;
}
} else if (hasEjsWorld) {
console.warn('[ST-Amily2-Chat-Optimisation][PlotOpt] 检测到未闭合的 EJS 标签（世界书），已跳过预处理。');
}
} catch (errWorld) {
try {
if (globalThis.EjsTemplate?.getSyntaxErrorInfo && typeof errWorld?.message === 'string') {
const extra = globalThis.EjsTemplate.getSyntaxErrorInfo(safeWorld);
console.error('[ST-Amily2-Chat-Optimisation][PlotOpt] EJS 預處理-世界书失败(含定位)：', errWorld?.message + (extra || ''));
} else {
console.error('[ST-Amily2-Chat-Optimisation][PlotOpt] EJS 預處理-世界书失败：', errWorld);
}
// 打印世界书片段（限長）
try {
const maxLen = 2000;
const snippet = typeof safeWorld === 'string' ? safeWorld.slice(0, maxLen) : String(safeWorld).slice(0, maxLen);
const isTruncated = (safeWorld?.length || 0) > maxLen;
// 存入全局以便用户在控制台直接读取
try {
// @ts-ignore
window.Amily2PlotOptDebug = window.Amily2PlotOptDebug || {};
// @ts-ignore
window.Amily2PlotOptDebug.worldErrorMessage = (errWorld?.message || String(errWorld)) + '';
// @ts-ignore
window.Amily2PlotOptDebug.worldSnippet = snippet;
// @ts-ignore
window.Amily2PlotOptDebug.worldSnippetTruncated = isTruncated;
// @ts-ignore
window.Amily2PlotOptDebug.worldOpenClose = { open: openWorld, close: closeWorld };
} catch (_) {}
// 多级别日志，避免特定环境过滤
console.groupCollapsed('[ST-Amily2-Chat-Optimisation][PlotOpt] 失败世界书片段 (截断=' + isTruncated + ')');
console.log(snippet);
console.groupEnd();
console.warn('[ST-Amily2-Chat-Optimisation][PlotOpt] worldOpenClose:', { open: openWorld, close: closeWorld });
console.error('[ST-Amily2-Chat-Optimisation][PlotOpt] 以上即失败世界书片段。');
} catch (logErr) {
console.error('[ST-Amily2-Chat-Optimisation][PlotOpt] 打印失败世界书片段时出错：', logErr);
}
} catch (sub) {
console.error('[ST-Amily2-Chat-Optimisation][PlotOpt] 记录语法位置信息失败：', sub);
}
toastr.error('EJS 预处理世界书失败，已中止。', 'Amily2号');
return null;
}
}
}
} catch (e) {
console.error('[ST-Amily2-Chat-Optimisation][PlotOpt] EJS 預處理初始化失败（可能是上下文环境）：', e);
toastr.error('EJS 预处理初始化失败，已中止。', 'Amily2号');
return null; // 直接中止，不送出訊息
}
let tableContent = '';
if (settings.plotOpt_tableEnabled) {
@@ -310,7 +403,7 @@ export async function processPlotOptimization(currentUserMessage, contextMessage
}
break;
case 'coreContent':
messages.push({ role: 'user', content: `[核心处理内容]:\n${userMessageContent}` });
messages.push({ role: 'user', content: `[核心处理内容]:\n${currentUserMessage.mes}` });
break;
case 'plotTag':
messages.push({ role: 'assistant', content: '<plot>' });