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 11:15:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update cwb_lorebookManager.js

Browse Source
This commit is contained in:
Wx-2025
2025-10-26 04:58:08 +08:00
committed by GitHub
parent 4239c8f86e
commit 30c3e953e2
1 changed files with 10 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
CharacterWorldBook/src/cwb_lorebookManager.js
View File

@@ -1,13 +1,7 @@
import { state } from './cwb_state.js'; import { state } from './cwb_state.js';
import { logError, logDebug, showToastr, parseCustomFormat } from './cwb_utils.js'; import { logError, logDebug, showToastr, parseCustomFormat } from './cwb_utils.js';
import {
safeLorebooks,
safeCharLorebooks,
safeLorebookEntries,
safeUpdateLorebookEntries,
compatibleWriteToLorebook,
} from '../../core/tavernhelper-compatibility.js';
import { amilyHelper } from '../../core/tavern-helper/main.js'; import { amilyHelper } from '../../core/tavern-helper/main.js';
import { loadWorldInfo, saveWorldInfo } from "/scripts/world-info.js";
const { SillyTavern } = window; const { SillyTavern } = window;
@@ -20,7 +14,7 @@ export async function getTargetWorldBook() {
return state.customWorldBook; return state.customWorldBook;
} }
try { try {
const charLorebooks = await safeCharLorebooks(); const charLorebooks = await amilyHelper.getCharLorebooks();
const primaryBook = charLorebooks.primary; const primaryBook = charLorebooks.primary;
if (!primaryBook) { if (!primaryBook) {
showToastr('error', '当前角色未设置主世界书。'); showToastr('error', '当前角色未设置主世界书。');
@@ -44,12 +38,12 @@ export async function deleteLorebookEntries(uids) {
const book = await getTargetWorldBook(); const book = await getTargetWorldBook();
if (!book) throw new Error('未找到目标世界书。'); if (!book) throw new Error('未找到目标世界书。');
const bookData = await amilyHelper.loadWorldInfo(book); const bookData = await loadWorldInfo(book);
if (!bookData) throw new Error(`World book "${book}" not found.`); if (!bookData) throw new Error(`World book "${book}" not found.`);
uids.forEach(uid => { uids.forEach(uid => {
delete bookData.entries[uid]; delete bookData.entries[uid];
}); });
await amilyHelper.saveWorldInfo(book, bookData, true); await saveWorldInfo(book, bookData, true);
} catch (error) { } catch (error) {
logError('删除世界书条目失败:', error); logError('删除世界书条目失败:', error);
showToastr('error', `删除失败: ${error.message}`); showToastr('error', `删除失败: ${error.message}`);
@@ -80,7 +74,7 @@ export async function saveDescriptionToLorebook(characterName, newDescription, s
return false; return false;
} }
const entries = (await safeLorebookEntries(bookName)) || []; const entries = await amilyHelper.getLorebookEntries(bookName);
let existing = entries.find(e => let existing = entries.find(e =>
Array.isArray(e.keys) && Array.isArray(e.keys) &&
e.keys.includes(chatIdentifier) && e.keys.includes(chatIdentifier) &&
@@ -97,7 +91,7 @@ export async function saveDescriptionToLorebook(characterName, newDescription, s
}; };
if (existing) { if (existing) {
await safeUpdateLorebookEntries(bookName, [{ uid: existing.uid, ...entryData }]); await amilyHelper.setLorebookEntries(bookName, [{ uid: existing.uid, ...entryData }]);
} else { } else {
const cwbEntries = entries.filter(e => const cwbEntries = entries.filter(e =>
Array.isArray(e.keys) && Array.isArray(e.keys) &&
@@ -180,7 +174,7 @@ export async function updateCharacterRosterLorebookEntry(processedCharacterNames
return false; return false;
} }
let entries = (await safeLorebookEntries(bookName)) || []; let entries = await amilyHelper.getLorebookEntries(bookName);
let existingRosterEntry = entries.find(entry => let existingRosterEntry = entries.find(entry =>
entry.comment === rosterEntryComment || entry.comment === rosterEntryComment ||
entry.comment === `Amily2角色总集-${chatIdentifier}-角色总览` entry.comment === `Amily2角色总集-${chatIdentifier}-角色总览`
@@ -245,7 +239,7 @@ export async function updateCharacterRosterLorebookEntry(processedCharacterNames
}; };
if (existingRosterEntry) { if (existingRosterEntry) {
await safeUpdateLorebookEntries(bookName, [ await amilyHelper.setLorebookEntries(bookName, [
{ uid: existingRosterEntry.uid, comment: rosterEntryComment, ...entryData }, { uid: existingRosterEntry.uid, comment: rosterEntryComment, ...entryData },
]); ]);
} else { } else {
@@ -276,7 +270,7 @@ export async function manageAutoCardUpdateLorebookEntry() {
const bookName = await getTargetWorldBook(); const bookName = await getTargetWorldBook();
if (!bookName) return; if (!bookName) return;
const entries = (await safeLorebookEntries(bookName)) || []; const entries = await amilyHelper.getLorebookEntries(bookName);
const currentChatId = state.currentChatFileIdentifier; const currentChatId = state.currentChatFileIdentifier;
if (!currentChatId || currentChatId.startsWith('unknown_chat')) { if (!currentChatId || currentChatId.startsWith('unknown_chat')) {
@@ -305,7 +299,7 @@ export async function manageAutoCardUpdateLorebookEntry() {
} }
if (entriesToUpdate.length > 0) { if (entriesToUpdate.length > 0) {
await safeUpdateLorebookEntries(bookName, entriesToUpdate); await amilyHelper.setLorebookEntries(bookName, entriesToUpdate);
logDebug(`已为聊天: ${cleanChatId} 管理了 ${entriesToUpdate.length} 个世界书条目的状态。`); logDebug(`已为聊天: ${cleanChatId} 管理了 ${entriesToUpdate.length} 个世界书条目的状态。`);
} }