@probelabs/probe 0.6.0-rc242 → 0.6.0-rc245

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

@@ -84874,12 +84874,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/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",
@@ -113890,12 +113892,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/cjs/index.cjs

@@ -7835,6 +7835,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.");
           }
@@ -19361,7 +19363,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",
@@ -19391,9 +19393,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",
@@ -25963,7 +25965,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",
@@ -25999,7 +26001,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",
@@ -110612,12 +110614,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/package.json

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

package/src/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 = [];
       }