mirror of
https://github.com/SilenceLurker/ST-Amily2-Chat-Optimisation.git
synced 2026-06-06 07:45:51 +00:00
Initial commit with CC BY-NC-ND 4.0 license
This commit is contained in:
276
core/super-memory/manager.js
Normal file
276
core/super-memory/manager.js
Normal file
@@ -0,0 +1,276 @@
|
||||
import { extension_settings, getContext } from "/scripts/extensions.js";
|
||||
import { extensionName } from "../../utils/settings.js";
|
||||
import { amilyHelper } from "../tavern-helper/main.js";
|
||||
import { generateIndex } from "./smart-indexer.js";
|
||||
import { syncToLorebook, ensureMemoryBook, updateTransientHint, getMemoryBookName } from "./lorebook-bridge.js";
|
||||
import { getMemoryState, loadMemoryState, saveMemoryState } from "../table-system/manager.js";
|
||||
import { eventSource, event_types } from "/script.js";
|
||||
|
||||
let isInitialized = false;
|
||||
let updateQueue = [];
|
||||
let isProcessing = false;
|
||||
let lastChatId = null;
|
||||
|
||||
const METADATA_KEY = 'Amily2_Memory_Data';
|
||||
|
||||
export async function initializeSuperMemory() {
|
||||
const userType = parseInt(localStorage.getItem("plugin_user_type") || "0");
|
||||
if (userType < 2) {
|
||||
console.warn('[Amily2-SuperMemory] 权限不足 (Type < 2)，拒绝初始化超级记忆系统。');
|
||||
if (window.$) $('#sm-system-status').text('未授权').css('color', 'red');
|
||||
return;
|
||||
}
|
||||
|
||||
const settings = extension_settings[extensionName] || {};
|
||||
if (settings.super_memory_enabled === false) {
|
||||
console.log('[Amily2-SuperMemory] 功能已禁用 (super_memory_enabled = false)。');
|
||||
if (window.$) $('#sm-system-status').text('已禁用').css('color', 'gray');
|
||||
return;
|
||||
}
|
||||
|
||||
if (isInitialized) {
|
||||
if (window.$) $('#sm-system-status').text('运行中').css('color', '#4caf50');
|
||||
return;
|
||||
}
|
||||
console.log('[Amily2-SuperMemory] 初始化核心管理器...');
|
||||
|
||||
if (!amilyHelper) {
|
||||
console.error('[Amily2-SuperMemory] 致命错误：AmilyHelper 未就绪。');
|
||||
return;
|
||||
}
|
||||
|
||||
document.addEventListener('AMILY2_TABLE_UPDATED', handleTableUpdate);
|
||||
|
||||
eventSource.on(event_types.CHAT_CHANGED, async () => {
|
||||
const settings = extension_settings[extensionName] || {};
|
||||
if (settings.super_memory_enabled === false) return;
|
||||
|
||||
console.log('[Amily2-SuperMemory] 检测到聊天切换，正在刷新记忆状态...');
|
||||
await checkWorldBookStatus();
|
||||
|
||||
await tryRestoreStateFromMetadata();
|
||||
|
||||
await forceSyncAll();
|
||||
});
|
||||
|
||||
await checkWorldBookStatus();
|
||||
|
||||
await tryRestoreStateFromMetadata();
|
||||
|
||||
await forceSyncAll();
|
||||
|
||||
isInitialized = true;
|
||||
console.log('[Amily2-SuperMemory] 核心管理器初始化完成。');
|
||||
|
||||
if (window.$) {
|
||||
$('#sm-system-status').text('运行中').css('color', '#4caf50');
|
||||
}
|
||||
}
|
||||
|
||||
async function checkWorldBookStatus() {
|
||||
try {
|
||||
await ensureMemoryBook();
|
||||
} catch (error) {
|
||||
console.error('[Amily2-SuperMemory] 检查世界书状态失败:', error);
|
||||
}
|
||||
}
|
||||
|
||||
function handleTableUpdate(event) {
|
||||
const settings = extension_settings[extensionName] || {};
|
||||
if (settings.super_memory_enabled === false) return;
|
||||
|
||||
const { tableName, data, role, hint, headers, rowStatuses } = event.detail;
|
||||
console.log(`[Amily2-SuperMemory] 检测到表格更新: ${tableName} (Role: ${role})`);
|
||||
|
||||
updateQueue.push({ tableName, data, role, hint, headers, rowStatuses });
|
||||
processQueue();
|
||||
}
|
||||
|
||||
async function processQueue() {
|
||||
if (isProcessing || updateQueue.length === 0) return;
|
||||
isProcessing = true;
|
||||
|
||||
try {
|
||||
while (updateQueue.length > 0) {
|
||||
|
||||
const consolidatedTasks = new Map();
|
||||
const currentBatch = [...updateQueue];
|
||||
updateQueue.length = 0; // 清空队列
|
||||
|
||||
for (const task of currentBatch) {
|
||||
consolidatedTasks.set(task.tableName, task);
|
||||
}
|
||||
|
||||
if (currentBatch.length > consolidatedTasks.size) {
|
||||
console.log(`[Amily2-SuperMemory] 队列优化: 将 ${currentBatch.length} 个事件合并为 ${consolidatedTasks.size} 个操作。`);
|
||||
}
|
||||
|
||||
for (const task of consolidatedTasks.values()) {
|
||||
await processUpdateTask(task);
|
||||
}
|
||||
}
|
||||
|
||||
await saveStateToMetadata();
|
||||
|
||||
} catch (error) {
|
||||
console.error('[Amily2-SuperMemory] 处理更新队列失败:', error);
|
||||
} finally {
|
||||
isProcessing = false;
|
||||
if (updateQueue.length > 0) {
|
||||
processQueue();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
async function processUpdateTask(task) {
|
||||
const { tableName, data, role, hint, headers, rowStatuses } = task;
|
||||
|
||||
const settings = extension_settings[extensionName] || {};
|
||||
const tableSettings = settings.superMemory_tableSettings?.[tableName] || {};
|
||||
|
||||
if (tableSettings.sync === false) {
|
||||
console.log(`[Amily2-SuperMemory] 表格 ${tableName} 已配置为不写入世界书，跳过同步。`);
|
||||
return;
|
||||
}
|
||||
|
||||
const isIndexConstant = tableSettings.constant !== false;
|
||||
|
||||
const activeData = data.filter((_, i) => !rowStatuses || rowStatuses[i] !== 'pending-deletion');
|
||||
const indexText = generateIndex(activeData, headers, role, tableName);
|
||||
|
||||
const allTables = getMemoryState();
|
||||
const tableIndex = allTables.findIndex(t => t.name === tableName);
|
||||
const depth = 8001 + (tableIndex >= 0 ? tableIndex : 99);
|
||||
|
||||
await syncToLorebook(tableName, data, indexText, role, headers, rowStatuses, depth, isIndexConstant);
|
||||
|
||||
if (hint) {
|
||||
console.log(`[Amily2-SuperMemory] 应用主动记忆提示: ${hint}`);
|
||||
await updateTransientHint(hint);
|
||||
}
|
||||
|
||||
console.log(`[Amily2-SuperMemory] 任务完成: ${tableName}`);
|
||||
|
||||
updateDashboardCounters();
|
||||
}
|
||||
|
||||
async function saveStateToMetadata() {
|
||||
const context = getContext();
|
||||
if (!context.chat || context.chat.length === 0) return;
|
||||
|
||||
const lastMsgIndex = context.chat.length - 1;
|
||||
const lastMsg = context.chat[lastMsgIndex];
|
||||
|
||||
const currentState = getMemoryState();
|
||||
|
||||
if (!lastMsg.metadata) lastMsg.metadata = {};
|
||||
|
||||
lastMsg.metadata[METADATA_KEY] = JSON.parse(JSON.stringify(currentState));
|
||||
|
||||
if (context.saveChat) {
|
||||
await context.saveChat();
|
||||
}
|
||||
|
||||
console.log(`[Amily2-SuperMemory] 状态已保存至消息 #${lastMsgIndex}`);
|
||||
}
|
||||
|
||||
export async function tryRestoreStateFromMetadata() {
|
||||
const context = getContext();
|
||||
if (!context.chat || context.chat.length === 0) return;
|
||||
|
||||
let foundState = null;
|
||||
let foundIndex = -1;
|
||||
|
||||
for (let i = context.chat.length - 1; i >= 0; i--) {
|
||||
const msg = context.chat[i];
|
||||
if (msg.metadata && msg.metadata[METADATA_KEY]) {
|
||||
foundState = msg.metadata[METADATA_KEY];
|
||||
foundIndex = i;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (foundState) {
|
||||
console.log(`[Amily2-SuperMemory] 发现历史状态 (Msg #${foundIndex})，正在恢复...`);
|
||||
if (typeof loadMemoryState === 'function') {
|
||||
loadMemoryState(foundState);
|
||||
await forceSyncAll();
|
||||
} else {
|
||||
console.warn('[Amily2-SuperMemory] table-system 缺少 loadMemoryState 方法，无法恢复状态。');
|
||||
}
|
||||
} else {
|
||||
console.log('[Amily2-SuperMemory] 未在聊天记录中发现历史状态，使用默认/当前状态。');
|
||||
}
|
||||
}
|
||||
|
||||
function updateDashboardCounters() {
|
||||
const tables = getMemoryState();
|
||||
if (tables && window.$) {
|
||||
$('#sm-index-count').text(`${tables.length} 个索引`);
|
||||
const totalRows = tables.reduce((acc, t) => acc + (t.rows ? t.rows.length : 0), 0);
|
||||
$('#sm-detail-count').text(`${totalRows} 个详情`);
|
||||
}
|
||||
}
|
||||
|
||||
export async function forceSyncAll() {
|
||||
console.log('[Amily2-SuperMemory] 正在执行全量同步...');
|
||||
const tables = getMemoryState();
|
||||
|
||||
if (!tables || tables.length === 0) {
|
||||
console.warn('[Amily2-SuperMemory] 没有可同步的表格数据。');
|
||||
return;
|
||||
}
|
||||
|
||||
for (const table of tables) {
|
||||
let role = 'database';
|
||||
if (table.name.includes('时空') || table.name.includes('世界钟')) role = 'anchor';
|
||||
if (table.name.includes('日志') || table.name.includes('Log')) role = 'log';
|
||||
|
||||
updateQueue.push({
|
||||
tableName: table.name,
|
||||
data: table.rows,
|
||||
headers: table.headers,
|
||||
rowStatuses: table.rowStatuses || [],
|
||||
role: role
|
||||
});
|
||||
}
|
||||
|
||||
await processQueue();
|
||||
console.log('[Amily2-SuperMemory] 全量同步完成。');
|
||||
}
|
||||
|
||||
export async function purgeSuperMemory() {
|
||||
try {
|
||||
console.log('[Amily2-SuperMemory] 开始清空记忆...');
|
||||
const bookName = getMemoryBookName();
|
||||
const entries = await amilyHelper.getLorebookEntries(bookName);
|
||||
|
||||
if (!entries || entries.length === 0) {
|
||||
console.log('[Amily2-SuperMemory] 世界书为空，无需清理。');
|
||||
return;
|
||||
}
|
||||
|
||||
const entriesToDelete = [];
|
||||
const prefixes = ['[Amily2]', '【Amily2'];
|
||||
|
||||
for (const entry of entries) {
|
||||
if (entry.comment && prefixes.some(p => entry.comment.startsWith(p))) {
|
||||
entriesToDelete.push(entry.uid);
|
||||
}
|
||||
}
|
||||
|
||||
if (entriesToDelete.length > 0) {
|
||||
await amilyHelper.deleteLorebookEntries(bookName, entriesToDelete);
|
||||
console.log(`[Amily2-SuperMemory] 已清空 ${entriesToDelete.length} 个条目。`);
|
||||
if (window.toastr) toastr.success(`已清空 ${entriesToDelete.length} 条记忆数据`);
|
||||
} else {
|
||||
if (window.toastr) toastr.info('没有发现需要清空的Amily2记忆数据');
|
||||
}
|
||||
|
||||
updateDashboardCounters();
|
||||
|
||||
} catch (error) {
|
||||
console.error('[Amily2-SuperMemory] 清空失败:', error);
|
||||
if (window.toastr) toastr.error('清空失败: ' + error.message);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user