From 94f562f1bb5464ab3d261144042930528e54ea5b Mon Sep 17 00:00:00 2001 From: tomt610 Date: Fri, 9 Jan 2026 20:20:04 +0000 Subject: [PATCH] Refactor message restoration logic to use a one-time event listener --- src/systems/generation/injector.js | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/systems/generation/injector.js b/src/systems/generation/injector.js index 2801a56..f7db4bd 100644 --- a/src/systems/generation/injector.js +++ b/src/systems/generation/injector.js @@ -561,12 +561,10 @@ Ensure these details naturally reflect and influence the narrative. Character be injectHistoricalContextIntoChat(); // Register a one-time listener to restore messages after prompt is built - // This ensures messages are restored even if generation is cancelled or for dry runs - const restoreAfterPrompt = () => { + // Using .once() so it auto-removes after firing + eventSource.once(event_types.GENERATE_AFTER_COMBINE_PROMPTS, () => { restoreOriginalMessageContent(); - eventSource.off(event_types.GENERATE_AFTER_COMBINE_PROMPTS, restoreAfterPrompt); - }; - eventSource.once(event_types.GENERATE_AFTER_COMBINE_PROMPTS, restoreAfterPrompt); + }); } /**