@probelabs/probe 0.6.0-rc132 → 0.6.0-rc133

package/build/agent/index.js

@@ -46498,19 +46498,16 @@ function cleanSchemaResponse(response) {
       return trimmed.substring(startIndex, endIndex);
     }
   }
-  const firstBracket = Math.min(
-    trimmed.indexOf("{") >= 0 ? trimmed.indexOf("{") : Infinity,
-    trimmed.indexOf("[") >= 0 ? trimmed.indexOf("[") : Infinity
-  );
-  const lastBracket = Math.max(
-    trimmed.lastIndexOf("}"),
-    trimmed.lastIndexOf("]")
-  );
-  if (firstBracket < Infinity && lastBracket >= 0 && firstBracket < lastBracket) {
-    const beforeFirstBracket = trimmed.substring(0, firstBracket).trim();
-    if (beforeFirstBracket === "" || beforeFirstBracket.match(/^```\w*$/) || beforeFirstBracket.split("\n").length <= 2) {
-      return trimmed.substring(firstBracket, lastBracket + 1);
-    }
+  let cleaned = trimmed;
+  cleaned = cleaned.replace(/^```(?:json)?\s*\n?/i, "");
+  cleaned = cleaned.replace(/\n?```\s*$/, "");
+  cleaned = cleaned.trim();
+  const firstChar = cleaned[0];
+  const lastChar = cleaned[cleaned.length - 1];
+  const isJsonObject = firstChar === "{" && lastChar === "}";
+  const isJsonArray = firstChar === "[" && lastChar === "]";
+  if (isJsonObject || isJsonArray) {
+    return cleaned;
   }
   return response;
 }

package/build/agent/schemaUtils.js

@@ -106,28 +106,28 @@ export function cleanSchemaResponse(response) {
     }
   }
 
-  // Fallback: Find JSON boundaries anywhere in the text
-  const firstBracket = Math.min(
-    trimmed.indexOf('{') >= 0 ? trimmed.indexOf('{') : Infinity,
-    trimmed.indexOf('[') >= 0 ? trimmed.indexOf('[') : Infinity
-  );
-
-  const lastBracket = Math.max(
-    trimmed.lastIndexOf('}'),
-    trimmed.lastIndexOf(']')
-  );
-
-  // Only extract if we found valid JSON boundaries
-  if (firstBracket < Infinity && lastBracket >= 0 && firstBracket < lastBracket) {
-    // Check if the response likely starts with JSON (directly or after minimal content)
-    const beforeFirstBracket = trimmed.substring(0, firstBracket).trim();
-
-    // If there's minimal content before the first bracket, extract the JSON
-    if (beforeFirstBracket === '' ||
-        beforeFirstBracket.match(/^```\w*$/) ||
-        beforeFirstBracket.split('\n').length <= 2) {
-      return trimmed.substring(firstBracket, lastBracket + 1);
-    }
+  // Fallback: Check if response is JSON after removing code block markers and whitespace
+  // First, remove common code block markers from top and bottom
+  let cleaned = trimmed;
+
+  // Remove opening code block markers (```json, ```, etc.)
+  cleaned = cleaned.replace(/^```(?:json)?\s*\n?/i, '');
+
+  // Remove closing code block markers
+  cleaned = cleaned.replace(/\n?```\s*$/, '');
+
+  // Trim whitespace and newlines
+  cleaned = cleaned.trim();
+
+  // Now check if first and last characters are valid JSON boundaries
+  const firstChar = cleaned[0];
+  const lastChar = cleaned[cleaned.length - 1];
+
+  const isJsonObject = firstChar === '{' && lastChar === '}';
+  const isJsonArray = firstChar === '[' && lastChar === ']';
+
+  if (isJsonObject || isJsonArray) {
+    return cleaned;
   }
 
   return response; // Return original if no extractable JSON found

package/cjs/agent/ProbeAgent.cjs

@@ -71893,19 +71893,16 @@ function cleanSchemaResponse(response) {
       return trimmed.substring(startIndex, endIndex);
     }
   }
-  const firstBracket = Math.min(
-    trimmed.indexOf("{") >= 0 ? trimmed.indexOf("{") : Infinity,
-    trimmed.indexOf("[") >= 0 ? trimmed.indexOf("[") : Infinity
-  );
-  const lastBracket = Math.max(
-    trimmed.lastIndexOf("}"),
-    trimmed.lastIndexOf("]")
-  );
-  if (firstBracket < Infinity && lastBracket >= 0 && firstBracket < lastBracket) {
-    const beforeFirstBracket = trimmed.substring(0, firstBracket).trim();
-    if (beforeFirstBracket === "" || beforeFirstBracket.match(/^```\w*$/) || beforeFirstBracket.split("\n").length <= 2) {
-      return trimmed.substring(firstBracket, lastBracket + 1);
-    }
+  let cleaned = trimmed;
+  cleaned = cleaned.replace(/^```(?:json)?\s*\n?/i, "");
+  cleaned = cleaned.replace(/\n?```\s*$/, "");
+  cleaned = cleaned.trim();
+  const firstChar = cleaned[0];
+  const lastChar = cleaned[cleaned.length - 1];
+  const isJsonObject = firstChar === "{" && lastChar === "}";
+  const isJsonArray = firstChar === "[" && lastChar === "]";
+  if (isJsonObject || isJsonArray) {
+    return cleaned;
   }
   return response;
 }

package/cjs/index.cjs

@@ -72073,19 +72073,16 @@ function cleanSchemaResponse(response) {
       return trimmed.substring(startIndex, endIndex);
     }
   }
-  const firstBracket = Math.min(
-    trimmed.indexOf("{") >= 0 ? trimmed.indexOf("{") : Infinity,
-    trimmed.indexOf("[") >= 0 ? trimmed.indexOf("[") : Infinity
-  );
-  const lastBracket = Math.max(
-    trimmed.lastIndexOf("}"),
-    trimmed.lastIndexOf("]")
-  );
-  if (firstBracket < Infinity && lastBracket >= 0 && firstBracket < lastBracket) {
-    const beforeFirstBracket = trimmed.substring(0, firstBracket).trim();
-    if (beforeFirstBracket === "" || beforeFirstBracket.match(/^```\w*$/) || beforeFirstBracket.split("\n").length <= 2) {
-      return trimmed.substring(firstBracket, lastBracket + 1);
-    }
+  let cleaned = trimmed;
+  cleaned = cleaned.replace(/^```(?:json)?\s*\n?/i, "");
+  cleaned = cleaned.replace(/\n?```\s*$/, "");
+  cleaned = cleaned.trim();
+  const firstChar = cleaned[0];
+  const lastChar = cleaned[cleaned.length - 1];
+  const isJsonObject = firstChar === "{" && lastChar === "}";
+  const isJsonArray = firstChar === "[" && lastChar === "]";
+  if (isJsonObject || isJsonArray) {
+    return cleaned;
   }
   return response;
 }

package/package.json

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

package/src/agent/schemaUtils.js

@@ -106,28 +106,28 @@ export function cleanSchemaResponse(response) {
     }
   }
 
-  // Fallback: Find JSON boundaries anywhere in the text
-  const firstBracket = Math.min(
-    trimmed.indexOf('{') >= 0 ? trimmed.indexOf('{') : Infinity,
-    trimmed.indexOf('[') >= 0 ? trimmed.indexOf('[') : Infinity
-  );
-
-  const lastBracket = Math.max(
-    trimmed.lastIndexOf('}'),
-    trimmed.lastIndexOf(']')
-  );
-
-  // Only extract if we found valid JSON boundaries
-  if (firstBracket < Infinity && lastBracket >= 0 && firstBracket < lastBracket) {
-    // Check if the response likely starts with JSON (directly or after minimal content)
-    const beforeFirstBracket = trimmed.substring(0, firstBracket).trim();
-
-    // If there's minimal content before the first bracket, extract the JSON
-    if (beforeFirstBracket === '' ||
-        beforeFirstBracket.match(/^```\w*$/) ||
-        beforeFirstBracket.split('\n').length <= 2) {
-      return trimmed.substring(firstBracket, lastBracket + 1);
-    }
+  // Fallback: Check if response is JSON after removing code block markers and whitespace
+  // First, remove common code block markers from top and bottom
+  let cleaned = trimmed;
+
+  // Remove opening code block markers (```json, ```, etc.)
+  cleaned = cleaned.replace(/^```(?:json)?\s*\n?/i, '');
+
+  // Remove closing code block markers
+  cleaned = cleaned.replace(/\n?```\s*$/, '');
+
+  // Trim whitespace and newlines
+  cleaned = cleaned.trim();
+
+  // Now check if first and last characters are valid JSON boundaries
+  const firstChar = cleaned[0];
+  const lastChar = cleaned[cleaned.length - 1];
+
+  const isJsonObject = firstChar === '{' && lastChar === '}';
+  const isJsonArray = firstChar === '[' && lastChar === ']';
+
+  if (isJsonObject || isJsonArray) {
+    return cleaned;
   }
 
   return response; // Return original if no extractable JSON found