@probelabs/probe 0.6.0-rc154 → 0.6.0-rc161

package/build/tools/vercel.js

@@ -250,23 +250,64 @@ export const delegateTool = (options = {}) => {
 		name: 'delegate',
 		description: delegateDescription,
 		inputSchema: delegateSchema,
-		execute: async ({ task }) => {
-			try {
-				if (debug) {
-					console.error(`Executing delegate with task: "${task}"`);
-				}
+		execute: async ({ task, currentIteration, maxIterations, parentSessionId, path, provider, model, tracer }) => {
+			// Validate required parameters - throw errors for consistency
+			if (!task || typeof task !== 'string') {
+				throw new Error('Task parameter is required and must be a non-empty string');
+			}
 
-				const result = await delegate({
-					task,
-					timeout,
-					debug
-				});
+			if (task.trim().length === 0) {
+				throw new Error('Task parameter cannot be empty or whitespace only');
+			}
 
-				return result;
-			} catch (error) {
-				console.error('Error executing delegate command:', error);
-				return `Error executing delegate command: ${error.message}`;
+			// Validate optional numeric parameters
+			if (currentIteration !== undefined && (typeof currentIteration !== 'number' || currentIteration < 0)) {
+				throw new Error('currentIteration must be a non-negative number');
+			}
+
+			if (maxIterations !== undefined && (typeof maxIterations !== 'number' || maxIterations < 1)) {
+				throw new Error('maxIterations must be a positive number');
+			}
+
+			// Validate optional string parameters for type consistency
+			if (parentSessionId !== undefined && parentSessionId !== null && typeof parentSessionId !== 'string') {
+				throw new TypeError('parentSessionId must be a string, null, or undefined');
 			}
+
+			if (path !== undefined && path !== null && typeof path !== 'string') {
+				throw new TypeError('path must be a string, null, or undefined');
+			}
+
+			if (provider !== undefined && provider !== null && typeof provider !== 'string') {
+				throw new TypeError('provider must be a string, null, or undefined');
+			}
+
+			if (model !== undefined && model !== null && typeof model !== 'string') {
+				throw new TypeError('model must be a string, null, or undefined');
+			}
+
+			if (debug) {
+				console.error(`Executing delegate with task: "${task.substring(0, 100)}${task.length > 100 ? '...' : ''}"`);
+				if (parentSessionId) {
+					console.error(`Parent session: ${parentSessionId}`);
+				}
+			}
+
+			// Execute delegation - let errors propagate naturally
+			const result = await delegate({
+				task,
+				timeout,
+				debug,
+				currentIteration: currentIteration || 0,
+				maxIterations: maxIterations || 30,
+				parentSessionId,
+				path,
+				provider,
+				model,
+				tracer
+			});
+
+			return result;
 		}
 	});
 };

package/build/utils.js

@@ -25,9 +25,9 @@ let probeBinaryPath = '';
  * @returns {Promise<string>} - Path to the binary
  */
 export async function getBinaryPath(options = {}) {
-	const { forceDownload = false, version } = options;
+    const { forceDownload = false, version } = options;
 
-	// Check environment variable first (user override)
+    // Check environment variable first (user override)
 	if (process.env.PROBE_PATH && fs.existsSync(process.env.PROBE_PATH) && !forceDownload) {
 		probeBinaryPath = process.env.PROBE_PATH;
 		return probeBinaryPath;
@@ -40,13 +40,22 @@ export async function getBinaryPath(options = {}) {
 		return probeBinaryPath;
 	}
 
-	// Get dynamic bin directory (handles CI, npx, Docker scenarios)
-	const binDir = await getPackageBinDir();
-
-	// Check postinstall binary in package directory (most up-to-date with npm package version)
-	const isWindows = process.platform === 'win32';
-	const binaryName = isWindows ? 'probe.exe' : 'probe-binary';
-	const binaryPath = path.join(binDir, binaryName);
+    // Prefer local package bin if available (avoids network during tests/monorepo)
+    const isWindows = process.platform === 'win32';
+    const binaryName = isWindows ? 'probe.exe' : 'probe-binary';
+    const localPackageBin = path.resolve(__dirname, '..', 'bin');
+    const localBinaryPath = path.join(localPackageBin, binaryName);
+    if (fs.existsSync(localBinaryPath) && !forceDownload) {
+        // Use committed binary bundled with the repo/package
+        probeBinaryPath = localBinaryPath;
+        return probeBinaryPath;
+    }
+
+    // Get dynamic bin directory (handles CI, npx, Docker scenarios)
+    const binDir = await getPackageBinDir();
+
+    // Check postinstall binary in resolved directory
+    const binaryPath = path.join(binDir, binaryName);
 
 	if (fs.existsSync(binaryPath) && !forceDownload) {
 		probeBinaryPath = binaryPath;