@probelabs/probe 0.6.0-rc169 → 0.6.0-rc170

package/cjs/agent/ProbeAgent.cjs

@@ -11080,6 +11080,50 @@ var init_ConfigurableSerdeContext = __esm({
   }
 });
 
+// node_modules/@aws-sdk/core/dist-es/submodules/protocols/structIterator.js
+function* serializingStructIterator(ns, sourceObject) {
+  if (ns.isUnitSchema()) {
+    return;
+  }
+  const struct2 = ns.getSchema();
+  for (let i4 = 0; i4 < struct2[4].length; ++i4) {
+    const key = struct2[4][i4];
+    const memberNs = new NormalizedSchema([struct2[5][i4], 0], key);
+    if (!(key in sourceObject) && !memberNs.isIdempotencyToken()) {
+      continue;
+    }
+    yield [key, memberNs];
+  }
+}
+function* deserializingStructIterator(ns, sourceObject, nameTrait) {
+  if (ns.isUnitSchema()) {
+    return;
+  }
+  const struct2 = ns.getSchema();
+  let keysRemaining = Object.keys(sourceObject).length;
+  for (let i4 = 0; i4 < struct2[4].length; ++i4) {
+    if (keysRemaining === 0) {
+      break;
+    }
+    const key = struct2[4][i4];
+    const memberNs = new NormalizedSchema([struct2[5][i4], 0], key);
+    let serializationKey = key;
+    if (nameTrait) {
+      serializationKey = memberNs.getMergedTraits()[nameTrait] ?? key;
+    }
+    if (!(serializationKey in sourceObject)) {
+      continue;
+    }
+    yield [key, memberNs];
+    keysRemaining -= 1;
+  }
+}
+var init_structIterator = __esm({
+  "node_modules/@aws-sdk/core/dist-es/submodules/protocols/structIterator.js"() {
+    init_schema();
+  }
+});
+
 // node_modules/@aws-sdk/core/dist-es/submodules/protocols/json/jsonReviver.js
 function jsonReviver(key, value, context) {
   if (context?.source) {
@@ -11182,6 +11226,7 @@ var init_JsonShapeDeserializer = __esm({
     init_serde();
     import_util_base644 = __toESM(require_dist_cjs12());
     init_ConfigurableSerdeContext();
+    init_structIterator();
     init_jsonReviver();
     init_parseJsonBody();
     JsonShapeDeserializer = class extends SerdeContextConfig {
@@ -11221,7 +11266,7 @@ var init_JsonShapeDeserializer = __esm({
           return out;
         } else if (ns.isStructSchema() && isObject2) {
           const out = {};
-          for (const [memberName, memberSchema] of ns.structIterator()) {
+          for (const [memberName, memberSchema] of deserializingStructIterator(ns, value, this.settings.jsonName ? "jsonName" : false)) {
             const fromKey = this.settings.jsonName ? memberSchema.getMergedTraits().jsonName ?? memberName : memberName;
             const deserializedValue = this._read(memberSchema, value[fromKey]);
             if (deserializedValue != null) {
@@ -11360,6 +11405,7 @@ var init_JsonShapeSerializer = __esm({
     init_serde();
     import_util_base645 = __toESM(require_dist_cjs12());
     init_ConfigurableSerdeContext();
+    init_structIterator();
     init_jsonReplacer();
     JsonShapeSerializer = class extends SerdeContextConfig {
       settings;
@@ -11413,10 +11459,10 @@ var init_JsonShapeSerializer = __esm({
           return out;
         } else if (ns.isStructSchema() && isObject2) {
           const out = {};
-          for (const [memberName, memberSchema] of ns.structIterator()) {
-            const targetKey = this.settings.jsonName ? memberSchema.getMergedTraits().jsonName ?? memberName : memberName;
+          for (const [memberName, memberSchema] of serializingStructIterator(ns, value)) {
             const serializableValue = this._write(memberSchema, value[memberName], ns);
             if (serializableValue !== void 0) {
+              const targetKey = this.settings.jsonName ? memberSchema.getMergedTraits().jsonName ?? memberName : memberName;
               out[targetKey] = serializableValue;
             }
           }
@@ -12833,6 +12879,7 @@ var init_QueryShapeSerializer = __esm({
     import_smithy_client5 = __toESM(require_dist_cjs27());
     import_util_base646 = __toESM(require_dist_cjs12());
     init_ConfigurableSerdeContext();
+    init_structIterator();
     QueryShapeSerializer = class extends SerdeContextConfig {
       settings;
       buffer;
@@ -12932,7 +12979,7 @@ var init_QueryShapeSerializer = __esm({
           }
         } else if (ns.isStructSchema()) {
           if (value && typeof value === "object") {
-            for (const [memberName, member2] of ns.structIterator()) {
+            for (const [memberName, member2] of serializingStructIterator(ns, value)) {
               if (value[memberName] == null && !member2.isIdempotencyToken()) {
                 continue;
               }
@@ -13208,6 +13255,7 @@ var init_XmlShapeSerializer = __esm({
     import_smithy_client7 = __toESM(require_dist_cjs27());
     import_util_base647 = __toESM(require_dist_cjs12());
     init_ConfigurableSerdeContext();
+    init_structIterator();
     XmlShapeSerializer = class extends SerdeContextConfig {
       settings;
       stringBuffer;
@@ -13259,7 +13307,7 @@ var init_XmlShapeSerializer = __esm({
         }
         const structXmlNode = import_xml_builder3.XmlNode.of(name14);
         const [xmlnsAttr, xmlns] = this.getXmlnsAttribute(ns, parentXmlns);
-        for (const [memberName, memberSchema] of ns.structIterator()) {
+        for (const [memberName, memberSchema] of serializingStructIterator(ns, value)) {
           const val = value[memberName];
           if (val != null || memberSchema.isIdempotencyToken()) {
             if (memberSchema.getMergedTraits().xmlAttribute) {
@@ -17003,7 +17051,7 @@ var require_package2 = __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.938.0",
+      version: "3.940.0",
       scripts: {
         build: "concurrently 'yarn:build:cjs' 'yarn:build:es' 'yarn:build:types'",
         "build:cjs": "node ../../scripts/compilation/inline client-bedrock-runtime",
@@ -17022,21 +17070,21 @@ var require_package2 = __commonJS({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.936.0",
-        "@aws-sdk/credential-provider-node": "3.936.0",
+        "@aws-sdk/core": "3.940.0",
+        "@aws-sdk/credential-provider-node": "3.940.0",
         "@aws-sdk/eventstream-handler-node": "3.936.0",
         "@aws-sdk/middleware-eventstream": "3.936.0",
         "@aws-sdk/middleware-host-header": "3.936.0",
         "@aws-sdk/middleware-logger": "3.936.0",
         "@aws-sdk/middleware-recursion-detection": "3.936.0",
-        "@aws-sdk/middleware-user-agent": "3.936.0",
+        "@aws-sdk/middleware-user-agent": "3.940.0",
         "@aws-sdk/middleware-websocket": "3.936.0",
         "@aws-sdk/region-config-resolver": "3.936.0",
-        "@aws-sdk/token-providers": "3.936.0",
+        "@aws-sdk/token-providers": "3.940.0",
         "@aws-sdk/types": "3.936.0",
         "@aws-sdk/util-endpoints": "3.936.0",
         "@aws-sdk/util-user-agent-browser": "3.936.0",
-        "@aws-sdk/util-user-agent-node": "3.936.0",
+        "@aws-sdk/util-user-agent-node": "3.940.0",
         "@smithy/config-resolver": "^4.4.3",
         "@smithy/core": "^3.18.5",
         "@smithy/eventstream-serde-browser": "^4.2.5",
@@ -17783,7 +17831,7 @@ var init_package = __esm({
   "node_modules/@aws-sdk/nested-clients/package.json"() {
     package_default = {
       name: "@aws-sdk/nested-clients",
-      version: "3.936.0",
+      version: "3.940.0",
       description: "Nested clients for AWS SDK packages.",
       main: "./dist-cjs/index.js",
       module: "./dist-es/index.js",
@@ -17812,16 +17860,16 @@ var init_package = __esm({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.936.0",
+        "@aws-sdk/core": "3.940.0",
         "@aws-sdk/middleware-host-header": "3.936.0",
         "@aws-sdk/middleware-logger": "3.936.0",
         "@aws-sdk/middleware-recursion-detection": "3.936.0",
-        "@aws-sdk/middleware-user-agent": "3.936.0",
+        "@aws-sdk/middleware-user-agent": "3.940.0",
         "@aws-sdk/region-config-resolver": "3.936.0",
         "@aws-sdk/types": "3.936.0",
         "@aws-sdk/util-endpoints": "3.936.0",
         "@aws-sdk/util-user-agent-browser": "3.936.0",
-        "@aws-sdk/util-user-agent-node": "3.936.0",
+        "@aws-sdk/util-user-agent-node": "3.940.0",
         "@smithy/config-resolver": "^4.4.3",
         "@smithy/core": "^3.18.5",
         "@smithy/fetch-http-handler": "^5.3.6",
@@ -19213,7 +19261,7 @@ var require_package3 = __commonJS({
     module2.exports = {
       name: "@aws-sdk/client-sso",
       description: "AWS SDK for JavaScript Sso Client for Node.js, Browser and React Native",
-      version: "3.936.0",
+      version: "3.940.0",
       scripts: {
         build: "concurrently 'yarn:build:cjs' 'yarn:build:es' 'yarn:build:types'",
         "build:cjs": "node ../../scripts/compilation/inline client-sso",
@@ -19232,16 +19280,16 @@ var require_package3 = __commonJS({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.936.0",
+        "@aws-sdk/core": "3.940.0",
         "@aws-sdk/middleware-host-header": "3.936.0",
         "@aws-sdk/middleware-logger": "3.936.0",
         "@aws-sdk/middleware-recursion-detection": "3.936.0",
-        "@aws-sdk/middleware-user-agent": "3.936.0",
+        "@aws-sdk/middleware-user-agent": "3.940.0",
         "@aws-sdk/region-config-resolver": "3.936.0",
         "@aws-sdk/types": "3.936.0",
         "@aws-sdk/util-endpoints": "3.936.0",
         "@aws-sdk/util-user-agent-browser": "3.936.0",
-        "@aws-sdk/util-user-agent-node": "3.936.0",
+        "@aws-sdk/util-user-agent-node": "3.940.0",
         "@smithy/config-resolver": "^4.4.3",
         "@smithy/core": "^3.18.5",
         "@smithy/fetch-http-handler": "^5.3.6",
@@ -28051,7 +28099,27 @@ var init_HookManager = __esm({
 });
 
 // src/agent/imageConfig.js
-var SUPPORTED_IMAGE_EXTENSIONS, IMAGE_MIME_TYPES;
+function isFormatSupportedByProvider(extension, provider) {
+  if (!extension || typeof extension !== "string") {
+    return false;
+  }
+  if (extension.includes("/") || extension.includes("\\") || extension.includes("..")) {
+    return false;
+  }
+  const ext2 = extension.toLowerCase();
+  if (!SUPPORTED_IMAGE_EXTENSIONS.includes(ext2)) {
+    return false;
+  }
+  if (!provider || typeof provider !== "string") {
+    return true;
+  }
+  const unsupportedFormats = PROVIDER_UNSUPPORTED_FORMATS[provider];
+  if (unsupportedFormats && unsupportedFormats.includes(ext2)) {
+    return false;
+  }
+  return true;
+}
+var SUPPORTED_IMAGE_EXTENSIONS, IMAGE_MIME_TYPES, PROVIDER_UNSUPPORTED_FORMATS;
 var init_imageConfig = __esm({
   "src/agent/imageConfig.js"() {
     "use strict";
@@ -28064,6 +28132,10 @@ var init_imageConfig = __esm({
       "bmp": "image/bmp",
       "svg": "image/svg+xml"
     };
+    PROVIDER_UNSUPPORTED_FORMATS = {
+      "google": ["svg"]
+      // Google Gemini doesn't support image/svg+xml
+    };
   }
 });
 
@@ -41212,7 +41284,7 @@ var init_esm4 = __esm({
        *
        * @internal
        */
-      constructor(cwd = process.cwd(), pathImpl, sep3, { nocase, childrenCacheSize = 16 * 1024, fs: fs8 = defaultFS } = {}) {
+      constructor(cwd = process.cwd(), pathImpl, sep4, { nocase, childrenCacheSize = 16 * 1024, fs: fs8 = defaultFS } = {}) {
         this.#fs = fsFromOption(fs8);
         if (cwd instanceof URL || cwd.startsWith("file://")) {
           cwd = (0, import_node_url.fileURLToPath)(cwd);
@@ -41223,7 +41295,7 @@ var init_esm4 = __esm({
         this.#resolveCache = new ResolveCache();
         this.#resolvePosixCache = new ResolveCache();
         this.#children = new ChildrenCache(childrenCacheSize);
-        const split = cwdPath.substring(this.rootPath.length).split(sep3);
+        const split = cwdPath.substring(this.rootPath.length).split(sep4);
         if (split.length === 1 && !split[0]) {
           split.pop();
         }
@@ -62140,7 +62212,7 @@ var init_graph_builder = __esm({
       applyLinkStyles() {
         if (!this.pendingLinkStyles.length || !this.edges.length)
           return;
-        const normalize2 = (s4) => {
+        const normalize3 = (s4) => {
           const out = {};
           for (const [kRaw, vRaw] of Object.entries(s4)) {
             const k4 = kRaw.trim().toLowerCase();
@@ -62161,7 +62233,7 @@ var init_graph_builder = __esm({
           return out;
         };
         for (const cmd of this.pendingLinkStyles) {
-          const style = normalize2(cmd.props);
+          const style = normalize3(cmd.props);
           for (const idx of cmd.indices) {
             if (idx >= 0 && idx < this.edges.length) {
               const e4 = this.edges[idx];
@@ -69253,7 +69325,7 @@ var require_bk = __commonJS({
       return xs;
     }
     function buildBlockGraph(g4, layering, root2, reverseSep) {
-      var blockGraph = new Graph(), graphLabel = g4.graph(), sepFn = sep3(graphLabel.nodesep, graphLabel.edgesep, reverseSep);
+      var blockGraph = new Graph(), graphLabel = g4.graph(), sepFn = sep4(graphLabel.nodesep, graphLabel.edgesep, reverseSep);
       _.forEach(layering, function(layer) {
         var u4;
         _.forEach(layer, function(v4) {
@@ -69343,7 +69415,7 @@ var require_bk = __commonJS({
       alignCoordinates(xss, smallestWidth);
       return balance(xss, g4.graph().align);
     }
-    function sep3(nodeSep, edgeSep, reverseSep) {
+    function sep4(nodeSep, edgeSep, reverseSep) {
       return function(g4, v4, w4) {
         var vLabel = g4.node(v4);
         var wLabel = g4.node(w4);
@@ -69428,7 +69500,7 @@ var require_layout = __commonJS({
     "use strict";
     var _ = require_lodash2();
     var acyclic = require_acyclic();
-    var normalize2 = require_normalize();
+    var normalize3 = require_normalize();
     var rank = require_rank();
     var normalizeRanks = require_util().normalizeRanks;
     var parentDummyChains = require_parent_dummy_chains();
@@ -69490,7 +69562,7 @@ var require_layout = __commonJS({
         removeEdgeLabelProxies(g4);
       });
       time2("    normalize.run", function() {
-        normalize2.run(g4);
+        normalize3.run(g4);
       });
       time2("    parentDummyChains", function() {
         parentDummyChains(g4);
@@ -69517,7 +69589,7 @@ var require_layout = __commonJS({
         removeBorderNodes(g4);
       });
       time2("    normalize.undo", function() {
-        normalize2.undo(g4);
+        normalize3.undo(g4);
       });
       time2("    fixupEdgeLabelCoords", function() {
         fixupEdgeLabelCoords(g4);
@@ -75888,8 +75960,8 @@ var require_resolve = __commonJS({
       }
       return count;
     }
-    function getFullPath(resolver, id = "", normalize2) {
-      if (normalize2 !== false)
+    function getFullPath(resolver, id = "", normalize3) {
+      if (normalize3 !== false)
         id = normalizeId(id);
       const p4 = resolver.parse(id);
       return _getFullPath(resolver, p4);
@@ -77229,7 +77301,7 @@ var require_fast_uri = __commonJS({
     "use strict";
     var { normalizeIPv6, removeDotSegments, recomposeAuthority, normalizeComponentEncoding, isIPv4, nonSimpleDomain } = require_utils();
     var { SCHEMES, getSchemeHandler } = require_schemes();
-    function normalize2(uri, options) {
+    function normalize3(uri, options) {
       if (typeof uri === "string") {
         uri = /** @type {T} */
         serialize(parse6(uri, options), options);
@@ -77465,7 +77537,7 @@ var require_fast_uri = __commonJS({
     }
     var fastUri = {
       SCHEMES,
-      normalize: normalize2,
+      normalize: normalize3,
       resolve: resolve5,
       resolveComponent,
       equal,
@@ -85158,6 +85230,14 @@ var init_ProbeAgent = __esm({
               if (!imagePath) {
                 throw new Error("Image path is required");
               }
+              const filename = (0, import_path12.basename)(imagePath);
+              const extension = filename.toLowerCase().split(".").pop();
+              if (!extension || !SUPPORTED_IMAGE_EXTENSIONS.includes(extension)) {
+                throw new Error(`Invalid or unsupported image extension: ${extension}. Supported formats: ${SUPPORTED_IMAGE_EXTENSIONS.join(", ")}`);
+              }
+              if (this.apiType && !isFormatSupportedByProvider(extension, this.apiType)) {
+                throw new Error(`Image format '${extension}' is not supported by the current AI provider (${this.apiType}). Try using a different image format like PNG or JPEG.`);
+              }
               const loaded = await this.loadImageIfValid(imagePath);
               if (!loaded) {
                 throw new Error(`Failed to load image: ${imagePath}. The file may not exist, be too large, have an unsupported format, or be outside allowed directories.`);
@@ -85776,12 +85856,16 @@ var init_ProbeAgent = __esm({
           let absolutePath;
           let isPathAllowed2 = false;
           if ((0, import_path12.isAbsolute)(imagePath)) {
-            absolutePath = imagePath;
-            isPathAllowed2 = allowedDirs.some((dir) => absolutePath.startsWith((0, import_path12.resolve)(dir)));
+            absolutePath = (0, import_path12.normalize)((0, import_path12.resolve)(imagePath));
+            isPathAllowed2 = allowedDirs.some((dir) => {
+              const normalizedDir = (0, import_path12.normalize)((0, import_path12.resolve)(dir));
+              return absolutePath === normalizedDir || absolutePath.startsWith(normalizedDir + import_path12.sep);
+            });
           } else {
             for (const dir of allowedDirs) {
-              const resolvedPath2 = (0, import_path12.resolve)(dir, imagePath);
-              if (resolvedPath2.startsWith((0, import_path12.resolve)(dir))) {
+              const normalizedDir = (0, import_path12.normalize)((0, import_path12.resolve)(dir));
+              const resolvedPath2 = (0, import_path12.normalize)((0, import_path12.resolve)(dir, imagePath));
+              if (resolvedPath2 === normalizedDir || resolvedPath2.startsWith(normalizedDir + import_path12.sep)) {
                 absolutePath = resolvedPath2;
                 isPathAllowed2 = true;
                 break;
@@ -86789,8 +86873,8 @@ ${toolResultContent}
                           // Pass parent session ID for tracking
                           path: this.searchPath,
                           // Inherit search path
-                          provider: this.provider,
-                          // Inherit AI provider
+                          provider: this.apiType,
+                          // Inherit AI provider (string identifier)
                           model: this.model,
                           // Inherit model
                           debug: this.debug,
@@ -86799,7 +86883,7 @@ ${toolResultContent}
                         if (this.debug) {
                           console.log(`[DEBUG] Executing delegate tool at iteration ${currentIteration}/${maxIterations}`);
                           console.log(`[DEBUG] Parent session: ${this.sessionId}`);
-                          console.log(`[DEBUG] Inherited config: path=${this.searchPath}, provider=${this.provider}, model=${this.model}`);
+                          console.log(`[DEBUG] Inherited config: path=${this.searchPath}, provider=${this.apiType}, model=${this.model}`);
                           console.log(`[DEBUG] Delegate task: ${toolParams.task?.substring(0, 100)}...`);
                         }
                         if (this.tracer) {
@@ -86866,9 +86950,6 @@ ${toolResultContent}
                       role: "user",
                       content: toolResultMessage
                     });
-                    if (toolResultContent) {
-                      await this.processImageReferences(toolResultContent);
-                    }
                     if (this.debug) {
                       console.log(`[DEBUG] Tool ${toolName} executed successfully. Result length: ${typeof toolResult === "string" ? toolResult.length : JSON.stringify(toolResult).length}`);
                     }