@probelabs/probe 0.6.0-rc188 → 0.6.0-rc190

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());
@@ -62141,6 +62145,7 @@ Convert your previous response content into actual JSON data that follows this s
 
 // src/agent/index.js
 init_ProbeAgent();
+init_index();
 init_simpleTelemetry();
 init_schemaUtils();
 import dotenv3 from "dotenv";
@@ -63148,11 +63153,30 @@ var ProbeAgentMcpServer = class {
             },
             required: ["query"]
           }
+        },
+        {
+          name: "extract_code",
+          description: "Extract full code blocks from files using tree-sitter AST parsing. Use this to get complete code content based on file paths and symbols returned by search_code. Each file path can include optional line numbers or symbol names to extract specific code blocks.",
+          inputSchema: {
+            type: "object",
+            properties: {
+              path: {
+                type: "string",
+                description: "Absolute path to the project root directory (used as working directory for relative file paths)."
+              },
+              files: {
+                type: "array",
+                items: { type: "string" },
+                description: 'Array of file paths to extract from. Formats: "file.js" (entire file), "file.js:42" (code block at line 42), "file.js:10-20" (lines 10-20), "file.js#funcName" (specific symbol). Line numbers and symbols are part of the path string, not separate parameters. Paths can be absolute or relative to the project directory.'
+              }
+            },
+            required: ["path", "files"]
+          }
         }
       ]
     }));
     this.server.setRequestHandler(CallToolRequestSchema2, async (request) => {
-      if (request.params.name !== "search_code") {
+      if (request.params.name !== "search_code" && request.params.name !== "extract_code") {
         throw new McpError(
           ErrorCode2.MethodNotFound,
           `Unknown tool: ${request.params.name}`
@@ -63160,6 +63184,33 @@ var ProbeAgentMcpServer = class {
       }
       try {
         const args = request.params.arguments;
+        if (request.params.name === "extract_code") {
+          if (!args.path) {
+            throw new Error("Path is required");
+          }
+          if (!args.files || !Array.isArray(args.files) || args.files.length === 0) {
+            throw new Error("Files array is required and must not be empty");
+          }
+          const options = {
+            files: args.files,
+            path: args.path,
+            format: "xml",
+            allowTests: true
+            // Include test files by default
+          };
+          if (process.env.DEBUG === "1") {
+            console.error("[DEBUG] Executing extract_code with options:", JSON.stringify(options, null, 2));
+          }
+          const result2 = await extract(options);
+          return {
+            content: [
+              {
+                type: "text",
+                text: result2
+              }
+            ]
+          };
+        }
         if (!args.query) {
           throw new Error("Query is required");
         }
@@ -63276,7 +63327,7 @@ Please reformat your previous response to match this schema exactly. Only return
           ]
         };
       } catch (error) {
-        console.error(`Error executing search_code:`, error);
+        console.error(`Error executing ${request.params.name}:`, error);
         return {
           content: [
             {

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

@@ -6144,6 +6144,9 @@ var init_HttpBindingProtocol = __esm({
           const memberTraits = memberNs.getMergedTraits() ?? {};
           const inputMemberValue = input[memberName];
           if (inputMemberValue == null && !memberNs.isIdempotencyToken()) {
+            if (memberTraits.httpLabel) {
+              throw new Error(`No value provided for input HTTP label: ${memberName}.`);
+            }
             continue;
           }
           if (memberTraits.httpPayload) {
@@ -17357,7 +17360,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.970.0",
+      version: "3.971.0",
       scripts: {
         build: "concurrently 'yarn:build:types' 'yarn:build:es' && yarn build:cjs",
         "build:cjs": "node ../../scripts/compilation/inline client-bedrock-runtime",
@@ -17378,20 +17381,20 @@ var require_package2 = __commonJS({
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
         "@aws-sdk/core": "3.970.0",
-        "@aws-sdk/credential-provider-node": "3.970.0",
-        "@aws-sdk/eventstream-handler-node": "3.969.0",
+        "@aws-sdk/credential-provider-node": "3.971.0",
+        "@aws-sdk/eventstream-handler-node": "3.971.0",
         "@aws-sdk/middleware-eventstream": "3.969.0",
         "@aws-sdk/middleware-host-header": "3.969.0",
         "@aws-sdk/middleware-logger": "3.969.0",
         "@aws-sdk/middleware-recursion-detection": "3.969.0",
         "@aws-sdk/middleware-user-agent": "3.970.0",
-        "@aws-sdk/middleware-websocket": "3.969.0",
+        "@aws-sdk/middleware-websocket": "3.971.0",
         "@aws-sdk/region-config-resolver": "3.969.0",
-        "@aws-sdk/token-providers": "3.970.0",
+        "@aws-sdk/token-providers": "3.971.0",
         "@aws-sdk/types": "3.969.0",
         "@aws-sdk/util-endpoints": "3.970.0",
         "@aws-sdk/util-user-agent-browser": "3.969.0",
-        "@aws-sdk/util-user-agent-node": "3.970.0",
+        "@aws-sdk/util-user-agent-node": "3.971.0",
         "@smithy/config-resolver": "^4.4.6",
         "@smithy/core": "^3.20.6",
         "@smithy/eventstream-serde-browser": "^4.2.8",
@@ -18138,7 +18141,7 @@ var init_package = __esm({
   "node_modules/@aws-sdk/nested-clients/package.json"() {
     package_default = {
       name: "@aws-sdk/nested-clients",
-      version: "3.970.0",
+      version: "3.971.0",
       description: "Nested clients for AWS SDK packages.",
       main: "./dist-cjs/index.js",
       module: "./dist-es/index.js",
@@ -18176,7 +18179,7 @@ var init_package = __esm({
         "@aws-sdk/types": "3.969.0",
         "@aws-sdk/util-endpoints": "3.970.0",
         "@aws-sdk/util-user-agent-browser": "3.969.0",
-        "@aws-sdk/util-user-agent-node": "3.970.0",
+        "@aws-sdk/util-user-agent-node": "3.971.0",
         "@smithy/config-resolver": "^4.4.6",
         "@smithy/core": "^3.20.6",
         "@smithy/fetch-http-handler": "^5.3.9",
@@ -19520,7 +19523,7 @@ var require_package3 = __commonJS({
     module2.exports = {
       name: "@aws-sdk/client-sso",
       description: "AWS SDK for JavaScript Sso Client for Node.js, Browser and React Native",
-      version: "3.970.0",
+      version: "3.971.0",
       scripts: {
         build: "concurrently 'yarn:build:types' 'yarn:build:es' && yarn build:cjs",
         "build:cjs": "node ../../scripts/compilation/inline client-sso",
@@ -19549,7 +19552,7 @@ var require_package3 = __commonJS({
         "@aws-sdk/types": "3.969.0",
         "@aws-sdk/util-endpoints": "3.970.0",
         "@aws-sdk/util-user-agent-browser": "3.969.0",
-        "@aws-sdk/util-user-agent-node": "3.970.0",
+        "@aws-sdk/util-user-agent-node": "3.971.0",
         "@smithy/config-resolver": "^4.4.6",
         "@smithy/core": "^3.20.6",
         "@smithy/fetch-http-handler": "^5.3.9",
@@ -81416,6 +81419,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

@@ -7671,6 +7671,9 @@ var init_HttpBindingProtocol = __esm({
           const memberTraits = memberNs.getMergedTraits() ?? {};
           const inputMemberValue = input[memberName];
           if (inputMemberValue == null && !memberNs.isIdempotencyToken()) {
+            if (memberTraits.httpLabel) {
+              throw new Error(`No value provided for input HTTP label: ${memberName}.`);
+            }
             continue;
           }
           if (memberTraits.httpPayload) {
@@ -18884,7 +18887,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.970.0",
+      version: "3.971.0",
       scripts: {
         build: "concurrently 'yarn:build:types' 'yarn:build:es' && yarn build:cjs",
         "build:cjs": "node ../../scripts/compilation/inline client-bedrock-runtime",
@@ -18905,20 +18908,20 @@ var require_package2 = __commonJS({
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
         "@aws-sdk/core": "3.970.0",
-        "@aws-sdk/credential-provider-node": "3.970.0",
-        "@aws-sdk/eventstream-handler-node": "3.969.0",
+        "@aws-sdk/credential-provider-node": "3.971.0",
+        "@aws-sdk/eventstream-handler-node": "3.971.0",
         "@aws-sdk/middleware-eventstream": "3.969.0",
         "@aws-sdk/middleware-host-header": "3.969.0",
         "@aws-sdk/middleware-logger": "3.969.0",
         "@aws-sdk/middleware-recursion-detection": "3.969.0",
         "@aws-sdk/middleware-user-agent": "3.970.0",
-        "@aws-sdk/middleware-websocket": "3.969.0",
+        "@aws-sdk/middleware-websocket": "3.971.0",
         "@aws-sdk/region-config-resolver": "3.969.0",
-        "@aws-sdk/token-providers": "3.970.0",
+        "@aws-sdk/token-providers": "3.971.0",
         "@aws-sdk/types": "3.969.0",
         "@aws-sdk/util-endpoints": "3.970.0",
         "@aws-sdk/util-user-agent-browser": "3.969.0",
-        "@aws-sdk/util-user-agent-node": "3.970.0",
+        "@aws-sdk/util-user-agent-node": "3.971.0",
         "@smithy/config-resolver": "^4.4.6",
         "@smithy/core": "^3.20.6",
         "@smithy/eventstream-serde-browser": "^4.2.8",
@@ -19665,7 +19668,7 @@ var init_package = __esm({
   "node_modules/@aws-sdk/nested-clients/package.json"() {
     package_default = {
       name: "@aws-sdk/nested-clients",
-      version: "3.970.0",
+      version: "3.971.0",
       description: "Nested clients for AWS SDK packages.",
       main: "./dist-cjs/index.js",
       module: "./dist-es/index.js",
@@ -19703,7 +19706,7 @@ var init_package = __esm({
         "@aws-sdk/types": "3.969.0",
         "@aws-sdk/util-endpoints": "3.970.0",
         "@aws-sdk/util-user-agent-browser": "3.969.0",
-        "@aws-sdk/util-user-agent-node": "3.970.0",
+        "@aws-sdk/util-user-agent-node": "3.971.0",
         "@smithy/config-resolver": "^4.4.6",
         "@smithy/core": "^3.20.6",
         "@smithy/fetch-http-handler": "^5.3.9",
@@ -21047,7 +21050,7 @@ var require_package3 = __commonJS({
     module2.exports = {
       name: "@aws-sdk/client-sso",
       description: "AWS SDK for JavaScript Sso Client for Node.js, Browser and React Native",
-      version: "3.970.0",
+      version: "3.971.0",
       scripts: {
         build: "concurrently 'yarn:build:types' 'yarn:build:es' && yarn build:cjs",
         "build:cjs": "node ../../scripts/compilation/inline client-sso",
@@ -21076,7 +21079,7 @@ var require_package3 = __commonJS({
         "@aws-sdk/types": "3.969.0",
         "@aws-sdk/util-endpoints": "3.970.0",
         "@aws-sdk/util-user-agent-browser": "3.969.0",
-        "@aws-sdk/util-user-agent-node": "3.970.0",
+        "@aws-sdk/util-user-agent-node": "3.971.0",
         "@smithy/config-resolver": "^4.4.6",
         "@smithy/core": "^3.20.6",
         "@smithy/fetch-http-handler": "^5.3.9",
@@ -79038,6 +79041,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-rc190",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",

package/src/agent/index.js

@@ -14,6 +14,7 @@ import {
 } from '@modelcontextprotocol/sdk/types.js';
 import { readFileSync, existsSync } from 'fs';
 import { resolve } from 'path';
+import { extract } from '../index.js';
 import { initializeSimpleTelemetryFromOptions, SimpleAppTracer } from './simpleTelemetry.js';
 import { 
   cleanSchemaResponse, 
@@ -395,11 +396,30 @@ class ProbeAgentMcpServer {
             required: ['query']
           },
         },
+        {
+          name: 'extract_code',
+          description: "Extract full code blocks from files using tree-sitter AST parsing. Use this to get complete code content based on file paths and symbols returned by search_code. Each file path can include optional line numbers or symbol names to extract specific code blocks.",
+          inputSchema: {
+            type: 'object',
+            properties: {
+              path: {
+                type: 'string',
+                description: 'Absolute path to the project root directory (used as working directory for relative file paths).',
+              },
+              files: {
+                type: 'array',
+                items: { type: 'string' },
+                description: 'Array of file paths to extract from. Formats: "file.js" (entire file), "file.js:42" (code block at line 42), "file.js:10-20" (lines 10-20), "file.js#funcName" (specific symbol). Line numbers and symbols are part of the path string, not separate parameters. Paths can be absolute or relative to the project directory.',
+              }
+            },
+            required: ['path', 'files'],
+          },
+        },
       ],
     }));
 
     this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
-      if (request.params.name !== 'search_code') {
+      if (request.params.name !== 'search_code' && request.params.name !== 'extract_code') {
         throw new McpError(
           ErrorCode.MethodNotFound,
           `Unknown tool: ${request.params.name}`
@@ -409,6 +429,42 @@ class ProbeAgentMcpServer {
       try {
         const args = request.params.arguments;
 
+        // Handle extract_code tool
+        if (request.params.name === 'extract_code') {
+          // Validate required parameters
+          if (!args.path) {
+            throw new Error("Path is required");
+          }
+          if (!args.files || !Array.isArray(args.files) || args.files.length === 0) {
+            throw new Error("Files array is required and must not be empty");
+          }
+
+          // Build options with smart defaults
+          const options = {
+            files: args.files,
+            path: args.path,
+            format: 'xml',
+            allowTests: true,  // Include test files by default
+          };
+
+          if (process.env.DEBUG === '1') {
+            console.error('[DEBUG] Executing extract_code with options:', JSON.stringify(options, null, 2));
+          }
+
+          // Execute the extract command
+          const result = await extract(options);
+
+          return {
+            content: [
+              {
+                type: 'text',
+                text: result,
+              },
+            ],
+          };
+        }
+
+        // Handle search_code tool
         // Validate required fields
         if (!args.query) {
           throw new Error("Query is required");
@@ -554,7 +610,7 @@ class ProbeAgentMcpServer {
           ],
         };
       } catch (error) {
-        console.error(`Error executing search_code:`, error);
+        console.error(`Error executing ${request.params.name}:`, error);
         return {
           content: [
             {

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```/);