@probelabs/probe 0.6.0-rc210 → 0.6.0-rc212

package/package.json

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

package/src/agent/ProbeAgent.js

@@ -24,6 +24,7 @@ import {
   queryToolDefinition,
   extractToolDefinition,
   delegateToolDefinition,
+  analyzeAllToolDefinition,
   bashToolDefinition,
   listFilesToolDefinition,
   searchFilesToolDefinition,
@@ -561,6 +562,9 @@ export class ProbeAgent {
     if (this.enableDelegate && wrappedTools.delegateToolInstance && isToolAllowed('delegate')) {
       this.toolImplementations.delegate = wrappedTools.delegateToolInstance;
     }
+    if (wrappedTools.analyzeAllToolInstance && isToolAllowed('analyze_all')) {
+      this.toolImplementations.analyze_all = wrappedTools.analyzeAllToolInstance;
+    }
 
     // File browsing tools
     if (isToolAllowed('listFiles')) {
@@ -2065,6 +2069,11 @@ ${extractGuidance}
       toolDefinitions += `${delegateToolDefinition}\n`;
     }
 
+    // Analyze All tool for bulk data processing
+    if (isToolAllowed('analyze_all')) {
+      toolDefinitions += `${analyzeAllToolDefinition}\n`;
+    }
+
     // Build XML tool guidelines with dynamic examples based on allowed tools
     // Build examples only for allowed tools
     let toolExamples = '';
@@ -2129,6 +2138,9 @@ The configuration is loaded from src/config.js lines 15-25 which contains the da
     if (this.enableDelegate && isToolAllowed('delegate')) {
       availableToolsList += '- delegate: Delegate big distinct tasks to specialized probe subagents.\n';
     }
+    if (isToolAllowed('analyze_all')) {
+      availableToolsList += '- analyze_all: Process ALL data matching a query using map-reduce (for aggregate questions needing 100% coverage).\n';
+    }
     if (this.enableBash && isToolAllowed('bash')) {
       availableToolsList += '- bash: Execute bash commands for system operations.\n';
     }
@@ -2642,11 +2654,11 @@ Follow these instructions carefully:
         if (!maxResponseTokens) {
           // Use model-based defaults if not explicitly configured
           maxResponseTokens = 4000;
-          if (this.model.includes('opus') || this.model.includes('sonnet') || this.model.startsWith('gpt-4-')) {
+          if (this.model && this.model.includes('opus') || this.model && this.model.includes('sonnet') || this.model && this.model.startsWith('gpt-4-')) {
             maxResponseTokens = 8192;
-          } else if (this.model.startsWith('gpt-4o')) {
+          } else if (this.model && this.model.startsWith('gpt-4o')) {
             maxResponseTokens = 8192;
-          } else if (this.model.startsWith('gemini')) {
+          } else if (this.model && this.model.startsWith('gemini')) {
             maxResponseTokens = 32000;
           }
         }
@@ -2820,6 +2832,10 @@ Follow these instructions carefully:
           if (this.enableDelegate && this.allowedTools.isEnabled('delegate')) {
             validTools.push('delegate');
           }
+          // Analyze All tool (for bulk data processing with map-reduce)
+          if (this.allowedTools.isEnabled('analyze_all')) {
+            validTools.push('analyze_all');
+          }
           // Task tool (require both enableTasks flag AND allowedTools permission)
           if (this.enableTasks && this.allowedTools.isEnabled('task')) {
             validTools.push('task');

package/src/agent/probeTool.js

@@ -196,12 +196,21 @@ export function createWrappedTools(baseTools) {
   // Wrap delegate tool
   if (baseTools.delegateTool) {
     wrappedTools.delegateToolInstance = wrapToolWithEmitter(
-      baseTools.delegateTool, 
-      'delegate', 
+      baseTools.delegateTool,
+      'delegate',
       baseTools.delegateTool.execute
     );
   }
 
+  // Wrap analyze_all tool
+  if (baseTools.analyzeAllTool) {
+    wrappedTools.analyzeAllToolInstance = wrapToolWithEmitter(
+      baseTools.analyzeAllTool,
+      'analyze_all',
+      baseTools.analyzeAllTool.execute
+    );
+  }
+
   // Wrap bash tool
   if (baseTools.bashTool) {
     wrappedTools.bashToolInstance = wrapToolWithEmitter(

package/src/agent/tools.js

@@ -4,6 +4,7 @@ import {
   queryTool,
   extractTool,
   delegateTool,
+  analyzeAllTool,
   bashTool,
   editTool,
   createTool,
@@ -14,6 +15,7 @@ import {
   querySchema,
   extractSchema,
   delegateSchema,
+  analyzeAllSchema,
   bashSchema,
   editSchema,
   createSchema,
@@ -21,6 +23,7 @@ import {
   queryToolDefinition,
   extractToolDefinition,
   delegateToolDefinition,
+  analyzeAllToolDefinition,
   bashToolDefinition,
   editToolDefinition,
   createToolDefinition,
@@ -53,6 +56,9 @@ export function createTools(configOptions) {
   if (configOptions.enableDelegate && isToolAllowed('delegate')) {
     tools.delegateTool = delegateTool(configOptions);
   }
+  if (isToolAllowed('analyze_all')) {
+    tools.analyzeAllTool = analyzeAllTool(configOptions);
+  }
 
   // Add bash tool if enabled
   if (configOptions.enableBash && isToolAllowed('bash')) {
@@ -88,6 +94,7 @@ export {
   querySchema,
   extractSchema,
   delegateSchema,
+  analyzeAllSchema,
   bashSchema,
   editSchema,
   createSchema,
@@ -96,6 +103,7 @@ export {
   queryToolDefinition,
   extractToolDefinition,
   delegateToolDefinition,
+  analyzeAllToolDefinition,
   bashToolDefinition,
   editToolDefinition,
   createToolDefinition,

package/src/index.js

@@ -25,12 +25,14 @@ import {
 	querySchema,
 	extractSchema,
 	delegateSchema,
+	analyzeAllSchema,
 	attemptCompletionSchema,
 	bashSchema,
 	searchToolDefinition,
 	queryToolDefinition,
 	extractToolDefinition,
 	delegateToolDefinition,
+	analyzeAllToolDefinition,
 	attemptCompletionToolDefinition,
 	bashToolDefinition,
 	parseXmlToolCall
@@ -41,7 +43,7 @@ import {
 	editToolDefinition,
 	createToolDefinition
 } from './tools/edit.js';
-import { searchTool, queryTool, extractTool, delegateTool } from './tools/vercel.js';
+import { searchTool, queryTool, extractTool, delegateTool, analyzeAllTool } from './tools/vercel.js';
 import { bashTool } from './tools/bash.js';
 import { editTool, createTool } from './tools/edit.js';
 import { ProbeAgent } from './agent/ProbeAgent.js';
@@ -85,6 +87,7 @@ export {
 	queryTool,
 	extractTool,
 	delegateTool,
+	analyzeAllTool,
 	bashTool,
 	editTool,
 	createTool,
@@ -96,6 +99,7 @@ export {
 	querySchema,
 	extractSchema,
 	delegateSchema,
+	analyzeAllSchema,
 	attemptCompletionSchema,
 	bashSchema,
 	editSchema,
@@ -105,6 +109,7 @@ export {
 	queryToolDefinition,
 	extractToolDefinition,
 	delegateToolDefinition,
+	analyzeAllToolDefinition,
 	attemptCompletionToolDefinition,
 	bashToolDefinition,
 	editToolDefinition,

package/src/search.js

@@ -94,8 +94,8 @@ export async function search(options) {
 
 	// Set default maxTokens if not provided
 	if (!options.maxTokens) {
-		options.maxTokens = 10000;
-		cliArgs.push('--max-tokens', '10000');
+		options.maxTokens = 20000;
+		cliArgs.push('--max-tokens', '20000');
 	}
 
 	// Set default timeout if not provided