@probelabs/probe 0.6.0-rc258 → 0.6.0-rc260

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');

package/build/agent/bashCommandUtils.js

@@ -127,6 +127,8 @@ export function parseSimpleCommand(command) {
     /&&/,           // Logical AND
     /\|\|/,         // Logical OR
     /(?<!\\);/,     // Command separator (but not escaped \;)
+    /\n/,           // Newline command separator (multi-line commands)
+    /\r/,           // Carriage return (CRLF line endings)
     /&$/,           // Background execution
     /\$\(/,         // Command substitution $()
     /`/,            // Command substitution ``
@@ -260,6 +262,7 @@ export function isComplexPattern(pattern) {
     /&&/,           // Logical AND
     /\|\|/,         // Logical OR
     /;/,            // Command separator
+    /\n/,           // Newline command separator
     /&$/,           // Background execution
     /\$\(/,         // Command substitution $()
     /`/,            // Command substitution ``

package/build/agent/bashPermissions.js

@@ -402,6 +402,29 @@ export class BashPermissionChecker {
           i++;
           continue;
         }
+        // Check for newline (command separator in multi-line scripts)
+        // Also handle \r\n (CRLF) line endings
+        if (char === '\n' || (char === '\r' && nextChar === '\n')) {
+          if (current.trim()) {
+            components.push(current.trim());
+          }
+          current = '';
+          if (char === '\r') {
+            i += 2; // Skip \r\n
+          } else {
+            i++;
+          }
+          continue;
+        }
+        if (char === '\r') {
+          // Bare \r without \n
+          if (current.trim()) {
+            components.push(current.trim());
+          }
+          current = '';
+          i++;
+          continue;
+        }
       }
 
       current += char;

package/build/agent/index.js

@@ -9427,7 +9427,7 @@ async function handleLineEdit({ resolvedPath, file_path, start_line, end_line, n
     return buildLineEditResponse(file_path, startLine, endLine, newLines.length, fileLines, startLine - 1, action, modifications);
   }
 }
-var editTool, createTool, editSchema, createSchema, editDescription, createDescription, editToolDefinition, createToolDefinition;
+var editTool, createTool, multiEditTool, editSchema, createSchema, multiEditSchema, editDescription, createDescription, multiEditDescription, editToolDefinition, createToolDefinition, multiEditToolDefinition;
 var init_edit = __esm({
   "src/tools/edit.js"() {
     "use strict";
@@ -9497,7 +9497,7 @@ Parameters:
           },
           required: ["file_path", "new_string"]
         },
-        execute: async ({ file_path, old_string, new_string, replace_all = false, symbol, position, start_line, end_line }) => {
+        execute: async ({ file_path, old_string, new_string, replace_all = false, symbol, position, start_line, end_line, workingDirectory }) => {
           try {
             if (!file_path || typeof file_path !== "string" || file_path.trim() === "") {
               return `Error editing file: Invalid file_path - must be a non-empty string. Provide an absolute path or a path relative to the working directory (e.g. "src/main.js").`;
@@ -9505,7 +9505,8 @@ Parameters:
             if (new_string === void 0 || new_string === null || typeof new_string !== "string") {
               return `Error editing file: Invalid new_string - must be a string. Provide the replacement content as a string value (empty string "" is valid for deletions).`;
             }
-            const resolvedPath = isAbsolute(file_path) ? file_path : resolve(cwd || process.cwd(), file_path);
+            const effectiveCwd = workingDirectory || cwd || process.cwd();
+            const resolvedPath = isAbsolute(file_path) ? file_path : resolve(effectiveCwd, file_path);
             if (debug) {
               console.error(`[Edit] Attempting to edit file: ${resolvedPath}`);
             }
@@ -9612,7 +9613,7 @@ Important:
           },
           required: ["file_path", "content"]
         },
-        execute: async ({ file_path, content, overwrite = false }) => {
+        execute: async ({ file_path, content, overwrite = false, workingDirectory }) => {
           try {
             if (!file_path || typeof file_path !== "string" || file_path.trim() === "") {
               return `Error creating file: Invalid file_path - must be a non-empty string. Provide an absolute path or a path relative to the working directory (e.g. "src/newFile.js").`;
@@ -9620,7 +9621,8 @@ Important:
             if (content === void 0 || content === null || typeof content !== "string") {
               return `Error creating file: Invalid content - must be a string. Provide the file content as a string value (empty string "" is valid for an empty file).`;
             }
-            const resolvedPath = isAbsolute(file_path) ? file_path : resolve(cwd || process.cwd(), file_path);
+            const effectiveCwd = workingDirectory || cwd || process.cwd();
+            const resolvedPath = isAbsolute(file_path) ? file_path : resolve(effectiveCwd, file_path);
             if (debug) {
               console.error(`[Create] Attempting to create file: ${resolvedPath}`);
             }
@@ -9649,6 +9651,70 @@ Important:
         }
       });
     };
+    multiEditTool = (options = {}) => {
+      const editInstance = editTool(options);
+      return tool({
+        name: "multi_edit",
+        description: "Apply multiple file edits in a single tool call. Accepts a JSON array of edit operations.",
+        inputSchema: {
+          type: "object",
+          properties: {
+            edits: {
+              type: "string",
+              description: "JSON array of edit operations. Each object supports: file_path, old_string, new_string, replace_all, symbol, position, start_line, end_line."
+            }
+          },
+          required: ["edits"]
+        },
+        execute: async ({ edits: rawEdits }) => {
+          let edits;
+          if (typeof rawEdits === "string") {
+            try {
+              edits = JSON.parse(rawEdits);
+            } catch (e) {
+              return `Error: Invalid JSON in edits parameter - ${e.message}. Provide a raw JSON array between <edits> tags.`;
+            }
+          } else if (Array.isArray(rawEdits)) {
+            edits = rawEdits;
+          } else {
+            return "Error: edits must be a JSON array of edit operations.";
+          }
+          if (!Array.isArray(edits) || edits.length === 0) {
+            return "Error: edits must be a non-empty JSON array.";
+          }
+          if (edits.length > 50) {
+            return `Error: Too many edits (${edits.length}). Maximum 50 per batch.`;
+          }
+          const results = [];
+          let successCount = 0;
+          let failCount = 0;
+          for (let i = 0; i < edits.length; i++) {
+            const editOp = edits[i];
+            if (!editOp || typeof editOp !== "object" || Array.isArray(editOp)) {
+              results.push(`[${i + 1}] FAIL: Invalid edit operation - must be an object`);
+              failCount++;
+              continue;
+            }
+            try {
+              const result = await editInstance.execute(editOp);
+              const isError = typeof result === "string" && result.startsWith("Error");
+              if (isError) {
+                results.push(`[${i + 1}] FAIL: ${result}`);
+                failCount++;
+              } else {
+                results.push(`[${i + 1}] OK: ${result}`);
+                successCount++;
+              }
+            } catch (error) {
+              results.push(`[${i + 1}] FAIL: ${error.message}`);
+              failCount++;
+            }
+          }
+          const summary = `Multi-edit: ${successCount}/${edits.length} succeeded` + (failCount > 0 ? `, ${failCount} failed` : "");
+          return summary + "\n\n" + results.join("\n");
+        }
+      });
+    };
     editSchema = {
       type: "object",
       properties: {
@@ -9706,8 +9772,19 @@ Important:
       },
       required: ["file_path", "content"]
     };
+    multiEditSchema = {
+      type: "object",
+      properties: {
+        edits: {
+          type: "string",
+          description: "JSON array of edit operations"
+        }
+      },
+      required: ["edits"]
+    };
     editDescription = "Edit files using text replacement, AST-aware symbol operations, or line-targeted editing. Supports fuzzy matching for text edits and optional hash-based integrity verification for line edits.";
     createDescription = "Create new files with specified content. Will create parent directories if needed.";
+    multiEditDescription = "Apply multiple file edits in a single tool call. Accepts a JSON array of edit operations, each supporting the same modes as the edit tool.";
     editToolDefinition = `
 ## edit
 Description: ${editDescription}
@@ -9863,6 +9940,44 @@ Examples:
 This is a new project.</content>
 <overwrite>true</overwrite>
 </create>`;
+    multiEditToolDefinition = `
+## multi_edit
+Description: ${multiEditDescription}
+
+Apply multiple edits in one call. Each operation in the array uses the same parameters as the edit tool:
+- file_path, old_string, new_string (text mode)
+- file_path, symbol, new_string (symbol replace)
+- file_path, symbol, new_string, position (symbol insert)
+- file_path, start_line, new_string (line-targeted)
+
+Edits are applied sequentially. Failures do not stop remaining edits. Maximum 50 edits per call.
+
+Parameters:
+- edits: (required) JSON array of edit objects. Place raw JSON between tags.
+
+When to use multi_edit vs edit:
+- Use edit for a single change to one file
+- Use multi_edit when making 2+ related changes across files (e.g., rename a function and update all call sites)
+- Use multi_edit for coordinated multi-file refactoring where order matters
+
+Examples:
+
+Multiple text replacements across files:
+<multi_edit>
+<edits>[
+  {"file_path": "src/main.js", "old_string": "return false;", "new_string": "return true;"},
+  {"file_path": "src/config.js", "old_string": "debug: false", "new_string": "debug: true"}
+]</edits>
+</multi_edit>
+
+Mixed edit modes in one batch:
+<multi_edit>
+<edits>[
+  {"file_path": "src/utils.js", "symbol": "oldHelper", "new_string": "function newHelper() { return 42; }"},
+  {"file_path": "src/main.js", "old_string": "oldHelper()", "new_string": "newHelper()", "replace_all": true},
+  {"file_path": "src/index.js", "start_line": "10", "end_line": "12", "new_string": "export { newHelper };"}
+]</edits>
+</multi_edit>`;
   }
 });
 
@@ -10356,7 +10471,8 @@ function getValidParamsForTool(toolName) {
     task: taskSchema,
     attempt_completion: attemptCompletionSchema,
     edit: editSchema,
-    create: createSchema
+    create: createSchema,
+    multi_edit: multiEditSchema
   };
   const schema = schemaMap[toolName];
   if (!schema) {
@@ -10396,7 +10512,7 @@ function parseXmlToolCall(xmlString, validTools = DEFAULT_VALID_TOOLS) {
   const closeTag = `</${toolName}>`;
   const openIndex = earliestOpenIndex;
   let closeIndex;
-  if (toolName === "attempt_completion") {
+  if (LAST_INDEX_TOOLS.has(toolName)) {
     closeIndex = xmlString.lastIndexOf(closeTag);
     if (closeIndex !== -1 && closeIndex <= openIndex + openTag.length) {
       closeIndex = -1;
@@ -10421,7 +10537,15 @@ function parseXmlToolCall(xmlString, validTools = DEFAULT_VALID_TOOLS) {
     if (paramOpenIndex === -1) {
       continue;
     }
-    let paramCloseIndex = innerContent.indexOf(paramCloseTag, paramOpenIndex + paramOpenTag.length);
+    let paramCloseIndex;
+    if (RAW_CONTENT_PARAMS.has(paramName)) {
+      paramCloseIndex = innerContent.lastIndexOf(paramCloseTag);
+      if (paramCloseIndex !== -1 && paramCloseIndex <= paramOpenIndex + paramOpenTag.length) {
+        paramCloseIndex = -1;
+      }
+    } else {
+      paramCloseIndex = innerContent.indexOf(paramCloseTag, paramOpenIndex + paramOpenTag.length);
+    }
     if (paramCloseIndex === -1) {
       let nextTagIndex = innerContent.length;
       for (const nextParam of validParams) {
@@ -10433,18 +10557,26 @@ function parseXmlToolCall(xmlString, validTools = DEFAULT_VALID_TOOLS) {
       }
       paramCloseIndex = nextTagIndex;
     }
-    let paramValue = unescapeXmlEntities(innerContent.substring(
+    const rawValue = innerContent.substring(
       paramOpenIndex + paramOpenTag.length,
       paramCloseIndex
-    ).trim());
-    if (paramValue.toLowerCase() === "true") {
-      paramValue = true;
-    } else if (paramValue.toLowerCase() === "false") {
-      paramValue = false;
-    } else if (!isNaN(paramValue) && paramValue.trim() !== "") {
-      const num = Number(paramValue);
-      if (Number.isFinite(num)) {
-        paramValue = num;
+    );
+    let paramValue;
+    if (RAW_CONTENT_PARAMS.has(paramName)) {
+      paramValue = unescapeXmlEntities(rawValue.replace(/^\n/, "").replace(/\n$/, ""));
+    } else {
+      paramValue = unescapeXmlEntities(rawValue.trim());
+    }
+    if (!RAW_CONTENT_PARAMS.has(paramName)) {
+      if (paramValue.toLowerCase() === "true") {
+        paramValue = true;
+      } else if (paramValue.toLowerCase() === "false") {
+        paramValue = false;
+      } else if (!isNaN(paramValue) && paramValue.trim() !== "") {
+        const num = Number(paramValue);
+        if (Number.isFinite(num)) {
+          paramValue = num;
+        }
       }
     }
     params[paramName] = paramValue;
@@ -10487,6 +10619,7 @@ function detectUnrecognizedToolCall(xmlString, validTools) {
     "readImage",
     "edit",
     "create",
+    "multi_edit",
     "delegate",
     "bash",
     "task",
@@ -10616,7 +10749,7 @@ function resolveTargetPath(target, cwd) {
   }
   return filePart + suffix;
 }
-var searchSchema, searchAllSchema, querySchema, extractSchema, delegateSchema, listSkillsSchema, useSkillSchema, bashSchema, analyzeAllSchema, executePlanSchema, cleanupExecutePlanSchema, attemptCompletionSchema, searchToolDefinition, queryToolDefinition, extractToolDefinition, delegateToolDefinition, attemptCompletionToolDefinition, analyzeAllToolDefinition, bashToolDefinition, googleSearchToolDefinition, urlContextToolDefinition, searchDescription, queryDescription, extractDescription, delegateDescription, analyzeAllDescription, DEFAULT_VALID_TOOLS;
+var searchSchema, searchAllSchema, querySchema, extractSchema, delegateSchema, listSkillsSchema, useSkillSchema, bashSchema, analyzeAllSchema, executePlanSchema, cleanupExecutePlanSchema, attemptCompletionSchema, searchToolDefinition, queryToolDefinition, extractToolDefinition, delegateToolDefinition, attemptCompletionToolDefinition, analyzeAllToolDefinition, bashToolDefinition, googleSearchToolDefinition, urlContextToolDefinition, searchDescription, queryDescription, extractDescription, delegateDescription, analyzeAllDescription, DEFAULT_VALID_TOOLS, RAW_CONTENT_PARAMS, LAST_INDEX_TOOLS;
 var init_common = __esm({
   "src/tools/common.js"() {
     "use strict";
@@ -11034,6 +11167,8 @@ Capabilities:
       "task",
       "attempt_completion"
     ];
+    RAW_CONTENT_PARAMS = /* @__PURE__ */ new Set(["content", "new_string", "old_string"]);
+    LAST_INDEX_TOOLS = /* @__PURE__ */ new Set(["attempt_completion", "create", "edit"]);
   }
 });
 
@@ -12103,6 +12238,10 @@ function parseSimpleCommand(command) {
     // Logical OR
     /(?<!\\);/,
     // Command separator (but not escaped \;)
+    /\n/,
+    // Newline command separator (multi-line commands)
+    /\r/,
+    // Carriage return (CRLF line endings)
     /&$/,
     // Background execution
     /\$\(/,
@@ -12209,6 +12348,8 @@ function isComplexPattern(pattern) {
     // Logical OR
     /;/,
     // Command separator
+    /\n/,
+    // Newline command separator
     /&$/,
     // Background execution
     /\$\(/,
@@ -12581,6 +12722,26 @@ var init_bashPermissions = __esm({
               i++;
               continue;
             }
+            if (char === "\n" || char === "\r" && nextChar === "\n") {
+              if (current2.trim()) {
+                components.push(current2.trim());
+              }
+              current2 = "";
+              if (char === "\r") {
+                i += 2;
+              } else {
+                i++;
+              }
+              continue;
+            }
+            if (char === "\r") {
+              if (current2.trim()) {
+                components.push(current2.trim());
+              }
+              current2 = "";
+              i++;
+              continue;
+            }
           }
           current2 += char;
           i++;
@@ -24659,7 +24820,11 @@ var init_esm3 = __esm({
       #matchGlobstar(file, pattern, partial, fileIndex, patternIndex) {
         const firstgs = pattern.indexOf(GLOBSTAR, patternIndex);
         const lastgs = pattern.lastIndexOf(GLOBSTAR);
-        const [head2, body, tail] = [
+        const [head2, body, tail] = partial ? [
+          pattern.slice(patternIndex, firstgs),
+          pattern.slice(firstgs + 1),
+          []
+        ] : [
           pattern.slice(patternIndex, firstgs),
           pattern.slice(firstgs + 1, lastgs),
           pattern.slice(lastgs + 1)
@@ -24696,7 +24861,7 @@ var init_esm3 = __esm({
               return false;
             }
           }
-          return sawSome;
+          return partial || sawSome;
         }
         const bodySegments = [[[], 0]];
         let currentBody = bodySegments[0];
@@ -24745,7 +24910,7 @@ var init_esm3 = __esm({
           }
           fileIndex++;
         }
-        return null;
+        return partial || null;
       }
       #matchOne(file, pattern, partial, fileIndex, patternIndex) {
         let fi;
@@ -32036,6 +32201,13 @@ function createWrappedTools(baseTools) {
       baseTools.createTool.execute
     );
   }
+  if (baseTools.multiEditTool) {
+    wrappedTools.multiEditToolInstance = wrapToolWithEmitter(
+      baseTools.multiEditTool,
+      "multi_edit",
+      baseTools.multiEditTool.execute
+    );
+  }
   return wrappedTools;
 }
 var toolCallEmitter, activeToolExecutions, wrapToolWithEmitter, listFilesTool, searchFilesTool, listFilesToolInstance, searchFilesToolInstance;
@@ -32954,6 +33126,9 @@ function createTools(configOptions) {
   if (configOptions.allowEdit && isToolAllowed("create")) {
     tools2.createTool = createTool(configOptions);
   }
+  if (configOptions.allowEdit && isToolAllowed("multi_edit")) {
+    tools2.multiEditTool = multiEditTool(configOptions);
+  }
   return tools2;
 }
 function parseXmlToolCallWithThinking(xmlString, validTools) {
@@ -69965,6 +70140,15 @@ function normalizeJsonQuotes(str) {
   }
   return result;
 }
+function isCodeBlockEmbeddedInDocument(text, match2) {
+  const beforeBlock = text.substring(0, match2.index).trim();
+  const afterBlock = text.substring(match2.index + match2[0].length).trim();
+  const hasMarkdownHeadings = /^#{1,6}\s/m.test(beforeBlock) || /^#{1,6}\s/m.test(afterBlock);
+  if (hasMarkdownHeadings) {
+    return true;
+  }
+  return false;
+}
 function cleanSchemaResponse(response) {
   if (!response || typeof response !== "string") {
     return response;
@@ -69984,11 +70168,11 @@ function cleanSchemaResponse(response) {
     return cleanSchemaResponse(innerContent);
   }
   const jsonBlockMatch = trimmed.match(/```json\s*\n([\s\S]*?)\n```/);
-  if (jsonBlockMatch) {
+  if (jsonBlockMatch && !isCodeBlockEmbeddedInDocument(trimmed, jsonBlockMatch)) {
     return normalizeJsonQuotes(jsonBlockMatch[1].trim());
   }
   const anyBlockMatch = trimmed.match(/```\s*\n([{\[][\s\S]*?[}\]])\s*```/);
-  if (anyBlockMatch) {
+  if (anyBlockMatch && !isCodeBlockEmbeddedInDocument(trimmed, anyBlockMatch)) {
     return normalizeJsonQuotes(anyBlockMatch[1].trim());
   }
   const codeBlockPatterns = [
@@ -69997,7 +70181,7 @@ function cleanSchemaResponse(response) {
   ];
   for (const pattern of codeBlockPatterns) {
     const match2 = trimmed.match(pattern);
-    if (match2) {
+    if (match2 && !isCodeBlockEmbeddedInDocument(trimmed, match2)) {
       return normalizeJsonQuotes(match2[1].trim());
     }
   }
@@ -70008,7 +70192,7 @@ function cleanSchemaResponse(response) {
   }
   const codeBlockStartPattern = /```(?:json)?\s*\n?\s*([{\[])/;
   const codeBlockMatch = trimmed.match(codeBlockStartPattern);
-  if (codeBlockMatch) {
+  if (codeBlockMatch && !isCodeBlockEmbeddedInDocument(trimmed, codeBlockMatch)) {
     const startIndex = codeBlockMatch.index + codeBlockMatch[0].length - 1;
     const openChar = codeBlockMatch[1];
     const closeChar = openChar === "{" ? "}" : "]";
@@ -83452,6 +83636,9 @@ var init_ProbeAgent = __esm({
           if (wrappedTools.createToolInstance && isToolAllowed("create")) {
             this.toolImplementations.create = wrappedTools.createToolInstance;
           }
+          if (wrappedTools.multiEditToolInstance && isToolAllowed("multi_edit")) {
+            this.toolImplementations.multi_edit = wrappedTools.multiEditToolInstance;
+          }
         }
         this.wrappedTools = wrappedTools;
         if (this.debug) {
@@ -84741,6 +84928,10 @@ Workspace: ${this.allowedFolders.join(", ")}`;
         }
         if (this.allowEdit && isToolAllowed("create")) {
           toolDefinitions += `${createToolDefinition}
+`;
+        }
+        if (this.allowEdit && isToolAllowed("multi_edit")) {
+          toolDefinitions += `${multiEditToolDefinition}
 `;
         }
         if (this.enableBash && isToolAllowed("bash")) {
@@ -84837,6 +85028,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";
         }
@@ -85431,6 +85625,9 @@ You are working with a workspace. Available paths: ${workspaceDesc}
               if (this.allowEdit && this.allowedTools.isEnabled("create")) {
                 validTools.push("create");
               }
+              if (this.allowEdit && this.allowedTools.isEnabled("multi_edit")) {
+                validTools.push("multi_edit");
+              }
               if (this.enableBash && this.allowedTools.isEnabled("bash")) {
                 validTools.push("bash");
               }

package/build/agent/probeTool.js

@@ -256,6 +256,15 @@ export function createWrappedTools(baseTools) {
     );
   }
 
+  // Wrap multi_edit tool
+  if (baseTools.multiEditTool) {
+    wrappedTools.multiEditToolInstance = wrapToolWithEmitter(
+      baseTools.multiEditTool,
+      'multi_edit',
+      baseTools.multiEditTool.execute
+    );
+  }
+
   return wrappedTools;
 }
 

package/build/agent/schemaUtils.js

@@ -266,6 +266,36 @@ function normalizeJsonQuotes(str) {
   return result;
 }
 
+/**
+ * Check if a code block is embedded within a structured markdown document.
+ * Used to avoid extracting embedded JSON examples from markdown text.
+ * Issue #456: attempt_completion results containing markdown with JSON code block
+ * documentation examples were having the examples extracted as structured output.
+ *
+ * Normal AI behavior: AI wraps JSON answer in a code block with brief explanation text.
+ * Issue #456: AI returns a markdown document that contains JSON code blocks as examples.
+ *
+ * We distinguish these by checking if the surrounding text contains markdown structural
+ * elements (headings) which indicate a document rather than a brief explanation.
+ *
+ * @param {string} text - The full trimmed text
+ * @param {RegExpMatchArray} match - The regex match for the code block
+ * @returns {boolean} - true if the code block is embedded in a markdown document
+ */
+function isCodeBlockEmbeddedInDocument(text, match) {
+  const beforeBlock = text.substring(0, match.index).trim();
+  const afterBlock = text.substring(match.index + match[0].length).trim();
+
+  // Check if surrounding text contains markdown headings (lines starting with #)
+  // This is a strong signal that the content is a structured document, not just explanation text
+  const hasMarkdownHeadings = /^#{1,6}\s/m.test(beforeBlock) || /^#{1,6}\s/m.test(afterBlock);
+  if (hasMarkdownHeadings) {
+    return true;
+  }
+
+  return false;
+}
+
 /**
  * Clean AI response by extracting JSON content when response contains JSON
  * Only processes responses that contain JSON structures { or [
@@ -311,15 +341,20 @@ export function cleanSchemaResponse(response) {
   }
 
   // First, look for JSON after code block markers - similar to mermaid extraction
+  // Only extract from code blocks when they are the primary content (not embedded examples in markdown).
+  // Issue #456: When attempt_completion result contains markdown with embedded JSON code blocks
+  // as documentation examples, extracting those blocks produces wrong structured output.
+  // We check that there's no significant text content outside the code block.
+
   // Try with json language specifier
   const jsonBlockMatch = trimmed.match(/```json\s*\n([\s\S]*?)\n```/);
-  if (jsonBlockMatch) {
+  if (jsonBlockMatch && !isCodeBlockEmbeddedInDocument(trimmed, jsonBlockMatch)) {
     return normalizeJsonQuotes(jsonBlockMatch[1].trim());
   }
 
   // Try any code block with JSON content
   const anyBlockMatch = trimmed.match(/```\s*\n([{\[][\s\S]*?[}\]])\s*```/);
-  if (anyBlockMatch) {
+  if (anyBlockMatch && !isCodeBlockEmbeddedInDocument(trimmed, anyBlockMatch)) {
     return normalizeJsonQuotes(anyBlockMatch[1].trim());
   }
 
@@ -331,7 +366,7 @@ export function cleanSchemaResponse(response) {
 
   for (const pattern of codeBlockPatterns) {
     const match = trimmed.match(pattern);
-    if (match) {
+    if (match && !isCodeBlockEmbeddedInDocument(trimmed, match)) {
       return normalizeJsonQuotes(match[1].trim());
     }
   }
@@ -345,10 +380,11 @@ export function cleanSchemaResponse(response) {
   }
 
   // Look for code block start followed immediately by JSON
+  // Only extract if the code block is not embedded in a structured markdown document
   const codeBlockStartPattern = /```(?:json)?\s*\n?\s*([{\[])/;
   const codeBlockMatch = trimmed.match(codeBlockStartPattern);
 
-  if (codeBlockMatch) {
+  if (codeBlockMatch && !isCodeBlockEmbeddedInDocument(trimmed, codeBlockMatch)) {
     const startIndex = codeBlockMatch.index + codeBlockMatch[0].length - 1; // Position of the bracket
 
     // Find the matching closing bracket