@probelabs/probe 0.6.0-rc285 → 0.6.0-rc287

package/cjs/agent/ProbeAgent.cjs

@@ -24981,7 +24981,11 @@ async function extractBinary(assetPath, outputDir) {
       if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
         console.log(`Extracting zip to ${extractDir}...`);
       }
-      await exec(`unzip -q "${assetPath}" -d "${extractDir}"`);
+      if (isWindows) {
+        await exec(`powershell -NoProfile -Command "Expand-Archive -Path '${assetPath}' -DestinationPath '${extractDir}' -Force"`);
+      } else {
+        await exec(`unzip -q "${assetPath}" -d "${extractDir}"`);
+      }
     } else {
       if (process.env.DEBUG === "1" || process.env.VERBOSE === "1") {
         console.log(`Copying binary directly to ${binaryPath}`);
@@ -27066,7 +27070,14 @@ function parseTargets(targets) {
 }
 function parseAndResolvePaths(pathStr, cwd) {
   if (!pathStr) return [];
-  const paths = pathStr.split(",").map((p) => p.trim()).filter((p) => p.length > 0);
+  let paths = pathStr.split(",").map((p) => p.trim()).filter((p) => p.length > 0);
+  paths = paths.flatMap((p) => {
+    if (!/\s/.test(p)) return [p];
+    const parts = p.split(/\s+/).filter(Boolean);
+    if (parts.length <= 1) return [p];
+    const allLookLikePaths = parts.every((part) => /[/\\]/.test(part) || /\.\w+/.test(part));
+    return allLookLikePaths ? parts : [p];
+  });
   return paths.map((p) => {
     if ((0, import_path5.isAbsolute)(p)) {
       return p;
@@ -27299,11 +27310,24 @@ function normalizeTargets(targets) {
     if (typeof target !== "string") continue;
     const trimmed = target.trim();
     if (!trimmed || seen.has(trimmed)) continue;
-    seen.add(trimmed);
-    normalized.push(trimmed);
+    const subTargets = splitSpaceSeparatedPaths(trimmed);
+    for (const sub of subTargets) {
+      if (!seen.has(sub)) {
+        seen.add(sub);
+        normalized.push(sub);
+      }
+    }
   }
   return normalized;
 }
+function splitSpaceSeparatedPaths(target) {
+  if (!/\s/.test(target)) return [target];
+  const parts = target.split(/\s+/).filter(Boolean);
+  if (parts.length <= 1) return [target];
+  const allLookLikePaths = parts.every((p) => /[/\\]/.test(p) || /\.\w+/.test(p));
+  if (allLookLikePaths) return parts;
+  return [target];
+}
 function extractJsonSnippet(text) {
   const jsonBlockMatch = text.match(/```json\s*([\s\S]*?)```/i);
   if (jsonBlockMatch) {
@@ -27789,14 +27813,15 @@ var init_vercel = __esm({
               const parsedTargets = parseTargets(targets);
               extractFiles = parsedTargets.map((target) => resolveTargetPath(target, effectiveCwd));
               if (options.allowedFolders && options.allowedFolders.length > 0) {
+                const { join: pathJoin, sep: pathSep } = await import("path");
                 extractFiles = extractFiles.map((target) => {
                   const { filePart, suffix } = splitTargetSuffix(target);
                   if ((0, import_fs4.existsSync)(filePart)) return target;
-                  const cwdPrefix = effectiveCwd.endsWith("/") ? effectiveCwd : effectiveCwd + "/";
+                  const cwdPrefix = effectiveCwd.endsWith(pathSep) ? effectiveCwd : effectiveCwd + pathSep;
                   const relativePart = filePart.startsWith(cwdPrefix) ? filePart.slice(cwdPrefix.length) : null;
                   if (relativePart) {
                     for (const folder of options.allowedFolders) {
-                      const candidate = folder + "/" + relativePart;
+                      const candidate = pathJoin(folder, relativePart);
                       if ((0, import_fs4.existsSync)(candidate)) {
                         if (debug) console.error(`[extract] Auto-fixed path: ${filePart} \u2192 ${candidate}`);
                         return candidate + suffix;
@@ -27804,11 +27829,12 @@ var init_vercel = __esm({
                     }
                   }
                   for (const folder of options.allowedFolders) {
-                    const folderPrefix = folder.endsWith("/") ? folder : folder + "/";
-                    const wsParent = folderPrefix.replace(/[^/]+\/$/, "");
+                    const folderPrefix = folder.endsWith(pathSep) ? folder : folder + pathSep;
+                    const sepEscaped = pathSep === "\\" ? "\\\\" : pathSep;
+                    const wsParent = folderPrefix.replace(new RegExp("[^" + sepEscaped + "]+" + sepEscaped + "$"), "");
                     if (filePart.startsWith(wsParent)) {
                       const tail = filePart.slice(wsParent.length);
-                      const candidate = folderPrefix + tail;
+                      const candidate = pathJoin(folderPrefix, tail);
                       if (candidate !== filePart && (0, import_fs4.existsSync)(candidate)) {
                         if (debug) console.error(`[extract] Auto-fixed path via workspace: ${filePart} \u2192 ${candidate}`);
                         return candidate + suffix;
@@ -62006,6 +62032,7 @@ var init_parser2 = __esm({
                     { ALT: () => this.CONSUME(Identifier) },
                     { ALT: () => this.CONSUME(Text) },
                     { ALT: () => this.CONSUME(NumberLiteral) },
+                    { ALT: () => this.CONSUME(ColorValue) },
                     // Note: RoundOpen and RoundClose (parentheses) are NOT allowed in unquoted labels
                     // to match Mermaid's behavior - use quoted labels like ["text (with parens)"] instead
                     // Allow HTML-like tags (e.g., <br/>) inside labels
@@ -62097,6 +62124,7 @@ var init_parser2 = __esm({
               { ALT: () => this.CONSUME(Identifier) },
               { ALT: () => this.CONSUME(Text) },
               { ALT: () => this.CONSUME(NumberLiteral) },
+              { ALT: () => this.CONSUME(ColorValue) },
               // Allow HTML-like angle brackets and slashes for <br/>, <i>, etc.
               { ALT: () => this.CONSUME(AngleLess) },
               { ALT: () => this.CONSUME(AngleOpen) },
@@ -63272,13 +63300,24 @@ function mapFlowchartParserError(err, text) {
       length: len
     };
   }
-  if (tokType === "QuotedString" || tokType === "SquareOpen" || tokType === "SquareClose") {
+  if (tokType === "QuotedString" || tokType === "SquareOpen" || tokType === "SquareClose" || tokType === "DiamondOpen" || tokType === "DiamondClose") {
     const context = err?.context;
     const inLinkRule = context?.ruleStack?.includes("linkTextInline") || context?.ruleStack?.includes("link") || false;
     const lineContent = allLines[Math.max(0, line - 1)] || "";
     const beforeQuote = lineContent.slice(0, Math.max(0, column - 1));
     const hasLinkBefore = beforeQuote.match(/--\s*$|==\s*$|-\.\s*$|-\.-\s*$|\[\s*$/);
     if (inLinkRule || hasLinkBefore) {
+      if (tokType === "DiamondOpen" || tokType === "DiamondClose") {
+        return {
+          line,
+          column,
+          severity: "error",
+          code: "FL-EDGE-LABEL-CURLY-IN-PIPES",
+          message: "Curly braces { } are not supported inside pipe-delimited edge labels.",
+          hint: "Use HTML entities &#123; and &#125; inside |...|, e.g., --|Resolve $&#123;VAR&#125;|-->",
+          length: len
+        };
+      }
       if (tokType === "SquareOpen" || tokType === "SquareClose") {
         return {
           line,
@@ -63340,6 +63379,17 @@ function mapFlowchartParserError(err, text) {
           length: len
         };
       }
+      if (tokType === "DiamondOpen" || tokType === "DiamondClose") {
+        return {
+          line,
+          column,
+          severity: "error",
+          code: "FL-LABEL-CURLY-IN-UNQUOTED",
+          message: "Curly braces are not supported inside unquoted node labels.",
+          hint: "Use &#123; and &#125; for literal braces, e.g., C[Substitute &#123;params&#125;].",
+          length: len
+        };
+      }
       {
         const caret0 = Math.max(0, column - 1);
         const openIdx = lineStr.lastIndexOf("[", caret0);
@@ -63381,9 +63431,20 @@ function mapFlowchartParserError(err, text) {
               length: len
             };
           }
+          if (seg.includes("{") || seg.includes("}")) {
+            return {
+              line,
+              column,
+              severity: "error",
+              code: "FL-LABEL-CURLY-IN-UNQUOTED",
+              message: "Curly braces are not supported inside unquoted node labels.",
+              hint: "Use &#123; and &#125; for literal braces, e.g., C[Substitute &#123;params&#125;].",
+              length: len
+            };
+          }
         }
       }
-      if (tokType === "QuotedString" || tokType === "SquareOpen" || tokType === "SquareClose") {
+      if (tokType === "QuotedString") {
         return {
           line,
           column,
@@ -63394,6 +63455,17 @@ function mapFlowchartParserError(err, text) {
           length: len
         };
       }
+      if (tokType === "SquareOpen" || tokType === "SquareClose") {
+        return {
+          line,
+          column,
+          severity: "error",
+          code: "FL-LABEL-BRACKET-IN-UNQUOTED",
+          message: "Square brackets are not supported inside unquoted node labels.",
+          hint: 'Use &#91; and &#93; for literal brackets or wrap the label in quotes, e.g., C["bind_paths[]"]',
+          length: len
+        };
+      }
       const q = findInnerQuoteIssue("[");
       if (q?.kind === "escaped") {
         return { line, column: q.column, severity: "error", code: "FL-LABEL-ESCAPED-QUOTE", message: 'Escaped quotes (\\") in node labels are not supported by Mermaid. Use &quot; instead.', hint: 'Prefer "He said &quot;Hi&quot;".', length: 2 };
@@ -63404,7 +63476,7 @@ function mapFlowchartParserError(err, text) {
       return { line, column, severity: "error", code: "FL-NODE-UNCLOSED-BRACKET", message: "Unclosed '['. Add a matching ']' before the arrow or newline.", hint: "Example: A[Label] --> B", length: 1 };
     }
     if (expecting(err, "RoundClose")) {
-      if (tokType === "QuotedString" || tokType === "SquareOpen" || tokType === "SquareClose") {
+      if (tokType === "QuotedString") {
         return {
           line,
           column,
@@ -63415,6 +63487,17 @@ function mapFlowchartParserError(err, text) {
           length: len
         };
       }
+      if (tokType === "SquareOpen" || tokType === "SquareClose") {
+        return {
+          line,
+          column,
+          severity: "error",
+          code: "FL-LABEL-BRACKET-IN-UNQUOTED",
+          message: "Square brackets are not supported inside unquoted node labels.",
+          hint: 'Use &#91; and &#93; for literal brackets or wrap the label in quotes, e.g., C["bind_paths[]"]',
+          length: len
+        };
+      }
       {
         const caret0 = Math.max(0, column - 1);
         const openIdx = lineStr.lastIndexOf("(", caret0);
@@ -63443,7 +63526,7 @@ function mapFlowchartParserError(err, text) {
       return { line, column, severity: "error", code: "FL-NODE-UNCLOSED-BRACKET", message: "Unclosed '('. Add a matching ')'.", hint: "Example: B(Label)", length: 1 };
     }
     if (expecting(err, "DiamondClose")) {
-      if (tokType === "QuotedString" || tokType === "SquareOpen" || tokType === "SquareClose") {
+      if (tokType === "QuotedString") {
         return {
           line,
           column,
@@ -63454,6 +63537,17 @@ function mapFlowchartParserError(err, text) {
           length: len
         };
       }
+      if (tokType === "SquareOpen" || tokType === "SquareClose") {
+        return {
+          line,
+          column,
+          severity: "error",
+          code: "FL-LABEL-BRACKET-IN-UNQUOTED",
+          message: "Square brackets are not supported inside unquoted node labels.",
+          hint: 'Use &#91; and &#93; for literal brackets or wrap the label in quotes, e.g., C["bind_paths[]"]',
+          length: len
+        };
+      }
       {
         const caret0 = Math.max(0, column - 1);
         const openIdx = lineStr.lastIndexOf("{", caret0);
@@ -64296,7 +64390,7 @@ function validateFlowchart(text, options = {}) {
         const byLine = /* @__PURE__ */ new Map();
         const collect = (arr) => {
           for (const e of arr || []) {
-            if (e && (e.code === "FL-LABEL-PARENS-UNQUOTED" || e.code === "FL-LABEL-AT-IN-UNQUOTED" || e.code === "FL-LABEL-QUOTE-IN-UNQUOTED" || e.code === "FL-LABEL-SLASH-UNQUOTED")) {
+            if (e && (e.code === "FL-LABEL-PARENS-UNQUOTED" || e.code === "FL-LABEL-AT-IN-UNQUOTED" || e.code === "FL-LABEL-QUOTE-IN-UNQUOTED" || e.code === "FL-LABEL-SLASH-UNQUOTED" || e.code === "FL-LABEL-CURLY-IN-UNQUOTED" || e.code === "FL-LABEL-BRACKET-IN-UNQUOTED")) {
               const ln = e.line ?? 0;
               const col = e.column ?? 1;
               const list = byLine.get(ln) || [];
@@ -64378,6 +64472,9 @@ function validateFlowchart(text, options = {}) {
                 const hasParens = seg.includes("(") || seg.includes(")");
                 const hasAt = seg.includes("@");
                 const hasQuote = seg.includes('"');
+                const hasCurly = seg.includes("{") || seg.includes("}");
+                const hasBracket = seg.includes("[") || seg.includes("]");
+                const isDoubleSquare = raw.slice(i, i + 2) === "[[" && raw.slice(j - 2, j) === "]]";
                 const isSingleQuoted = /^'[^]*'$/.test(trimmed);
                 const hasLeadingSlash = lsp === "/" || lsp === "\\";
                 if (!covered && !isQuoted && !isParenWrapped && hasParens) {
@@ -64395,6 +64492,16 @@ function validateFlowchart(text, options = {}) {
                   existing.push({ start: startCol, end: endCol });
                   byLine.set(ln, existing);
                 }
+                if (!covered && !isQuoted && !isSlashPair && hasCurly) {
+                  errs.push({ line: ln, column: startCol, severity: "error", code: "FL-LABEL-CURLY-IN-UNQUOTED", message: "Curly braces are not supported inside unquoted node labels.", hint: "Use &#123; and &#125; for literal braces, e.g., C[Substitute &#123;params&#125;]." });
+                  existing.push({ start: startCol, end: endCol });
+                  byLine.set(ln, existing);
+                }
+                if (!covered && !isQuoted && !isSlashPair && !isDoubleSquare && hasBracket) {
+                  errs.push({ line: ln, column: startCol, severity: "error", code: "FL-LABEL-BRACKET-IN-UNQUOTED", message: "Square brackets are not supported inside unquoted node labels.", hint: 'Use &#91; and &#93; for literal brackets or wrap the label in quotes, e.g., C["bind_paths[]"]' });
+                  existing.push({ start: startCol, end: endCol });
+                  byLine.set(ln, existing);
+                }
                 if (!covered && !isQuoted && !isSlashPair && hasQuote && !isSingleQuoted) {
                   errs.push({ line: ln, column: startCol, severity: "error", code: "FL-LABEL-QUOTE-IN-UNQUOTED", message: "Quotes are not allowed inside unquoted node labels. Use &quot; for quotes or wrap the entire label in quotes.", hint: 'Example: C["HTML Output: data-trigger-visibility=&quot;true&quot;"]' });
                   existing.push({ start: startCol, end: endCol });
@@ -66755,7 +66862,7 @@ function computeFixes(text, errors, level = "safe") {
       }
       continue;
     }
-    if (is("FL-EDGE-LABEL-BRACKET", e)) {
+    if (is("FL-EDGE-LABEL-BRACKET", e) || is("FL-EDGE-LABEL-CURLY-IN-PIPES", e)) {
       const lineText = lineTextAt(text, e.line);
       const firstBar = lineText.indexOf("|");
       const secondBar = firstBar >= 0 ? lineText.indexOf("|", firstBar + 1) : -1;
@@ -66763,7 +66870,8 @@ function computeFixes(text, errors, level = "safe") {
         const before = lineText.slice(0, firstBar + 1);
         const label = lineText.slice(firstBar + 1, secondBar);
         const after = lineText.slice(secondBar);
-        const fixedLabel = label.replace(/\[/g, "&#91;").replace(/\]/g, "&#93;");
+        let fixedLabel = label.replace(/\[/g, "&#91;").replace(/\]/g, "&#93;");
+        fixedLabel = fixedLabel.replace(/\{/g, "&#123;").replace(/\}/g, "&#125;");
         const fixedLine = before + fixedLabel + after;
         const finalLine = fixedLine.replace(/\[([^\]]*)\]/g, (m, seg) => "[" + String(seg).replace(/`/g, "") + "]");
         edits.push({ start: { line: e.line, column: 1 }, end: { line: e.line, column: lineText.length + 1 }, newText: finalLine });
@@ -67403,7 +67511,7 @@ function computeFixes(text, errors, level = "safe") {
       }
       continue;
     }
-    if (is("FL-LABEL-PARENS-UNQUOTED", e) || is("FL-LABEL-AT-IN-UNQUOTED", e) || is("FL-LABEL-SLASH-UNQUOTED", e)) {
+    if (is("FL-LABEL-PARENS-UNQUOTED", e) || is("FL-LABEL-AT-IN-UNQUOTED", e) || is("FL-LABEL-SLASH-UNQUOTED", e) || is("FL-LABEL-CURLY-IN-UNQUOTED", e) || is("FL-LABEL-BRACKET-IN-UNQUOTED", e)) {
       if (level === "safe" || level === "all") {
         if (patchedLines.has(e.line))
           continue;
@@ -67444,7 +67552,7 @@ function computeFixes(text, errors, level = "safe") {
             if (openIdx === -1)
               break;
             const contentStart = openIdx + shape.open.length;
-            const closeIdx = shape.open === "(" && shape.close === ")" ? findMatchingCloser(lineText, openIdx, shape.open, shape.close) : lineText.indexOf(shape.close, contentStart);
+            const closeIdx = shape.open === "(" && shape.close === ")" || shape.open === "[" && shape.close === "]" ? findMatchingCloser(lineText, openIdx, shape.open, shape.close) : lineText.indexOf(shape.close, contentStart);
             if (closeIdx === -1)
               break;
             if (openIdx <= caret0 && caret0 < closeIdx) {
@@ -67464,11 +67572,21 @@ function computeFixes(text, errors, level = "safe") {
               const isSlashPair = (l, r) => l === "/" && r === "/" || l === "\\" && r === "\\" || l === "/" && r === "\\" || l === "\\" && r === "/";
               const isParallelogramShape = core.length >= 2 && isSlashPair(left, right);
               let replaced;
-              if (!isParallelogramShape) {
-                const escaped = inner.replace(/`/g, "").replace(/\"/g, "&quot;").replace(/"/g, "&quot;");
+              if (is("FL-LABEL-BRACKET-IN-UNQUOTED", e) && !isParallelogramShape) {
+                const hasOtherHazards = /[(){}@]/.test(inner) || inner.includes('"') || inner.includes('\\"');
+                if (hasOtherHazards) {
+                  const escaped = inner.replace(/`/g, "").replace(/\\"/g, "&quot;").replace(/"/g, "&quot;");
+                  replaced = '"' + escaped + '"';
+                } else {
+                  replaced = inner.replace(/\[/g, "&#91;").replace(/\]/g, "&#93;");
+                }
+              } else if (is("FL-LABEL-CURLY-IN-UNQUOTED", e)) {
+                replaced = inner.replace(/\{/g, "&#123;").replace(/\}/g, "&#125;");
+              } else if (!isParallelogramShape) {
+                const escaped = inner.replace(/`/g, "").replace(/\\"/g, "&quot;").replace(/"/g, "&quot;");
                 replaced = '"' + escaped + '"';
               } else {
-                replaced = inner.replace(/`/g, "").replace(/\(/g, "&#40;").replace(/\)/g, "&#41;").replace(/\"/g, "&quot;").replace(/"/g, "&quot;");
+                replaced = inner.replace(/`/g, "").replace(/\(/g, "&#40;").replace(/\)/g, "&#41;").replace(/\[/g, "&#91;").replace(/\]/g, "&#93;").replace(/\\"/g, "&quot;").replace(/"/g, "&quot;");
               }
               if (replaced !== inner) {
                 edits.push({ start: { line: e.line, column: contentStart + 1 }, end: { line: e.line, column: closeIdx + 1 }, newText: replaced });
@@ -97556,14 +97674,6 @@ var init_FallbackManager = __esm({
 });
 
 // src/agent/contextCompactor.js
-var contextCompactor_exports = {};
-__export(contextCompactor_exports, {
-  calculateCompactionStats: () => calculateCompactionStats,
-  compactMessages: () => compactMessages,
-  handleContextLimitError: () => handleContextLimitError,
-  identifyMessageSegments: () => identifyMessageSegments,
-  isContextLimitError: () => isContextLimitError
-});
 function isContextLimitError(error40) {
   if (!error40) return false;
   const errorMessage = (typeof error40 === "string" ? error40 : error40?.message || "").toLowerCase();
@@ -101879,6 +101989,16 @@ You are working with a workspace. Available paths: ${workspaceDesc}
               userMessage
             ];
           }
+          if (this.history.length > 0) {
+            const compacted = compactMessages(currentMessages, { keepLastSegment: true, minSegmentsToKeep: 1 });
+            if (compacted.length < currentMessages.length) {
+              const stats = calculateCompactionStats(currentMessages, compacted);
+              if (this.debug) {
+                console.log(`[DEBUG] Proactive history compaction: ${currentMessages.length} \u2192 ${compacted.length} messages (${stats.reductionPercent}% reduction, ~${stats.tokensSaved} tokens saved)`);
+              }
+              currentMessages = compacted;
+            }
+          }
           let currentIteration = 0;
           let finalResult = "I was unable to complete your request due to reaching the maximum number of tool iterations.";
           const baseMaxIterations = options._maxIterationsOverride || this.maxIterations || MAX_TOOL_ITERATIONS;
@@ -102579,7 +102699,6 @@ Double-check your response based on the criteria above. If everything looks good
        * @returns {Object} Compaction statistics
        */
       async compactHistory(options = {}) {
-        const { compactMessages: compactMessages2, calculateCompactionStats: calculateCompactionStats2 } = await Promise.resolve().then(() => (init_contextCompactor(), contextCompactor_exports));
         if (this.history.length === 0) {
           if (this.debug) {
             console.log(`[DEBUG] No history to compact for session ${this.sessionId}`);
@@ -102594,8 +102713,8 @@ Double-check your response based on the criteria above. If everything looks good
             tokensSaved: 0
           };
         }
-        const compactedMessages = compactMessages2(this.history, options);
-        const stats = calculateCompactionStats2(this.history, compactedMessages);
+        const compactedMessages = compactMessages(this.history, options);
+        const stats = calculateCompactionStats(this.history, compactedMessages);
         this.history = compactedMessages;
         try {
           await this.storageAdapter.clearHistory(this.sessionId);