@probelabs/probe 0.6.0-rc237 → 0.6.0-rc238

package/build/agent/dsl/environment.js

@@ -117,6 +117,23 @@ function traceToolCall(toolName, fn, tracer, logFn) {
   };
 }
 
+/**
+ * Try to parse a string as JSON if it looks like a JSON object or array.
+ * Checks if the first non-whitespace character is '{' or '[' before attempting parse.
+ * Returns the original string if it's not JSON.
+ *
+ * @param {string} text - The text to try parsing
+ * @returns {any} Parsed JSON value, or the original string
+ */
+function tryParseJSONValue(text) {
+  if (typeof text !== 'string') return text;
+  const firstChar = text.trimStart()[0];
+  if (firstChar === '{' || firstChar === '[') {
+    try { return JSON.parse(text); } catch (_) { /* not valid JSON */ }
+  }
+  return text;
+}
+
 /**
  * Generate sandbox globals that bridge DSL function calls to real tool implementations.
  *
@@ -195,7 +212,15 @@ export function generateSandboxGlobals(options) {
   if (mcpBridge) {
     for (const [name, tool] of Object.entries(mcpTools)) {
       const rawMcpFn = async (params = {}) => {
-        return tool.execute(params);
+        const result = await tool.execute(params);
+        // Extract text from MCP response envelope: { content: [{ type: 'text', text: '...' }] }
+        const text = result?.content?.[0]?.text;
+        if (text === undefined) {
+          // No envelope — if raw result is a JSON-like string, try parsing it
+          if (typeof result === 'string') return tryParseJSONValue(result);
+          return result;
+        }
+        return tryParseJSONValue(text);
       };
       globals[name] = traceToolCall(name, rawMcpFn, tracer, logFn);
     }

package/build/agent/index.js

@@ -21763,6 +21763,17 @@ function traceToolCall(toolName, fn, tracer, logFn) {
     }
   };
 }
+function tryParseJSONValue(text) {
+  if (typeof text !== "string") return text;
+  const firstChar = text.trimStart()[0];
+  if (firstChar === "{" || firstChar === "[") {
+    try {
+      return JSON.parse(text);
+    } catch (_) {
+    }
+  }
+  return text;
+}
 function generateSandboxGlobals(options) {
   const {
     toolImplementations = {},
@@ -21814,7 +21825,13 @@ function generateSandboxGlobals(options) {
   if (mcpBridge) {
     for (const [name, tool5] of Object.entries(mcpTools)) {
       const rawMcpFn = async (params = {}) => {
-        return tool5.execute(params);
+        const result = await tool5.execute(params);
+        const text = result?.content?.[0]?.text;
+        if (text === void 0) {
+          if (typeof result === "string") return tryParseJSONValue(result);
+          return result;
+        }
+        return tryParseJSONValue(text);
       };
       globals[name] = traceToolCall(name, rawMcpFn, tracer, logFn);
     }

package/cjs/agent/ProbeAgent.cjs

@@ -51212,6 +51212,17 @@ function traceToolCall(toolName, fn, tracer, logFn) {
     }
   };
 }
+function tryParseJSONValue(text) {
+  if (typeof text !== "string") return text;
+  const firstChar = text.trimStart()[0];
+  if (firstChar === "{" || firstChar === "[") {
+    try {
+      return JSON.parse(text);
+    } catch (_) {
+    }
+  }
+  return text;
+}
 function generateSandboxGlobals(options) {
   const {
     toolImplementations = {},
@@ -51263,7 +51274,13 @@ function generateSandboxGlobals(options) {
   if (mcpBridge) {
     for (const [name14, tool5] of Object.entries(mcpTools)) {
       const rawMcpFn = async (params = {}) => {
-        return tool5.execute(params);
+        const result = await tool5.execute(params);
+        const text = result?.content?.[0]?.text;
+        if (text === void 0) {
+          if (typeof result === "string") return tryParseJSONValue(result);
+          return result;
+        }
+        return tryParseJSONValue(text);
       };
       globals[name14] = traceToolCall(name14, rawMcpFn, tracer, logFn);
     }

package/cjs/index.cjs

@@ -112627,6 +112627,17 @@ function traceToolCall(toolName, fn, tracer, logFn) {
     }
   };
 }
+function tryParseJSONValue(text) {
+  if (typeof text !== "string") return text;
+  const firstChar = text.trimStart()[0];
+  if (firstChar === "{" || firstChar === "[") {
+    try {
+      return JSON.parse(text);
+    } catch (_) {
+    }
+  }
+  return text;
+}
 function generateSandboxGlobals(options) {
   const {
     toolImplementations = {},
@@ -112678,7 +112689,13 @@ function generateSandboxGlobals(options) {
   if (mcpBridge) {
     for (const [name14, tool5] of Object.entries(mcpTools)) {
       const rawMcpFn = async (params = {}) => {
-        return tool5.execute(params);
+        const result = await tool5.execute(params);
+        const text = result?.content?.[0]?.text;
+        if (text === void 0) {
+          if (typeof result === "string") return tryParseJSONValue(result);
+          return result;
+        }
+        return tryParseJSONValue(text);
       };
       globals[name14] = traceToolCall(name14, rawMcpFn, tracer, logFn);
     }

package/package.json

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

package/src/agent/dsl/environment.js

@@ -117,6 +117,23 @@ function traceToolCall(toolName, fn, tracer, logFn) {
   };
 }
 
+/**
+ * Try to parse a string as JSON if it looks like a JSON object or array.
+ * Checks if the first non-whitespace character is '{' or '[' before attempting parse.
+ * Returns the original string if it's not JSON.
+ *
+ * @param {string} text - The text to try parsing
+ * @returns {any} Parsed JSON value, or the original string
+ */
+function tryParseJSONValue(text) {
+  if (typeof text !== 'string') return text;
+  const firstChar = text.trimStart()[0];
+  if (firstChar === '{' || firstChar === '[') {
+    try { return JSON.parse(text); } catch (_) { /* not valid JSON */ }
+  }
+  return text;
+}
+
 /**
  * Generate sandbox globals that bridge DSL function calls to real tool implementations.
  *
@@ -195,7 +212,15 @@ export function generateSandboxGlobals(options) {
   if (mcpBridge) {
     for (const [name, tool] of Object.entries(mcpTools)) {
       const rawMcpFn = async (params = {}) => {
-        return tool.execute(params);
+        const result = await tool.execute(params);
+        // Extract text from MCP response envelope: { content: [{ type: 'text', text: '...' }] }
+        const text = result?.content?.[0]?.text;
+        if (text === undefined) {
+          // No envelope — if raw result is a JSON-like string, try parsing it
+          if (typeof result === 'string') return tryParseJSONValue(result);
+          return result;
+        }
+        return tryParseJSONValue(text);
       };
       globals[name] = traceToolCall(name, rawMcpFn, tracer, logFn);
     }