slvccans/ST-Amily2-Chat-Optimisation-SecFix
1
0
Fork 0
mirror of https://github.com/SilenceLurker/ST-Amily2-Chat-Optimisation.git synced 2026-06-06 17:05:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update summarizer.js

Browse Source
This commit is contained in:
Wx-2025
2025-12-09 00:55:11 +08:00
committed by GitHub
parent 6c24bbf8cc
commit 6c84656d6d
1 changed files with 21 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
core/summarizer.js
View File

@@ -4,7 +4,7 @@ import { world_info } from "/scripts/world-info.js";
import { extensionName } from "../utils/settings.js"; import { extensionName } from "../utils/settings.js";
import { extractContentByTag, replaceContentByTag, extractFullTagBlock } from '../utils/tagProcessor.js'; import { extractContentByTag, replaceContentByTag, extractFullTagBlock } from '../utils/tagProcessor.js';
import { isGoogleEndpoint, convertToGoogleRequest, parseGoogleResponse, buildGoogleApiUrl, buildPlotOptimizationGoogleRequest, parsePlotOptimizationGoogleResponse } from './utils/googleAdapter.js'; import { isGoogleEndpoint, convertToGoogleRequest, parseGoogleResponse, buildGoogleApiUrl, buildPlotOptimizationGoogleRequest, parsePlotOptimizationGoogleResponse } from './utils/googleAdapter.js';
import { applyExclusionRules } from './utils/rag-tag-extractor.js'; import { applyExclusionRules, extractBlocksByTags } from './utils/rag-tag-extractor.js';
import { import {
getCombinedWorldbookContent, getPlotOptimizedWorldbookContent, getOptimizationWorldbookContent, getCombinedWorldbookContent, getPlotOptimizedWorldbookContent, getOptimizationWorldbookContent,
} from "./lore.js"; } from "./lore.js";
@@ -353,12 +353,29 @@ export async function processPlotOptimization(currentUserMessage, contextMessage
const contextLimit = settings.plotOpt_contextLimit || 0; const contextLimit = settings.plotOpt_contextLimit || 0;
if (contextLimit > 0 && contextMessages.length > 0) { if (contextLimit > 0 && contextMessages.length > 0) {
const historyMessages = contextMessages.slice(-contextLimit); const historyMessages = contextMessages.slice(-contextLimit);
// 复刻 Historiographer 的标签提取与内容排除逻辑
const useTagExtraction = settings.historiographyTagExtractionEnabled ?? false;
const tagsToExtract = useTagExtraction ? (settings.historiographyTags || '').split(',').map(t => t.trim()).filter(Boolean) : [];
const exclusionRules = settings.historiographyExclusionRules || [];
history = historyMessages history = historyMessages
.map(msg => { .map(msg => {
if (msg.mes && msg.mes.trim()) { if (msg.mes && msg.mes.trim()) {
const commentExclusionRules = [{ start: '<!--', end: '-->' }]; let content = msg.mes.trim();
const cleanedMessage = applyExclusionRules(msg.mes.trim(), commentExclusionRules);
return cleanedMessage ? `${msg.is_user ? userName : charName}: ${cleanedMessage}` : null; // 1. 标签提取
if (useTagExtraction && tagsToExtract.length > 0) {
const blocks = extractBlocksByTags(content, tagsToExtract);
if (blocks.length > 0) {
content = blocks.join('\n\n');
}
}
// 2. 内容排除
content = applyExclusionRules(content, exclusionRules);
return content ? `${msg.is_user ? userName : charName}: ${content}` : null;
} }
return null; return null;
}) })
@@ -405,9 +422,6 @@ export async function processPlotOptimization(currentUserMessage, contextMessage
case 'coreContent': case 'coreContent':
messages.push({ role: 'user', content: `[核心处理内容]:\n${currentUserMessage.mes}` }); messages.push({ role: 'user', content: `[核心处理内容]:\n${currentUserMessage.mes}` });
break; break;
case 'plotTag':
messages.push({ role: 'assistant', content: '<plot>' });
break;
} }
} }
} }