@probelabs/probe 0.6.0-rc242 → 0.6.0-rc246

package/build/agent/ProbeAgent.js

@@ -4765,12 +4765,19 @@ Convert your previous response content into actual JSON data that follows this s
       // Append DSL output buffer directly to response (bypasses LLM rewriting)
       if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted) {
         const outputContent = this._outputBuffer.items.join('\n\n');
-        finalResult = (finalResult || '') + '\n\n' + outputContent;
+        if (options.schema) {
+          // Schema response — the finalResult is JSON. Wrap output in RAW_OUTPUT
+          // delimiters so clients (visor, etc.) can extract and propagate the
+          // content separately from the JSON.
+          finalResult = (finalResult || '') + '\n<<<RAW_OUTPUT>>>\n' + outputContent + '\n<<<END_RAW_OUTPUT>>>';
+        } else {
+          finalResult = (finalResult || '') + '\n\n' + outputContent;
+        }
         if (options.onStream) {
           options.onStream('\n\n' + outputContent);
         }
         if (this.debug) {
-          console.log(`[DEBUG] Appended ${this._outputBuffer.items.length} output buffer items (${outputContent.length} chars) to final result`);
+          console.log(`[DEBUG] Appended ${this._outputBuffer.items.length} output buffer items (${outputContent.length} chars) to final result${options.schema ? ' (with RAW_OUTPUT delimiters)' : ''}`);
         }
         this._outputBuffer.items = [];
       }

package/build/agent/index.js

@@ -29060,6 +29060,25 @@ function buildToolImplementations(configOptions) {
       }
     }
   };
+  if (configOptions.enableBash) {
+    const bashToolInstance = bashTool({
+      bashConfig: configOptions.bashConfig || {},
+      debug: configOptions.debug || false,
+      cwd,
+      allowedFolders: configOptions.allowedFolders || [],
+      workspaceRoot: configOptions.workspaceRoot || cwd,
+      tracer: configOptions.tracer || null
+    });
+    tools2.bash = {
+      execute: async (params) => {
+        try {
+          return await bashToolInstance.execute(params);
+        } catch (e) {
+          return `Bash error: ${e.message}`;
+        }
+      }
+    };
+  }
   return tools2;
 }
 function buildLLMCall(configOptions) {
@@ -29786,6 +29805,7 @@ var init_executePlan = __esm({
     init_extract();
     init_delegate();
     init_esm5();
+    init_bash();
     RAW_OUTPUT_START = "<<<RAW_OUTPUT>>>";
     RAW_OUTPUT_END = "<<<END_RAW_OUTPUT>>>";
   }
@@ -84874,12 +84894,16 @@ Convert your previous response content into actual JSON data that follows this s
           }
           if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted) {
             const outputContent = this._outputBuffer.items.join("\n\n");
-            finalResult = (finalResult || "") + "\n\n" + outputContent;
+            if (options.schema) {
+              finalResult = (finalResult || "") + "\n<<<RAW_OUTPUT>>>\n" + outputContent + "\n<<<END_RAW_OUTPUT>>>";
+            } else {
+              finalResult = (finalResult || "") + "\n\n" + outputContent;
+            }
             if (options.onStream) {
               options.onStream("\n\n" + outputContent);
             }
             if (this.debug) {
-              console.log(`[DEBUG] Appended ${this._outputBuffer.items.length} output buffer items (${outputContent.length} chars) to final result`);
+              console.log(`[DEBUG] Appended ${this._outputBuffer.items.length} output buffer items (${outputContent.length} chars) to final result${options.schema ? " (with RAW_OUTPUT delimiters)" : ""}`);
             }
             this._outputBuffer.items = [];
           }

package/build/tools/executePlan.js

@@ -13,6 +13,7 @@ import { query } from '../query.js';
 import { extract } from '../extract.js';
 import { delegate } from '../delegate.js';
 import { glob } from 'glob';
+import { bashTool } from './bash.js';
 
 export { executePlanSchema };
 
@@ -156,6 +157,27 @@ function buildToolImplementations(configOptions) {
     },
   };
 
+  // Add bash tool if enabled
+  if (configOptions.enableBash) {
+    const bashToolInstance = bashTool({
+      bashConfig: configOptions.bashConfig || {},
+      debug: configOptions.debug || false,
+      cwd: cwd,
+      allowedFolders: configOptions.allowedFolders || [],
+      workspaceRoot: configOptions.workspaceRoot || cwd,
+      tracer: configOptions.tracer || null,
+    });
+    tools.bash = {
+      execute: async (params) => {
+        try {
+          return await bashToolInstance.execute(params);
+        } catch (e) {
+          return `Bash error: ${e.message}`;
+        }
+      },
+    };
+  }
+
   return tools;
 }
 

package/cjs/agent/ProbeAgent.cjs

@@ -5978,6 +5978,8 @@ var init_EventStreamSerde = __esm({
             } else {
               serializer.write(eventSchema, event[unionMember]);
             }
+          } else if (eventSchema.isUnitSchema()) {
+            serializer.write(eventSchema, {});
           } else {
             throw new Error("@smithy/core/event-streams - non-struct member not supported in event stream union.");
           }
@@ -17504,7 +17506,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.991.0",
+      version: "3.992.0",
       scripts: {
         build: "concurrently 'yarn:build:types' 'yarn:build:es' && yarn build:cjs",
         "build:cjs": "node ../../scripts/compilation/inline client-bedrock-runtime",
@@ -17534,9 +17536,9 @@ var require_package2 = __commonJS({
         "@aws-sdk/middleware-user-agent": "^3.972.10",
         "@aws-sdk/middleware-websocket": "^3.972.6",
         "@aws-sdk/region-config-resolver": "^3.972.3",
-        "@aws-sdk/token-providers": "3.991.0",
+        "@aws-sdk/token-providers": "3.992.0",
         "@aws-sdk/types": "^3.973.1",
-        "@aws-sdk/util-endpoints": "3.991.0",
+        "@aws-sdk/util-endpoints": "3.992.0",
         "@aws-sdk/util-user-agent-browser": "^3.972.3",
         "@aws-sdk/util-user-agent-node": "^3.972.8",
         "@smithy/config-resolver": "^4.4.6",
@@ -24106,7 +24108,7 @@ var init_package2 = __esm({
   "node_modules/@aws-sdk/token-providers/node_modules/@aws-sdk/nested-clients/package.json"() {
     package_default2 = {
       name: "@aws-sdk/nested-clients",
-      version: "3.991.0",
+      version: "3.992.0",
       description: "Nested clients for AWS SDK packages.",
       main: "./dist-cjs/index.js",
       module: "./dist-es/index.js",
@@ -24142,7 +24144,7 @@ var init_package2 = __esm({
         "@aws-sdk/middleware-user-agent": "^3.972.10",
         "@aws-sdk/region-config-resolver": "^3.972.3",
         "@aws-sdk/types": "^3.973.1",
-        "@aws-sdk/util-endpoints": "3.991.0",
+        "@aws-sdk/util-endpoints": "3.992.0",
         "@aws-sdk/util-user-agent-browser": "^3.972.3",
         "@aws-sdk/util-user-agent-node": "^3.972.8",
         "@smithy/config-resolver": "^4.4.6",
@@ -58510,6 +58512,25 @@ function buildToolImplementations(configOptions) {
       }
     }
   };
+  if (configOptions.enableBash) {
+    const bashToolInstance = bashTool({
+      bashConfig: configOptions.bashConfig || {},
+      debug: configOptions.debug || false,
+      cwd,
+      allowedFolders: configOptions.allowedFolders || [],
+      workspaceRoot: configOptions.workspaceRoot || cwd,
+      tracer: configOptions.tracer || null
+    });
+    tools2.bash = {
+      execute: async (params) => {
+        try {
+          return await bashToolInstance.execute(params);
+        } catch (e5) {
+          return `Bash error: ${e5.message}`;
+        }
+      }
+    };
+  }
   return tools2;
 }
 function buildLLMCall(configOptions) {
@@ -59237,6 +59258,7 @@ var init_executePlan = __esm({
     init_extract();
     init_delegate();
     init_esm5();
+    init_bash();
     RAW_OUTPUT_START = "<<<RAW_OUTPUT>>>";
     RAW_OUTPUT_END = "<<<END_RAW_OUTPUT>>>";
   }
@@ -113890,12 +113912,16 @@ Convert your previous response content into actual JSON data that follows this s
           }
           if (this._outputBuffer && this._outputBuffer.items.length > 0 && !options._schemaFormatted) {
             const outputContent = this._outputBuffer.items.join("\n\n");
-            finalResult = (finalResult || "") + "\n\n" + outputContent;
+            if (options.schema) {
+              finalResult = (finalResult || "") + "\n<<<RAW_OUTPUT>>>\n" + outputContent + "\n<<<END_RAW_OUTPUT>>>";
+            } else {
+              finalResult = (finalResult || "") + "\n\n" + outputContent;
+            }
             if (options.onStream) {
               options.onStream("\n\n" + outputContent);
             }
             if (this.debug) {
-              console.log(`[DEBUG] Appended ${this._outputBuffer.items.length} output buffer items (${outputContent.length} chars) to final result`);
+              console.log(`[DEBUG] Appended ${this._outputBuffer.items.length} output buffer items (${outputContent.length} chars) to final result${options.schema ? " (with RAW_OUTPUT delimiters)" : ""}`);
             }
             this._outputBuffer.items = [];
           }