@probelabs/probe 0.6.0-rc277 → 0.6.0-rc279

package/src/tools/analyzeAll.js

@@ -192,7 +192,8 @@ Instructions:
 			promptType: 'code-researcher',
 			allowedTools: ['extract'],
 			maxIterations: 5,
-			delegationManager: options.delegationManager  // Per-instance delegation limits
+			delegationManager: options.delegationManager,  // Per-instance delegation limits
+			parentAbortSignal: options.parentAbortSignal || null
 			// timeout removed - inherit default from delegate (300s)
 		});
 
@@ -328,7 +329,8 @@ Organize all findings into clear categories with items listed under each.${compl
 			promptType: 'code-researcher',
 			allowedTools: [],
 			maxIterations: 5,
-			delegationManager: options.delegationManager  // Per-instance delegation limits
+			delegationManager: options.delegationManager,  // Per-instance delegation limits
+			parentAbortSignal: options.parentAbortSignal || null
 			// timeout removed - inherit default from delegate (300s)
 		});
 
@@ -404,7 +406,8 @@ CRITICAL: Do NOT guess keywords. Actually run searches and see what returns resu
 			promptType: 'code-researcher',
 			// Full tool access for exploration and experimentation
 			maxIterations: 15,
-			delegationManager: options.delegationManager  // Per-instance delegation limits
+			delegationManager: options.delegationManager,  // Per-instance delegation limits
+			parentAbortSignal: options.parentAbortSignal || null
 			// timeout removed - inherit default from delegate (300s)
 		});
 
@@ -475,7 +478,8 @@ When done, use the attempt_completion tool with your answer as the result.`;
 			promptType: 'code-researcher',
 			allowedTools: [],
 			maxIterations: 5,
-			delegationManager: options.delegationManager  // Per-instance delegation limits
+			delegationManager: options.delegationManager,  // Per-instance delegation limits
+			parentAbortSignal: options.parentAbortSignal || null
 			// timeout removed - inherit default from delegate (300s)
 		});
 

package/src/tools/vercel.js

@@ -277,7 +277,8 @@ export const searchTool = (options = {}) => {
 					promptType: 'code-searcher',
 					allowedTools: ['search', 'extract', 'listFiles', 'attempt_completion'],
 					searchDelegate: false,
-					schema: CODE_SEARCH_SCHEMA
+					schema: CODE_SEARCH_SCHEMA,
+					parentAbortSignal: options.parentAbortSignal || null
 				});
 
 				const delegateResult = options.tracer?.withSpan
@@ -581,7 +582,7 @@ export const delegateTool = (options = {}) => {
 		name: 'delegate',
 		description: delegateDescription,
 		inputSchema: delegateSchema,
-		execute: async ({ task, currentIteration, maxIterations, parentSessionId, path, provider, model, tracer, searchDelegate }) => {
+		execute: async ({ task, currentIteration, maxIterations, parentSessionId, path, provider, model, tracer, searchDelegate, parentAbortSignal }) => {
 			// 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');
@@ -673,7 +674,8 @@ export const delegateTool = (options = {}) => {
 				enableMcp,
 				mcpConfig,
 				mcpConfigPath,
-				delegationManager  // Per-instance delegation limits
+				delegationManager,  // Per-instance delegation limits
+				parentAbortSignal
 			});
 
 			return result;
@@ -733,7 +735,8 @@ export const analyzeAllTool = (options = {}) => {
 					provider: options.provider,
 					model: options.model,
 					tracer: options.tracer,
-					delegationManager  // Per-instance delegation limits
+					delegationManager,  // Per-instance delegation limits
+					parentAbortSignal: options.parentAbortSignal || null
 				});
 
 				return result;