mirror of
https://github.com/SilenceLurker/ST-Amily2-Chat-Optimisation.git
synced 2026-06-06 19:25:50 +00:00
186 lines
8.3 KiB
JavaScript
186 lines
8.3 KiB
JavaScript
import { callSybdAI } from '../core/api/SybdApi.js';
|
||
import { extensionName } from '../utils/settings.js';
|
||
import { getPresetPrompts, getMixedOrder } from '../PresetSettings/index.js';
|
||
import { generateRandomSeed } from '../core/api.js';
|
||
import { safeLorebookEntries, safeUpdateLorebookEntries, compatibleWriteToLorebook } from '../core/tavernhelper-compatibility.js';
|
||
|
||
function buildContextFromEntries(entries) {
|
||
if (!entries || entries.length === 0) {
|
||
return '当前世界书为空。';
|
||
}
|
||
|
||
const mappedContent = entries.map(entry => {
|
||
if (!Array.isArray(entry.keys) || entry.keys.length < 2) {
|
||
return null;
|
||
}
|
||
const name = entry.keys[1];
|
||
return `[--START_TABLE--]\n[name]:${name}\n${entry.content}\n[--END_TABLE--]`;
|
||
}).filter(Boolean).join('\n\n');
|
||
|
||
return mappedContent || '当前世界书为空。';
|
||
}
|
||
|
||
function parseStructuredResponse(responseText) {
|
||
const entries = [];
|
||
const entryRegex = /\[--START_TABLE--\]\s*\[name\]:(.*?)\n([\s\S]*?)\[--END_TABLE--\]/g;
|
||
let match;
|
||
|
||
while ((match = entryRegex.exec(responseText)) !== null) {
|
||
const title = match[1].trim();
|
||
const content = match[2].trim();
|
||
if (title && content) {
|
||
entries.push({ title, content });
|
||
}
|
||
}
|
||
|
||
return entries;
|
||
}
|
||
|
||
|
||
export async function executeNovelProcessing(processingState, updateStatusCallback) {
|
||
const { chunks: recognizedChapters, batchSize, forceNew, selectedWorldBook } = processingState;
|
||
|
||
if (recognizedChapters.length === 0) {
|
||
updateStatusCallback('没有可处理的章节。', 'error');
|
||
throw new Error('没有可处理的章节。');
|
||
}
|
||
|
||
updateStatusCallback('开始处理小说...', 'info');
|
||
|
||
try {
|
||
const bookName = selectedWorldBook;
|
||
if (!bookName) {
|
||
throw new Error('请先在设置中选择一个目标世界书。');
|
||
}
|
||
|
||
const allEntries = (await safeLorebookEntries(bookName)) || [];
|
||
const managedEntries = allEntries.filter(e => e.comment?.startsWith('[Amily2小说处理]') || e.comment?.startsWith('[Amily2-Glossary]'));
|
||
const localManagedEntries = [...managedEntries];
|
||
|
||
let existingEntriesContent = '当前世界书为空。';
|
||
if (!forceNew) {
|
||
existingEntriesContent = buildContextFromEntries(localManagedEntries);
|
||
}
|
||
|
||
for (let i = processingState.currentIndex; i < recognizedChapters.length; i += batchSize) {
|
||
if (processingState.isAborted) {
|
||
updateStatusCallback(`处理已中止。当前进度: ${i}/${recognizedChapters.length}`, 'info');
|
||
return 'paused';
|
||
}
|
||
processingState.currentIndex = i;
|
||
|
||
const batch = recognizedChapters.slice(i, i + batchSize);
|
||
const progress = `(${i + batch.length}/${recognizedChapters.length})`;
|
||
updateStatusCallback(`正在处理批次 ${Math.floor(i / batchSize) + 1}... ${progress}`, 'info');
|
||
|
||
const chapterContent = batch.map(c => `## ${c.title}\n${c.content}`).join('\n\n---\n\n');
|
||
const order = getMixedOrder('novel_processor') || [];
|
||
const presetPrompts = await getPresetPrompts('novel_processor');
|
||
const messages = [{ role: 'system', content: generateRandomSeed() }];
|
||
|
||
let promptCounter = 0;
|
||
for (const item of order) {
|
||
if (item.type === 'prompt') {
|
||
if (presetPrompts && presetPrompts[promptCounter]) {
|
||
messages.push(presetPrompts[promptCounter]);
|
||
promptCounter++;
|
||
}
|
||
} else if (item.type === 'conditional') {
|
||
if (item.id === 'existingLore') {
|
||
messages.push({ role: 'user', content: `# 已有世界书条目\n\n${existingEntriesContent}` });
|
||
} else if (item.id === 'chapterContent') {
|
||
messages.push({ role: 'user', content: `# 最新章节内容\n\n${chapterContent}\n\n请根据以上信息，分析并输出需要新增或更新的世界书条目。` });
|
||
}
|
||
}
|
||
}
|
||
|
||
if (messages.length <= 1) throw new Error('未能根据预设构建有效的API请求。');
|
||
|
||
const response = await callSybdAI(messages);
|
||
if (!response) {
|
||
throw new Error(`API调用失败，批次 ${Math.floor(i / batchSize) + 1} 未收到响应。`);
|
||
}
|
||
if (response.trim() === '无需更新') {
|
||
updateStatusCallback(`批次 ${Math.floor(i / batchSize) + 1} 无需更新。`, 'info');
|
||
continue;
|
||
}
|
||
|
||
const structuredData = parseStructuredResponse(response);
|
||
if (structuredData.length === 0) {
|
||
throw new Error(`未能从API响应中提取有效信息，批次 ${Math.floor(i / batchSize) + 1}。`);
|
||
}
|
||
|
||
const entriesToUpdate = [];
|
||
const entriesToCreate = [];
|
||
const fixedNovelEntries = ['世界观设定', '时间线', '角色关系网', '角色总览'];
|
||
|
||
let maxPart = 0;
|
||
localManagedEntries.forEach(entry => {
|
||
const match = entry.comment.match(/章节内容概述-第(\d+)部分/);
|
||
if (match && parseInt(match[1], 10) > maxPart) maxPart = parseInt(match[1], 10);
|
||
});
|
||
let nextPart = maxPart + 1;
|
||
|
||
for (const entry of structuredData) {
|
||
const { title, content } = entry;
|
||
let comment;
|
||
let keys;
|
||
|
||
if (title === '章节内容概述') {
|
||
comment = `[Amily2小说处理] ${title}-第${nextPart}部分`;
|
||
keys = [`小说处理`, title, `第${nextPart}部分`];
|
||
const newEntryData = { keys, content, comment, enabled: true, order: 100, position: 'before_char' };
|
||
entriesToCreate.push(newEntryData);
|
||
localManagedEntries.push({ uid: -1, ...newEntryData, keys: keys });
|
||
nextPart++;
|
||
continue;
|
||
}
|
||
|
||
if (fixedNovelEntries.includes(title)) {
|
||
comment = `[Amily2小说处理] ${title}`;
|
||
keys = [`小说处理`, title];
|
||
} else {
|
||
comment = `[Amily2-Glossary] ${title}`;
|
||
keys = [`自定义条目`, title];
|
||
}
|
||
|
||
const existingEntry = localManagedEntries.find(e => e.comment === comment);
|
||
const loreData = { keys, content, comment, enabled: true, order: 100, position: 'before_char' };
|
||
|
||
if (existingEntry) {
|
||
entriesToUpdate.push({ uid: existingEntry.uid, ...loreData });
|
||
Object.assign(existingEntry, { ...loreData, keys: keys });
|
||
} else {
|
||
entriesToCreate.push(loreData);
|
||
localManagedEntries.push({ uid: -1, ...loreData, keys: keys });
|
||
}
|
||
}
|
||
|
||
if (entriesToUpdate.length > 0) {
|
||
await safeUpdateLorebookEntries(bookName, entriesToUpdate);
|
||
updateStatusCallback(`更新了 ${entriesToUpdate.length} 个世界书条目。`, 'info');
|
||
}
|
||
if (entriesToCreate.length > 0) {
|
||
for (const entry of entriesToCreate) {
|
||
await compatibleWriteToLorebook(bookName, entry.comment, () => entry.content, {
|
||
keys: entry.keys,
|
||
isConstant: false,
|
||
insertion_position: 'before_char',
|
||
depth: 100,
|
||
});
|
||
}
|
||
updateStatusCallback(`创建了 ${entriesToCreate.length} 个新世界书条目。`, 'success');
|
||
}
|
||
|
||
existingEntriesContent = buildContextFromEntries(localManagedEntries);
|
||
}
|
||
|
||
updateStatusCallback('小说处理完成！', 'success');
|
||
return 'success';
|
||
} catch (error) {
|
||
console.error('处理小说时发生严重错误:', error);
|
||
updateStatusCallback(`处理失败: ${error.message}`, 'error');
|
||
throw error;
|
||
}
|
||
}
|