taozuqin/ST-Bionic-Memory-Ecology
1
0
Fork 0
mirror of https://github.com/Youzini-afk/ST-Bionic-Memory-Ecology.git synced 2026-05-15 22:30:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Guard against Tavern Helper prompt viewer fake generations

Browse Source
This commit is contained in:
Hao19911125
2026-04-11 22:09:28 +08:00
parent 0d4a692bbb
commit 943c398fb9
2 changed files with 98 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
host/event-binding.js
View File

@@ -24,6 +24,15 @@ function toSafeFloor(value, fallback = null) {
return Number.isFinite(numeric) ? Math.floor(numeric) : fallback;
}
function isTavernHelperPromptViewerSyntheticGeneration(runtime) {
if (!runtime.isTavernHelperPromptViewerRefreshActive?.()) {
return false;
}
const pendingSendIntent = runtime.getPendingRecallSendIntent?.();
return !runtime.isFreshRecallInputRecord?.(pendingSendIntent);
}
export function registerBeforeCombinePromptsController(runtime, listener) {
const makeFirst = runtime.getEventMakeFirst();
if (typeof makeFirst === "function") {
@@ -333,6 +342,21 @@ export function onGenerationStartedController(
const generationType = String(type || "normal").trim() || "normal";
if (generationType !== "normal") return null;
if (isTavernHelperPromptViewerSyntheticGeneration(runtime)) {
const context = runtime.getContext?.() || {};
runtime.markCurrentGenerationTrivialSkip?.({
reason: "tavern-helper-prompt-viewer",
chatId: context?.chatId || "",
chatLength: Array.isArray(context?.chat) ? context.chat.length : 0,
});
runtime.clearPendingRecallSendIntent?.();
runtime.clearPendingHostGenerationInputSnapshot?.();
console.debug?.(
"[ST-BME] skip: tavern-helper-prompt-viewer hook=GENERATION_STARTED",
);
return null;
}
const pendingSendIntent = runtime.getPendingRecallSendIntent?.();
const pendingIntentText = runtime.isFreshRecallInputRecord?.(
pendingSendIntent,
@@ -445,6 +469,8 @@ export async function onGenerationAfterCommandsController(
return;
}
const generationType = String(type || "normal").trim() || "normal";
if (runtime.isMvuExtraAnalysisGuardActive?.()) {
console.debug?.(
"[ST-BME] skip: mvu-extra-analysis hook=GENERATION_AFTER_COMMANDS",
@@ -452,7 +478,24 @@ export async function onGenerationAfterCommandsController(
return;
}
const generationType = String(type || "normal").trim() || "normal";
if (
generationType === "normal" &&
isTavernHelperPromptViewerSyntheticGeneration(runtime)
) {
const context = runtime.getContext?.() || {};
runtime.markCurrentGenerationTrivialSkip?.({
reason: "tavern-helper-prompt-viewer",
chatId: runtime.getCurrentChatId?.() || context?.chatId || "",
chatLength: Array.isArray(context?.chat) ? context.chat.length : 0,
});
runtime.clearPendingRecallSendIntent?.();
runtime.clearPendingHostGenerationInputSnapshot?.();
console.debug?.(
"[ST-BME] skip: tavern-helper-prompt-viewer hook=GENERATION_AFTER_COMMANDS",
);
return;
}
const frozenInputSnapshot =
generationType === "normal"
? runtime.consumeHostGenerationInputSnapshot?.({ preserve: true }) ||
@@ -595,6 +638,24 @@ export async function onBeforeCombinePromptsController(
};
}
if (isTavernHelperPromptViewerSyntheticGeneration(runtime)) {
const context = runtime.getContext?.() || {};
runtime.markCurrentGenerationTrivialSkip?.({
reason: "tavern-helper-prompt-viewer",
chatId: runtime.getCurrentChatId?.() || context?.chatId || "",
chatLength: Array.isArray(context?.chat) ? context.chat.length : 0,
});
runtime.clearPendingRecallSendIntent?.();
runtime.clearPendingHostGenerationInputSnapshot?.();
console.debug?.(
"[ST-BME] skip: tavern-helper-prompt-viewer hook=GENERATE_BEFORE_COMBINE_PROMPTS",
);
return {
skipped: true,
reason: "tavern-helper-prompt-viewer",
};
}
const frozenInputSnapshot =
runtime.consumeHostGenerationInputSnapshot?.() ||
runtime.getPendingHostGenerationInputSnapshot?.() ||