@probelabs/probe 0.6.0-rc291 → 0.6.0-rc293

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}`);
@@ -29908,7 +29909,14 @@ function lineTrimmedMatch(contentLines, searchLines) {
       }
     }
     if (allMatch) {
-      const matchedText = contentLines.slice(i, i + windowSize).join("\n");
+      const windowLines = contentLines.slice(i, i + windowSize);
+      const windowMinIndent = getMinIndent(windowLines);
+      const searchMinIndent = getMinIndent(searchLines);
+      const indentDiff = Math.abs(windowMinIndent - searchMinIndent);
+      if (isIndentDiffTooLarge(windowLines, searchLines, indentDiff)) {
+        continue;
+      }
+      const matchedText = windowLines.join("\n");
       matches.push(matchedText);
     }
   }
@@ -29938,6 +29946,15 @@ function whitespaceNormalizedMatch(content, search2) {
       actualEnd++;
     }
     const matchedText = content.substring(originalStart, actualEnd);
+    const matchedLines = matchedText.split("\n");
+    const searchLines = search2.split("\n");
+    const matchMinIndent = getMinIndent(matchedLines);
+    const searchMinIndent = getMinIndent(searchLines);
+    const indentDiff = Math.abs(matchMinIndent - searchMinIndent);
+    if (isIndentDiffTooLarge(matchedLines, searchLines, indentDiff)) {
+      searchStart = idx + 1;
+      continue;
+    }
     matches.push(matchedText);
     searchStart = idx + 1;
   }
@@ -29989,6 +30006,10 @@ function indentFlexibleMatch(contentLines, searchLines) {
       }
     }
     if (allMatch) {
+      const indentDiff = Math.abs(windowMinIndent - searchMinIndent);
+      if (isIndentDiffTooLarge(windowLines, searchLines, indentDiff)) {
+        continue;
+      }
       const matchedText = windowLines.join("\n");
       matches.push(matchedText);
     }
@@ -29999,6 +30020,14 @@ function indentFlexibleMatch(contentLines, searchLines) {
     count: matches.length
   };
 }
+function isIndentDiffTooLarge(linesA, linesB, indentDiff) {
+  if (indentDiff <= 0) return false;
+  const sampleA = linesA.find((l) => l.trim().length > 0) || "";
+  const sampleB = linesB.find((l) => l.trim().length > 0) || "";
+  const useTabs = sampleA.startsWith("	") || sampleB.startsWith("	");
+  const maxAllowedDiff = useTabs ? 1 : 4;
+  return indentDiff > maxAllowedDiff;
+}
 function getMinIndent(lines) {
   let min = Infinity;
   for (const line of lines) {
@@ -30158,6 +30187,12 @@ function restoreIndentation(newStr, originalLines) {
   const targetIndent = detectBaseIndent(originalCode);
   const newIndent = detectBaseIndent(newStr);
   if (targetIndent !== newIndent) {
+    const indentDiff = Math.abs(targetIndent.length - newIndent.length);
+    const useTabs = targetIndent.includes("	") || newIndent.includes("	");
+    const maxAllowedDiff = useTabs ? 1 : 4;
+    if (indentDiff > maxAllowedDiff) {
+      return { result: newStr, modifications };
+    }
     const reindented = reindent(newStr, targetIndent);
     if (reindented !== newStr) {
       modifications.push(`reindented from "${newIndent}" to "${targetIndent}"`);
@@ -40531,7 +40566,12 @@ function generateSandboxGlobals(options) {
         }
         return tryParseJSONValue(text);
       };
-      globals[name15] = traceToolCall(name15, rawMcpFn, tracer, logFn);
+      const tracedFn = traceToolCall(name15, rawMcpFn, tracer, logFn);
+      globals[name15] = tracedFn;
+      const sanitized = name15.replace(/[^a-zA-Z0-9_$]/g, "_");
+      if (sanitized !== name15) {
+        globals[sanitized] = tracedFn;
+      }
     }
   }
   if (llmCall) {
@@ -40886,9 +40926,17 @@ ${validation.errors.join("\n")}`,
         "dsl.duration_ms": elapsed,
         "dsl.error": e.message?.substring(0, 500)
       });
+      let errorMsg = `Execution failed: ${e.message}`;
+      if (e.message && e.message.includes("is not defined")) {
+        const globalNames = Object.keys(toolGlobals).sort();
+        errorMsg += `
+Available functions: ${globalNames.join(", ")}`;
+        errorMsg += `
+Note: Tools with hyphens (e.g. "my-tool") are available with underscores: my_tool()`;
+      }
       return {
         status: "error",
-        error: `Execution failed: ${e.message}`,
+        error: errorMsg,
         logs
       };
     }
@@ -48591,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);
@@ -48653,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}`);
         }
       }
       /**
@@ -48665,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.
@@ -48677,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,
@@ -48698,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.
@@ -48708,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 };
@@ -48785,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",
@@ -48800,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.
@@ -48809,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}`);
         }
       }
       /**
@@ -48821,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,
@@ -48850,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);
@@ -48866,7 +48921,7 @@ var init_fileTracker = __esm({
        * @returns {boolean}
        */
       isTracked(resolvedPath) {
-        return this.isFileSeen(resolvedPath);
+        return this.isFileSeen(normalizePath(resolvedPath));
       }
       /**
        * Clear all tracking state.
@@ -68979,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();
@@ -69000,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];
@@ -76267,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();
@@ -76329,7 +76384,7 @@ var require_layout = __commonJS({
         removeEdgeLabelProxies(g);
       });
       time3("    normalize.run", function() {
-        normalize3.run(g);
+        normalize4.run(g);
       });
       time3("    parentDummyChains", function() {
         parentDummyChains(g);
@@ -76356,7 +76411,7 @@ var require_layout = __commonJS({
         removeBorderNodes(g);
       });
       time3("    normalize.undo", function() {
-        normalize3.undo(g);
+        normalize4.undo(g);
       });
       time3("    fixupEdgeLabelCoords", function() {
         fixupEdgeLabelCoords(g);
@@ -82727,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);
@@ -84068,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);
@@ -84304,7 +84359,7 @@ var require_fast_uri = __commonJS({
     }
     var fastUri = {
       SCHEMES,
-      normalize: normalize3,
+      normalize: normalize4,
       resolve: resolve9,
       resolveComponent,
       equal,
@@ -88511,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
 
@@ -88538,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.
@@ -88564,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>\`.