@probelabs/probe 0.6.0-rc257 → 0.6.0-rc259

package/build/agent/ProbeAgent.js

@@ -59,6 +59,7 @@ import {
   attemptCompletionToolDefinition,
   editToolDefinition,
   createToolDefinition,
+  multiEditToolDefinition,
   googleSearchToolDefinition,
   urlContextToolDefinition,
   attemptCompletionSchema,
@@ -964,6 +965,9 @@ export class ProbeAgent {
       if (wrappedTools.createToolInstance && isToolAllowed('create')) {
         this.toolImplementations.create = wrappedTools.createToolInstance;
       }
+      if (wrappedTools.multiEditToolInstance && isToolAllowed('multi_edit')) {
+        this.toolImplementations.multi_edit = wrappedTools.multiEditToolInstance;
+      }
     }
 
     // Store wrapped tools for ACP system
@@ -2565,6 +2569,9 @@ ${extractGuidance}
     if (this.allowEdit && isToolAllowed('create')) {
       toolDefinitions += `${createToolDefinition}\n`;
     }
+    if (this.allowEdit && isToolAllowed('multi_edit')) {
+      toolDefinitions += `${multiEditToolDefinition}\n`;
+    }
     // Bash tool (require both enableBash flag AND allowedTools permission)
     if (this.enableBash && isToolAllowed('bash')) {
       toolDefinitions += `${bashToolDefinition}\n`;
@@ -2670,6 +2677,9 @@ The configuration is loaded from src/config.js lines 15-25 which contains the da
     if (this.allowEdit && isToolAllowed('create')) {
       availableToolsList += '- create: Create new files with specified content.\n';
     }
+    if (this.allowEdit && isToolAllowed('multi_edit')) {
+      availableToolsList += '- multi_edit: Apply multiple file edits in one call using a JSON array of operations.\n';
+    }
     if (this.enableDelegate && isToolAllowed('delegate')) {
       availableToolsList += '- delegate: Delegate big distinct tasks to specialized probe subagents.\n';
     }
@@ -3430,6 +3440,9 @@ Follow these instructions carefully:
           if (this.allowEdit && this.allowedTools.isEnabled('create')) {
             validTools.push('create');
           }
+          if (this.allowEdit && this.allowedTools.isEnabled('multi_edit')) {
+            validTools.push('multi_edit');
+          }
           // Bash tool (require both enableBash flag AND allowedTools permission)
           if (this.enableBash && this.allowedTools.isEnabled('bash')) {
             validTools.push('bash');