@probelabs/probe 0.6.0-rc221 → 0.6.0-rc223

package/build/agent/acp/tools.js

@@ -287,6 +287,18 @@ export class ACPToolManager {
             allow_tests: {
               type: 'boolean',
               description: 'Include test files in results (default: true)'
+            },
+            exact: {
+              type: 'boolean',
+              description: 'Default (false) enables stemming and keyword splitting for exploratory search. Set true for precise symbol lookup where the query matches only the exact term. Use true when you know the exact symbol name.'
+            },
+            session: {
+              type: 'string',
+              description: 'Session ID for result caching and pagination. Pass the session ID from a previous search to get additional results (next page). Results already shown in a session are automatically excluded.'
+            },
+            nextPage: {
+              type: 'boolean',
+              description: 'Set to true when requesting the next page of results. Requires passing the same session ID from the previous search output.'
             }
           },
           required: ['query']

package/build/agent/index.js

@@ -3748,7 +3748,7 @@ var init_grep = __esm({
 import { randomUUID } from "crypto";
 async function delegate({
   task,
-  timeout = 300,
+  timeout = DEFAULT_DELEGATE_TIMEOUT,
   debug = false,
   currentIteration = 0,
   maxIterations = 30,
@@ -3926,7 +3926,7 @@ async function delegate({
     throw new Error(`Delegation failed: ${error.message}`);
   }
 }
-var DelegationManager, defaultDelegationManager;
+var DelegationManager, defaultDelegationManager, DEFAULT_DELEGATE_TIMEOUT;
 var init_delegate = __esm({
   "src/delegate.js"() {
     "use strict";
@@ -4063,9 +4063,12 @@ var init_delegate = __esm({
       }
       /**
        * Process the wait queue - grant slot to next waiting delegation
+       * Uses setImmediate to avoid blocking the event loop when resolving promises.
        * @private
        */
       _processQueue(debug = false) {
+        const toResolve = [];
+        const toReject = [];
         while (this.waitQueue.length > 0 && this.globalActive < this.maxConcurrent) {
           const next = this.waitQueue.shift();
           if (!next) break;
@@ -4077,7 +4080,7 @@ var init_delegate = __esm({
               if (debug) {
                 console.error(`[DelegationManager] Session limit (${this.maxPerSession}) reached for queued item, rejecting`);
               }
-              reject2(new Error(`Maximum delegations per session (${this.maxPerSession}) reached for session ${parentSessionId}`));
+              toReject.push({ reject: reject2, error: new Error(`Maximum delegations per session (${this.maxPerSession}) reached for session ${parentSessionId}`) });
               continue;
             }
           }
@@ -4086,7 +4089,17 @@ var init_delegate = __esm({
             const waitTime = Date.now() - queuedAt;
             console.error(`[DelegationManager] Granted slot from queue (waited ${waitTime}ms). Active: ${this.globalActive}/${this.maxConcurrent}`);
           }
-          resolve8(true);
+          toResolve.push(resolve8);
+        }
+        if (toResolve.length > 0 || toReject.length > 0) {
+          setImmediate(() => {
+            for (const resolve8 of toResolve) {
+              resolve8(true);
+            }
+            for (const { reject: reject2, error } of toReject) {
+              reject2(error);
+            }
+          });
         }
       }
       /**
@@ -4135,6 +4148,7 @@ var init_delegate = __esm({
       }
     };
     defaultDelegationManager = new DelegationManager();
+    DEFAULT_DELEGATE_TIMEOUT = parseInt(process.env.DELEGATE_TIMEOUT, 10) || 300;
   }
 });
 
@@ -9715,7 +9729,10 @@ var init_common = __esm({
     init_taskTool();
     searchSchema = external_exports.object({
       query: external_exports.string().describe("Search query with Elasticsearch syntax. Use quotes for exact matches, AND/OR for boolean logic, - for negation."),
-      path: external_exports.string().optional().default(".").describe('Path to search in. For dependencies use "go:github.com/owner/repo", "js:package_name", or "rust:cargo_name" etc.')
+      path: external_exports.string().optional().default(".").describe('Path to search in. For dependencies use "go:github.com/owner/repo", "js:package_name", or "rust:cargo_name" etc.'),
+      exact: external_exports.boolean().optional().default(false).describe('Default (false) enables stemming and keyword splitting for exploratory search - "getUserData" matches "get", "user", "data", etc. Set true for precise symbol lookup where "getUserData" matches only "getUserData". Use true when you know the exact symbol name.'),
+      session: external_exports.string().optional().describe("Session ID for result caching and pagination. Pass the session ID from a previous search to get additional results (next page). Results already shown in a session are automatically excluded. Omit for a fresh search."),
+      nextPage: external_exports.boolean().optional().default(false).describe("Set to true when requesting the next page of results. Requires passing the same session ID from the previous search output.")
     });
     querySchema = external_exports.object({
       pattern: external_exports.string().describe("AST pattern to search for. Use $NAME for variable names, $$$PARAMS for parameter lists, etc."),
@@ -9814,6 +9831,9 @@ You need to focus on main keywords when constructing the query, and always use e
 Parameters:
 - query: (required) Search query. Free-form questions are accepted, but for best results prefer Elasticsearch-style syntax with quotes for exact matches ("functionName"), AND/OR for boolean logic, - for negation, + for important terms.
 - path: (optional, default: '.') Path to search in. All dependencies located in /dep folder, under language sub folders, like this: "/dep/go/github.com/owner/repo", "/dep/js/package_name", or "/dep/rust/cargo_name" etc.
+- exact: (optional, default: false) Set to true for precise symbol lookup without stemming/tokenization. Use when you know the exact symbol name (e.g., "getUserData" matches only "getUserData", not "get", "user", "data").
+- session: (optional) Session ID for pagination. Pass the session ID returned from a previous search to get the next page of results. Results already shown are automatically excluded.
+- nextPage: (optional, default: false) Set to true when requesting the next page of results. Requires passing the same session ID from the previous search.
 
 **Workflow:** Always start with search, then use extract for detailed context when needed.
 
@@ -10220,7 +10240,7 @@ var init_vercel = __esm({
         name: "search",
         description: searchDelegate ? `${searchDescription} (delegates code search to a subagent and returns extracted code blocks)` : searchDescription,
         inputSchema: searchSchema,
-        execute: async ({ query: searchQuery, path: path9, allow_tests, exact, maxTokens: paramMaxTokens, language }) => {
+        execute: async ({ query: searchQuery, path: path9, allow_tests, exact, maxTokens: paramMaxTokens, language, session, nextPage }) => {
           const effectiveMaxTokens = paramMaxTokens || maxTokens;
           let searchPaths;
           if (path9) {
@@ -10239,8 +10259,10 @@ var init_vercel = __esm({
             exact,
             json: false,
             maxTokens: effectiveMaxTokens,
-            session: sessionId,
-            // Pass session ID if provided
+            session: session || sessionId,
+            // Use explicit session param, or fall back to options sessionId
+            nextPage,
+            // Pass nextPage parameter for pagination
             language
             // Pass language parameter if provided
           };

package/build/delegate.js

@@ -194,9 +194,14 @@ class DelegationManager {
 
 	/**
 	 * Process the wait queue - grant slot to next waiting delegation
+	 * Uses setImmediate to avoid blocking the event loop when resolving promises.
 	 * @private
 	 */
 	_processQueue(debug = false) {
+		// Collect callbacks to invoke after the loop to avoid blocking the event loop
+		const toResolve = [];
+		const toReject = [];
+
 		// Process queue items one at a time when slots are available
 		// Items are only removed when they can be granted or must be rejected
 		while (this.waitQueue.length > 0 && this.globalActive < this.maxConcurrent) {
@@ -216,7 +221,7 @@ class DelegationManager {
 					if (debug) {
 						console.error(`[DelegationManager] Session limit (${this.maxPerSession}) reached for queued item, rejecting`);
 					}
-					reject(new Error(`Maximum delegations per session (${this.maxPerSession}) reached for session ${parentSessionId}`));
+					toReject.push({ reject, error: new Error(`Maximum delegations per session (${this.maxPerSession}) reached for session ${parentSessionId}`) });
 					// Continue to process next item in queue
 					continue;
 				}
@@ -230,7 +235,21 @@ class DelegationManager {
 				console.error(`[DelegationManager] Granted slot from queue (waited ${waitTime}ms). Active: ${this.globalActive}/${this.maxConcurrent}`);
 			}
 
-			resolve(true);
+			toResolve.push(resolve);
+		}
+
+		// Defer promise resolutions/rejections to next tick to avoid blocking the event loop.
+		// This is critical: synchronous resolve()/reject() calls in a tight loop can saturate
+		// the microtask queue and prevent other async operations from proceeding.
+		if (toResolve.length > 0 || toReject.length > 0) {
+			setImmediate(() => {
+				for (const resolve of toResolve) {
+					resolve(true);
+				}
+				for (const { reject, error } of toReject) {
+					reject(error);
+				}
+			});
 		}
 	}
 
@@ -293,6 +312,9 @@ const defaultDelegationManager = new DelegationManager();
 // Export the class for per-instance usage
 export { DelegationManager };
 
+// Default delegation timeout from environment variable or 300 seconds (5 minutes)
+const DEFAULT_DELEGATE_TIMEOUT = parseInt(process.env.DELEGATE_TIMEOUT, 10) || 300;
+
 /**
  * Delegate a big distinct task to a probe subagent (used automatically by AI agents)
  *
@@ -309,7 +331,7 @@ export { DelegationManager };
  *
  * @param {Object} options - Delegate options
  * @param {string} options.task - A complete, self-contained task for the subagent. Should be specific and focused on one area of expertise.
- * @param {number} [options.timeout=300] - Timeout in seconds (default: 5 minutes)
+ * @param {number} [options.timeout] - Timeout in seconds (default: DELEGATE_TIMEOUT env var or 300)
  * @param {boolean} [options.debug=false] - Enable debug logging
  * @param {number} [options.currentIteration=0] - Current tool iteration count from parent agent
  * @param {number} [options.maxIterations=30] - Maximum tool iterations allowed
@@ -335,7 +357,7 @@ export { DelegationManager };
  */
 export async function delegate({
 	task,
-	timeout = 300,
+	timeout = DEFAULT_DELEGATE_TIMEOUT,
 	debug = false,
 	currentIteration = 0,
 	maxIterations = 30,

package/build/tools/common.js

@@ -11,7 +11,10 @@ import { taskSchema } from '../agent/tasks/taskTool.js';
 // Common schemas for tool parameters (used for internal execution after XML parsing)
 export const searchSchema = z.object({
 	query: z.string().describe('Search query with Elasticsearch syntax. Use quotes for exact matches, AND/OR for boolean logic, - for negation.'),
-	path: z.string().optional().default('.').describe('Path to search in. For dependencies use "go:github.com/owner/repo", "js:package_name", or "rust:cargo_name" etc.')
+	path: z.string().optional().default('.').describe('Path to search in. For dependencies use "go:github.com/owner/repo", "js:package_name", or "rust:cargo_name" etc.'),
+	exact: z.boolean().optional().default(false).describe('Default (false) enables stemming and keyword splitting for exploratory search - "getUserData" matches "get", "user", "data", etc. Set true for precise symbol lookup where "getUserData" matches only "getUserData". Use true when you know the exact symbol name.'),
+	session: z.string().optional().describe('Session ID for result caching and pagination. Pass the session ID from a previous search to get additional results (next page). Results already shown in a session are automatically excluded. Omit for a fresh search.'),
+	nextPage: z.boolean().optional().default(false).describe('Set to true when requesting the next page of results. Requires passing the same session ID from the previous search output.')
 });
 
 export const querySchema = z.object({
@@ -130,6 +133,9 @@ You need to focus on main keywords when constructing the query, and always use e
 Parameters:
 - query: (required) Search query. Free-form questions are accepted, but for best results prefer Elasticsearch-style syntax with quotes for exact matches ("functionName"), AND/OR for boolean logic, - for negation, + for important terms.
 - path: (optional, default: '.') Path to search in. All dependencies located in /dep folder, under language sub folders, like this: "/dep/go/github.com/owner/repo", "/dep/js/package_name", or "/dep/rust/cargo_name" etc.
+- exact: (optional, default: false) Set to true for precise symbol lookup without stemming/tokenization. Use when you know the exact symbol name (e.g., "getUserData" matches only "getUserData", not "get", "user", "data").
+- session: (optional) Session ID for pagination. Pass the session ID returned from a previous search to get the next page of results. Results already shown are automatically excluded.
+- nextPage: (optional, default: false) Set to true when requesting the next page of results. Requires passing the same session ID from the previous search.
 
 **Workflow:** Always start with search, then use extract for detailed context when needed.
 

package/build/tools/langchain.js

@@ -16,7 +16,7 @@ export function createSearchTool(options = {}) {
 		name: 'search',
 		description: searchDescription,
 		schema: searchSchema,
-		func: async ({ query: searchQuery, path, allow_tests, exact, maxResults, maxTokens = 20000, language }) => {
+		func: async ({ query: searchQuery, path, allow_tests, exact, maxResults, maxTokens = 20000, language, session, nextPage }) => {
 			try {
 				const results = await search({
 					query: searchQuery,
@@ -27,7 +27,9 @@ export function createSearchTool(options = {}) {
 					json: false,
 					maxResults,
 					maxTokens,
-					language
+					language,
+					session,
+					nextPage
 				});
 
 				return results;

package/build/tools/vercel.js

@@ -169,7 +169,7 @@ export const searchTool = (options = {}) => {
 			? `${searchDescription} (delegates code search to a subagent and returns extracted code blocks)`
 			: searchDescription,
 		inputSchema: searchSchema,
-		execute: async ({ query: searchQuery, path, allow_tests, exact, maxTokens: paramMaxTokens, language }) => {
+		execute: async ({ query: searchQuery, path, allow_tests, exact, maxTokens: paramMaxTokens, language, session, nextPage }) => {
 			// Use parameter maxTokens if provided, otherwise use the default
 			const effectiveMaxTokens = paramMaxTokens || maxTokens;
 
@@ -195,7 +195,8 @@ export const searchTool = (options = {}) => {
 				exact,
 				json: false,
 				maxTokens: effectiveMaxTokens,
-				session: sessionId, // Pass session ID if provided
+				session: session || sessionId, // Use explicit session param, or fall back to options sessionId
+				nextPage, // Pass nextPage parameter for pagination
 				language // Pass language parameter if provided
 			};