@caupulican/pi-adaptative 0.81.9 → 0.81.12

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

@@ -1043,21 +1043,21 @@ export class InteractiveMode {
             return;
         }
     }
-    attachToolExecutionComponent(toolName, toolCallId, args) {
+    attachToolExecutionComponent(toolName, toolCallId, args, repair) {
         const actionKey = getToolPanelActionKey(this.getToolPanelScope(), toolName, args);
         const toolDefinition = this.getRegisteredToolDefinition(toolName);
         const reuseInPlace = shouldReuseToolPanelInPlace(toolName, args);
         const existing = this.toolPanels.getReusable(actionKey, { allowActive: reuseInPlace });
         if (existing) {
             if (reuseInPlace && actionKey) {
-                existing.resetInvocation(toolName, toolCallId, args, toolDefinition);
+                existing.resetInvocation(toolName, toolCallId, args, toolDefinition, repair);
                 existing.setExpanded(this.toolOutputExpanded);
                 this.toolPanels.replaceActiveForAction(toolCallId, existing, actionKey);
                 this.ui.requestRender();
                 return existing;
             }
             this.detachToolExecutionComponent(existing);
-            existing.resetInvocation(toolName, toolCallId, args, toolDefinition);
+            existing.resetInvocation(toolName, toolCallId, args, toolDefinition, repair);
             existing.setExpanded(this.toolOutputExpanded);
             this.appendToolExecutionComponent(existing, true);
             this.toolPanels.register(toolCallId, existing, actionKey);
@@ -1066,6 +1066,7 @@ export class InteractiveMode {
         const component = new ToolExecutionComponent(toolName, toolCallId, args, {
             showImages: this.settingsManager.getShowImages(),
             imageWidthCells: this.settingsManager.getImageWidthCells(),
+            repair,
         }, toolDefinition, this.ui, this.sessionManager.getCwd());
         component.setExpanded(this.toolOutputExpanded);
         this.appendToolExecutionComponent(component, true);
@@ -1326,6 +1327,22 @@ export class InteractiveMode {
                 this.editor.setText("");
                 return;
             }
+            if (text === "/toolhealth") {
+                this.showStatus(this.session.formatToolRepairHealthReport());
+                this.editor.setText("");
+                return;
+            }
+            if (text.startsWith("/toolrule-remove")) {
+                const [model, mode] = text.slice("/toolrule-remove".length).trim().split(/\s+/, 2);
+                const message = model && mode
+                    ? this.session.removeToolRepairRule(model, mode)
+                        ? `Removed tool repair rule ${mode} for ${model}.`
+                        : `No tool repair rule ${mode} found for ${model}.`
+                    : "Usage: /toolrule-remove <provider/model> <mode>";
+                this.showStatus(message);
+                this.editor.setText("");
+                return;
+            }
             if (text === "/scoped-models") {
                 this.editor.setText("");
                 await this.showModelsSelector();
@@ -1657,14 +1674,17 @@ export class InteractiveMode {
             case "tool_execution_start": {
                 let component = this.toolPanels.getActive(event.toolCallId);
                 if (!component)
-                    component = this.attachToolExecutionComponent(event.toolName, event.toolCallId, event.args);
-                component.markExecutionStarted();
+                    component = this.attachToolExecutionComponent(event.toolName, event.toolCallId, event.args, event.repair);
+                else
+                    component.updateArgs(event.args, event.repair);
+                component.markExecutionStarted(event.repair);
                 this.ui.requestRender();
                 break;
             }
             case "tool_execution_update": {
                 const component = this.toolPanels.getActive(event.toolCallId);
                 if (component) {
+                    component.updateArgs(event.args, event.repair);
                     component.updateResult({ ...event.partialResult, isError: false }, true);
                     this.ui.requestRender();
                 }
@@ -1865,17 +1885,27 @@ export class InteractiveMode {
             }
         })();
     }
+    getToolCallRepairInfo(toolCall) {
+        if (!toolCall.rawArguments && !toolCall.repairNotes?.length)
+            return undefined;
+        return {
+            repaired: true,
+            ...(toolCall.rawArguments ? { rawArguments: toolCall.rawArguments } : {}),
+            ...(toolCall.repairNotes?.length ? { notes: toolCall.repairNotes } : {}),
+        };
+    }
     attachStreamingToolPanels(message) {
         for (const content of message.content) {
             if (content.type !== "toolCall")
                 continue;
+            const repair = this.getToolCallRepairInfo(content);
             if (!this.toolPanels.hasActive(content.id)) {
-                this.attachToolExecutionComponent(content.name, content.id, content.arguments);
+                this.attachToolExecutionComponent(content.name, content.id, content.arguments, repair);
             }
             else {
                 const component = this.toolPanels.getActive(content.id);
                 if (component) {
-                    component.updateArgs(content.arguments);
+                    component.updateArgs(content.arguments, repair);
                 }
             }
         }