@probelabs/probe 0.6.0-rc231 → 0.6.0-rc233

package/build/agent/probeTool.js

@@ -211,6 +211,15 @@ export function createWrappedTools(baseTools) {
     );
   }
 
+  // Wrap execute_plan tool
+  if (baseTools.executePlanTool) {
+    wrappedTools.executePlanToolInstance = wrapToolWithEmitter(
+      baseTools.executePlanTool,
+      'execute_plan',
+      baseTools.executePlanTool.execute
+    );
+  }
+
   // Wrap bash tool
   if (baseTools.bashTool) {
     wrappedTools.bashToolInstance = wrapToolWithEmitter(

package/build/agent/tools.js

@@ -5,6 +5,7 @@ import {
   extractTool,
   delegateTool,
   analyzeAllTool,
+  createExecutePlanTool,
   bashTool,
   editTool,
   createTool,
@@ -16,6 +17,7 @@ import {
   extractSchema,
   delegateSchema,
   analyzeAllSchema,
+  executePlanSchema,
   bashSchema,
   editSchema,
   createSchema,
@@ -24,6 +26,7 @@ import {
   extractToolDefinition,
   delegateToolDefinition,
   analyzeAllToolDefinition,
+  getExecutePlanToolDefinition,
   bashToolDefinition,
   editToolDefinition,
   createToolDefinition,
@@ -58,7 +61,10 @@ export function createTools(configOptions) {
   if (configOptions.enableDelegate && isToolAllowed('delegate')) {
     tools.delegateTool = delegateTool(configOptions);
   }
-  if (isToolAllowed('analyze_all')) {
+  if (configOptions.enableExecutePlan && isToolAllowed('execute_plan')) {
+    tools.executePlanTool = createExecutePlanTool(configOptions);
+  } else if (isToolAllowed('analyze_all')) {
+    // analyze_all is fallback when execute_plan is not enabled
     tools.analyzeAllTool = analyzeAllTool(configOptions);
   }
 
@@ -97,6 +103,7 @@ export {
   extractSchema,
   delegateSchema,
   analyzeAllSchema,
+  executePlanSchema,
   bashSchema,
   editSchema,
   createSchema,
@@ -106,6 +113,7 @@ export {
   extractToolDefinition,
   delegateToolDefinition,
   analyzeAllToolDefinition,
+  getExecutePlanToolDefinition,
   bashToolDefinition,
   editToolDefinition,
   createToolDefinition,

package/build/delegate.js

@@ -19,11 +19,14 @@ import { ProbeAgent } from './agent/ProbeAgent.js';
  * - For long-running processes, periodic cleanup of stale sessions may be needed
  */
 class DelegationManager {
-	constructor() {
-		this.maxConcurrent = parseInt(process.env.MAX_CONCURRENT_DELEGATIONS || '3', 10);
-		this.maxPerSession = parseInt(process.env.MAX_DELEGATIONS_PER_SESSION || '10', 10);
+	constructor(options = {}) {
+		this.maxConcurrent = options.maxConcurrent
+			?? parseInt(process.env.MAX_CONCURRENT_DELEGATIONS || '3', 10);
+		this.maxPerSession = options.maxPerSession
+			?? parseInt(process.env.MAX_DELEGATIONS_PER_SESSION || '10', 10);
 		// Default queue timeout: 60 seconds. Set DELEGATION_QUEUE_TIMEOUT=0 to disable.
-		this.defaultQueueTimeout = parseInt(process.env.DELEGATION_QUEUE_TIMEOUT || '60000', 10);
+		this.defaultQueueTimeout = options.queueTimeout
+			?? parseInt(process.env.DELEGATION_QUEUE_TIMEOUT || '60000', 10);
 
 		// Track delegations per session with timestamp for potential TTL cleanup
 		// Map<string, { count: number, lastUpdated: number }>
@@ -353,6 +356,7 @@ const DEFAULT_DELEGATE_TIMEOUT = parseInt(process.env.DELEGATE_TIMEOUT, 10) || 3
  * @param {boolean} [options.enableMcp=false] - Enable MCP tool integration (inherited from parent)
  * @param {Object} [options.mcpConfig] - MCP configuration object (inherited from parent)
  * @param {string} [options.mcpConfigPath] - Path to MCP configuration file (inherited from parent)
+ * @param {Object} [options.concurrencyLimiter=null] - Global AI concurrency limiter (DelegationManager instance)
  * @returns {Promise<string>} The response from the delegate agent
  */
 export async function delegate({
@@ -379,7 +383,8 @@ export async function delegate({
 	enableMcp = false,
 	mcpConfig = null,
 	mcpConfigPath = null,
-	delegationManager = null  // Optional per-instance manager, falls back to default singleton
+	delegationManager = null,  // Optional per-instance manager, falls back to default singleton
+	concurrencyLimiter = null  // Optional global AI concurrency limiter
 }) {
 	if (!task || typeof task !== 'string') {
 		throw new Error('Task parameter is required and must be a string');
@@ -464,7 +469,8 @@ export async function delegate({
 			enableTasks, // Inherit from parent (subagent gets isolated TaskManager)
 			enableMcp,   // Inherit from parent (subagent creates own MCPXmlBridge)
 			mcpConfig,   // Inherit from parent
-			mcpConfigPath // Inherit from parent
+			mcpConfigPath, // Inherit from parent
+			concurrencyLimiter // Inherit global AI concurrency limiter
 		});
 
 		if (debug) {

package/build/index.js

@@ -26,6 +26,7 @@ import {
 	extractSchema,
 	delegateSchema,
 	analyzeAllSchema,
+	executePlanSchema,
 	attemptCompletionSchema,
 	bashSchema,
 	searchToolDefinition,
@@ -46,6 +47,7 @@ import {
 	createToolDefinition
 } from './tools/edit.js';
 import { searchTool, queryTool, extractTool, delegateTool, analyzeAllTool } from './tools/vercel.js';
+import { createExecutePlanTool, getExecutePlanToolDefinition } from './tools/executePlan.js';
 import { bashTool } from './tools/bash.js';
 import { editTool, createTool } from './tools/edit.js';
 import { ProbeAgent } from './agent/ProbeAgent.js';
@@ -90,6 +92,7 @@ export {
 	extractTool,
 	delegateTool,
 	analyzeAllTool,
+	createExecutePlanTool,
 	bashTool,
 	editTool,
 	createTool,
@@ -102,6 +105,7 @@ export {
 	extractSchema,
 	delegateSchema,
 	analyzeAllSchema,
+	executePlanSchema,
 	attemptCompletionSchema,
 	bashSchema,
 	editSchema,
@@ -112,6 +116,7 @@ export {
 	extractToolDefinition,
 	delegateToolDefinition,
 	analyzeAllToolDefinition,
+	getExecutePlanToolDefinition,
 	attemptCompletionToolDefinition,
 	bashToolDefinition,
 	editToolDefinition,

package/build/tools/common.js

@@ -54,6 +54,11 @@ export const analyzeAllSchema = z.object({
 	path: z.string().optional().default('.').describe('Directory path to search in')
 });
 
+export const executePlanSchema = z.object({
+	code: z.string().min(1).describe('JavaScript DSL code to execute. All function calls look synchronous — do NOT use async/await. Use map(items, fn) for batch operations. Use LLM(instruction, data) for AI processing.'),
+	description: z.string().optional().describe('Human-readable description of what this plan does, for logging.')
+});
+
 // Schema for the attempt_completion tool - flexible validation for direct XML response
 export const attemptCompletionSchema = {
 	// Custom validation that requires result parameter but allows direct XML response
@@ -425,6 +430,7 @@ export const DEFAULT_VALID_TOOLS = [
 	'extract',
 	'delegate',
 	'analyze_all',
+	'execute_plan',
 	'listSkills',
 	'useSkill',
 	'listFiles',
@@ -463,6 +469,7 @@ function getValidParamsForTool(toolName) {
 		extract: extractSchema,
 		delegate: delegateSchema,
 		analyze_all: analyzeAllSchema,
+		execute_plan: executePlanSchema,
 		listSkills: listSkillsSchema,
 		useSkill: useSkillSchema,
 		bash: bashSchema,