@probelabs/probe 0.6.0-rc107 → 0.6.0-rc109

package/cjs/index.cjs

@@ -1145,9 +1145,10 @@ function parseXmlToolCall(xmlString, validTools = DEFAULT_VALID_TOOLS) {
     if (openIndex === -1) {
       continue;
     }
-    const closeIndex = xmlString.indexOf(closeTag, openIndex + openTag.length);
+    let closeIndex = xmlString.indexOf(closeTag, openIndex + openTag.length);
+    let hasClosingTag = closeIndex !== -1;
     if (closeIndex === -1) {
-      continue;
+      closeIndex = xmlString.length;
     }
     const innerContent = xmlString.substring(
       openIndex + openTag.length,
@@ -1189,9 +1190,17 @@ function parseXmlToolCall(xmlString, validTools = DEFAULT_VALID_TOOLS) {
       if (paramOpenIndex === -1) {
         continue;
       }
-      const paramCloseIndex = innerContent.indexOf(paramCloseTag, paramOpenIndex + paramOpenTag.length);
+      let paramCloseIndex = innerContent.indexOf(paramCloseTag, paramOpenIndex + paramOpenTag.length);
       if (paramCloseIndex === -1) {
-        continue;
+        let nextTagIndex = innerContent.length;
+        for (const nextParam of commonParams3) {
+          const nextOpenTag = `<${nextParam}>`;
+          const nextIndex = innerContent.indexOf(nextOpenTag, paramOpenIndex + paramOpenTag.length);
+          if (nextIndex !== -1 && nextIndex < nextTagIndex) {
+            nextTagIndex = nextIndex;
+          }
+        }
+        paramCloseIndex = nextTagIndex;
       }
       let paramValue = innerContent.substring(
         paramOpenIndex + paramOpenTag.length,
@@ -13753,9 +13762,10 @@ var init_ProtocolLib = __esm({
           return { errorSchema, errorMetadata };
         } catch (e3) {
           dataObject.message = dataObject.message ?? dataObject.Message ?? "UnknownError";
-          const baseExceptionSchema = TypeRegistry.for("smithy.ts.sdk.synthetic." + namespace).getBaseException();
+          const synthetic = TypeRegistry.for("smithy.ts.sdk.synthetic." + namespace);
+          const baseExceptionSchema = synthetic.getBaseException();
           if (baseExceptionSchema) {
-            const ErrorCtor = baseExceptionSchema.ctor;
+            const ErrorCtor = synthetic.getErrorCtor(baseExceptionSchema) ?? Error;
             throw Object.assign(new ErrorCtor({ name: errorName }), errorMetadata, dataObject);
           }
           throw Object.assign(new Error(errorName), errorMetadata, dataObject);
@@ -13822,7 +13832,8 @@ var init_AwsSmithyRpcV2CborProtocol = __esm({
         const { errorSchema, errorMetadata } = await this.mixin.getErrorSchemaOrThrowBaseException(errorName, this.options.defaultNamespace, response, dataObject, metadata);
         const ns = NormalizedSchema.of(errorSchema);
         const message = dataObject.message ?? dataObject.Message ?? "Unknown";
-        const exception = new errorSchema.ctor(message);
+        const ErrorCtor = TypeRegistry.for(errorSchema.namespace).getErrorCtor(errorSchema) ?? Error;
+        const exception = new ErrorCtor(message);
         const output = {};
         for (const [name14, member] of ns.structIterator()) {
           output[name14] = this.deserializer.readValue(member, dataObject[name14]);
@@ -15364,7 +15375,8 @@ var init_AwsJsonRpcProtocol = __esm({
         const { errorSchema, errorMetadata } = await this.mixin.getErrorSchemaOrThrowBaseException(errorIdentifier, this.options.defaultNamespace, response, dataObject, metadata);
         const ns = NormalizedSchema.of(errorSchema);
         const message = dataObject.message ?? dataObject.Message ?? "Unknown";
-        const exception = new errorSchema.ctor(message);
+        const ErrorCtor = TypeRegistry.for(errorSchema.namespace).getErrorCtor(errorSchema) ?? Error;
+        const exception = new ErrorCtor(message);
         const output = {};
         for (const [name14, member] of ns.structIterator()) {
           const target = member.getMergedTraits().jsonName ?? name14;
@@ -15493,7 +15505,8 @@ var init_AwsRestJsonProtocol = __esm({
         const { errorSchema, errorMetadata } = await this.mixin.getErrorSchemaOrThrowBaseException(errorIdentifier, this.options.defaultNamespace, response, dataObject, metadata);
         const ns = NormalizedSchema.of(errorSchema);
         const message = dataObject.message ?? dataObject.Message ?? "Unknown";
-        const exception = new errorSchema.ctor(message);
+        const ErrorCtor = TypeRegistry.for(errorSchema.namespace).getErrorCtor(errorSchema) ?? Error;
+        const exception = new ErrorCtor(message);
         await this.deserializeHttpMessage(errorSchema, context, response, dataObject);
         const output = {};
         for (const [name14, member] of ns.structIterator()) {
@@ -16905,7 +16918,8 @@ var init_AwsQueryProtocol = __esm({
         };
         const { errorSchema, errorMetadata } = await this.mixin.getErrorSchemaOrThrowBaseException(errorIdentifier, this.options.defaultNamespace, response, errorData, metadata, (registry, errorName) => registry.find((schema) => NormalizedSchema.of(schema).getMergedTraits().awsQueryError?.[0] === errorName));
         const ns = NormalizedSchema.of(errorSchema);
-        const exception = new errorSchema.ctor(message);
+        const ErrorCtor = TypeRegistry.for(errorSchema.namespace).getErrorCtor(errorSchema) ?? Error;
+        const exception = new ErrorCtor(message);
         const output = {
           Error: errorData.Error
         };
@@ -17367,7 +17381,8 @@ var init_AwsRestXmlProtocol = __esm({
         const { errorSchema, errorMetadata } = await this.mixin.getErrorSchemaOrThrowBaseException(errorIdentifier, this.options.defaultNamespace, response, dataObject, metadata);
         const ns = NormalizedSchema.of(errorSchema);
         const message = dataObject.Error?.message ?? dataObject.Error?.Message ?? dataObject.message ?? dataObject.Message ?? "Unknown";
-        const exception = new errorSchema.ctor(message);
+        const ErrorCtor = TypeRegistry.for(errorSchema.namespace).getErrorCtor(errorSchema) ?? Error;
+        const exception = new ErrorCtor(message);
         await this.deserializeHttpMessage(errorSchema, context, response, dataObject);
         const output = {};
         for (const [name14, member] of ns.structIterator()) {
@@ -21318,7 +21333,7 @@ var require_package = __commonJS({
     module2.exports = {
       name: "@aws-sdk/client-bedrock-runtime",
       description: "AWS SDK for JavaScript Bedrock Runtime Client for Node.js, Browser and React Native",
-      version: "3.896.0",
+      version: "3.899.0",
       scripts: {
         build: "concurrently 'yarn:build:cjs' 'yarn:build:es' 'yarn:build:types'",
         "build:cjs": "node ../../scripts/compilation/inline client-bedrock-runtime",
@@ -21337,23 +21352,23 @@ var require_package = __commonJS({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.896.0",
-        "@aws-sdk/credential-provider-node": "3.896.0",
+        "@aws-sdk/core": "3.899.0",
+        "@aws-sdk/credential-provider-node": "3.899.0",
         "@aws-sdk/eventstream-handler-node": "3.893.0",
         "@aws-sdk/middleware-eventstream": "3.893.0",
         "@aws-sdk/middleware-host-header": "3.893.0",
         "@aws-sdk/middleware-logger": "3.893.0",
         "@aws-sdk/middleware-recursion-detection": "3.893.0",
-        "@aws-sdk/middleware-user-agent": "3.896.0",
+        "@aws-sdk/middleware-user-agent": "3.899.0",
         "@aws-sdk/middleware-websocket": "3.893.0",
         "@aws-sdk/region-config-resolver": "3.893.0",
-        "@aws-sdk/token-providers": "3.896.0",
+        "@aws-sdk/token-providers": "3.899.0",
         "@aws-sdk/types": "3.893.0",
         "@aws-sdk/util-endpoints": "3.895.0",
         "@aws-sdk/util-user-agent-browser": "3.893.0",
-        "@aws-sdk/util-user-agent-node": "3.896.0",
+        "@aws-sdk/util-user-agent-node": "3.899.0",
         "@smithy/config-resolver": "^4.2.2",
-        "@smithy/core": "^3.12.0",
+        "@smithy/core": "^3.13.0",
         "@smithy/eventstream-serde-browser": "^4.1.1",
         "@smithy/eventstream-serde-config-resolver": "^4.2.1",
         "@smithy/eventstream-serde-node": "^4.1.1",
@@ -21361,21 +21376,21 @@ var require_package = __commonJS({
         "@smithy/hash-node": "^4.1.1",
         "@smithy/invalid-dependency": "^4.1.1",
         "@smithy/middleware-content-length": "^4.1.1",
-        "@smithy/middleware-endpoint": "^4.2.4",
-        "@smithy/middleware-retry": "^4.3.0",
+        "@smithy/middleware-endpoint": "^4.2.5",
+        "@smithy/middleware-retry": "^4.3.1",
         "@smithy/middleware-serde": "^4.1.1",
         "@smithy/middleware-stack": "^4.1.1",
         "@smithy/node-config-provider": "^4.2.2",
         "@smithy/node-http-handler": "^4.2.1",
         "@smithy/protocol-http": "^5.2.1",
-        "@smithy/smithy-client": "^4.6.4",
+        "@smithy/smithy-client": "^4.6.5",
         "@smithy/types": "^4.5.0",
         "@smithy/url-parser": "^4.1.1",
         "@smithy/util-base64": "^4.1.0",
         "@smithy/util-body-length-browser": "^4.1.0",
         "@smithy/util-body-length-node": "^4.1.0",
-        "@smithy/util-defaults-mode-browser": "^4.1.4",
-        "@smithy/util-defaults-mode-node": "^4.1.4",
+        "@smithy/util-defaults-mode-browser": "^4.1.5",
+        "@smithy/util-defaults-mode-node": "^4.1.5",
         "@smithy/util-endpoints": "^3.1.2",
         "@smithy/util-middleware": "^4.1.1",
         "@smithy/util-retry": "^4.1.2",
@@ -22167,7 +22182,7 @@ var require_package2 = __commonJS({
     module2.exports = {
       name: "@aws-sdk/client-sso",
       description: "AWS SDK for JavaScript Sso Client for Node.js, Browser and React Native",
-      version: "3.896.0",
+      version: "3.899.0",
       scripts: {
         build: "concurrently 'yarn:build:cjs' 'yarn:build:es' 'yarn:build:types'",
         "build:cjs": "node ../../scripts/compilation/inline client-sso",
@@ -22186,37 +22201,37 @@ var require_package2 = __commonJS({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.896.0",
+        "@aws-sdk/core": "3.899.0",
         "@aws-sdk/middleware-host-header": "3.893.0",
         "@aws-sdk/middleware-logger": "3.893.0",
         "@aws-sdk/middleware-recursion-detection": "3.893.0",
-        "@aws-sdk/middleware-user-agent": "3.896.0",
+        "@aws-sdk/middleware-user-agent": "3.899.0",
         "@aws-sdk/region-config-resolver": "3.893.0",
         "@aws-sdk/types": "3.893.0",
         "@aws-sdk/util-endpoints": "3.895.0",
         "@aws-sdk/util-user-agent-browser": "3.893.0",
-        "@aws-sdk/util-user-agent-node": "3.896.0",
+        "@aws-sdk/util-user-agent-node": "3.899.0",
         "@smithy/config-resolver": "^4.2.2",
-        "@smithy/core": "^3.12.0",
+        "@smithy/core": "^3.13.0",
         "@smithy/fetch-http-handler": "^5.2.1",
         "@smithy/hash-node": "^4.1.1",
         "@smithy/invalid-dependency": "^4.1.1",
         "@smithy/middleware-content-length": "^4.1.1",
-        "@smithy/middleware-endpoint": "^4.2.4",
-        "@smithy/middleware-retry": "^4.3.0",
+        "@smithy/middleware-endpoint": "^4.2.5",
+        "@smithy/middleware-retry": "^4.3.1",
         "@smithy/middleware-serde": "^4.1.1",
         "@smithy/middleware-stack": "^4.1.1",
         "@smithy/node-config-provider": "^4.2.2",
         "@smithy/node-http-handler": "^4.2.1",
         "@smithy/protocol-http": "^5.2.1",
-        "@smithy/smithy-client": "^4.6.4",
+        "@smithy/smithy-client": "^4.6.5",
         "@smithy/types": "^4.5.0",
         "@smithy/url-parser": "^4.1.1",
         "@smithy/util-base64": "^4.1.0",
         "@smithy/util-body-length-browser": "^4.1.0",
         "@smithy/util-body-length-node": "^4.1.0",
-        "@smithy/util-defaults-mode-browser": "^4.1.4",
-        "@smithy/util-defaults-mode-node": "^4.1.4",
+        "@smithy/util-defaults-mode-browser": "^4.1.5",
+        "@smithy/util-defaults-mode-node": "^4.1.5",
         "@smithy/util-endpoints": "^3.1.2",
         "@smithy/util-middleware": "^4.1.1",
         "@smithy/util-retry": "^4.1.2",
@@ -23448,7 +23463,7 @@ var init_package = __esm({
   "node_modules/@aws-sdk/nested-clients/package.json"() {
     package_default = {
       name: "@aws-sdk/nested-clients",
-      version: "3.896.0",
+      version: "3.899.0",
       description: "Nested clients for AWS SDK packages.",
       main: "./dist-cjs/index.js",
       module: "./dist-es/index.js",
@@ -23477,37 +23492,37 @@ var init_package = __esm({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.896.0",
+        "@aws-sdk/core": "3.899.0",
         "@aws-sdk/middleware-host-header": "3.893.0",
         "@aws-sdk/middleware-logger": "3.893.0",
         "@aws-sdk/middleware-recursion-detection": "3.893.0",
-        "@aws-sdk/middleware-user-agent": "3.896.0",
+        "@aws-sdk/middleware-user-agent": "3.899.0",
         "@aws-sdk/region-config-resolver": "3.893.0",
         "@aws-sdk/types": "3.893.0",
         "@aws-sdk/util-endpoints": "3.895.0",
         "@aws-sdk/util-user-agent-browser": "3.893.0",
-        "@aws-sdk/util-user-agent-node": "3.896.0",
+        "@aws-sdk/util-user-agent-node": "3.899.0",
         "@smithy/config-resolver": "^4.2.2",
-        "@smithy/core": "^3.12.0",
+        "@smithy/core": "^3.13.0",
         "@smithy/fetch-http-handler": "^5.2.1",
         "@smithy/hash-node": "^4.1.1",
         "@smithy/invalid-dependency": "^4.1.1",
         "@smithy/middleware-content-length": "^4.1.1",
-        "@smithy/middleware-endpoint": "^4.2.4",
-        "@smithy/middleware-retry": "^4.3.0",
+        "@smithy/middleware-endpoint": "^4.2.5",
+        "@smithy/middleware-retry": "^4.3.1",
         "@smithy/middleware-serde": "^4.1.1",
         "@smithy/middleware-stack": "^4.1.1",
         "@smithy/node-config-provider": "^4.2.2",
         "@smithy/node-http-handler": "^4.2.1",
         "@smithy/protocol-http": "^5.2.1",
-        "@smithy/smithy-client": "^4.6.4",
+        "@smithy/smithy-client": "^4.6.5",
         "@smithy/types": "^4.5.0",
         "@smithy/url-parser": "^4.1.1",
         "@smithy/util-base64": "^4.1.0",
         "@smithy/util-body-length-browser": "^4.1.0",
         "@smithy/util-body-length-node": "^4.1.0",
-        "@smithy/util-defaults-mode-browser": "^4.1.4",
-        "@smithy/util-defaults-mode-node": "^4.1.4",
+        "@smithy/util-defaults-mode-browser": "^4.1.5",
+        "@smithy/util-defaults-mode-node": "^4.1.5",
         "@smithy/util-endpoints": "^3.1.2",
         "@smithy/util-middleware": "^4.1.1",
         "@smithy/util-retry": "^4.1.2",
@@ -27977,6 +27992,7 @@ var require_dist_cjs65 = __commonJS({
       END_TURN: "end_turn",
       GUARDRAIL_INTERVENED: "guardrail_intervened",
       MAX_TOKENS: "max_tokens",
+      MODEL_CONTEXT_WINDOW_EXCEEDED: "model_context_window_exceeded",
       STOP_SEQUENCE: "stop_sequence",
       TOOL_USE: "tool_use"
     };
@@ -28906,6 +28922,9 @@ var require_dist_cjs65 = __commonJS({
         case "ServiceQuotaExceededException":
         case "com.amazonaws.bedrockruntime#ServiceQuotaExceededException":
           throw await de_ServiceQuotaExceededExceptionRes(parsedOutput, context);
+        case "ServiceUnavailableException":
+        case "com.amazonaws.bedrockruntime#ServiceUnavailableException":
+          throw await de_ServiceUnavailableExceptionRes(parsedOutput, context);
         case "ThrottlingException":
         case "com.amazonaws.bedrockruntime#ThrottlingException":
           throw await de_ThrottlingExceptionRes(parsedOutput, context);
@@ -28921,9 +28940,6 @@ var require_dist_cjs65 = __commonJS({
         case "ModelTimeoutException":
         case "com.amazonaws.bedrockruntime#ModelTimeoutException":
           throw await de_ModelTimeoutExceptionRes(parsedOutput, context);
-        case "ServiceUnavailableException":
-        case "com.amazonaws.bedrockruntime#ServiceUnavailableException":
-          throw await de_ServiceUnavailableExceptionRes(parsedOutput, context);
         case "ModelStreamErrorException":
         case "com.amazonaws.bedrockruntime#ModelStreamErrorException":
           throw await de_ModelStreamErrorExceptionRes(parsedOutput, context);
@@ -31270,13 +31286,42 @@ var init_tokenCounter = __esm({
 
 // src/agent/xmlParsingUtils.js
 function removeThinkingTags(xmlString) {
-  return xmlString.replace(/<thinking>[\s\S]*?<\/thinking>/g, "").trim();
+  let result = xmlString;
+  result = result.replace(/<thinking>[\s\S]*?<\/thinking>/g, "");
+  const thinkingIndex = result.indexOf("<thinking>");
+  if (thinkingIndex !== -1) {
+    const afterThinking = result.substring(thinkingIndex + "<thinking>".length);
+    const toolPattern = /<(search|query|extract|listFiles|searchFiles|implement|attempt_completion|attempt_complete)>/;
+    const toolMatch = afterThinking.match(toolPattern);
+    if (toolMatch) {
+      const toolStart = thinkingIndex + "<thinking>".length + toolMatch.index;
+      result = result.substring(0, thinkingIndex) + result.substring(toolStart);
+    } else {
+      result = result.substring(0, thinkingIndex);
+    }
+  }
+  return result.trim();
 }
 function extractThinkingContent(xmlString) {
   const thinkingMatch = xmlString.match(/<thinking>([\s\S]*?)<\/thinking>/);
   return thinkingMatch ? thinkingMatch[1].trim() : null;
 }
 function checkAttemptCompleteRecovery(cleanedXmlString, validTools = []) {
+  const attemptCompletionMatch = cleanedXmlString.match(/<attempt_completion>([\s\S]*?)(?:<\/attempt_completion>|$)/);
+  if (attemptCompletionMatch) {
+    const content = attemptCompletionMatch[1].trim();
+    const hasClosingTag = cleanedXmlString.includes("</attempt_completion>");
+    if (content) {
+      return {
+        toolName: "attempt_completion",
+        params: { result: content }
+      };
+    }
+    return {
+      toolName: "attempt_completion",
+      params: { result: hasClosingTag ? "" : "__PREVIOUS_RESPONSE__" }
+    };
+  }
   const attemptCompletePatterns = [
     // Standard shorthand with optional whitespace
     /^<attempt_complete>\s*$/,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc107",
+  "version": "0.6.0-rc109",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",
@@ -72,8 +72,6 @@
     "@ai-sdk/anthropic": "^2.0.8",
     "@ai-sdk/google": "^2.0.14",
     "@ai-sdk/openai": "^2.0.10",
-    "@mermaid-js/mermaid-cli": "^11.12.0",
-    "@mermaid-js/parser": "^0.6.2",
     "@modelcontextprotocol/sdk": "^1.0.0",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/core": "^1.30.1",
@@ -84,14 +82,8 @@
     "@opentelemetry/semantic-conventions": "^1.36.0",
     "ai": "^5.0.0",
     "axios": "^1.8.3",
-    "chalk": "^5.6.2",
-    "dompurify": "^3.2.7",
-    "dotenv": "^16.4.7",
     "fs-extra": "^11.1.1",
     "glob": "^10.3.10",
-    "isomorphic-dompurify": "^2.28.0",
-    "jsdom": "^27.0.0",
-    "mermaid": "^11.12.0",
     "tar": "^6.2.0",
     "tiktoken": "^1.0.20",
     "zod": "^3.24.2"

package/src/agent/xmlParsingUtils.js

@@ -5,11 +5,39 @@
 
 /**
  * Remove thinking tags and their content from XML string
+ * Handles both closed and unclosed thinking tags
  * @param {string} xmlString - The XML string to clean
  * @returns {string} - Cleaned XML string without thinking tags
  */
 export function removeThinkingTags(xmlString) {
-  return xmlString.replace(/<thinking>[\s\S]*?<\/thinking>/g, '').trim();
+  let result = xmlString;
+
+  // Remove all properly closed thinking tags first
+  result = result.replace(/<thinking>[\s\S]*?<\/thinking>/g, '');
+
+  // Handle unclosed thinking tags
+  // Find any remaining <thinking> tag (which means it's unclosed)
+  const thinkingIndex = result.indexOf('<thinking>');
+  if (thinkingIndex !== -1) {
+    // Check if there's a tool tag after the thinking tag
+    // We want to preserve tool tags even if they're after unclosed thinking
+    const afterThinking = result.substring(thinkingIndex + '<thinking>'.length);
+
+    // Look for any tool tags in the remaining content
+    const toolPattern = /<(search|query|extract|listFiles|searchFiles|implement|attempt_completion|attempt_complete)>/;
+    const toolMatch = afterThinking.match(toolPattern);
+
+    if (toolMatch) {
+      // Found a tool tag - remove thinking tag and its content up to the tool tag
+      const toolStart = thinkingIndex + '<thinking>'.length + toolMatch.index;
+      result = result.substring(0, thinkingIndex) + result.substring(toolStart);
+    } else {
+      // No tool tag found - remove everything from <thinking> onwards
+      result = result.substring(0, thinkingIndex);
+    }
+  }
+
+  return result.trim();
 }
 
 /**
@@ -29,6 +57,30 @@ export function extractThinkingContent(xmlString) {
  * @returns {Object|null} - Standardized attempt_completion result or null
  */
 export function checkAttemptCompleteRecovery(cleanedXmlString, validTools = []) {
+  // Check for <attempt_completion> with content (with or without closing tag)
+  // This handles: "<attempt_completion>content" or "<attempt_completion>content</attempt_completion>"
+  const attemptCompletionMatch = cleanedXmlString.match(/<attempt_completion>([\s\S]*?)(?:<\/attempt_completion>|$)/);
+  if (attemptCompletionMatch) {
+    const content = attemptCompletionMatch[1].trim();
+    const hasClosingTag = cleanedXmlString.includes('</attempt_completion>');
+
+    if (content) {
+      // If there's content after the tag, use it as the result
+      return {
+        toolName: 'attempt_completion',
+        params: { result: content }
+      };
+    }
+
+    // If the tag exists but is empty:
+    // - With closing tag (e.g., "<attempt_completion></attempt_completion>"): use empty string
+    // - Without closing tag (e.g., "<attempt_completion>"): use previous response
+    return {
+      toolName: 'attempt_completion',
+      params: { result: hasClosingTag ? '' : '__PREVIOUS_RESPONSE__' }
+    };
+  }
+
   // Enhanced recovery logic for attempt_complete shorthand
   const attemptCompletePatterns = [
     // Standard shorthand with optional whitespace
@@ -61,7 +113,7 @@ export function checkAttemptCompleteRecovery(cleanedXmlString, validTools = [])
   if (cleanedXmlString.includes('<attempt_complete') && !hasOtherToolTags(cleanedXmlString, validTools)) {
     // This handles malformed cases where attempt_complete appears but is broken
     return {
-      toolName: 'attempt_completion', 
+      toolName: 'attempt_completion',
       params: { result: '__PREVIOUS_RESPONSE__' }
     };
   }

package/src/tools/common.js

@@ -364,47 +364,62 @@ export function parseXmlToolCall(xmlString, validTools = DEFAULT_VALID_TOOLS) {
 	for (const toolName of validTools) {
 		const openTag = `<${toolName}>`;
 		const closeTag = `</${toolName}>`;
-		
+
 		const openIndex = xmlString.indexOf(openTag);
 		if (openIndex === -1) {
 			continue; // Tool not found, try next tool
 		}
-		
-		const closeIndex = xmlString.indexOf(closeTag, openIndex + openTag.length);
+
+		let closeIndex = xmlString.indexOf(closeTag, openIndex + openTag.length);
+		let hasClosingTag = closeIndex !== -1;
+
+		// If no closing tag found, use content until end of string
+		// This makes the parser more resilient to AI formatting errors
 		if (closeIndex === -1) {
-			continue; // No closing tag found, try next tool
+			closeIndex = xmlString.length;
 		}
-		
-		// Extract the content between tags
+
+		// Extract the content between tags (or until end if no closing tag)
 		const innerContent = xmlString.substring(
-			openIndex + openTag.length, 
+			openIndex + openTag.length,
 			closeIndex
 		);
-		
+
 		const params = {};
 
 		// Parse parameters using string-based approach for better safety
 		// Common parameter names to look for (can be extended as needed)
 		// Note: includes both camelCase and underscore_case variants to handle inconsistencies
-		const commonParams = ['query', 'file_path', 'line', 'end_line', 'path', 'recursive', 'includeHidden', 
+		const commonParams = ['query', 'file_path', 'line', 'end_line', 'path', 'recursive', 'includeHidden',
 		                      'max_results', 'maxResults', 'result', 'command', 'description', 'task', 'param', 'pattern',
 		                      'allow_tests', 'exact', 'maxTokens', 'language', 'input_content',
 		                      'context_lines', 'format', 'directory', 'autoCommits', 'files', 'targets'];
-		
+
 		for (const paramName of commonParams) {
 			const paramOpenTag = `<${paramName}>`;
 			const paramCloseTag = `</${paramName}>`;
-			
+
 			const paramOpenIndex = innerContent.indexOf(paramOpenTag);
 			if (paramOpenIndex === -1) {
 				continue; // Parameter not found
 			}
-			
-			const paramCloseIndex = innerContent.indexOf(paramCloseTag, paramOpenIndex + paramOpenTag.length);
+
+			let paramCloseIndex = innerContent.indexOf(paramCloseTag, paramOpenIndex + paramOpenTag.length);
+
+			// Handle unclosed parameter tags - use content until next tag or end of content
 			if (paramCloseIndex === -1) {
-				continue; // No closing tag found
+				// Find the next opening tag after this parameter
+				let nextTagIndex = innerContent.length;
+				for (const nextParam of commonParams) {
+					const nextOpenTag = `<${nextParam}>`;
+					const nextIndex = innerContent.indexOf(nextOpenTag, paramOpenIndex + paramOpenTag.length);
+					if (nextIndex !== -1 && nextIndex < nextTagIndex) {
+						nextTagIndex = nextIndex;
+					}
+				}
+				paramCloseIndex = nextTagIndex;
 			}
-			
+
 			let paramValue = innerContent.substring(
 				paramOpenIndex + paramOpenTag.length,
 				paramCloseIndex