@probelabs/probe 0.6.0-rc168 → 0.6.0-rc170

package/cjs/index.cjs

@@ -12522,6 +12522,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) {
@@ -12624,6 +12668,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 {
@@ -12663,7 +12708,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) {
@@ -12802,6 +12847,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;
@@ -12855,10 +12901,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;
             }
           }
@@ -14275,6 +14321,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;
@@ -14374,7 +14421,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;
               }
@@ -14650,6 +14697,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;
@@ -14701,7 +14749,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) {
@@ -18445,7 +18493,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.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-bedrock-runtime",
@@ -18464,21 +18512,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",
@@ -19225,7 +19273,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",
@@ -19254,16 +19302,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",
@@ -20655,7 +20703,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",
@@ -20674,16 +20722,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",
@@ -24718,6 +24766,7 @@ var require_dist_cjs66 = __commonJS({
     var _A2 = "Accept";
     var _ADE3 = "AccessDeniedException";
     var _AG = "ApplyGuardrail";
+    var _AGD = "AppliedGuardrailDetails";
     var _AGR = "ApplyGuardrailRequest";
     var _AGRp = "ApplyGuardrailResponse";
     var _AIM = "AsyncInvokeMessage";
@@ -24933,6 +24982,7 @@ var require_dist_cjs66 = __commonJS({
     var _XABST = "X-Amzn-Bedrock-Service-Tier";
     var _XABT = "X-Amzn-Bedrock-Trace";
     var _a16 = "action";
+    var _aGD = "appliedGuardrailDetails";
     var _aIS = "asyncInvokeSummaries";
     var _aMRF = "additionalModelRequestFields";
     var _aMRFP = "additionalModelResponseFieldPaths";
@@ -24996,10 +25046,14 @@ var require_dist_cjs66 = __commonJS({
     var _fi = "findings";
     var _fil = "filters";
     var _g = "guardrail";
+    var _gA = "guardrailArn";
     var _gC = "guardrailCoverage";
     var _gCu = "guardrailConfig";
     var _gCua = "guardContent";
-    var _gI = "guardrailIdentifier";
+    var _gI = "guardrailId";
+    var _gIu = "guardrailIdentifier";
+    var _gO = "guardrailOrigin";
+    var _gOu = "guardrailOwnership";
     var _gPL = "guardrailProcessingLatency";
     var _gV = "guardrailVersion";
     var _gu = "guarded";
@@ -25170,12 +25224,20 @@ var require_dist_cjs66 = __commonJS({
     ];
     schema.TypeRegistry.for(n04).registerError(AccessDeniedException5, AccessDeniedException$1);
     var AnyToolChoice = [3, n04, _ATC, 0, [], []];
+    var AppliedGuardrailDetails = [
+      3,
+      n04,
+      _AGD,
+      0,
+      [_gI, _gV, _gA, _gO, _gOu],
+      [0, 0, 0, 64 | 0, 0]
+    ];
     var ApplyGuardrailRequest = [
       3,
       n04,
       _AGR,
       0,
-      [_gI, _gV, _s4, _co3, _oS],
+      [_gIu, _gV, _s4, _co3, _oS],
       [[0, 1], [0, 1], 0, [() => GuardrailContentBlockList, 0], 0]
     ];
     var ApplyGuardrailResponse = [
@@ -25405,7 +25467,7 @@ var require_dist_cjs66 = __commonJS({
       n04,
       _GA,
       0,
-      [_tP, _cP, _wP, _sIP, _cGP, _aRP, _iM],
+      [_tP, _cP, _wP, _sIP, _cGP, _aRP, _iM, _aGD],
       [
         () => GuardrailTopicPolicyAssessment,
         () => GuardrailContentPolicyAssessment,
@@ -25413,7 +25475,8 @@ var require_dist_cjs66 = __commonJS({
         () => GuardrailSensitiveInformationPolicyAssessment,
         () => GuardrailContextualGroundingPolicyAssessment,
         [() => GuardrailAutomatedReasoningPolicyAssessment, 0],
-        () => GuardrailInvocationMetrics
+        () => GuardrailInvocationMetrics,
+        () => AppliedGuardrailDetails
       ]
     ];
     var GuardrailAutomatedReasoningImpossibleFinding = [
@@ -25545,7 +25608,7 @@ var require_dist_cjs66 = __commonJS({
         [() => GuardrailAutomatedReasoningLogicWarning, 0]
       ]
     ];
-    var GuardrailConfiguration = [3, n04, _GC, 0, [_gI, _gV, _tr], [0, 0, 0]];
+    var GuardrailConfiguration = [3, n04, _GC, 0, [_gIu, _gV, _tr], [0, 0, 0]];
     var GuardrailContentFilter = [3, n04, _GCF, 0, [_t, _conf, _fS, _a16, _de], [0, 0, 0, 0, 2]];
     var GuardrailContentPolicyAssessment = [
       3,
@@ -25618,7 +25681,7 @@ var require_dist_cjs66 = __commonJS({
       [_pE, _re],
       [() => GuardrailPiiEntityFilterList, () => GuardrailRegexFilterList]
     ];
-    var GuardrailStreamConfiguration = [3, n04, _GSC, 0, [_gI, _gV, _tr, _sPM], [0, 0, 0, 0]];
+    var GuardrailStreamConfiguration = [3, n04, _GSC, 0, [_gIu, _gV, _tr, _sPM], [0, 0, 0, 0]];
     var GuardrailTextBlock = [3, n04, _GTB, 0, [_te, _q], [0, 64 | 0]];
     var GuardrailTextCharactersCoverage = [3, n04, _GTCC, 0, [_gu, _to], [1, 1]];
     var GuardrailTopic = [3, n04, _GT, 0, [_n, _t, _a16, _de], [0, 0, 0, 2]];
@@ -25673,7 +25736,7 @@ var require_dist_cjs66 = __commonJS({
       n04,
       _IMR,
       0,
-      [_bo, _cT, _ac, _mI, _tr, _gI, _gV, _pCL, _sTe],
+      [_bo, _cT, _ac, _mI, _tr, _gIu, _gV, _pCL, _sTe],
       [
         [() => Body, 16],
         [
@@ -25774,7 +25837,7 @@ var require_dist_cjs66 = __commonJS({
       n04,
       _IMWRSR,
       0,
-      [_bo, _cT, _ac, _mI, _tr, _gI, _gV, _pCL, _sTe],
+      [_bo, _cT, _ac, _mI, _tr, _gIu, _gV, _pCL, _sTe],
       [
         [() => Body, 16],
         [
@@ -26593,6 +26656,15 @@ var require_dist_cjs66 = __commonJS({
       GUARDRAIL_INTERVENED: "GUARDRAIL_INTERVENED",
       NONE: "NONE"
     };
+    var GuardrailOrigin = {
+      ACCOUNT_ENFORCED: "ACCOUNT_ENFORCED",
+      ORGANIZATION_ENFORCED: "ORGANIZATION_ENFORCED",
+      REQUEST: "REQUEST"
+    };
+    var GuardrailOwnership = {
+      CROSS_ACCOUNT: "CROSS_ACCOUNT",
+      SELF: "SELF"
+    };
     var GuardrailAutomatedReasoningLogicWarningType = {
       ALWAYS_FALSE: "ALWAYS_FALSE",
       ALWAYS_TRUE: "ALWAYS_TRUE"
@@ -26804,7 +26876,9 @@ var require_dist_cjs66 = __commonJS({
     exports2.GuardrailConverseImageFormat = GuardrailConverseImageFormat;
     exports2.GuardrailImageFormat = GuardrailImageFormat;
     exports2.GuardrailManagedWordType = GuardrailManagedWordType;
+    exports2.GuardrailOrigin = GuardrailOrigin;
     exports2.GuardrailOutputScope = GuardrailOutputScope;
+    exports2.GuardrailOwnership = GuardrailOwnership;
     exports2.GuardrailPiiEntityType = GuardrailPiiEntityType;
     exports2.GuardrailSensitiveInformationPolicyAction = GuardrailSensitiveInformationPolicyAction;
     exports2.GuardrailStreamProcessingMode = GuardrailStreamProcessingMode;
@@ -29467,7 +29541,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";
@@ -29480,6 +29574,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
+    };
   }
 });
 
@@ -29596,17 +29694,30 @@ var init_xmlParsingUtils = __esm({
 
 // src/agent/tools.js
 function createTools(configOptions) {
-  const tools2 = {
-    searchTool: searchTool(configOptions),
-    queryTool: queryTool(configOptions),
-    extractTool: extractTool(configOptions),
-    delegateTool: delegateTool(configOptions)
-  };
-  if (configOptions.enableBash) {
+  const tools2 = {};
+  const isToolAllowed = configOptions.isToolAllowed || ((toolName) => {
+    if (!configOptions.allowedTools) return true;
+    return configOptions.allowedTools.isEnabled(toolName);
+  });
+  if (isToolAllowed("search")) {
+    tools2.searchTool = searchTool(configOptions);
+  }
+  if (isToolAllowed("query")) {
+    tools2.queryTool = queryTool(configOptions);
+  }
+  if (isToolAllowed("extract")) {
+    tools2.extractTool = extractTool(configOptions);
+  }
+  if (configOptions.enableDelegate && isToolAllowed("delegate")) {
+    tools2.delegateTool = delegateTool(configOptions);
+  }
+  if (configOptions.enableBash && isToolAllowed("bash")) {
     tools2.bashTool = bashTool(configOptions);
   }
-  if (configOptions.allowEdit) {
+  if (configOptions.allowEdit && isToolAllowed("edit")) {
     tools2.editTool = editTool(configOptions);
+  }
+  if (configOptions.allowEdit && isToolAllowed("create")) {
     tools2.createTool = createTool(configOptions);
   }
   return tools2;
@@ -39216,7 +39327,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);
@@ -39227,7 +39338,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();
         }
@@ -59843,7 +59954,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();
@@ -59864,7 +59975,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];
@@ -66956,7 +67067,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) {
@@ -67046,7 +67157,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);
@@ -67131,7 +67242,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();
@@ -67193,7 +67304,7 @@ var require_layout = __commonJS({
         removeEdgeLabelProxies(g4);
       });
       time2("    normalize.run", function() {
-        normalize2.run(g4);
+        normalize3.run(g4);
       });
       time2("    parentDummyChains", function() {
         parentDummyChains(g4);
@@ -67220,7 +67331,7 @@ var require_layout = __commonJS({
         removeBorderNodes(g4);
       });
       time2("    normalize.undo", function() {
-        normalize2.undo(g4);
+        normalize3.undo(g4);
       });
       time2("    fixupEdgeLabelCoords", function() {
         fixupEdgeLabelCoords(g4);
@@ -73591,8 +73702,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);
@@ -74932,7 +75043,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);
@@ -75168,7 +75279,7 @@ var require_fast_uri = __commonJS({
     }
     var fastUri = {
       SCHEMES,
-      normalize: normalize2,
+      normalize: normalize3,
       resolve: resolve5,
       resolveComponent,
       equal,
@@ -82819,46 +82930,72 @@ var init_ProbeAgent = __esm({
        * Initialize tools with configuration
        */
       initializeTools() {
+        const isToolAllowed = (toolName) => this.allowedTools.isEnabled(toolName);
         const configOptions = {
           sessionId: this.sessionId,
           debug: this.debug,
           defaultPath: this.allowedFolders.length > 0 ? this.allowedFolders[0] : process.cwd(),
           allowedFolders: this.allowedFolders,
           outline: this.outline,
+          allowEdit: this.allowEdit,
+          enableDelegate: this.enableDelegate,
           enableBash: this.enableBash,
-          bashConfig: this.bashConfig
+          bashConfig: this.bashConfig,
+          allowedTools: this.allowedTools,
+          isToolAllowed
         };
         const baseTools = createTools(configOptions);
         const wrappedTools = createWrappedTools(baseTools);
-        this.toolImplementations = {
-          search: wrappedTools.searchToolInstance,
-          query: wrappedTools.queryToolInstance,
-          extract: wrappedTools.extractToolInstance,
-          delegate: wrappedTools.delegateToolInstance,
-          listFiles: listFilesToolInstance,
-          searchFiles: searchFilesToolInstance,
-          readImage: {
+        this.toolImplementations = {};
+        if (wrappedTools.searchToolInstance && isToolAllowed("search")) {
+          this.toolImplementations.search = wrappedTools.searchToolInstance;
+        }
+        if (wrappedTools.queryToolInstance && isToolAllowed("query")) {
+          this.toolImplementations.query = wrappedTools.queryToolInstance;
+        }
+        if (wrappedTools.extractToolInstance && isToolAllowed("extract")) {
+          this.toolImplementations.extract = wrappedTools.extractToolInstance;
+        }
+        if (this.enableDelegate && wrappedTools.delegateToolInstance && isToolAllowed("delegate")) {
+          this.toolImplementations.delegate = wrappedTools.delegateToolInstance;
+        }
+        if (isToolAllowed("listFiles")) {
+          this.toolImplementations.listFiles = listFilesToolInstance;
+        }
+        if (isToolAllowed("searchFiles")) {
+          this.toolImplementations.searchFiles = searchFilesToolInstance;
+        }
+        if (isToolAllowed("readImage")) {
+          this.toolImplementations.readImage = {
             execute: async (params) => {
               const imagePath = params.path;
               if (!imagePath) {
                 throw new Error("Image path is required");
               }
+              const filename = (0, import_path7.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.`);
               }
               return `Image loaded successfully: ${imagePath}. The image is now available for analysis in the conversation.`;
             }
-          }
-        };
-        if (this.enableBash && wrappedTools.bashToolInstance) {
+          };
+        }
+        if (this.enableBash && wrappedTools.bashToolInstance && isToolAllowed("bash")) {
           this.toolImplementations.bash = wrappedTools.bashToolInstance;
         }
         if (this.allowEdit) {
-          if (wrappedTools.editToolInstance) {
+          if (wrappedTools.editToolInstance && isToolAllowed("edit")) {
             this.toolImplementations.edit = wrappedTools.editToolInstance;
           }
-          if (wrappedTools.createToolInstance) {
+          if (wrappedTools.createToolInstance && isToolAllowed("create")) {
             this.toolImplementations.create = wrappedTools.createToolInstance;
           }
         }
@@ -83461,12 +83598,16 @@ var init_ProbeAgent = __esm({
           let absolutePath;
           let isPathAllowed2 = false;
           if ((0, import_path7.isAbsolute)(imagePath)) {
-            absolutePath = imagePath;
-            isPathAllowed2 = allowedDirs.some((dir) => absolutePath.startsWith((0, import_path7.resolve)(dir)));
+            absolutePath = (0, import_path7.normalize)((0, import_path7.resolve)(imagePath));
+            isPathAllowed2 = allowedDirs.some((dir) => {
+              const normalizedDir = (0, import_path7.normalize)((0, import_path7.resolve)(dir));
+              return absolutePath === normalizedDir || absolutePath.startsWith(normalizedDir + import_path7.sep);
+            });
           } else {
             for (const dir of allowedDirs) {
-              const resolvedPath2 = (0, import_path7.resolve)(dir, imagePath);
-              if (resolvedPath2.startsWith((0, import_path7.resolve)(dir))) {
+              const normalizedDir = (0, import_path7.normalize)((0, import_path7.resolve)(dir));
+              const resolvedPath2 = (0, import_path7.normalize)((0, import_path7.resolve)(dir, imagePath));
+              if (resolvedPath2 === normalizedDir || resolvedPath2.startsWith(normalizedDir + import_path7.sep)) {
                 absolutePath = resolvedPath2;
                 isPathAllowed2 = true;
                 break;
@@ -84474,8 +84615,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,
@@ -84484,7 +84625,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) {
@@ -84551,9 +84692,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}`);
                     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc168",
+  "version": "0.6.0-rc170",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",