@probelabs/probe 0.6.0-rc154 → 0.6.0-rc159

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc154",
+  "version": "0.6.0-rc159",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",
@@ -78,7 +78,7 @@
     "@ai-sdk/google": "^2.0.14",
     "@ai-sdk/openai": "^2.0.10",
     "@modelcontextprotocol/sdk": "^1.0.0",
-    "@probelabs/maid": "^0.0.19",
+    "@probelabs/maid": "^0.0.20",
     "adm-zip": "^0.5.16",
     "ai": "^5.0.0",
     "ajv": "^8.17.1",

package/src/agent/ProbeAgent.d.ts

@@ -13,6 +13,8 @@ export interface ProbeAgentOptions {
   promptType?: 'code-explorer' | 'engineer' | 'code-review' | 'support' | 'architect';
   /** Allow the use of the 'implement' tool for code editing */
   allowEdit?: boolean;
+  /** Enable the delegate tool for task distribution to subagents */
+  enableDelegate?: boolean;
   /** Search directory path */
   path?: string;
   /** Force specific AI provider */

package/src/agent/ProbeAgent.js

@@ -18,11 +18,12 @@ import { TokenCounter } from './tokenCounter.js';
 import { InMemoryStorageAdapter } from './storage/InMemoryStorageAdapter.js';
 import { HookManager, HOOK_TYPES } from './hooks/HookManager.js';
 import { SUPPORTED_IMAGE_EXTENSIONS, IMAGE_MIME_TYPES } from './imageConfig.js';
-import { 
+import {
   createTools,
   searchToolDefinition,
   queryToolDefinition,
   extractToolDefinition,
+  delegateToolDefinition,
   listFilesToolDefinition,
   searchFilesToolDefinition,
   attemptCompletionToolDefinition,
@@ -75,6 +76,7 @@ export class ProbeAgent {
    * @param {string} [options.customPrompt] - Custom prompt to replace the default system message
    * @param {string} [options.promptType] - Predefined prompt type (architect, code-review, support)
    * @param {boolean} [options.allowEdit=false] - Allow the use of the 'implement' tool
+   * @param {boolean} [options.enableDelegate=false] - Enable the delegate tool for task distribution to subagents
    * @param {string} [options.path] - Search directory path
    * @param {string} [options.provider] - Force specific AI provider
    * @param {string} [options.model] - Override model name
@@ -97,6 +99,7 @@ export class ProbeAgent {
     this.customPrompt = options.customPrompt || null;
     this.promptType = options.promptType || 'code-explorer';
     this.allowEdit = !!options.allowEdit;
+    this.enableDelegate = !!options.enableDelegate;
     this.debug = options.debug || process.env.DEBUG === '1';
     this.cancelled = false;
     this.tracer = options.tracer || null;
@@ -889,6 +892,9 @@ ${attemptCompletionToolDefinition}
     if (this.allowEdit) {
       toolDefinitions += `${implementToolDefinition}\n`;
     }
+    if (this.enableDelegate) {
+      toolDefinitions += `${delegateToolDefinition}\n`;
+    }
 
     // Build XML tool guidelines
     let xmlToolGuidelines = `
@@ -952,7 +958,7 @@ Available Tools:
 - extract: Extract specific code blocks or lines from files.
 - listFiles: List files and directories in a specified location.
 - searchFiles: Find files matching a glob pattern with recursive search capability.
-${this.allowEdit ? '- implement: Implement a feature or fix a bug using aider.\n' : ''}
+${this.allowEdit ? '- implement: Implement a feature or fix a bug using aider.\n' : ''}${this.enableDelegate ? '- delegate: Delegate big distinct tasks to specialized probe subagents.\n' : ''}
 - attempt_completion: Finalize the task and provide the result to the user.
 - attempt_complete: Quick completion using previous response (shorthand).
 `;
@@ -1323,6 +1329,9 @@ When troubleshooting:
         if (this.allowEdit) {
           validTools.push('implement');
         }
+        if (this.enableDelegate) {
+          validTools.push('delegate');
+        }
         
         // Try parsing with hybrid parser that supports both native and MCP tools
         const nativeTools = validTools;
@@ -1462,18 +1471,24 @@ When troubleshooting:
                 
                 // Execute tool with tracing if available
                 const executeToolCall = async () => {
-                  // For delegate tool, pass current iteration and max iterations
+                  // For delegate tool, pass current iteration, max iterations, session ID, and config
                   if (toolName === 'delegate') {
                     const enhancedParams = {
                       ...toolParams,
                       currentIteration,
                       maxIterations,
+                      parentSessionId: this.sessionId,  // Pass parent session ID for tracking
+                      path: this.searchPath,            // Inherit search path
+                      provider: this.provider,          // Inherit AI provider
+                      model: this.model,                // Inherit model
                       debug: this.debug,
                       tracer: this.tracer
                     };
-                    
+
                     if (this.debug) {
                       console.log(`[DEBUG] Executing delegate tool at iteration ${currentIteration}/${maxIterations}`);
+                      console.log(`[DEBUG] Parent session: ${this.sessionId}`);
+                      console.log(`[DEBUG] Inherited config: path=${this.searchPath}, provider=${this.provider}, model=${this.model}`);
                       console.log(`[DEBUG] Delegate task: ${toolParams.task?.substring(0, 100)}...`);
                     }
                     
@@ -2205,6 +2220,7 @@ Convert your previous response content into actual JSON data that follows this s
       customPrompt: this.customPrompt,
       promptType: this.promptType,
       allowEdit: this.allowEdit,
+      enableDelegate: this.enableDelegate,
       path: this.allowedFolders[0], // Use first allowed folder as primary path
       allowedFolders: [...this.allowedFolders],
       provider: this.clientApiProvider,

package/src/agent/acp/server.js

@@ -329,6 +329,7 @@ export class ACPServer {
       provider: this.options.provider,
       model: this.options.model,
       allowEdit: this.options.allowEdit,
+      enableDelegate: this.options.enableDelegate,
       debug: this.options.debug,
       enableMcp: this.options.enableMcp,
       mcpConfig: this.options.mcpConfig,

package/src/agent/index.js

@@ -123,6 +123,7 @@ function parseArgs() {
     provider: null,
     model: null,
     allowEdit: false,
+    enableDelegate: false,
     verbose: false,
     help: false,
     maxIterations: null,
@@ -156,6 +157,10 @@ function parseArgs() {
       config.verbose = true;
     } else if (arg === '--allow-edit') {
       config.allowEdit = true;
+    } else if (arg === '--enable-delegate') {
+      config.enableDelegate = true;
+    } else if (arg === '--no-delegate') {
+      config.enableDelegate = false; // Explicitly disable delegation (used by subagents)
     } else if (arg === '--path' && i + 1 < args.length) {
       config.path = args[++i];
     } else if (arg === '--allowed-folders' && i + 1 < args.length) {
@@ -237,6 +242,7 @@ Options:
   --provider <name>                Force AI provider: anthropic, openai, google
   --model <name>                   Override model name
   --allow-edit                     Enable code modification capabilities
+  --enable-delegate                Enable delegate tool for task distribution to subagents
   --verbose                        Enable verbose output
   --outline                        Use outline-xml format for code search results
   --mcp                           Run as MCP server
@@ -565,6 +571,7 @@ async function main() {
       model: config.model,
       path: config.path,
       allowEdit: config.allowEdit,
+      enableDelegate: config.enableDelegate,
       debug: config.verbose
     });
     await server.start();
@@ -713,6 +720,7 @@ async function main() {
       promptType: config.prompt,
       customPrompt: systemPrompt,
       allowEdit: config.allowEdit,
+      enableDelegate: config.enableDelegate,
       debug: config.verbose,
       tracer: appTracer,
       outline: config.outline,