@caupulican/pi-adaptative 0.80.0 → 0.80.2

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

@@ -1190,6 +1190,7 @@ export class InteractiveMode {
         const uiContext = this.createExtensionUIContext();
         await this.session.bindExtensions({
             uiContext,
+            mode: "tui",
             abortHandler: () => {
                 this.restoreQueuedMessagesToEditor({ abort: true });
             },
@@ -1346,15 +1347,38 @@ export class InteractiveMode {
         }
         this.chatContainer.addChild(component);
     }
+    detachToolExecutionComponent(component) {
+        const children = this.chatContainer.children;
+        const directIndex = children.indexOf(component);
+        if (directIndex !== -1) {
+            children.splice(directIndex, 1);
+            return;
+        }
+        for (let i = 0; i < children.length; i++) {
+            const child = children[i];
+            if (!(child instanceof ToolGroupComponent) || !child.removeTool(component))
+                continue;
+            const remaining = child.getToolCount();
+            if (remaining === 0) {
+                children.splice(i, 1);
+            }
+            else if (remaining === 1) {
+                const onlyTool = child.getOnlyTool();
+                if (onlyTool)
+                    children[i] = onlyTool;
+            }
+            return;
+        }
+    }
     attachToolExecutionComponent(toolName, toolCallId, args) {
         const actionKey = getToolPanelActionKey(this.getToolPanelScope(), toolName, args);
         const toolDefinition = this.getRegisteredToolDefinition(toolName);
         const existing = this.toolPanels.getReusable(actionKey);
         if (existing) {
-            this.chatContainer.removeChild(existing);
+            this.detachToolExecutionComponent(existing);
             existing.resetInvocation(toolName, toolCallId, args, toolDefinition);
             existing.setExpanded(this.toolOutputExpanded);
-            this.chatContainer.addChild(existing);
+            this.appendToolExecutionComponent(existing, true);
             this.toolPanels.register(toolCallId, existing, actionKey);
             return existing;
         }
@@ -1363,7 +1387,7 @@ export class InteractiveMode {
             imageWidthCells: this.settingsManager.getImageWidthCells(),
         }, toolDefinition, this.ui, this.sessionManager.getCwd());
         component.setExpanded(this.toolOutputExpanded);
-        this.appendToolExecutionComponent(component, !actionKey);
+        this.appendToolExecutionComponent(component, true);
         this.toolPanels.register(toolCallId, component, actionKey);
         return component;
     }
@@ -1384,6 +1408,7 @@ export class InteractiveMode {
         const createContext = () => ({
             ui: this.createExtensionUIContext(),
             hasUI: true,
+            mode: "tui",
             cwd: this.sessionManager.getCwd(),
             sessionManager: this.sessionManager,
             modelRegistry: this.session.modelRegistry,