@probelabs/probe 0.6.0-rc108 → 0.6.0-rc109

package/cjs/agent/ProbeAgent.cjs

@@ -10236,9 +10236,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);
@@ -10305,7 +10306,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]);
@@ -11847,7 +11849,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;
@@ -11976,7 +11979,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()) {
@@ -13388,7 +13392,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
         };
@@ -13850,7 +13855,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()) {
@@ -17801,7 +17807,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",
@@ -17820,23 +17826,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",
@@ -17844,21 +17850,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",
@@ -18650,7 +18656,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",
@@ -18669,37 +18675,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",
@@ -19931,7 +19937,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",
@@ -19960,37 +19966,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",
@@ -24460,6 +24466,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"
     };
@@ -25389,6 +25396,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);
@@ -25404,9 +25414,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);
@@ -28857,9 +28864,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,
@@ -28901,9 +28909,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,
@@ -31125,13 +31141,42 @@ var init_index = __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*$/,