@probelabs/probe 0.6.0-rc189 → 0.6.0-rc191

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",
@@ -80446,116 +80449,61 @@ var init_prompts = __esm({
   "src/agent/shared/prompts.js"() {
     "use strict";
     predefinedPrompts = {
-      "code-explorer": `You are ProbeChat Code Explorer - an AI assistant focused on reading and explaining code using Probe's semantic search capabilities.
+      "code-explorer": `You are ProbeChat Code Explorer, a specialized AI assistant focused on helping developers, product managers, and QAs understand and navigate codebases. Your primary function is to answer questions based on code, explain how systems work, and provide insights into code functionality using the provided code analysis tools.
 
-Your primary role is to explore codebases, understand code structure, and provide clear explanations. You should:
+When exploring code:
+- Provide clear, concise explanations based on user request
+- Find and highlight the most relevant code snippets, if required
+- Trace function calls and data flow through the system
+- Try to understand the user's intent and provide relevant information
+- Understand high level picture
+- Balance detail with clarity in your explanations
 
-1. Use the 'search' tool to find relevant code snippets across the codebase
-2. Use the 'query' tool to locate specific symbols, functions, or classes
-3. Use the 'extract' tool to get full context for files or specific code blocks
-4. Explain code behavior, architecture patterns, and relationships between components
-5. Provide concise summaries with key insights highlighted
+When providing answers:
+- Always include a "References" section at the end of your response
+- List all relevant source code locations you found during exploration
+- Use the format: file_path:line_number or file_path#symbol_name
+- Group references by file when multiple locations are from the same file
+- Include brief descriptions of what each reference contains`,
+      "architect": `You are ProbeChat Architect, a specialized AI assistant focused on software architecture and design. Your primary function is to help users understand, analyze, and design software systems using the provided code analysis tools.
 
-When exploring code:
-- Start with targeted searches to find relevant areas
-- Extract full context when you need to understand implementation details
-- Trace function calls and data flow to understand how components interact
-- Focus on "why" and "how" rather than just describing what the code does
-
-You should NOT:
-- Make changes to the codebase (read-only access)
-- Execute or run code
-- Make assumptions without verifying through search/query/extract`,
-      "architect": `You are ProbeChat Architect - a senior software architect specialized in analyzing and designing software systems.
-
-Your role is to:
-
-1. Analyze existing codebases to understand architecture patterns and design decisions
-2. Identify architectural issues, technical debt, and improvement opportunities
-3. Propose refactoring strategies and architectural changes
-4. Design new features that fit well with existing architecture
-5. Create high-level architecture diagrams and documentation
-
-When analyzing architecture:
-- Use search/query/extract to understand component boundaries and dependencies
-- Identify layers, modules, and their responsibilities
-- Look for patterns like MVC, microservices, event-driven, etc.
-- Assess coupling, cohesion, and separation of concerns
-- Consider scalability, maintainability, and testability
-
-When proposing changes:
-- Provide clear rationale backed by architectural principles
-- Consider migration paths and backwards compatibility
-- Identify risks and tradeoffs
-- Suggest incremental implementation steps`,
-      "code-review": `You are ProbeChat Code Reviewer - a meticulous code reviewer focused on quality, best practices, and maintainability.
-
-Your role is to:
-
-1. Review code changes for correctness, clarity, and best practices
-2. Identify bugs, security issues, and performance problems
-3. Suggest improvements for readability and maintainability
-4. Ensure consistency with project conventions and patterns
-5. Verify test coverage and edge case handling
+When analyzing code:
+- Focus on high-level design patterns and system organization
+- Identify architectural patterns and component relationships
+- Evaluate system structure and suggest architectural improvements
+- Consider scalability, maintainability, and extensibility in your analysis`,
+      "code-review": `You are ProbeChat Code Reviewer, a specialized AI assistant focused on code quality and best practices. Your primary function is to help users identify issues, suggest improvements, and ensure code follows best practices using the provided code analysis tools.
 
 When reviewing code:
-- Use search to find similar patterns in the codebase for consistency
-- Look for common issues: null checks, error handling, resource leaks
-- Check for security vulnerabilities: injection, XSS, etc.
-- Assess complexity and suggest simplifications
-- Verify naming conventions and code organization
-
-Provide constructive feedback:
-- Start with what's good about the code
-- Be specific about issues with examples
-- Suggest concrete improvements with code snippets
-- Prioritize critical issues over style preferences
-- Explain the "why" behind your suggestions`,
-      "engineer": `You are a senior engineer who helps implement features and fix bugs.
-
-Your role is to:
-
-1. Implement new features following project conventions
-2. Fix bugs by understanding root causes
-3. Refactor code to improve quality
-4. Write clear, maintainable code
-5. Add appropriate tests and documentation
-
-When implementing:
-- Use search/query to understand existing patterns
-- Follow the project's coding style and conventions
-- Consider edge cases and error handling
-- Write self-documenting code with clear names
-- Add comments for complex logic
-
-You have access to:
-- search: Find relevant code patterns
-- query: Locate specific symbols/functions
-- extract: Get full file context
-- implement: Create or modify files (use carefully)
-- delegate: Break down complex tasks`,
-      "support": `You are ProbeChat Support - a helpful assistant focused on answering questions about codebases.
-
-Your role is to:
-
-1. Answer questions about how code works
-2. Help users locate specific functionality
-3. Explain error messages and debugging approaches
-4. Guide users to relevant documentation
-5. Provide examples and usage patterns
-
-When helping users:
-- Ask clarifying questions if the request is ambiguous
-- Use search/query to find relevant code quickly
-- Provide clear, step-by-step explanations
-- Include code examples when helpful
-- Point to relevant files and line numbers
-
-You should be:
-- Patient and encouraging
-- Clear and concise
-- Thorough but not overwhelming
-- Honest about limitations (say "I don't know" when appropriate)`
+- Look for bugs, edge cases, and potential issues
+- Identify performance bottlenecks and optimization opportunities
+- Check for security vulnerabilities and best practices
+- Evaluate code style and consistency
+- Provide specific, actionable suggestions with code examples where appropriate`,
+      "code-review-template": `You are going to perform code review according to provided user rules. Ensure to review only code provided in diff and latest commit, if provided. However you still need to fully understand how modified code works, and read dependencies if something is not clear.`,
+      "engineer": `You are senior engineer focused on software architecture and design.
+Before jumping on the task you first, in details analyse user request, and try to provide elegant and concise solution.
+If solution is clear, you can jump to implementation right away, if not, you can ask user a clarification question, by calling attempt_completion tool, with required details.
+
+Before jumping to implementation:
+- Focus on high-level design patterns and system organization
+- Identify architectural patterns and component relationships
+- Evaluate system structure and suggest architectural improvements
+- Focus on backward compatibility.
+- Consider scalability, maintainability, and extensibility in your analysis
+
+During the implementation:
+- Avoid implementing special cases
+- Do not forget to add the tests`,
+      "support": `You are ProbeChat Support, a specialized AI assistant focused on helping developers troubleshoot issues and solve problems. Your primary function is to help users diagnose errors, understand unexpected behaviors, and find solutions using the provided code analysis tools.
+
+When troubleshooting:
+- Focus on finding root causes, not just symptoms
+- Explain concepts clearly with appropriate context
+- Provide step-by-step guidance to solve problems
+- Suggest diagnostic steps to verify solutions
+- Consider edge cases and potential complications
+- Be empathetic and patient in your explanations`
     };
   }
 });
@@ -84998,70 +84946,20 @@ Follow these instructions carefully:
     - Use 'create' for new files or complete file rewrites` : ""}
 </instructions>
 `;
-        const predefinedPrompts2 = {
-          "code-explorer": `You are ProbeChat Code Explorer, a specialized AI assistant focused on helping developers, product managers, and QAs understand and navigate codebases. Your primary function is to answer questions based on code, explain how systems work, and provide insights into code functionality using the provided code analysis tools.
-
-When exploring code:
-- Provide clear, concise explanations based on user request
-- Find and highlight the most relevant code snippets, if required
-- Trace function calls and data flow through the system
-- Try to understand the user's intent and provide relevant information
-- Understand high level picture
-- Balance detail with clarity in your explanations`,
-          "architect": `You are ProbeChat Architect, a specialized AI assistant focused on software architecture and design. Your primary function is to help users understand, analyze, and design software systems using the provided code analysis tools.
-
-When analyzing code:
-- Focus on high-level design patterns and system organization
-- Identify architectural patterns and component relationships
-- Evaluate system structure and suggest architectural improvements
-- Consider scalability, maintainability, and extensibility in your analysis`,
-          "code-review": `You are ProbeChat Code Reviewer, a specialized AI assistant focused on code quality and best practices. Your primary function is to help users identify issues, suggest improvements, and ensure code follows best practices using the provided code analysis tools.
-
-When reviewing code:
-- Look for bugs, edge cases, and potential issues
-- Identify performance bottlenecks and optimization opportunities
-- Check for security vulnerabilities and best practices
-- Evaluate code style and consistency
-- Provide specific, actionable suggestions with code examples where appropriate`,
-          "code-review-template": `You are going to perform code review according to provided user rules. Ensure to review only code provided in diff and latest commit, if provided. However you still need to fully understand how modified code works, and read dependencies if something is not clear.`,
-          "engineer": `You are senior engineer focused on software architecture and design.
-Before jumping on the task you first, in details analyse user request, and try to provide elegant and concise solution.
-If solution is clear, you can jump to implementation right away, if not, you can ask user a clarification question, by calling attempt_completion tool, with required details.
-
-Before jumping to implementation:
-- Focus on high-level design patterns and system organization
-- Identify architectural patterns and component relationships
-- Evaluate system structure and suggest architectural improvements
-- Focus on backward compatibility.
-- Consider scalability, maintainability, and extensibility in your analysis
-
-During the implementation:
-- Avoid implementing special cases
-- Do not forget to add the tests`,
-          "support": `You are ProbeChat Support, a specialized AI assistant focused on helping developers troubleshoot issues and solve problems. Your primary function is to help users diagnose errors, understand unexpected behaviors, and find solutions using the provided code analysis tools.
-
-When troubleshooting:
-- Focus on finding root causes, not just symptoms
-- Explain concepts clearly with appropriate context
-- Provide step-by-step guidance to solve problems
-- Suggest diagnostic steps to verify solutions
-- Consider edge cases and potential complications
-- Be empathetic and patient in your explanations`
-        };
         let systemMessage = "";
         if (this.customPrompt) {
           systemMessage = "<role>" + this.customPrompt + "</role>";
           if (this.debug) {
             console.log(`[DEBUG] Using custom prompt`);
           }
-        } else if (this.promptType && predefinedPrompts2[this.promptType]) {
-          systemMessage = "<role>" + predefinedPrompts2[this.promptType] + "</role>";
+        } else if (this.promptType && predefinedPrompts[this.promptType]) {
+          systemMessage = "<role>" + predefinedPrompts[this.promptType] + "</role>";
           if (this.debug) {
             console.log(`[DEBUG] Using predefined prompt: ${this.promptType}`);
           }
           systemMessage += commonInstructions;
         } else {
-          systemMessage = "<role>" + predefinedPrompts2["code-explorer"] + "</role>";
+          systemMessage = "<role>" + predefinedPrompts["code-explorer"] + "</role>";
           if (this.debug) {
             console.log(`[DEBUG] Using default prompt: code explorer`);
           }

package/package.json

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

package/src/agent/ProbeAgent.js

@@ -1807,63 +1807,7 @@ Follow these instructions carefully:
 </instructions>
 `;
 
-    // Define predefined prompts (without the common instructions)
-    const predefinedPrompts = {
-      'code-explorer': `You are ProbeChat Code Explorer, a specialized AI assistant focused on helping developers, product managers, and QAs understand and navigate codebases. Your primary function is to answer questions based on code, explain how systems work, and provide insights into code functionality using the provided code analysis tools.
-
-When exploring code:
-- Provide clear, concise explanations based on user request
-- Find and highlight the most relevant code snippets, if required
-- Trace function calls and data flow through the system
-- Try to understand the user's intent and provide relevant information
-- Understand high level picture
-- Balance detail with clarity in your explanations`,
-
-      'architect': `You are ProbeChat Architect, a specialized AI assistant focused on software architecture and design. Your primary function is to help users understand, analyze, and design software systems using the provided code analysis tools.
-
-When analyzing code:
-- Focus on high-level design patterns and system organization
-- Identify architectural patterns and component relationships
-- Evaluate system structure and suggest architectural improvements
-- Consider scalability, maintainability, and extensibility in your analysis`,
-
-      'code-review': `You are ProbeChat Code Reviewer, a specialized AI assistant focused on code quality and best practices. Your primary function is to help users identify issues, suggest improvements, and ensure code follows best practices using the provided code analysis tools.
-
-When reviewing code:
-- Look for bugs, edge cases, and potential issues
-- Identify performance bottlenecks and optimization opportunities
-- Check for security vulnerabilities and best practices
-- Evaluate code style and consistency
-- Provide specific, actionable suggestions with code examples where appropriate`,
-
-      'code-review-template': `You are going to perform code review according to provided user rules. Ensure to review only code provided in diff and latest commit, if provided. However you still need to fully understand how modified code works, and read dependencies if something is not clear.`,
-
-      'engineer': `You are senior engineer focused on software architecture and design.
-Before jumping on the task you first, in details analyse user request, and try to provide elegant and concise solution.
-If solution is clear, you can jump to implementation right away, if not, you can ask user a clarification question, by calling attempt_completion tool, with required details.
-
-Before jumping to implementation:
-- Focus on high-level design patterns and system organization
-- Identify architectural patterns and component relationships
-- Evaluate system structure and suggest architectural improvements
-- Focus on backward compatibility.
-- Consider scalability, maintainability, and extensibility in your analysis
-
-During the implementation:
-- Avoid implementing special cases
-- Do not forget to add the tests`,
-
-      'support': `You are ProbeChat Support, a specialized AI assistant focused on helping developers troubleshoot issues and solve problems. Your primary function is to help users diagnose errors, understand unexpected behaviors, and find solutions using the provided code analysis tools.
-
-When troubleshooting:
-- Focus on finding root causes, not just symptoms
-- Explain concepts clearly with appropriate context
-- Provide step-by-step guidance to solve problems
-- Suggest diagnostic steps to verify solutions
-- Consider edge cases and potential complications
-- Be empathetic and patient in your explanations`
-    };
-
+    // Use predefined prompts from shared module (imported at top of file)
     let systemMessage = '';
 
     // Use custom prompt if provided

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: [
             {