@probelabs/probe 0.6.0-rc232 → 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/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,