@probelabs/probe 0.6.0-rc188 → 0.6.0-rc189

package/build/agent/index.js

@@ -54730,6 +54730,10 @@ function cleanSchemaResponse(response) {
     return response;
   }
   const trimmed = response.trim();
+  const resultWrapperMatch = trimmed.match(/^<result>\s*([\s\S]*?)\s*<\/result>$/);
+  if (resultWrapperMatch) {
+    return cleanSchemaResponse(resultWrapperMatch[1]);
+  }
   const jsonBlockMatch = trimmed.match(/```json\s*\n([\s\S]*?)\n```/);
   if (jsonBlockMatch) {
     return normalizeJsonQuotes(jsonBlockMatch[1].trim());

package/build/agent/schemaUtils.js

@@ -253,6 +253,14 @@ export function cleanSchemaResponse(response) {
 
   const trimmed = response.trim();
 
+  // Strip <result>...</result> wrapper if present (legacy XML format from some AI models)
+  // Some models like Google Gemini wrap JSON in <result> tags despite instructions not to
+  const resultWrapperMatch = trimmed.match(/^<result>\s*([\s\S]*?)\s*<\/result>$/);
+  if (resultWrapperMatch) {
+    // Recursively clean the inner content in case there are nested patterns
+    return cleanSchemaResponse(resultWrapperMatch[1]);
+  }
+
   // First, look for JSON after code block markers - similar to mermaid extraction
   // Try with json language specifier
   const jsonBlockMatch = trimmed.match(/```json\s*\n([\s\S]*?)\n```/);

package/cjs/agent/ProbeAgent.cjs

@@ -81416,6 +81416,10 @@ function cleanSchemaResponse(response) {
     return response;
   }
   const trimmed = response.trim();
+  const resultWrapperMatch = trimmed.match(/^<result>\s*([\s\S]*?)\s*<\/result>$/);
+  if (resultWrapperMatch) {
+    return cleanSchemaResponse(resultWrapperMatch[1]);
+  }
   const jsonBlockMatch = trimmed.match(/```json\s*\n([\s\S]*?)\n```/);
   if (jsonBlockMatch) {
     return normalizeJsonQuotes(jsonBlockMatch[1].trim());

package/cjs/index.cjs

@@ -79038,6 +79038,10 @@ function cleanSchemaResponse(response) {
     return response;
   }
   const trimmed = response.trim();
+  const resultWrapperMatch = trimmed.match(/^<result>\s*([\s\S]*?)\s*<\/result>$/);
+  if (resultWrapperMatch) {
+    return cleanSchemaResponse(resultWrapperMatch[1]);
+  }
   const jsonBlockMatch = trimmed.match(/```json\s*\n([\s\S]*?)\n```/);
   if (jsonBlockMatch) {
     return normalizeJsonQuotes(jsonBlockMatch[1].trim());

package/package.json

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

package/src/agent/schemaUtils.js

@@ -253,6 +253,14 @@ export function cleanSchemaResponse(response) {
 
   const trimmed = response.trim();
 
+  // Strip <result>...</result> wrapper if present (legacy XML format from some AI models)
+  // Some models like Google Gemini wrap JSON in <result> tags despite instructions not to
+  const resultWrapperMatch = trimmed.match(/^<result>\s*([\s\S]*?)\s*<\/result>$/);
+  if (resultWrapperMatch) {
+    // Recursively clean the inner content in case there are nested patterns
+    return cleanSchemaResponse(resultWrapperMatch[1]);
+  }
+
   // First, look for JSON after code block markers - similar to mermaid extraction
   // Try with json language specifier
   const jsonBlockMatch = trimmed.match(/```json\s*\n([\s\S]*?)\n```/);