Refactor message restoration logic to use a one-time event listener
This commit is contained in:
@@ -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);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|||||||
Reference in New Issue
Block a user