mirror of
https://github.com/Wx-2025/ST-Amily2-Chat-Optimisation.git
synced 2026-06-06 06:55:51 +00:00
9bc2f694b0
### 新功能 - Function Call 填表开关下方新增公益站风险提示横幅：部分公益站会屏蔽 tools 参数，请确认支持情况避免被意外封禁 ### 修复 - **Function Call 填表**： - 修复 ST 代理以 HTTP 200 + error body 形式返回错误、导致降级重试机制从未触发的问题 - 修复思考模式模型（如 DeepSeek v4-flash）因 tool_choice 不兼容返回 Bad Request 后正确降级并重试 - 重试时自动追加强制调用指令，防止思考模型绕过工具直接输出文本造成无效二次开销 - **超级记忆 / 翰林院**： - 修复 `getRagSettings()` 读写顶层路径而非嵌套路径，导致打开超级记忆面板后向量化、归档等开关在重载时被全默认值覆盖的问题 - 修复自动归档失效问题 - 修复归档管理器在同一事件中被三次触发的回归问题 - 修复翰林院设置旧版迁移逻辑异常
646 lines
28 KiB
JavaScript
646 lines
28 KiB
JavaScript
import { getContext, extension_settings } from '/scripts/extensions.js';
|
||
import { characters } from '/script.js';
|
||
import { loadWorldInfo } from '/scripts/world-info.js';
|
||
import { log } from './logger.js';
|
||
import { updateTableFromText } from './manager.js';
|
||
import { extensionName } from '../../utils/settings.js';
|
||
import { renderTables } from '../../ui/table-bindings.js';
|
||
import { getPresetPrompts, getMixedOrder } from '../../PresetSettings/index.js';
|
||
import { callAI, callAIForTools, generateRandomSeed } from '../api.js';
|
||
import { callNccsAI } from '../api/NccsApi.js';
|
||
import { TABLE_FILL_TOOL, parseToolCallArgs } from './formatters/tool-call.js';
|
||
import { updateTableFromOps } from './manager.js';
|
||
import { extractBlocksByTags, applyExclusionRules } from '../utils/rag-tag-extractor.js';
|
||
import { resolveTableRuleConfig } from '../../utils/config/RuleProfileManager.js';
|
||
import { showTableFillReviewModal } from '../../ui/page-window.js';
|
||
|
||
import { getBatchFillerRuleTemplate, getBatchFillerFlowTemplate, convertTablesToCsvString } from './manager.js';
|
||
|
||
const CONTINUE_PROMPT = '上一条回复不完整或缺少 <Amily2Edit> 指令块。请直接从中断处继续生成剩余内容，不要重复已输出的文本，也不要添加任何解释或寒暄，确保最终输出中包含完整的 <Amily2Edit>...</Amily2Edit> 指令块。';
|
||
|
||
async function requestContinuation(baseMessages, partialResponse) {
|
||
const continueMessages = [
|
||
...baseMessages,
|
||
{ role: 'assistant', content: partialResponse || '' },
|
||
{ role: 'user', content: CONTINUE_PROMPT },
|
||
];
|
||
const continued = await callTableModel(continueMessages);
|
||
if (!continued) return null;
|
||
return `${partialResponse || ''}${continued}`;
|
||
}
|
||
|
||
let isFilling = false;
|
||
let manualStopRequested = false;
|
||
let currentBatch = 0;
|
||
let totalBatches = 0;
|
||
let chatHistoryLength = 0;
|
||
let threshold = 30;
|
||
const MAX_RETRIES = 2;
|
||
|
||
|
||
async function getWorldBookContext() {
|
||
const settings = extension_settings[extensionName] || {};
|
||
if (!settings.table_worldbook_enabled) {
|
||
return '';
|
||
}
|
||
|
||
const context = getContext();
|
||
let bookNames = [];
|
||
let content = '';
|
||
|
||
if (settings.table_worldbook_source === 'character') {
|
||
const characterId = context.characterId;
|
||
const character = characters[characterId];
|
||
const characterBook = character?.data?.extensions?.world;
|
||
if (characterBook) {
|
||
bookNames.push(characterBook);
|
||
}
|
||
} else {
|
||
bookNames = settings.table_selected_worldbooks || [];
|
||
}
|
||
|
||
if (bookNames.length === 0) {
|
||
return '';
|
||
}
|
||
|
||
const selectedEntriesConfig = settings.table_selected_entries || {};
|
||
|
||
for (const bookName of bookNames) {
|
||
try {
|
||
const bookData = await loadWorldInfo(bookName);
|
||
if (!bookData || !bookData.entries) continue;
|
||
|
||
const entriesToInclude = settings.table_worldbook_source === 'manual'
|
||
? (selectedEntriesConfig[bookName] || []).map(uid => String(uid))
|
||
: Object.values(bookData.entries).map(entry => String(entry.uid));
|
||
|
||
for (const entry of Object.values(bookData.entries)) {
|
||
if (entriesToInclude.includes(String(entry.uid))) {
|
||
content += `[来源：世界书，条目名字：${entry.comment || '无标题条目'}]\n${entry.content}\n\n`;
|
||
}
|
||
}
|
||
} catch (error) {
|
||
log(`加载世界书 "${bookName}" 失败: ${error.message}`, 'error');
|
||
}
|
||
}
|
||
|
||
if (content.length > settings.table_worldbook_char_limit) {
|
||
content = content.substring(0, settings.table_worldbook_char_limit);
|
||
}
|
||
|
||
return content.trim() ? `<世界书>\n${content.trim()}\n</世界书>` : '';
|
||
}
|
||
|
||
const fillButton = () => document.getElementById('fill-table-now-btn');
|
||
|
||
function updateButtonState(state, batchNum = 0, attemptNum = 0) {
|
||
const button = fillButton();
|
||
if (!button) return;
|
||
|
||
switch (state) {
|
||
case 'processing':
|
||
let attemptText = attemptNum > 0 ? ` (尝试 ${attemptNum + 1})` : '';
|
||
button.textContent = `点击停止 (${batchNum}/${totalBatches})${attemptText}`;
|
||
button.disabled = false;
|
||
isFilling = true;
|
||
break;
|
||
case 'stopping':
|
||
button.textContent = '正在停止...';
|
||
button.disabled = true;
|
||
break;
|
||
case 'paused':
|
||
button.textContent = '继续填表';
|
||
button.disabled = false;
|
||
isFilling = true;
|
||
break;
|
||
case 'error':
|
||
button.textContent = '继续填表 (出错)';
|
||
button.disabled = false;
|
||
isFilling = true;
|
||
break;
|
||
case 'idle':
|
||
default:
|
||
button.textContent = '立即填表';
|
||
button.disabled = false;
|
||
isFilling = false;
|
||
currentBatch = 0;
|
||
manualStopRequested = false;
|
||
break;
|
||
}
|
||
}
|
||
|
||
async function callTableModel(messages) {
|
||
try {
|
||
const settings = extension_settings[extensionName] || {};
|
||
|
||
if (settings.nccsEnabled) {
|
||
log('使用 Nccs API 进行表格填充...', 'info');
|
||
const result = await callNccsAI(messages);
|
||
if (!result) {
|
||
throw new Error('Nccs API返回内容为空。');
|
||
}
|
||
return result;
|
||
} else {
|
||
log('使用 tableFilling slot 进行表格填充...', 'info');
|
||
const result = await callAI(messages, { slot: 'tableFilling' });
|
||
if (!result) {
|
||
throw new Error('API返回内容为空。');
|
||
}
|
||
return result;
|
||
}
|
||
} catch (error) {
|
||
log(`与模型通讯时发生异常: ${error.message}`, "error");
|
||
toastr.error(`与模型通讯时发生异常: ${error.message}`, "通讯异常");
|
||
return null;
|
||
}
|
||
}
|
||
|
||
function getRawMessagesForSummary(startFloor, endFloor) {
|
||
const context = getContext();
|
||
const chat = context.chat;
|
||
const settings = extension_settings[extensionName] || {};
|
||
|
||
const historySlice = chat.slice(startFloor - 1, endFloor);
|
||
if (historySlice.length === 0) return null;
|
||
|
||
const userName = context.name1 || '用户';
|
||
const characterName = context.name2 || '角色';
|
||
|
||
let tagsToExtract = [];
|
||
let exclusionRules = [];
|
||
|
||
const tableRuleConfig = resolveTableRuleConfig(settings);
|
||
if (tableRuleConfig.tags || (tableRuleConfig.exclusionRules && tableRuleConfig.exclusionRules.length)) {
|
||
log('批量填表：使用提取规则配置。', 'info');
|
||
tagsToExtract = (tableRuleConfig.tags || '').split(',').map(t => t.trim()).filter(Boolean);
|
||
exclusionRules = tableRuleConfig.exclusionRules || [];
|
||
}
|
||
|
||
const messages = historySlice.map((msg, index) => {
|
||
let content = msg.mes;
|
||
|
||
if (tagsToExtract.length > 0) {
|
||
const blocks = extractBlocksByTags(content, tagsToExtract);
|
||
content = blocks.length > 0 ? blocks.join('\n\n') : '';
|
||
}
|
||
|
||
if (content) {
|
||
content = applyExclusionRules(content, exclusionRules);
|
||
}
|
||
|
||
if (!content.trim()) return null;
|
||
|
||
return {
|
||
floor: startFloor + index,
|
||
author: msg.is_user ? userName : characterName,
|
||
authorType: msg.is_user ? 'user' : 'char',
|
||
content: content.trim()
|
||
};
|
||
}).filter(Boolean);
|
||
|
||
return messages;
|
||
}
|
||
|
||
async function runBatchAttempt(batchNum, attemptNum) {
|
||
try {
|
||
if (manualStopRequested) {
|
||
log(`任务已在批次 ${batchNum} 开始前手动暂停。`, 'warn');
|
||
updateButtonState('paused');
|
||
return;
|
||
}
|
||
|
||
updateButtonState('processing', batchNum, attemptNum);
|
||
|
||
const startFloor = (batchNum - 1) * threshold + 1;
|
||
const endFloor = Math.min(startFloor + threshold - 1, chatHistoryLength);
|
||
|
||
log(`正在处理批次 ${batchNum}/${totalBatches} (楼层 ${startFloor}-${endFloor}, 尝试 ${attemptNum + 1}/${MAX_RETRIES + 1})`, 'info');
|
||
|
||
const purifiedMessages = getRawMessagesForSummary(startFloor, endFloor);
|
||
if (!purifiedMessages || purifiedMessages.length === 0) {
|
||
throw new Error('净化后无有效内容可处理。');
|
||
}
|
||
|
||
const batchContent = purifiedMessages.map(m => `【第 ${m.floor} 楼】 ${m.author}: ${m.content}`).join('\n');
|
||
const ruleTemplate = getBatchFillerRuleTemplate();
|
||
const flowTemplate = getBatchFillerFlowTemplate();
|
||
const currentTableDataString = convertTablesToCsvString();
|
||
const finalFlowPrompt = flowTemplate.replace('{{{Amily2TableData}}}', currentTableDataString);
|
||
|
||
let mixedOrder;
|
||
try {
|
||
const savedOrder = localStorage.getItem('amily2_prompt_presets_v2_mixed_order');
|
||
if (savedOrder) {
|
||
mixedOrder = JSON.parse(savedOrder);
|
||
}
|
||
} catch (e) {
|
||
console.error("[批量填表] 加载混合顺序失败:", e);
|
||
}
|
||
const order = getMixedOrder('batch_filler') || [];
|
||
|
||
const presetPrompts = await getPresetPrompts('batch_filler');
|
||
|
||
const worldBookContext = await getWorldBookContext();
|
||
|
||
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') {
|
||
switch (item.id) {
|
||
case 'worldbook':
|
||
if (worldBookContext) {
|
||
messages.push({ role: 'system', content: worldBookContext });
|
||
}
|
||
break;
|
||
case 'ruleTemplate':
|
||
messages.push({ role: "system", content: ruleTemplate });
|
||
break;
|
||
case 'flowTemplate':
|
||
messages.push({ role: "system", content: finalFlowPrompt });
|
||
break;
|
||
case 'coreContent':
|
||
messages.push({ role: 'user', content: `请严格根据以下"对话记录"中的内容进行填写表格，并按照指定的格式输出，不要添加任何额外信息。\n\n<对话记录>\n${batchContent}\n</对话记录>` });
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
if (!presetPrompts || presetPrompts.length === 0) {
|
||
const defaultPrompts = [
|
||
{ role: 'system', content: generateRandomSeed() }
|
||
];
|
||
messages.splice(1, 0, ...defaultPrompts);
|
||
}
|
||
|
||
console.groupCollapsed(`[Amily2 立即远征] 批次 ${batchNum}/${totalBatches} - 即将发送至 API 的内容`);
|
||
console.dir(messages);
|
||
console.groupEnd();
|
||
|
||
const batchSettings = extension_settings[extensionName] || {};
|
||
if (batchSettings.tableFillFunctionCall) {
|
||
// Function Call 路径：结构化输出，无需检查 <Amily2Edit>
|
||
const argsString = await callAIForTools(messages, TABLE_FILL_TOOL, { slot: 'tableFilling' });
|
||
if (!argsString) throw new Error('Function Call 返回为空。');
|
||
const ops = parseToolCallArgs(argsString);
|
||
if (ops.length === 0) {
|
||
log(`批次 ${batchNum} 的 Function Call 返回操作列表为空，AI 判断此批次无需变更。`, 'warn');
|
||
toastr.info('AI 判断此批次无需修改。', `批次 ${batchNum}`);
|
||
} else {
|
||
await updateTableFromOps(ops, { immediateDelete: true });
|
||
renderTables();
|
||
log(`批次 ${batchNum} Function Call 处理成功（${ops.length} 条操作）。`, 'success');
|
||
}
|
||
} else {
|
||
// Legacy 文本路径
|
||
const resultText = await callTableModel(messages);
|
||
console.log(`[Amily2 立即远征] 批次 ${batchNum}/${totalBatches} - 收到 API 原始回复:`, resultText);
|
||
if (!resultText) throw new Error('API返回内容为空。');
|
||
|
||
if (!resultText.includes('<Amily2Edit>')) {
|
||
log(`批次 ${batchNum} 的响应未包含 <Amily2Edit> 指令块，弹出检查窗口等待用户处理。`, 'warn');
|
||
updateButtonState('paused');
|
||
showTableFillReviewModal(resultText, {
|
||
title: `填表响应检查 - 批次 ${batchNum}/${totalBatches}`,
|
||
subtitle: `批次 ${batchNum}/${totalBatches}（楼层 ${startFloor}-${endFloor}）的 AI 响应未包含有效的 <Amily2Edit> 指令块。请检查原始响应并选择处理方式。`,
|
||
onContinue: async (currentText) => {
|
||
const merged = await requestContinuation(messages, currentText);
|
||
if (!merged) { toastr.error('补全请求失败或返回为空。', '继续补全'); return null; }
|
||
if (!merged.includes('<Amily2Edit>')) {
|
||
toastr.warning('补全后仍未包含 <Amily2Edit> 指令块，可继续补全、手动应用或重新填表。', '继续补全');
|
||
} else {
|
||
toastr.success('已获得包含指令块的补全内容，可点击”手动应用”写入。', '继续补全');
|
||
}
|
||
return merged;
|
||
},
|
||
onApply: (editedText) => {
|
||
if (!editedText || !editedText.includes('<Amily2Edit>')) {
|
||
toastr.warning('应用的文本中未检测到 <Amily2Edit> 指令块，已按原文尝试写入。', '手动应用');
|
||
}
|
||
try {
|
||
updateTableFromText(editedText, { immediateDelete: true });
|
||
renderTables();
|
||
log(`批次 ${batchNum} 已由用户手动处理完成。`, 'success');
|
||
} catch (err) {
|
||
log(`批次 ${batchNum} 手动应用失败: ${err.message}`, 'error');
|
||
toastr.error(`手动应用失败: ${err.message}`, '写入异常');
|
||
currentBatch = batchNum - 1;
|
||
updateButtonState('error');
|
||
return;
|
||
}
|
||
currentBatch = batchNum;
|
||
setTimeout(processNextBatch, 500);
|
||
},
|
||
onRetry: () => {
|
||
log(`用户选择重新填表，批次 ${batchNum} 将重新执行。`, 'warn');
|
||
setTimeout(() => runBatchAttempt(batchNum, 0), 300);
|
||
},
|
||
onCancel: () => {
|
||
log(`用户取消了批次 ${batchNum} 的处理，任务已暂停。`, 'warn');
|
||
currentBatch = batchNum - 1;
|
||
updateButtonState('error');
|
||
},
|
||
});
|
||
return;
|
||
}
|
||
|
||
updateTableFromText(resultText, { immediateDelete: true });
|
||
renderTables();
|
||
log(`批次 ${batchNum} 处理成功。`, 'success');
|
||
}
|
||
|
||
currentBatch = batchNum;
|
||
setTimeout(processNextBatch, 1000);
|
||
|
||
} catch (error) {
|
||
log(`批次 ${batchNum} 尝试 ${attemptNum + 1} 失败: ${error.message}`, 'error');
|
||
if (attemptNum >= MAX_RETRIES) {
|
||
log(`批次 ${batchNum} 已达到最大重试次数，任务暂停。`, 'error');
|
||
toastr.error(`批次 ${batchNum} 多次失败，请检查网络或API设置后手动继续。`, '任务暂停');
|
||
currentBatch = batchNum - 1;
|
||
updateButtonState('error');
|
||
} else {
|
||
log(`将在3秒后自动重试批次 ${batchNum}...`, 'warn');
|
||
setTimeout(() => runBatchAttempt(batchNum, attemptNum + 1), 3000);
|
||
}
|
||
}
|
||
}
|
||
|
||
async function processNextBatch() {
|
||
if (manualStopRequested) {
|
||
log(`任务已在批次 ${currentBatch + 1} 开始前手动暂停。`, 'warn');
|
||
updateButtonState('paused');
|
||
return;
|
||
}
|
||
|
||
if (currentBatch >= totalBatches) {
|
||
log('所有批次处理完毕！', 'success');
|
||
updateButtonState('idle');
|
||
return;
|
||
}
|
||
|
||
runBatchAttempt(currentBatch + 1, 0);
|
||
}
|
||
|
||
export function startBatchFilling() {
|
||
const button = fillButton();
|
||
if (!button) return;
|
||
|
||
const settings = extension_settings[extensionName] || {};
|
||
const tableSystemEnabled = settings.table_system_enabled !== false;
|
||
if (!tableSystemEnabled) {
|
||
log('表格系统总开关已关闭，跳过批量填表。', 'info');
|
||
toastr.info('表格系统总开关已关闭，无法执行批量填表。');
|
||
return;
|
||
}
|
||
|
||
if (isFilling) {
|
||
if (button.textContent.startsWith('点击停止')) {
|
||
manualStopRequested = true;
|
||
updateButtonState('stopping');
|
||
log('停战敕令已下达！将在当前批次完成后暂停。', 'warn');
|
||
} else if (button.textContent.startsWith('继续填表')) {
|
||
manualStopRequested = false;
|
||
log('从上次暂停处继续处理...', 'info');
|
||
processNextBatch();
|
||
}
|
||
return;
|
||
}
|
||
|
||
manualStopRequested = false;
|
||
const context = getContext();
|
||
chatHistoryLength = context.chat.length;
|
||
threshold = extension_settings[extensionName]?.batch_filling_threshold
|
||
?? parseInt(/** @type {HTMLInputElement|null} */ (document.getElementById('batch-filling-threshold'))?.value, 10)
|
||
?? 30;
|
||
|
||
const ruleTemplate = getBatchFillerRuleTemplate();
|
||
const flowTemplate = getBatchFillerFlowTemplate();
|
||
|
||
if (!ruleTemplate || !flowTemplate) {
|
||
log('规则或流程提示词为空，无法开始填表。', 'error');
|
||
toastr.error('请确保"规则提示词"和"流程提示词"都已填写。', '无法开始');
|
||
return;
|
||
}
|
||
|
||
if (chatHistoryLength === 0) {
|
||
log('聊天记录为空，无需填表。', 'info');
|
||
return;
|
||
}
|
||
|
||
totalBatches = Math.ceil(chatHistoryLength / threshold);
|
||
currentBatch = 0;
|
||
|
||
const startFloorInput = document.getElementById('floor-start-input');
|
||
console.log('[Amily2 Debug] startFloorInput found:', !!startFloorInput);
|
||
if (startFloorInput) {
|
||
console.log('[Amily2 Debug] startFloorInput value:', startFloorInput.value);
|
||
const val = parseInt(startFloorInput.value, 10);
|
||
console.log('[Amily2 Debug] Parsed val:', val, 'Threshold:', threshold);
|
||
|
||
if (!isNaN(val) && val > 1) {
|
||
const startBatch = Math.ceil(val / threshold);
|
||
console.log('[Amily2 Debug] Calculated startBatch:', startBatch);
|
||
currentBatch = startBatch - 1;
|
||
log(`根据设定，将从第 ${startBatch} 批次（包含楼层 ${val}）开始执行。`, 'info');
|
||
} else {
|
||
console.log('[Amily2 Debug] Value is NaN or <= 1');
|
||
}
|
||
} else {
|
||
console.log('[Amily2 Debug] startFloorInput element not found');
|
||
}
|
||
|
||
log(`准备开始批量填表任务，共 ${totalBatches} 个批次。`, 'info');
|
||
processNextBatch();
|
||
}
|
||
|
||
|
||
export async function startFloorRangeFilling(startFloor, endFloor) {
|
||
const settings = extension_settings[extensionName] || {};
|
||
const tableSystemEnabled = settings.table_system_enabled !== false;
|
||
if (!tableSystemEnabled) {
|
||
log('表格系统总开关已关闭，跳过楼层填表。', 'info');
|
||
toastr.info('表格系统总开关已关闭，无法执行楼层填表。');
|
||
return;
|
||
}
|
||
|
||
const context = getContext();
|
||
const currentChatLength = context.chat.length;
|
||
|
||
if (endFloor > currentChatLength) {
|
||
toastr.warning(`结束楼层 ${endFloor} 超出了当前聊天记录长度 ${currentChatLength}。`);
|
||
return;
|
||
}
|
||
|
||
const ruleTemplate = getBatchFillerRuleTemplate();
|
||
const flowTemplate = getBatchFillerFlowTemplate();
|
||
|
||
if (!ruleTemplate || !flowTemplate) {
|
||
log('规则或流程提示词为空，无法开始楼层填表。', 'error');
|
||
toastr.error('请确保"规则提示词"和"流程提示词"都已填写。', '无法开始');
|
||
return;
|
||
}
|
||
|
||
try {
|
||
log(`开始处理楼层 ${startFloor}-${endFloor} 的内容...`, 'info');
|
||
|
||
const purifiedMessages = getRawMessagesForSummary(startFloor, endFloor);
|
||
if (!purifiedMessages || purifiedMessages.length === 0) {
|
||
toastr.warning('指定楼层范围内没有有效内容可处理。');
|
||
return;
|
||
}
|
||
|
||
const batchContent = purifiedMessages.map(m => `【第 ${m.floor} 楼】 ${m.author}: ${m.content}`).join('\n');
|
||
const currentTableDataString = convertTablesToCsvString();
|
||
const finalFlowPrompt = flowTemplate.replace('{{{Amily2TableData}}}', currentTableDataString);
|
||
|
||
let mixedOrder;
|
||
try {
|
||
const savedOrder = localStorage.getItem('amily2_prompt_presets_v2_mixed_order');
|
||
if (savedOrder) {
|
||
mixedOrder = JSON.parse(savedOrder);
|
||
}
|
||
} catch (e) {
|
||
console.error("[楼层填表] 加载混合顺序失败:", e);
|
||
}
|
||
const order = getMixedOrder('batch_filler') || [];
|
||
|
||
const presetPrompts = await getPresetPrompts('batch_filler');
|
||
|
||
const worldBookContext = await getWorldBookContext();
|
||
|
||
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') {
|
||
switch (item.id) {
|
||
case 'worldbook':
|
||
if (worldBookContext) {
|
||
messages.push({ role: 'system', content: worldBookContext });
|
||
}
|
||
break;
|
||
case 'ruleTemplate':
|
||
messages.push({ role: "system", content: ruleTemplate });
|
||
break;
|
||
case 'flowTemplate':
|
||
messages.push({ role: "system", content: finalFlowPrompt });
|
||
break;
|
||
case 'coreContent':
|
||
messages.push({ role: 'user', content: `请严格根据以下"对话记录"中的内容进行填写表格，并按照指定的格式输出，不要添加任何额外信息。\n\n<对话记录>\n${batchContent}\n</对话记录>` });
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
if (!presetPrompts || presetPrompts.length === 0) {
|
||
const defaultPrompts = [
|
||
{ role: 'system', content: generateRandomSeed() }
|
||
];
|
||
messages.splice(1, 0, ...defaultPrompts);
|
||
}
|
||
|
||
console.groupCollapsed(`[Amily2 楼层填表] 楼层 ${startFloor}-${endFloor} - 即将发送至 API 的内容`);
|
||
console.dir(messages);
|
||
console.groupEnd();
|
||
|
||
const floorSettings = extension_settings[extensionName] || {};
|
||
if (floorSettings.tableFillFunctionCall) {
|
||
const argsString = await callAIForTools(messages, TABLE_FILL_TOOL, { slot: 'tableFilling' });
|
||
if (!argsString) throw new Error('Function Call 返回为空。');
|
||
const ops = parseToolCallArgs(argsString);
|
||
if (ops.length === 0) {
|
||
log(`楼层 ${startFloor}-${endFloor} Function Call 返回操作列表为空，无需变更。`, 'warn');
|
||
toastr.info('AI 判断此楼层范围无需修改。', `楼层 ${startFloor}-${endFloor}`);
|
||
} else {
|
||
await updateTableFromOps(ops, { immediateDelete: true });
|
||
renderTables();
|
||
toastr.success(`楼层 ${startFloor}-${endFloor} 填表完成！`);
|
||
log(`楼层 ${startFloor}-${endFloor} Function Call 处理成功（${ops.length} 条操作）。`, 'success');
|
||
}
|
||
} else {
|
||
const resultText = await callTableModel(messages);
|
||
console.log(`[Amily2 楼层填表] 楼层 ${startFloor}-${endFloor} - 收到 API 原始回复:`, resultText);
|
||
if (!resultText) throw new Error('API返回内容为空。');
|
||
|
||
if (!resultText.includes('<Amily2Edit>')) {
|
||
log(`楼层 ${startFloor}-${endFloor} 的响应未包含 <Amily2Edit> 指令块，弹出检查窗口等待用户处理。`, 'warn');
|
||
showTableFillReviewModal(resultText, {
|
||
title: `填表响应检查 - 楼层 ${startFloor}-${endFloor}`,
|
||
subtitle: `楼层 ${startFloor}-${endFloor} 的 AI 响应未包含有效的 <Amily2Edit> 指令块。请检查原始响应并选择处理方式。`,
|
||
onContinue: async (currentText) => {
|
||
const merged = await requestContinuation(messages, currentText);
|
||
if (!merged) { toastr.error('补全请求失败或返回为空。', '继续补全'); return null; }
|
||
if (!merged.includes('<Amily2Edit>')) {
|
||
toastr.warning('补全后仍未包含 <Amily2Edit> 指令块，可继续补全、手动应用或重新填表。', '继续补全');
|
||
} else {
|
||
toastr.success('已获得包含指令块的补全内容，可点击”手动应用”写入。', '继续补全');
|
||
}
|
||
return merged;
|
||
},
|
||
onApply: (editedText) => {
|
||
if (!editedText || !editedText.includes('<Amily2Edit>')) {
|
||
toastr.warning('应用的文本中未检测到 <Amily2Edit> 指令块，已按原文尝试写入。', '手动应用');
|
||
}
|
||
try {
|
||
updateTableFromText(editedText, { immediateDelete: true });
|
||
renderTables();
|
||
toastr.success(`楼层 ${startFloor}-${endFloor} 填表完成！`);
|
||
log(`楼层 ${startFloor}-${endFloor} 填表由用户手动处理完成。`, 'success');
|
||
} catch (err) {
|
||
log(`楼层 ${startFloor}-${endFloor} 手动应用失败: ${err.message}`, 'error');
|
||
toastr.error(`手动应用失败: ${err.message}`, '写入异常');
|
||
}
|
||
},
|
||
onRetry: () => {
|
||
log(`用户请求重新填写楼层 ${startFloor}-${endFloor}。`, 'warn');
|
||
setTimeout(() => startFloorRangeFilling(startFloor, endFloor), 300);
|
||
},
|
||
onCancel: () => {
|
||
log(`用户取消了楼层 ${startFloor}-${endFloor} 的填表。`, 'warn');
|
||
toastr.info(`已取消楼层 ${startFloor}-${endFloor} 的填表。`);
|
||
},
|
||
});
|
||
return;
|
||
}
|
||
|
||
updateTableFromText(resultText, { immediateDelete: true });
|
||
renderTables();
|
||
toastr.success(`楼层 ${startFloor}-${endFloor} 填表完成！`);
|
||
log(`楼层 ${startFloor}-${endFloor} 填表处理完成。`, 'success');
|
||
}
|
||
|
||
} catch (error) {
|
||
log(`楼层 ${startFloor}-${endFloor} 填表失败: ${error.message}`, 'error');
|
||
toastr.error(`楼层填表失败: ${error.message}`, '处理失败');
|
||
}
|
||
}
|
||
|
||
|
||
export async function startCurrentFloorFilling() {
|
||
const context = getContext();
|
||
const currentFloor = context.chat.length;
|
||
|
||
if (currentFloor === 0) {
|
||
toastr.info('当前没有聊天记录。');
|
||
return;
|
||
}
|
||
|
||
log(`准备填写当前楼层（第 ${currentFloor} 楼）...`, 'info');
|
||
await startFloorRangeFilling(currentFloor, currentFloor);
|
||
}
|