@astralyn/pi 0.82.6 → 0.82.7

package/dist/modes/interactive/interactive-mode.js

@@ -186,6 +186,8 @@ export class InteractiveMode {
     onInputCallback;
     pendingUserInputs = [];
     activeStatusIndicator = undefined;
+    activeSelector = null;
+    activeSelectorCleanup = null;
     idleStatus = new IdleStatus();
     workingMessage = undefined;
     workingVisible = true;
@@ -1252,7 +1254,7 @@ export class InteractiveMode {
                     if (result.cancelled) {
                         return { cancelled: true };
                     }
-                    this.chatContainer.clear();
+                    this.clearChat();
                     this.renderInitialMessages();
                     if (result.editorText && !this.editor.getText().trim()) {
                         this.editor.setText(result.editorText);
@@ -1331,14 +1333,34 @@ export class InteractiveMode {
         this.stop();
         process.exit(1);
     }
+    disposeChatToolComponents() {
+        for (const child of this.chatContainer.children) {
+            if (child instanceof ToolExecutionComponent ||
+                child instanceof ToolGroupComponent ||
+                child instanceof ArminComponent ||
+                child instanceof DaxnutsComponent) {
+                child.dispose();
+            }
+        }
+    }
+    clearChat() {
+        this.disposeChatToolComponents();
+        this.chatContainer.clear();
+    }
+    clearPendingTools() {
+        for (const component of this.pendingTools.values()) {
+            component.dispose();
+        }
+        this.pendingTools.clear();
+    }
     renderCurrentSessionState() {
         this.loadedResourcesContainer.clear();
-        this.chatContainer.clear();
+        this.clearChat();
         this.pendingMessagesContainer.clear();
         this.compactionQueuedMessages = [];
         this.streamingComponent = undefined;
         this.streamingMessage = undefined;
-        this.pendingTools.clear();
+        this.clearPendingTools();
         this.renderInitialMessages();
     }
     /**
@@ -2326,7 +2348,7 @@ export class InteractiveMode {
             case "agent_start":
                 this.agentRunActive = true;
                 this.coalescingTerminal.setScrollbackPreservation(true);
-                this.pendingTools.clear();
+                this.clearPendingTools();
                 if (this.settingsManager.getShowTerminalProgress()) {
                     this.ui.terminal.setProgress(true);
                 }
@@ -2495,7 +2517,7 @@ export class InteractiveMode {
                     this.streamingComponent = undefined;
                     this.streamingMessage = undefined;
                 }
-                this.pendingTools.clear();
+                this.clearPendingTools();
                 this.ui.requestRender();
                 break;
             case "agent_settled":
@@ -2537,7 +2559,7 @@ export class InteractiveMode {
                 }
                 else {
                     if (event.result) {
-                        this.chatContainer.clear();
+                        this.clearChat();
                         this.rebuildChatFromMessages();
                         this.addMessageToChat(createCompactionSummaryMessage(event.result.summary, event.result.tokensBefore, new Date().toISOString()));
                         this.footer.invalidate();
@@ -2733,7 +2755,7 @@ export class InteractiveMode {
         }
     }
     renderSessionItems(items, options = {}) {
-        this.pendingTools.clear();
+        this.clearPendingTools();
         const renderedPendingTools = new Map();
         // Cache-miss notices are not persisted; re-derive them from the full entry
         // list and re-inject them after the assistant messages that paid for them.
@@ -2882,7 +2904,7 @@ export class InteractiveMode {
         });
     }
     rebuildChatFromMessages() {
-        this.chatContainer.clear();
+        this.clearChat();
         this.renderSessionEntries(this.sessionManager.buildContextEntries());
     }
     // =========================================================================
@@ -3174,7 +3196,7 @@ export class InteractiveMode {
         // Same reasoning as setToolsExpanded.
         this.coalescingTerminal.passNextFullRedraw();
         // Rebuild chat from session messages
-        this.chatContainer.clear();
+        this.clearChat();
         this.rebuildChatFromMessages();
         // If streaming, re-add the streaming component with updated visibility and re-render
         if (this.streamingComponent && this.streamingMessage) {
@@ -3422,15 +3444,42 @@ export class InteractiveMode {
      * @param create Factory that receives a `done` callback and returns the component and focus target
      */
     showSelector(create) {
+        this.activeSelectorCleanup?.();
+        this.activeSelectorCleanup = null;
+        this.activeSelector = null;
+        let selector;
+        let disposed = false;
+        let closed = false;
+        const dispose = () => {
+            if (disposed || !selector)
+                return;
+            disposed = true;
+            selector.dispose?.();
+        };
         const done = () => {
-            this.editorContainer.clear();
-            this.editorContainer.addChild(this.editor);
-            this.ui.setFocus(this.editor);
+            if (closed)
+                return;
+            closed = true;
+            dispose();
+            if (!selector || this.activeSelector === selector) {
+                this.activeSelector = null;
+                this.activeSelectorCleanup = null;
+                this.editorContainer.clear();
+                this.editorContainer.addChild(this.editor);
+                this.ui.setFocus(this.editor);
+            }
         };
-        const { component, focus } = create(done);
+        const created = create(done);
+        selector = created.component;
+        if (closed) {
+            dispose();
+            return;
+        }
+        this.activeSelector = selector;
+        this.activeSelectorCleanup = dispose;
         this.editorContainer.clear();
-        this.editorContainer.addChild(component);
-        this.ui.setFocus(focus);
+        this.editorContainer.addChild(selector);
+        this.ui.setFocus(created.focus);
         this.ui.requestRender();
     }
     showSettingsSelector() {
@@ -3530,7 +3579,7 @@ export class InteractiveMode {
                             child.setHideThinkingBlock(hidden);
                         }
                     }
-                    this.chatContainer.clear();
+                    this.clearChat();
                     this.rebuildChatFromMessages();
                 },
                 onShowCacheMissNoticesChange: (shown) => {
@@ -3952,7 +4001,7 @@ export class InteractiveMode {
                         return;
                     }
                     // Update UI
-                    this.chatContainer.clear();
+                    this.clearChat();
                     this.renderInitialMessages();
                     if (result.editorText && !this.editor.getText().trim()) {
                         this.editor.setText(result.editorText);
@@ -3975,7 +4024,7 @@ export class InteractiveMode {
             }, (entryId, label) => {
                 this.sessionManager.appendLabelChange(entryId, label);
                 this.ui.requestRender();
-            }, initialSelectedId, initialFilterMode);
+            }, initialSelectedId, initialFilterMode, () => this.ui.requestRender());
             selector.onCopy = async (text) => {
                 if (!text) {
                     this.showError("Selected entry has no text to copy");
@@ -5082,12 +5131,17 @@ export class InteractiveMode {
         }
     }
     stop() {
+        this.activeSelectorCleanup?.();
+        this.activeSelectorCleanup = null;
+        this.activeSelector = null;
         if (this.settingsManager.getShowTerminalProgress()) {
             this.ui.terminal.setProgress(false);
         }
         this.clearStatusIndicator();
         this.themeController.disableAutoSync();
         this.clearExtensionTerminalInputListeners();
+        this.clearPendingTools();
+        this.disposeChatToolComponents();
         this.footer.dispose();
         this.footerDataProvider.dispose();
         if (this.unsubscribe) {