Fix ephemeral injects clean-up

This commit is contained in:
Cohee
2024-05-24 21:53:29 +03:00
parent 66454bb711
commit 0e7eff155d

View File

@ -1212,9 +1212,16 @@ function injectCallback(args, value) {
saveMetadataDebounced();
if (ephemeral) {
let deleted = false;
const unsetInject = () => {
if (deleted) {
return;
}
console.log('Removing ephemeral script injection', id);
delete chat_metadata.script_injects[id];
setExtensionPrompt(prefixedId, '', position, depth, scan, role);
saveMetadataDebounced();
deleted = true;
};
eventSource.once(event_types.GENERATION_ENDED, unsetInject);
eventSource.once(event_types.GENERATION_STOPPED, unsetInject);