Pakobbix/rpg-companion-sillytavern
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
tomt610
2026-01-09 20:20:04 +00:00
parent 3d5fc5fee1
commit 94f562f1bb
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/systems/generation/injector.js
+3 -5
View File
@@ -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);
});
}
/**