@probelabs/probe 0.6.0-rc278 → 0.6.0-rc280

package/build/agent/index.js

@@ -2293,9 +2293,7 @@ async function acquireFileLock(lockPath, version2) {
   };
   try {
     await fs3.writeFile(lockPath, JSON.stringify(lockData), { flag: "wx" });
-    if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
-      console.log(`Acquired file lock: ${lockPath}`);
-    }
+    console.log(`Acquired file lock: ${lockPath}`);
     return true;
   } catch (error) {
     if (error.code === "EEXIST") {
@@ -2303,15 +2301,11 @@ async function acquireFileLock(lockPath, version2) {
         const existingLock = JSON.parse(await fs3.readFile(lockPath, "utf-8"));
         const lockAge = Date.now() - existingLock.timestamp;
         if (lockAge > LOCK_TIMEOUT_MS) {
-          if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
-            console.log(`Removing stale lock file (age: ${Math.round(lockAge / 1e3)}s, pid: ${existingLock.pid})`);
-          }
+          console.log(`Removing stale lock file (age: ${Math.round(lockAge / 1e3)}s, pid: ${existingLock.pid})`);
           await fs3.remove(lockPath);
           return false;
         }
-        if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
-          console.log(`Download in progress by process ${existingLock.pid}, waiting...`);
-        }
+        console.log(`Download in progress by process ${existingLock.pid}, waiting...`);
         return false;
       } catch (readError) {
         if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
@@ -2352,36 +2346,36 @@ async function releaseFileLock(lockPath) {
 }
 async function waitForFileLock(lockPath, binaryPath) {
   const startTime = Date.now();
+  let lastStatusTime = startTime;
+  console.log(`Waiting for file lock to clear: ${lockPath}`);
   while (Date.now() - startTime < MAX_LOCK_WAIT_MS) {
     if (await fs3.pathExists(binaryPath)) {
-      if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
-        console.log(`Binary now available at ${binaryPath}, download completed by another process`);
-      }
+      const waitedSeconds = Math.round((Date.now() - startTime) / 1e3);
+      console.log(`Binary now available at ${binaryPath}, download completed by another process (waited ${waitedSeconds}s)`);
       return true;
     }
     const lockExists = await fs3.pathExists(lockPath);
     if (!lockExists) {
-      if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
-        console.log(`Lock file removed but binary not found - download may have failed`);
-      }
+      console.log(`Lock file removed but binary not found - download may have failed`);
       return false;
     }
     try {
       const lockData = JSON.parse(await fs3.readFile(lockPath, "utf-8"));
       const lockAge = Date.now() - lockData.timestamp;
       if (lockAge > LOCK_TIMEOUT_MS) {
-        if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
-          console.log(`Lock expired (age: ${Math.round(lockAge / 1e3)}s), will retry download`);
-        }
+        console.log(`Lock expired (age: ${Math.round(lockAge / 1e3)}s), will retry download`);
         return false;
       }
     } catch {
     }
+    if (Date.now() - lastStatusTime >= 15e3) {
+      const elapsedSeconds = Math.round((Date.now() - startTime) / 1e3);
+      console.log(`Still waiting for file lock (${elapsedSeconds}s/${MAX_LOCK_WAIT_MS / 1e3}s max)`);
+      lastStatusTime = Date.now();
+    }
     await new Promise((resolve9) => setTimeout(resolve9, LOCK_POLL_INTERVAL_MS));
   }
-  if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
-    console.log(`Timeout waiting for file lock`);
-  }
+  console.log(`Timeout waiting for file lock after ${MAX_LOCK_WAIT_MS / 1e3}s`);
   return false;
 }
 async function withDownloadLock(version2, downloadFn) {
@@ -2395,9 +2389,7 @@ async function withDownloadLock(version2, downloadFn) {
       }
       downloadLocks.delete(lockKey);
     } else {
-      if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
-        console.log(`Download already in progress in this process for version ${lockKey}, waiting...`);
-      }
+      console.log(`Download already in progress in this process for version ${lockKey}, waiting...`);
       try {
         return await lock.promise;
       } catch (error) {
@@ -2407,10 +2399,16 @@ async function withDownloadLock(version2, downloadFn) {
       }
     }
   }
+  let timeoutId = null;
   const downloadPromise = Promise.race([
     downloadFn(),
     new Promise(
-      (_, reject2) => setTimeout(() => reject2(new Error(`Download timeout after ${LOCK_TIMEOUT_MS / 1e3}s`)), LOCK_TIMEOUT_MS)
+      (_, reject2) => {
+        timeoutId = setTimeout(() => reject2(new Error(`Download timeout after ${LOCK_TIMEOUT_MS / 1e3}s`)), LOCK_TIMEOUT_MS);
+        if (timeoutId.unref) {
+          timeoutId.unref();
+        }
+      }
     )
   ]);
   downloadLocks.set(lockKey, {
@@ -2421,6 +2419,9 @@ async function withDownloadLock(version2, downloadFn) {
     const result = await downloadPromise;
     return result;
   } finally {
+    if (timeoutId) {
+      clearTimeout(timeoutId);
+    }
     downloadLocks.delete(lockKey);
   }
 }
@@ -3862,12 +3863,17 @@ async function delegate({
   mcpConfigPath = null,
   delegationManager = null,
   // Optional per-instance manager, falls back to default singleton
-  concurrencyLimiter = null
+  concurrencyLimiter = null,
   // Optional global AI concurrency limiter
+  parentAbortSignal = null
+  // Optional AbortSignal from parent to cancel this delegation
 }) {
   if (!task || typeof task !== "string") {
     throw new Error("Task parameter is required and must be a string");
   }
+  if (parentAbortSignal?.aborted) {
+    throw new Error("Delegation cancelled: parent operation was aborted");
+  }
   const hasExplicitTimeout = Object.prototype.hasOwnProperty.call(arguments?.[0] ?? {}, "timeout");
   if (!hasExplicitTimeout) {
     const envTimeoutMs = parseInt(process.env.DELEGATION_TIMEOUT_MS || "", 10);
@@ -3952,12 +3958,37 @@ async function delegate({
     }
     const timeoutPromise = new Promise((_, reject2) => {
       timeoutId = setTimeout(() => {
+        subagent.cancel();
         reject2(new Error(`Delegation timed out after ${timeout} seconds`));
       }, timeout * 1e3);
     });
+    let parentAbortHandler;
+    const parentAbortPromise = new Promise((_, reject2) => {
+      if (parentAbortSignal) {
+        if (parentAbortSignal.aborted) {
+          subagent.cancel();
+          reject2(new Error("Delegation cancelled: parent operation was aborted"));
+          return;
+        }
+        parentAbortHandler = () => {
+          subagent.cancel();
+          reject2(new Error("Delegation cancelled: parent operation was aborted"));
+        };
+        parentAbortSignal.addEventListener("abort", parentAbortHandler, { once: true });
+      }
+    });
     const answerOptions = schema ? { schema } : void 0;
     const answerPromise = answerOptions ? subagent.answer(task, [], answerOptions) : subagent.answer(task);
-    const response = await Promise.race([answerPromise, timeoutPromise]);
+    const racers = [answerPromise, timeoutPromise];
+    if (parentAbortSignal) racers.push(parentAbortPromise);
+    let response;
+    try {
+      response = await Promise.race(racers);
+    } finally {
+      if (parentAbortHandler && parentAbortSignal) {
+        parentAbortSignal.removeEventListener("abort", parentAbortHandler);
+      }
+    }
     if (timeoutId !== null) {
       clearTimeout(timeoutId);
       timeoutId = null;
@@ -4112,10 +4143,9 @@ var init_delegate = __esm({
         if (this.tryAcquire(parentSessionId)) {
           return true;
         }
-        if (debug) {
-          console.error(`[DelegationManager] Slot unavailable (${this.globalActive}/${this.maxConcurrent}), queuing... (queue size: ${this.waitQueue.length}, timeout: ${effectiveTimeout}ms)`);
-        }
+        console.error(`[DelegationManager] Slot unavailable (${this.globalActive}/${this.maxConcurrent}), queuing... (queue size: ${this.waitQueue.length + 1}, timeout: ${effectiveTimeout}ms)`);
         return new Promise((resolve9, reject2) => {
+          const queuedAt = Date.now();
           const entry = {
             resolve: null,
             // Will be wrapped below
@@ -4123,20 +4153,23 @@ var init_delegate = __esm({
             // Will be wrapped below
             parentSessionId,
             debug,
-            queuedAt: Date.now(),
-            timeoutId: null
+            queuedAt,
+            timeoutId: null,
+            reminderId: null
           };
           let settled = false;
           entry.resolve = (value) => {
             if (settled) return;
             settled = true;
             if (entry.timeoutId) clearTimeout(entry.timeoutId);
+            if (entry.reminderId) clearInterval(entry.reminderId);
             resolve9(value);
           };
           entry.reject = (error) => {
             if (settled) return;
             settled = true;
             if (entry.timeoutId) clearTimeout(entry.timeoutId);
+            if (entry.reminderId) clearInterval(entry.reminderId);
             reject2(error);
           };
           if (effectiveTimeout > 0) {
@@ -4148,6 +4181,13 @@ var init_delegate = __esm({
               entry.reject(new Error(`Delegation queue timeout: waited ${effectiveTimeout}ms for an available slot`));
             }, effectiveTimeout);
           }
+          entry.reminderId = setInterval(() => {
+            const waitedSeconds = Math.round((Date.now() - queuedAt) / 1e3);
+            console.error(`[DelegationManager] Still waiting for slot (${waitedSeconds}s). ${this.globalActive}/${this.maxConcurrent} active, ${this.waitQueue.length} queued.`);
+          }, 15e3);
+          if (entry.reminderId.unref) {
+            entry.reminderId.unref();
+          }
           this.waitQueue.push(entry);
         });
       }
@@ -4186,18 +4226,14 @@ var init_delegate = __esm({
             const sessionData = this.sessionDelegations.get(parentSessionId);
             const sessionCount = sessionData?.count || 0;
             if (sessionCount >= this.maxPerSession) {
-              if (debug) {
-                console.error(`[DelegationManager] Session limit (${this.maxPerSession}) reached for queued item, rejecting`);
-              }
+              console.error(`[DelegationManager] Session limit (${this.maxPerSession}) reached for queued item, rejecting`);
               toReject.push({ reject: reject2, error: new Error(`Maximum delegations per session (${this.maxPerSession}) reached for session ${parentSessionId}`) });
               continue;
             }
           }
           this._incrementCounters(parentSessionId);
-          if (debug) {
-            const waitTime = Date.now() - queuedAt;
-            console.error(`[DelegationManager] Granted slot from queue (waited ${waitTime}ms). Active: ${this.globalActive}/${this.maxConcurrent}`);
-          }
+          const waitTime = Date.now() - queuedAt;
+          console.error(`[DelegationManager] Granted slot from queue (waited ${waitTime}ms). Active: ${this.globalActive}/${this.maxConcurrent}`);
           toResolve.push(resolve9);
         }
         if (toResolve.length > 0 || toReject.length > 0) {
@@ -4247,6 +4283,9 @@ var init_delegate = __esm({
           if (entry.timeoutId) {
             clearTimeout(entry.timeoutId);
           }
+          if (entry.reminderId) {
+            clearInterval(entry.reminderId);
+          }
           if (entry.reject) {
             entry.reject(new Error("DelegationManager was cleaned up"));
           }
@@ -4369,8 +4408,9 @@ Instructions:
       promptType: "code-researcher",
       allowedTools: ["extract"],
       maxIterations: 5,
-      delegationManager: options.delegationManager
+      delegationManager: options.delegationManager,
       // Per-instance delegation limits
+      parentAbortSignal: options.parentAbortSignal || null
       // timeout removed - inherit default from delegate (300s)
     });
     return { chunk, result };
@@ -4390,16 +4430,12 @@ async function processChunksParallel(chunks, extractionPrompt, maxWorkers, optio
         return result;
       });
       active.add(promise);
-      if (options.debug) {
-        console.error(`[analyze_all] Started processing chunk ${chunk.id}/${chunk.total}`);
-      }
+      console.error(`[analyze_all] Started processing chunk ${chunk.id}/${chunk.total}`);
     }
     if (active.size > 0) {
       const result = await Promise.race(active);
       results.push(result);
-      if (options.debug) {
-        console.error(`[analyze_all] Completed chunk ${result.chunk.id}/${result.chunk.total}`);
-      }
+      console.error(`[analyze_all] Completed chunk ${result.chunk.id}/${result.chunk.total}`);
     }
   }
   results.sort((a, b) => a.chunk.id - b.chunk.id);
@@ -4469,8 +4505,9 @@ Organize all findings into clear categories with items listed under each.${compl
       promptType: "code-researcher",
       allowedTools: [],
       maxIterations: 5,
-      delegationManager: options.delegationManager
+      delegationManager: options.delegationManager,
       // Per-instance delegation limits
+      parentAbortSignal: options.parentAbortSignal || null
       // timeout removed - inherit default from delegate (300s)
     });
     return result;
@@ -4534,8 +4571,9 @@ 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
+      delegationManager: options.delegationManager,
       // Per-instance delegation limits
+      parentAbortSignal: options.parentAbortSignal || null
       // timeout removed - inherit default from delegate (300s)
     });
     const plan = parsePlanningResult(stripResultTags(result));
@@ -4592,8 +4630,9 @@ When done, use the attempt_completion tool with your answer as the result.`;
       promptType: "code-researcher",
       allowedTools: [],
       maxIterations: 5,
-      delegationManager: options.delegationManager
+      delegationManager: options.delegationManager,
       // Per-instance delegation limits
+      parentAbortSignal: options.parentAbortSignal || null
       // timeout removed - inherit default from delegate (300s)
     });
     return stripResultTags(result);
@@ -8890,13 +8929,13 @@ function resolveTargetPath(target, cwd) {
   }
   return filePart + suffix;
 }
-var searchSchema, searchAllSchema, querySchema, extractSchema, delegateSchema, listSkillsSchema, useSkillSchema, listFilesSchema, searchFilesSchema, readImageSchema, bashSchema, analyzeAllSchema, executePlanSchema, cleanupExecutePlanSchema, searchDescription, queryDescription, extractDescription, delegateDescription, analyzeAllDescription;
+var searchSchema, searchAllSchema, querySchema, extractSchema, delegateSchema, listSkillsSchema, useSkillSchema, listFilesSchema, searchFilesSchema, readImageSchema, bashSchema, analyzeAllSchema, executePlanSchema, cleanupExecutePlanSchema, searchDescription, searchDelegateDescription, queryDescription, extractDescription, delegateDescription, analyzeAllDescription;
 var init_common = __esm({
   "src/tools/common.js"() {
     "use strict";
     init_zod();
     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."),
+      query: external_exports.string().describe("Search query \u2014 natural language questions or Elasticsearch-style keywords both work. For keywords: use quotes for exact phrases, AND/OR for boolean logic, - for negation. Probe handles stemming and camelCase/snake_case splitting automatically, so do NOT try case or style variations of the same keyword."),
       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.'),
       maxTokens: external_exports.number().nullable().optional().describe("Maximum tokens to return. Default is 20000. Set to null for unlimited results."),
@@ -8904,7 +8943,7 @@ var init_common = __esm({
       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.")
     });
     searchAllSchema = 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."),
+      query: external_exports.string().describe("Search query \u2014 natural language questions or Elasticsearch-style keywords both work. For keywords: use quotes for exact phrases, AND/OR for boolean logic, - for negation. Probe handles stemming and camelCase/snake_case splitting automatically, so do NOT try case or style variations of the same keyword."),
       path: external_exports.string().optional().default(".").describe("Path to search in."),
       exact: external_exports.boolean().optional().default(false).describe("Use exact matching instead of stemming."),
       maxTokensPerPage: external_exports.number().optional().default(2e4).describe("Tokens per page when paginating. Default 20000."),
@@ -8959,7 +8998,8 @@ var init_common = __esm({
       clearOutputBuffer: external_exports.boolean().optional().default(true).describe("Clear the output buffer from previous execute_plan calls"),
       clearSessionStore: external_exports.boolean().optional().default(false).describe("Clear the session store (persisted data across execute_plan calls)")
     });
-    searchDescription = "Search code in the repository. Free-form questions are accepted, but Elasticsearch-style keyword queries work best. Use this tool first for any code-related questions.";
+    searchDescription = 'Search code in the repository. Free-form questions are accepted, but Elasticsearch-style keyword queries work best. Use this tool first for any code-related questions. NOTE: By default, search handles stemming, case-insensitive matching, and camelCase/snake_case splitting automatically \u2014 do NOT manually try keyword variations like "getAllUsers" then "get_all_users" then "GetAllUsers". One search covers all variations.';
+    searchDelegateDescription = 'Search code in the repository by asking a question. Accepts natural language questions (e.g., "How does authentication work?", "Where is the user validation logic?"). A specialized subagent breaks down your question into targeted keyword searches and returns extracted code blocks. Do NOT formulate keyword queries yourself \u2014 just ask the question naturally.';
     queryDescription = "Search code using ast-grep structural pattern matching. Use this tool to find specific code structures like functions, classes, or methods.";
     extractDescription = "Extract code blocks from files based on file paths and optional line numbers. Use this tool to see complete context after finding relevant files. Line numbers from output can be used with edit start_line/end_line for precise editing.";
     delegateDescription = "Automatically delegate big distinct tasks to specialized probe subagents within the agentic loop. Used by AI agents to break down complex requests into focused, parallel tasks.";
@@ -9144,11 +9184,41 @@ function buildSearchDelegateTask({ searchQuery, searchPath, exact, language, all
     "- extract: Verify code snippets to ensure targets are actually relevant before including them.",
     "- listFiles: Understand directory structure to find where relevant code might live.",
     "",
-    "Strategy for complex queries:",
+    "CRITICAL - How probe search works (do NOT ignore):",
+    "- By default (exact=false), probe ALREADY handles stemming, case-insensitive matching, and camelCase/snake_case splitting.",
+    '- Searching "allowed_ips" ALREADY matches "AllowedIPs", "allowedIps", "allowed_ips", etc. Do NOT manually try case/style variations.',
+    '- Searching "getUserData" ALREADY matches "get", "user", "data" and their variations.',
+    "- NEVER repeat the same search query \u2014 you will get the same results.",
+    "- NEVER search trivial variations of the same keyword (e.g., AllowedIPs then allowedIps then allowed_ips). This is wasteful \u2014 probe handles it.",
+    "- If a search returns no results, the term likely does not exist in that path. Try a genuinely DIFFERENT keyword or concept, not a variation.",
+    "- If 2-3 consecutive searches return no results for a concept, STOP searching for it and move on.",
+    "",
+    "GOOD search strategy (do this):",
+    '  Query: "How does authentication work and how are sessions managed?"',
+    '  \u2192 search "authentication" \u2192 search "session management" (two different concepts)',
+    '  Query: "Find the IP allowlist middleware"',
+    '  \u2192 search "allowlist middleware" (one search, probe handles IP/ip/Ip variations)',
+    '  Query: "How does BM25 scoring work with SIMD optimization?"',
+    '  \u2192 search "BM25 scoring" \u2192 search "SIMD optimization" (two different concepts)',
+    "",
+    "BAD search strategy (never do this):",
+    '  \u2192 search "AllowedIPs" \u2192 search "allowedIps" \u2192 search "allowed_ips" (WRONG: these are trivial case variations, probe handles them)',
+    '  \u2192 search "CIDR" \u2192 search "cidr" \u2192 search "Cidr" \u2192 search "*cidr*" (WRONG: same keyword repeated with variations)',
+    '  \u2192 search "error handling" \u2192 search "error handling" \u2192 search "error handling" (WRONG: repeating exact same query)',
+    "",
+    "Keyword tips:",
+    "- Common programming keywords are filtered as stopwords when unquoted: function, class, return, new, struct, impl, var, let, const, etc.",
+    '- Avoid searching for these alone \u2014 combine with a specific term (e.g., "middleware function" is fine, "function" alone is too generic).',
+    '- To bypass stopword filtering: wrap terms in quotes ("return", "struct") or set exact=true. Both disable stemming and splitting too.',
+    "- Multiple words without operators use OR logic: foo bar = foo OR bar. Use AND explicitly if you need both: foo AND bar.",
+    '- camelCase terms are split: getUserData becomes "get", "user", "data" \u2014 so one search covers all naming styles.',
+    "",
+    "Strategy:",
     "1. Analyze the query - identify key concepts, entities, and relationships",
-    '2. Run focused searches for each independent concept (e.g., for "how do payments work and how are emails sent", search "payments" and "emails" separately since they are unrelated)',
-    "3. Use extract to verify relevance of promising results",
-    "4. Combine all relevant targets in your final response",
+    "2. Run ONE focused search per concept with the most natural keyword. Trust probe to handle variations.",
+    "3. If a search returns results, use extract to verify relevance",
+    "4. Only try a different keyword if the first one returned irrelevant results (not if it returned no results \u2014 that means the concept is absent)",
+    "5. Combine all relevant targets in your final response",
     "",
     `Query: ${searchQuery}`,
     `Search path(s): ${searchPath}`,
@@ -9199,9 +9269,12 @@ var init_vercel = __esm({
         }
         return result;
       };
+      const previousSearches = /* @__PURE__ */ new Set();
+      const paginationCounts = /* @__PURE__ */ new Map();
+      const MAX_PAGES_PER_QUERY = 3;
       return tool({
         name: "search",
-        description: searchDelegate ? `${searchDescription} (delegates code search to a subagent and returns extracted code blocks)` : searchDescription,
+        description: searchDelegate ? searchDelegateDescription : searchDescription,
         inputSchema: searchSchema,
         execute: async ({ query: searchQuery, path: path9, allow_tests, exact, maxTokens: paramMaxTokens, language, session, nextPage }) => {
           const effectiveMaxTokens = paramMaxTokens || maxTokens;
@@ -9239,6 +9312,26 @@ var init_vercel = __esm({
             return await search(searchOptions);
           };
           if (!searchDelegate) {
+            const searchKey = `${searchQuery}::${searchPath}::${exact || false}`;
+            if (!nextPage) {
+              if (previousSearches.has(searchKey)) {
+                if (debug) {
+                  console.error(`[DEDUP] Blocked duplicate search: "${searchQuery}" in "${searchPath}"`);
+                }
+                return "DUPLICATE SEARCH BLOCKED: You already searched for this exact query in this path. Do NOT repeat the same search. If you need more results, set nextPage=true with the session ID from the previous search. Otherwise, try a genuinely different keyword, use extract to examine results you already found, or use attempt_completion if you have enough information.";
+              }
+              previousSearches.add(searchKey);
+              paginationCounts.set(searchKey, 0);
+            } else {
+              const pageCount = (paginationCounts.get(searchKey) || 0) + 1;
+              paginationCounts.set(searchKey, pageCount);
+              if (pageCount > MAX_PAGES_PER_QUERY) {
+                if (debug) {
+                  console.error(`[DEDUP] Blocked excessive pagination (page ${pageCount}/${MAX_PAGES_PER_QUERY}): "${searchQuery}" in "${searchPath}"`);
+                }
+                return `PAGINATION LIMIT REACHED: You have already retrieved ${MAX_PAGES_PER_QUERY} pages of results for this query. You have enough results \u2014 use extract to examine specific files, or use attempt_completion to return your findings.`;
+              }
+            }
             try {
               const result = maybeAnnotate(await runRawSearch());
               if (options.fileTracker && typeof result === "string") {
@@ -9277,7 +9370,8 @@ var init_vercel = __esm({
               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 ? await options.tracer.withSpan("search.delegate", runDelegation, {
               "search.query": searchQuery,
@@ -9491,7 +9585,7 @@ var init_vercel = __esm({
         name: "delegate",
         description: delegateDescription,
         inputSchema: delegateSchema,
-        execute: async ({ task, currentIteration, maxIterations, parentSessionId, path: path9, provider, model, tracer, searchDelegate }) => {
+        execute: async ({ task, currentIteration, maxIterations, parentSessionId, path: path9, provider, model, tracer, searchDelegate, parentAbortSignal }) => {
           if (!task || typeof task !== "string") {
             throw new Error("Task parameter is required and must be a non-empty string");
           }
@@ -9549,8 +9643,9 @@ var init_vercel = __esm({
             enableMcp,
             mcpConfig,
             mcpConfigPath,
-            delegationManager
+            delegationManager,
             // Per-instance delegation limits
+            parentAbortSignal
           });
           return result;
         }
@@ -9588,8 +9683,9 @@ var init_vercel = __esm({
               provider: options.provider,
               model: options.model,
               tracer: options.tracer,
-              delegationManager
+              delegationManager,
               // Per-instance delegation limits
+              parentAbortSignal: options.parentAbortSignal || null
             });
             return result;
           } catch (error) {
@@ -22106,6 +22202,7 @@ function generateSandboxGlobals(options) {
       executing.add(p);
       results.push(p);
       if (executing.size >= mapConcurrency) {
+        console.error(`[map] Concurrency limit reached (${executing.size}/${mapConcurrency}), waiting for a slot...`);
         await Promise.race(executing);
       }
     }
@@ -81768,7 +81865,9 @@ __export(ProbeAgent_exports, {
   ENGINE_ACTIVITY_TIMEOUT_DEFAULT: () => ENGINE_ACTIVITY_TIMEOUT_DEFAULT,
   ENGINE_ACTIVITY_TIMEOUT_MAX: () => ENGINE_ACTIVITY_TIMEOUT_MAX,
   ENGINE_ACTIVITY_TIMEOUT_MIN: () => ENGINE_ACTIVITY_TIMEOUT_MIN,
-  ProbeAgent: () => ProbeAgent
+  ProbeAgent: () => ProbeAgent,
+  debugLogToolResults: () => debugLogToolResults,
+  debugTruncate: () => debugTruncate
 });
 import dotenv2 from "dotenv";
 import { createAnthropic as createAnthropic2 } from "@ai-sdk/anthropic";
@@ -81781,6 +81880,19 @@ import { EventEmitter as EventEmitter5 } from "events";
 import { existsSync as existsSync7 } from "fs";
 import { readFile as readFile3, stat, readdir as readdir3 } from "fs/promises";
 import { resolve as resolve7, isAbsolute as isAbsolute6, dirname as dirname5, basename, normalize as normalize2, sep as sep5 } from "path";
+function debugTruncate(s, limit = 200) {
+  if (s.length <= limit) return s;
+  const half = Math.floor(limit / 2);
+  return s.substring(0, half) + ` ... [${s.length} chars] ... ` + s.substring(s.length - half);
+}
+function debugLogToolResults(toolResults) {
+  if (!toolResults || toolResults.length === 0) return;
+  for (const tr of toolResults) {
+    const argsStr = JSON.stringify(tr.args || {});
+    const resultStr = typeof tr.result === "string" ? tr.result : JSON.stringify(tr.result || "");
+    console.log(`[DEBUG]   tool: ${tr.toolName} | args: ${debugTruncate(argsStr)} | result: ${debugTruncate(resultStr)}`);
+  }
+}
 var ENGINE_ACTIVITY_TIMEOUT_DEFAULT, ENGINE_ACTIVITY_TIMEOUT_MIN, ENGINE_ACTIVITY_TIMEOUT_MAX, MAX_TOOL_ITERATIONS, MAX_HISTORY_MESSAGES, MAX_IMAGE_FILE_SIZE, ProbeAgent;
 var init_ProbeAgent = __esm({
   "src/agent/ProbeAgent.js"() {
@@ -81889,6 +82001,7 @@ var init_ProbeAgent = __esm({
         this.enableExecutePlan = !!options.enableExecutePlan;
         this.debug = options.debug || process.env.DEBUG === "1";
         this.cancelled = false;
+        this._abortController = new AbortController();
         this.tracer = options.tracer || null;
         this.outline = !!options.outline;
         this.searchDelegate = options.searchDelegate !== void 0 ? !!options.searchDelegate : true;
@@ -81917,6 +82030,7 @@ var init_ProbeAgent = __esm({
         this.completionPrompt = options.completionPrompt || null;
         this.thinkingEffort = options.thinkingEffort || null;
         const effectiveAllowedTools = options.disableTools ? [] : options.allowedTools;
+        this._rawAllowedTools = options.allowedTools;
         this.allowedTools = this._parseAllowedTools(effectiveAllowedTools);
         this.storageAdapter = options.storageAdapter || new InMemoryStorageAdapter();
         this.hooks = new HookManager();
@@ -82059,6 +82173,16 @@ var init_ProbeAgent = __esm({
       _filterMcpTools(mcpToolNames) {
         return mcpToolNames.filter((toolName) => this._isMcpToolAllowed(toolName));
       }
+      /**
+       * Check if query tool was explicitly listed in allowedTools (not via wildcard).
+       * Query (ast-grep) is excluded by default because models struggle with AST pattern syntax.
+       * @returns {boolean}
+       * @private
+       */
+      _isQueryExplicitlyAllowed() {
+        if (!this._rawAllowedTools) return false;
+        return Array.isArray(this._rawAllowedTools) && this._rawAllowedTools.includes("query");
+      }
       /**
        * Check if tracer is AppTracer (expects sessionId as first param) vs SimpleAppTracer
        * @returns {boolean} - True if tracer is AppTracer style (requires sessionId)
@@ -82334,6 +82458,8 @@ var init_ProbeAgent = __esm({
           searchDelegateModel: this.searchDelegateModel,
           delegationManager: this.delegationManager,
           // Per-instance delegation limits
+          parentAbortSignal: this._abortController.signal,
+          // Propagate cancellation to delegations
           outputBuffer: this._outputBuffer,
           concurrencyLimiter: this.concurrencyLimiter,
           // Global AI concurrency limiter
@@ -82350,7 +82476,7 @@ var init_ProbeAgent = __esm({
         if (wrappedTools.searchToolInstance && isToolAllowed("search")) {
           this.toolImplementations.search = wrappedTools.searchToolInstance;
         }
-        if (wrappedTools.queryToolInstance && isToolAllowed("query")) {
+        if (wrappedTools.queryToolInstance && isToolAllowed("query") && this._isQueryExplicitlyAllowed()) {
           this.toolImplementations.query = wrappedTools.queryToolInstance;
         }
         if (wrappedTools.extractToolInstance && isToolAllowed("extract")) {
@@ -82781,6 +82907,15 @@ var init_ProbeAgent = __esm({
         }
         const controller = new AbortController();
         const timeoutState = { timeoutId: null };
+        if (this._abortController.signal.aborted) {
+          controller.abort();
+        } else {
+          const onAgentAbort = () => controller.abort();
+          this._abortController.signal.addEventListener("abort", onAgentAbort, { once: true });
+          controller.signal.addEventListener("abort", () => {
+            this._abortController.signal.removeEventListener("abort", onAgentAbort);
+          }, { once: true });
+        }
         if (this.maxOperationTimeout && this.maxOperationTimeout > 0) {
           timeoutState.timeoutId = setTimeout(() => {
             controller.abort();
@@ -83084,7 +83219,8 @@ var init_ProbeAgent = __esm({
                     allowEdit: this.allowEdit,
                     allowedTools: allowedToolsForDelegate,
                     debug: this.debug,
-                    tracer: this.tracer
+                    tracer: this.tracer,
+                    parentAbortSignal: this._abortController.signal
                   };
                   if (this.debug) {
                     console.log(`[DEBUG] Executing delegate tool`);
@@ -83279,12 +83415,13 @@ var init_ProbeAgent = __esm({
         const toolMap = {
           search: {
             schema: searchSchema,
-            description: "Search code in the repository using keyword queries with Elasticsearch syntax."
-          },
-          query: {
-            schema: querySchema,
-            description: "Search code using ast-grep structural pattern matching."
+            description: this.searchDelegate ? "Search code in the repository by asking a question. Accepts natural language questions \u2014 a subagent breaks them into targeted keyword searches and returns extracted code blocks. Do NOT formulate keyword queries yourself." : "Search code in the repository using keyword queries with Elasticsearch syntax. Handles stemming, case-insensitive matching, and camelCase/snake_case splitting automatically \u2014 do NOT try keyword variations manually."
           },
+          // query tool (ast-grep) removed from AI-facing tools — models struggle with pattern syntax
+          // query: {
+          //   schema: querySchema,
+          //   description: 'Search code using ast-grep structural pattern matching.'
+          // },
           extract: {
             schema: extractSchema,
             description: "Extract code blocks from files based on file paths and optional line numbers."
@@ -83952,25 +84089,27 @@ ${this.architectureContext.content}
         } else {
           systemPrompt += predefinedPrompts["code-explorer"] + "\n\n";
         }
+        const searchToolDesc1 = this.searchDelegate ? '- search: Ask natural language questions to find code (e.g., "How does authentication work?"). A subagent handles keyword searches and returns extracted code blocks. Do NOT formulate keyword queries \u2014 just ask questions.' : "- search: Find code patterns using keyword queries with Elasticsearch syntax. Handles stemming and case variations automatically \u2014 do NOT try manual keyword variations.";
         systemPrompt += `You have access to powerful code search and analysis tools through MCP:
-- search: Find code patterns using semantic search
+${searchToolDesc1}
 - extract: Extract specific code sections with context
-- query: Use AST patterns for structural code matching
 - listFiles: Browse directory contents
 - searchFiles: Find files by name patterns`;
         if (this.enableBash) {
           systemPrompt += `
 - bash: Execute bash commands for system operations`;
         }
-        const searchGuidance = this.searchDelegate ? "1. Start with search to retrieve extracted code blocks" : "1. Start with search to find relevant code patterns";
-        const extractGuidance = this.searchDelegate ? "2. Use extract only if you need more context or a full file" : "2. Use extract to get detailed context when needed";
+        const searchGuidance1 = this.searchDelegate ? "1. Start with search \u2014 ask a question about what you want to understand. It returns extracted code blocks directly." : "1. Start with search to find relevant code patterns. One search per concept is usually enough \u2014 probe handles stemming and case variations.";
+        const extractGuidance1 = this.searchDelegate ? "2. Use extract only if you need more context or a full file" : "2. Use extract to get detailed context when needed";
         systemPrompt += `
 
 When exploring code:
-${searchGuidance}
-${extractGuidance}
+${searchGuidance1}
+${extractGuidance1}
 3. Prefer focused, specific searches over broad queries
-4. Combine multiple tools to build complete understanding`;
+4. Do NOT repeat the same search or try trivial keyword variations \u2014 probe handles stemming and case variations automatically
+5. If 2-3 consecutive searches return no results for a concept, stop searching for it \u2014 the term likely does not exist in that codebase
+6. Combine multiple tools to build complete understanding`;
         if (this.allowedFolders && this.allowedFolders.length > 0) {
           systemPrompt += `
 
@@ -84005,25 +84144,27 @@ Workspace: ${this.allowedFolders.join(", ")}`;
         } else {
           systemPrompt += predefinedPrompts["code-explorer"] + "\n\n";
         }
+        const searchToolDesc2 = this.searchDelegate ? '- search: Ask natural language questions to find code (e.g., "How does authentication work?"). A subagent handles keyword searches and returns extracted code blocks. Do NOT formulate keyword queries \u2014 just ask questions.' : "- search: Find code patterns using keyword queries with Elasticsearch syntax. Handles stemming and case variations automatically \u2014 do NOT try manual keyword variations.";
         systemPrompt += `You have access to powerful code search and analysis tools through MCP:
-- search: Find code patterns using semantic search
+${searchToolDesc2}
 - extract: Extract specific code sections with context
-- query: Use AST patterns for structural code matching
 - listFiles: Browse directory contents
 - searchFiles: Find files by name patterns`;
         if (this.enableBash) {
           systemPrompt += `
 - bash: Execute bash commands for system operations`;
         }
-        const searchGuidance = this.searchDelegate ? "1. Start with search to retrieve extracted code blocks" : "1. Start with search to find relevant code patterns";
-        const extractGuidance = this.searchDelegate ? "2. Use extract only if you need more context or a full file" : "2. Use extract to get detailed context when needed";
+        const searchGuidance2 = this.searchDelegate ? "1. Start with search \u2014 ask a question about what you want to understand. It returns extracted code blocks directly." : "1. Start with search to find relevant code patterns. One search per concept is usually enough \u2014 probe handles stemming and case variations.";
+        const extractGuidance2 = this.searchDelegate ? "2. Use extract only if you need more context or a full file" : "2. Use extract to get detailed context when needed";
         systemPrompt += `
 
 When exploring code:
-${searchGuidance}
-${extractGuidance}
+${searchGuidance2}
+${extractGuidance2}
 3. Prefer focused, specific searches over broad queries
-4. Combine multiple tools to build complete understanding`;
+4. Do NOT repeat the same search or try trivial keyword variations \u2014 probe handles stemming and case variations automatically
+5. If 2-3 consecutive searches return no results for a concept, stop searching for it \u2014 the term likely does not exist in that codebase
+6. Combine multiple tools to build complete understanding`;
         if (this.allowedFolders && this.allowedFolders.length > 0) {
           systemPrompt += `
 
@@ -84074,10 +84215,10 @@ Workspace: ${this.allowedFolders.join(", ")}`;
 Follow these instructions carefully:
 1. Analyze the user's request.
 2. Use the available tools step-by-step to fulfill the request.
-3. You should always prefer the search tool for code-related questions.${this.searchDelegate ? " It already returns extracted code blocks; use extract only to expand context or read full files." : " Read full files only if really necessary."}
+3. You should always prefer the search tool for code-related questions.${this.searchDelegate ? " Ask natural language questions \u2014 the search subagent handles keyword formulation and returns extracted code blocks. Use extract only to expand context or read full files." : " Search handles stemming and case variations automatically \u2014 do NOT try keyword variations manually. Read full files only if really necessary."}
 4. Ensure to get really deep and understand the full picture before answering.
 5. Once the task is fully completed, use the attempt_completion tool to provide the final result.
-6. Prefer concise and focused search queries. Use specific keywords and phrases to narrow down results.${this.allowEdit ? `
+6. ${this.searchDelegate ? "Ask clear, specific questions when searching. Each search should target a distinct concept or question." : "Prefer concise and focused search queries. Use specific keywords and phrases to narrow down results."}${this.allowEdit ? `
 7. When modifying files, choose the appropriate tool:
     - Use 'edit' for all code modifications:
       * PREFERRED: Use start_line (and optionally end_line) for line-targeted editing \u2014 this is the safest and most precise approach.${this.hashLines ? ' Use the line:hash references from extract/search output (e.g. "42:ab") for integrity verification.' : ""} Always use extract first to see line numbers${this.hashLines ? " and hashes" : ""}, then edit by line reference.
@@ -84401,6 +84542,10 @@ You are working with a workspace. Available paths: ${workspaceDesc}
             completionResult = result;
             completionAttempted = true;
           }, toolContext);
+          if (this.debug) {
+            const toolNames = Object.keys(tools2);
+            console.log(`[DEBUG] Agent tools registered (${toolNames.length}): ${toolNames.join(", ")}`);
+          }
           let maxResponseTokens = this.maxResponseTokens;
           if (!maxResponseTokens) {
             maxResponseTokens = 4e3;
@@ -84440,6 +84585,7 @@ You are working with a workspace. Available paths: ${workspaceDesc}
                   }
                   if (this.debug) {
                     console.log(`[DEBUG] Step ${currentIteration}/${maxIterations} finished (reason: ${finishReason}, tools: ${toolResults?.length || 0})`);
+                    debugLogToolResults(toolResults);
                   }
                 }
               };
@@ -84596,6 +84742,7 @@ Double-check your response based on the criteria above. If everything looks good
                   }
                   if (this.debug) {
                     console.log(`[DEBUG] Completion prompt step finished (reason: ${finishReason}, tools: ${toolResults?.length || 0})`);
+                    debugLogToolResults(toolResults);
                   }
                 }
               };
@@ -85306,6 +85453,9 @@ Convert your previous response content into actual JSON data that follows this s
        * Clean up resources (including MCP connections)
        */
       async cleanup() {
+        if (!this._abortController.signal.aborted) {
+          this._abortController.abort();
+        }
         if (this.mcpBridge) {
           try {
             await this.mcpBridge.cleanup();
@@ -85329,14 +85479,25 @@ Convert your previous response content into actual JSON data that follows this s
         this.clearHistory();
       }
       /**
-       * Cancel the current request
+       * Cancel the current request and all in-flight delegations.
+       * Aborts the internal AbortController so streamText, subagents,
+       * and any code checking the signal will stop.
        */
       cancel() {
         this.cancelled = true;
+        this._abortController.abort();
         if (this.debug) {
           console.log(`[DEBUG] Agent cancelled for session ${this.sessionId}`);
         }
       }
+      /**
+       * Get the abort signal for this agent.
+       * Delegations and subagents should check this signal.
+       * @returns {AbortSignal}
+       */
+      get abortSignal() {
+        return this._abortController.signal;
+      }
     };
   }
 });