From 0e7eff155df1582dd8d57e7796b38fda1021088a Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Fri, 24 May 2024 21:53:29 +0300
Subject: [PATCH] Fix ephemeral injects clean-up

---
 public/scripts/slash-commands.js | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/public/scripts/slash-commands.js b/public/scripts/slash-commands.js
index 9ebf264c8..5baa11435 100644
--- a/public/scripts/slash-commands.js
+++ b/public/scripts/slash-commands.js
@@ -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);