Refactor message restoration logic to use a one-time event listener

This commit is contained in:
tomt610
2026-01-09 20:20:04 +00:00
parent 3d5fc5fee1
commit 94f562f1bb
+3 -5
View File
@@ -561,12 +561,10 @@ Ensure these details naturally reflect and influence the narrative. Character be
injectHistoricalContextIntoChat(); injectHistoricalContextIntoChat();
// Register a one-time listener to restore messages after prompt is built // 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 // Using .once() so it auto-removes after firing
const restoreAfterPrompt = () => { eventSource.once(event_types.GENERATE_AFTER_COMBINE_PROMPTS, () => {
restoreOriginalMessageContent(); restoreOriginalMessageContent();
eventSource.off(event_types.GENERATE_AFTER_COMBINE_PROMPTS, restoreAfterPrompt); });
};
eventSource.once(event_types.GENERATE_AFTER_COMBINE_PROMPTS, restoreAfterPrompt);
} }
/** /**