@probelabs/probe 0.6.0-rc292 → 0.6.0-rc294

package/cjs/agent/ProbeAgent.cjs

@@ -27566,7 +27566,7 @@ var init_vercel = __esm({
         name: "search",
         description: searchDelegate ? searchDelegateDescription : searchDescription,
         inputSchema: searchSchema,
-        execute: async ({ query: searchQuery, path: path9, allow_tests, exact, maxTokens: paramMaxTokens, language, session, nextPage }) => {
+        execute: async ({ query: searchQuery, path: path9, allow_tests, exact, maxTokens: paramMaxTokens, language, session, nextPage, workingDirectory }) => {
           if (!exact && searchQuery) {
             const originalQuery = searchQuery;
             searchQuery = autoQuoteSearchTerms(searchQuery);
@@ -27575,18 +27575,19 @@ var init_vercel = __esm({
             }
           }
           const effectiveMaxTokens = paramMaxTokens || maxTokens;
+          const effectiveSearchCwd = workingDirectory || options.cwd || ".";
           let searchPaths;
           if (path9) {
-            searchPaths = parseAndResolvePaths(path9, options.cwd);
+            searchPaths = parseAndResolvePaths(path9, effectiveSearchCwd);
           }
           if (!searchPaths || searchPaths.length === 0) {
-            searchPaths = [options.cwd || "."];
+            searchPaths = [effectiveSearchCwd];
           }
           const searchPath = searchPaths.join(" ");
           const searchOptions = {
             query: searchQuery,
             path: searchPath,
-            cwd: options.cwd,
+            cwd: effectiveSearchCwd,
             // Working directory for resolving relative paths
             allowTests: allow_tests ?? true,
             exact,
@@ -27637,7 +27638,7 @@ var init_vercel = __esm({
             try {
               const result = maybeAnnotate(await runRawSearch());
               if (options.fileTracker && typeof result === "string") {
-                options.fileTracker.trackFilesFromOutput(result, options.cwd || ".").catch(() => {
+                options.fileTracker.trackFilesFromOutput(result, effectiveSearchCwd).catch(() => {
                 });
               }
               return result;
@@ -27690,7 +27691,7 @@ var init_vercel = __esm({
               }
               const fallbackResult = maybeAnnotate(await runRawSearch());
               if (options.fileTracker && typeof fallbackResult === "string") {
-                options.fileTracker.trackFilesFromOutput(fallbackResult, options.cwd || ".").catch(() => {
+                options.fileTracker.trackFilesFromOutput(fallbackResult, effectiveSearchCwd).catch(() => {
                 });
               }
               return fallbackResult;
@@ -27753,7 +27754,7 @@ var init_vercel = __esm({
             try {
               const fallbackResult2 = maybeAnnotate(await runRawSearch());
               if (options.fileTracker && typeof fallbackResult2 === "string") {
-                options.fileTracker.trackFilesFromOutput(fallbackResult2, options.cwd || ".").catch(() => {
+                options.fileTracker.trackFilesFromOutput(fallbackResult2, effectiveSearchCwd).catch(() => {
                 });
               }
               return fallbackResult2;
@@ -27807,9 +27808,9 @@ var init_vercel = __esm({
         name: "extract",
         description: extractDescription,
         inputSchema: extractSchema,
-        execute: async ({ targets, input_content, line, end_line, allow_tests, context_lines, format }) => {
+        execute: async ({ targets, input_content, line, end_line, allow_tests, context_lines, format, workingDirectory }) => {
           try {
-            const effectiveCwd = options.cwd || ".";
+            const effectiveCwd = workingDirectory || options.cwd || ".";
             if (debug) {
               if (targets) {
                 console.error(`Executing extract with targets: "${targets}", cwd: "${effectiveCwd}", context lines: ${context_lines || 10}`);
@@ -48638,6 +48639,10 @@ var init_file_lister = __esm({
 });
 
 // src/tools/fileTracker.js
+function normalizePath(filePath) {
+  if (!filePath) return filePath;
+  return (0, import_path10.resolve)(filePath);
+}
 function computeContentHash(content) {
   const normalized = (content || "").split("\n").map((l) => l.trimEnd()).join("\n");
   return (0, import_crypto3.createHash)("sha256").update(normalized).digest("hex").slice(0, 16);
@@ -48700,10 +48705,11 @@ var init_fileTracker = __esm({
        * @param {string} resolvedPath - Absolute path to the file
        */
       markFileSeen(resolvedPath) {
-        this._seenFiles.add(resolvedPath);
-        this._textEditCounts.set(resolvedPath, 0);
+        const normalized = normalizePath(resolvedPath);
+        this._seenFiles.add(normalized);
+        this._textEditCounts.set(normalized, 0);
         if (this.debug) {
-          console.error(`[FileTracker] Marked as seen: ${resolvedPath}`);
+          console.error(`[FileTracker] Marked as seen: ${normalized}`);
         }
       }
       /**
@@ -48712,7 +48718,7 @@ var init_fileTracker = __esm({
        * @returns {boolean}
        */
       isFileSeen(resolvedPath) {
-        return this._seenFiles.has(resolvedPath);
+        return this._seenFiles.has(normalizePath(resolvedPath));
       }
       /**
        * Store a content hash for a symbol in a file.
@@ -48724,7 +48730,7 @@ var init_fileTracker = __esm({
        * @param {string} [source='extract'] - How the content was obtained
        */
       trackSymbolContent(resolvedPath, symbolName, code, startLine, endLine, source = "extract") {
-        const key = `${resolvedPath}#${symbolName}`;
+        const key = `${normalizePath(resolvedPath)}#${symbolName}`;
         const contentHash = computeContentHash(code);
         this._contentRecords.set(key, {
           contentHash,
@@ -48745,7 +48751,7 @@ var init_fileTracker = __esm({
        * @returns {Object|null} The stored record or null
        */
       getSymbolRecord(resolvedPath, symbolName) {
-        return this._contentRecords.get(`${resolvedPath}#${symbolName}`) || null;
+        return this._contentRecords.get(`${normalizePath(resolvedPath)}#${symbolName}`) || null;
       }
       /**
        * Check if a symbol's current content matches what was stored.
@@ -48755,7 +48761,7 @@ var init_fileTracker = __esm({
        * @returns {{ok: boolean, reason?: string, message?: string}}
        */
       checkSymbolContent(resolvedPath, symbolName, currentCode) {
-        const key = `${resolvedPath}#${symbolName}`;
+        const key = `${normalizePath(resolvedPath)}#${symbolName}`;
         const record2 = this._contentRecords.get(key);
         if (!record2) {
           return { ok: true };
@@ -48832,7 +48838,7 @@ var init_fileTracker = __esm({
        * @returns {{ok: boolean, reason?: string, message?: string}}
        */
       checkBeforeEdit(resolvedPath) {
-        if (!this._seenFiles.has(resolvedPath)) {
+        if (!this._seenFiles.has(normalizePath(resolvedPath))) {
           return {
             ok: false,
             reason: "untracked",
@@ -48847,8 +48853,9 @@ var init_fileTracker = __esm({
        * @param {string} resolvedPath - Absolute path to the file
        */
       async trackFileAfterWrite(resolvedPath) {
-        this._seenFiles.add(resolvedPath);
-        this.invalidateFileRecords(resolvedPath);
+        const normalized = normalizePath(resolvedPath);
+        this._seenFiles.add(normalized);
+        this.invalidateFileRecords(normalized);
       }
       /**
        * Record a text-mode edit (old_string/new_string) to a file.
@@ -48856,10 +48863,11 @@ var init_fileTracker = __esm({
        * @param {string} resolvedPath - Absolute path to the file
        */
       recordTextEdit(resolvedPath) {
-        const count = (this._textEditCounts.get(resolvedPath) || 0) + 1;
-        this._textEditCounts.set(resolvedPath, count);
+        const normalized = normalizePath(resolvedPath);
+        const count = (this._textEditCounts.get(normalized) || 0) + 1;
+        this._textEditCounts.set(normalized, count);
         if (this.debug) {
-          console.error(`[FileTracker] Text edit #${count} for ${resolvedPath}`);
+          console.error(`[FileTracker] Text edit #${count} for ${normalized}`);
         }
       }
       /**
@@ -48868,7 +48876,7 @@ var init_fileTracker = __esm({
        * @returns {{ok: boolean, editCount?: number, message?: string}}
        */
       checkTextEditStaleness(resolvedPath) {
-        const count = this._textEditCounts.get(resolvedPath) || 0;
+        const count = this._textEditCounts.get(normalizePath(resolvedPath)) || 0;
         if (count >= this.maxConsecutiveTextEdits) {
           return {
             ok: false,
@@ -48897,7 +48905,7 @@ var init_fileTracker = __esm({
        * @param {string} resolvedPath - Absolute path to the file
        */
       invalidateFileRecords(resolvedPath) {
-        const prefix = resolvedPath + "#";
+        const prefix = normalizePath(resolvedPath) + "#";
         for (const key of this._contentRecords.keys()) {
           if (key.startsWith(prefix)) {
             this._contentRecords.delete(key);
@@ -48913,7 +48921,7 @@ var init_fileTracker = __esm({
        * @returns {boolean}
        */
       isTracked(resolvedPath) {
-        return this.isFileSeen(resolvedPath);
+        return this.isFileSeen(normalizePath(resolvedPath));
       }
       /**
        * Clear all tracking state.
@@ -69026,7 +69034,7 @@ var init_graph_builder = __esm({
       applyLinkStyles() {
         if (!this.pendingLinkStyles.length || !this.edges.length)
           return;
-        const normalize3 = (s) => {
+        const normalize4 = (s) => {
           const out = {};
           for (const [kRaw, vRaw] of Object.entries(s)) {
             const k = kRaw.trim().toLowerCase();
@@ -69047,7 +69055,7 @@ var init_graph_builder = __esm({
           return out;
         };
         for (const cmd of this.pendingLinkStyles) {
-          const style = normalize3(cmd.props);
+          const style = normalize4(cmd.props);
           for (const idx of cmd.indices) {
             if (idx >= 0 && idx < this.edges.length) {
               const e = this.edges[idx];
@@ -76314,7 +76322,7 @@ var require_layout = __commonJS({
     "use strict";
     var _ = require_lodash2();
     var acyclic = require_acyclic();
-    var normalize3 = require_normalize();
+    var normalize4 = require_normalize();
     var rank = require_rank();
     var normalizeRanks = require_util().normalizeRanks;
     var parentDummyChains = require_parent_dummy_chains();
@@ -76376,7 +76384,7 @@ var require_layout = __commonJS({
         removeEdgeLabelProxies(g);
       });
       time3("    normalize.run", function() {
-        normalize3.run(g);
+        normalize4.run(g);
       });
       time3("    parentDummyChains", function() {
         parentDummyChains(g);
@@ -76403,7 +76411,7 @@ var require_layout = __commonJS({
         removeBorderNodes(g);
       });
       time3("    normalize.undo", function() {
-        normalize3.undo(g);
+        normalize4.undo(g);
       });
       time3("    fixupEdgeLabelCoords", function() {
         fixupEdgeLabelCoords(g);
@@ -82774,8 +82782,8 @@ var require_resolve = __commonJS({
       }
       return count;
     }
-    function getFullPath(resolver, id = "", normalize3) {
-      if (normalize3 !== false)
+    function getFullPath(resolver, id = "", normalize4) {
+      if (normalize4 !== false)
         id = normalizeId(id);
       const p = resolver.parse(id);
       return _getFullPath(resolver, p);
@@ -84115,7 +84123,7 @@ var require_fast_uri = __commonJS({
     "use strict";
     var { normalizeIPv6, removeDotSegments, recomposeAuthority, normalizeComponentEncoding, isIPv4, nonSimpleDomain } = require_utils2();
     var { SCHEMES, getSchemeHandler } = require_schemes();
-    function normalize3(uri, options) {
+    function normalize4(uri, options) {
       if (typeof uri === "string") {
         uri = /** @type {T} */
         serialize(parse11(uri, options), options);
@@ -84351,7 +84359,7 @@ var require_fast_uri = __commonJS({
     }
     var fastUri = {
       SCHEMES,
-      normalize: normalize3,
+      normalize: normalize4,
       resolve: resolve9,
       resolveComponent,
       equal,
@@ -88558,9 +88566,9 @@ If the solution is clear, you can jump to implementation right away. If not, ask
 - Do not add code comments unless the logic is genuinely complex and non-obvious.
 
 # Before Implementation
-- Focus on high-level design patterns and system organization
-- Identify architectural patterns and component relationships
-- Evaluate system structure and suggest architectural improvements
+- Read tests first \u2014 find existing test files for the module you're changing. They reveal expected behavior, edge cases, and the project's testing patterns.
+- Read neighboring files \u2014 understand naming conventions, error handling patterns, import style, and existing utilities before creating new ones.
+- Trace the call chain \u2014 follow how the code you're changing is called and what depends on it. Check interfaces, types, and consumers.
 - Focus on backward compatibility
 - Consider scalability, maintainability, and extensibility in your analysis
 
@@ -88585,6 +88593,20 @@ Before building or testing, determine the project's toolchain:
 - Read README for build/test instructions if the above are unclear
 - Common patterns: \`make build\`/\`make test\`, \`npm run build\`/\`npm test\`, \`cargo build\`/\`cargo test\`, \`go build ./...\`/\`go test ./...\`, \`python -m pytest\`
 
+# File Editing Rules
+You have access to the \`edit\`, \`create\`, and \`multi_edit\` tools for modifying files. You MUST use these tools for ALL code changes. They are purpose-built, atomic, and safe.
+
+DO NOT use sed, awk, echo/cat redirection, or heredocs to modify source code. These commands cause real damage in practice: truncated lines, duplicate code blocks, broken syntax. Every bad edit wastes iterations on fix-up commits.
+
+Use the right tool:
+1. To MODIFY existing code \u2192 \`edit\` tool (old_string \u2192 new_string, or start_line/end_line)
+2. To CREATE a new file \u2192 \`create\` tool
+3. To CHANGE multiple files at once \u2192 \`multi_edit\` tool
+4. To READ code \u2192 \`extract\` or \`search\` tools
+5. If \`edit\` fails with "file has not been read yet" \u2192 use \`extract\` with the EXACT same file path you will pass to \`edit\`. Relative vs absolute path mismatch causes this error. Use the same path format consistently. If it still fails, use bash \`cat\` to read the file, then use \`create\` to write the entire modified file. Do NOT fall back to sed.
+
+Bash is fine for: formatters (gofmt, prettier, black), build/test/lint commands, git operations, and read-only file inspection (cat, head, tail). sed/awk should ONLY be used for trivial non-code tasks (e.g., config file tweaks) where the replacement is a simple literal string swap.
+
 # During Implementation
 - Always create a new branch before making changes to the codebase.
 - Fix problems at the root cause, not with surface-level patches. Prefer general solutions over special cases.
@@ -88611,6 +88633,22 @@ Before committing or creating a PR, run through this checklist:
 
 Do NOT skip verification. Do NOT proceed to PR creation with a broken build or failing tests.
 
+# Output Integrity
+Your final output MUST accurately reflect what ACTUALLY happened. Do NOT fabricate, hallucinate, or report aspirational results.
+
+- Only report PR URLs you actually created or updated with \`gh pr create\` or \`git push\`. If you checked out an existing PR but did NOT push changes to it, do NOT claim you updated it.
+- Describe what you ACTUALLY DID, not what you planned or intended to do. If you ran out of iterations, say so. If tests failed, say so.
+- Only list files you actually modified AND committed.
+- If you could not complete the task \u2014 ran out of iterations, tests failed, build broken, push rejected \u2014 report the real reason honestly.
+
+NEVER claim success when:
+- You did not run \`git push\` successfully
+- Tests failed and you did not fix them
+- You hit the iteration limit before completing the work
+- You only analyzed/investigated but did not implement changes
+
+A false success report is WORSE than an honest failure \u2014 it misleads the user into thinking work is done when it is not.
+
 # GitHub Integration
 - Use the \`gh\` CLI for all GitHub operations: issues, pull requests, checks, releases.
 - To view issues or PRs: \`gh issue view <number>\`, \`gh pr view <number>\`.
@@ -100582,9 +100620,13 @@ var init_ProbeAgent = __esm({
         }
         return await this.fallbackManager.executeWithFallback(
           async (provider, model, config2) => {
+            let fallbackModel = provider(model);
+            if (this.concurrencyLimiter) {
+              fallbackModel = _ProbeAgent._wrapModelWithLimiter(fallbackModel, this.concurrencyLimiter, this.debug);
+            }
             const fallbackOptions = {
               ...options,
-              model: provider(model),
+              model: fallbackModel,
               abortSignal: controller.signal
             };
             if (config2.provider !== "google" && fallbackOptions.tools) {
@@ -100612,6 +100654,132 @@ var init_ProbeAgent = __esm({
           }
         );
       }
+      /**
+       * Wrap a LanguageModelV1 model so each doStream/doGenerate call acquires and
+       * releases a concurrency limiter slot. This gates individual LLM API calls
+       * (seconds each) instead of entire multi-step agent sessions (minutes).
+       *
+       * @param {Object} model - LanguageModelV1 model instance
+       * @param {Object} limiter - Concurrency limiter with acquire/release/getStats
+       * @param {boolean} debug - Enable debug logging
+       * @returns {Object} Wrapped model with per-call concurrency gating
+       * @private
+       */
+      static _wrapModelWithLimiter(model, limiter, debug) {
+        return new Proxy(model, {
+          get(target, prop) {
+            if (prop === "doStream") {
+              return async function(...args) {
+                await limiter.acquire(null);
+                if (debug) {
+                  const stats = limiter.getStats();
+                  console.log(`[DEBUG] Acquired AI slot for LLM call (${stats.globalActive}/${stats.maxConcurrent}, queue: ${stats.queueSize})`);
+                }
+                try {
+                  const result = await target.doStream(...args);
+                  const originalStream = result.stream;
+                  const originalReader = originalStream.getReader();
+                  let released = false;
+                  const releaseOnce = () => {
+                    if (released) return;
+                    released = true;
+                    limiter.release(null);
+                  };
+                  const wrappedStream = new ReadableStream({
+                    async pull(controller) {
+                      try {
+                        const { done, value: value2 } = await originalReader.read();
+                        if (done) {
+                          controller.close();
+                          releaseOnce();
+                          if (debug) {
+                            const stats = limiter.getStats();
+                            console.log(`[DEBUG] Released AI slot after LLM stream complete (${stats.globalActive}/${stats.maxConcurrent})`);
+                          }
+                        } else {
+                          controller.enqueue(value2);
+                        }
+                      } catch (err) {
+                        releaseOnce();
+                        if (debug) {
+                          console.log(`[DEBUG] Released AI slot on LLM stream error`);
+                        }
+                        controller.error(err);
+                      }
+                    },
+                    cancel() {
+                      releaseOnce();
+                      if (debug) {
+                        console.log(`[DEBUG] Released AI slot on LLM stream cancel`);
+                      }
+                      originalReader.cancel();
+                    }
+                  });
+                  return { ...result, stream: wrappedStream };
+                } catch (err) {
+                  limiter.release(null);
+                  if (debug) {
+                    console.log(`[DEBUG] Released AI slot on doStream error`);
+                  }
+                  throw err;
+                }
+              };
+            }
+            if (prop === "doGenerate") {
+              return async function(...args) {
+                await limiter.acquire(null);
+                if (debug) {
+                  const stats = limiter.getStats();
+                  console.log(`[DEBUG] Acquired AI slot for LLM generate (${stats.globalActive}/${stats.maxConcurrent})`);
+                }
+                try {
+                  const result = await target.doGenerate(...args);
+                  return result;
+                } finally {
+                  limiter.release(null);
+                  if (debug) {
+                    const stats = limiter.getStats();
+                    console.log(`[DEBUG] Released AI slot after LLM generate (${stats.globalActive}/${stats.maxConcurrent})`);
+                  }
+                }
+              };
+            }
+            const value = target[prop];
+            return typeof value === "function" ? value.bind(target) : value;
+          }
+        });
+      }
+      /**
+       * Wrap an engine stream result so its textStream async generator acquires
+       * and releases a concurrency limiter slot. Acquire happens when iteration
+       * begins; release happens in finally (completion, error, or break).
+       *
+       * @param {Object} result - Engine result with { textStream, usage, ... }
+       * @param {Object} limiter - Concurrency limiter with acquire/release/getStats
+       * @param {boolean} debug - Enable debug logging
+       * @returns {Object} Result with wrapped textStream
+       * @private
+       */
+      static _wrapEngineStreamWithLimiter(result, limiter, debug) {
+        const originalStream = result.textStream;
+        async function* gatedStream() {
+          await limiter.acquire(null);
+          if (debug) {
+            const stats = limiter.getStats();
+            console.log(`[DEBUG] Acquired AI slot for engine stream (${stats.globalActive}/${stats.maxConcurrent}, queue: ${stats.queueSize})`);
+          }
+          try {
+            yield* originalStream;
+          } finally {
+            limiter.release(null);
+            if (debug) {
+              const stats = limiter.getStats();
+              console.log(`[DEBUG] Released AI slot after engine stream (${stats.globalActive}/${stats.maxConcurrent})`);
+            }
+          }
+        }
+        return { ...result, textStream: gatedStream() };
+      }
       /**
        * Execute streamText with retry and fallback support
        * @param {Object} options - streamText options
@@ -100620,12 +100788,8 @@ var init_ProbeAgent = __esm({
        */
       async streamTextWithRetryAndFallback(options) {
         const limiter = this.concurrencyLimiter;
-        if (limiter) {
-          await limiter.acquire(null);
-          if (this.debug) {
-            const stats = limiter.getStats();
-            console.log(`[DEBUG] Acquired global AI concurrency slot (${stats.globalActive}/${stats.maxConcurrent}, queue: ${stats.queueSize})`);
-          }
+        if (limiter && options.model) {
+          options = { ...options, model: _ProbeAgent._wrapModelWithLimiter(options.model, limiter, this.debug) };
         }
         const controller = new AbortController();
         const timeoutState = { timeoutId: null };
@@ -100653,6 +100817,9 @@ var init_ProbeAgent = __esm({
           if (useClaudeCode || useCodex) {
             try {
               result = await this._tryEngineStreamPath(options, controller, timeoutState);
+              if (result && limiter) {
+                result = _ProbeAgent._wrapEngineStreamWithLimiter(result, limiter, this.debug);
+              }
             } catch (error40) {
               if (this.debug) {
                 const engineType = useClaudeCode ? "Claude Code" : "Codex";
@@ -100663,41 +100830,7 @@ var init_ProbeAgent = __esm({
           if (!result) {
             result = await this._executeWithVercelProvider(options, controller);
           }
-          if (limiter && result.textStream) {
-            const originalStream = result.textStream;
-            const debug = this.debug;
-            const wrappedStream = (async function* () {
-              try {
-                for await (const chunk of originalStream) {
-                  yield chunk;
-                }
-              } finally {
-                limiter.release(null);
-                if (debug) {
-                  const stats = limiter.getStats();
-                  console.log(`[DEBUG] Released global AI concurrency slot (${stats.globalActive}/${stats.maxConcurrent}, queue: ${stats.queueSize})`);
-                }
-              }
-            })();
-            return new Proxy(result, {
-              get(target, prop) {
-                if (prop === "textStream") return wrappedStream;
-                const value = target[prop];
-                return typeof value === "function" ? value.bind(target) : value;
-              }
-            });
-          } else if (limiter) {
-            limiter.release(null);
-          }
           return result;
-        } catch (error40) {
-          if (limiter) {
-            limiter.release(null);
-            if (this.debug) {
-              console.log(`[DEBUG] Released global AI concurrency slot on error`);
-            }
-          }
-          throw error40;
         } finally {
           if (timeoutState.timeoutId) {
             clearTimeout(timeoutState.timeoutId);