@probelabs/probe 0.6.0-rc296 → 0.6.0-rc298

package/cjs/agent/ProbeAgent.cjs

@@ -63618,7 +63618,22 @@ function mapFlowchartParserError(err, text) {
     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*$/);
+    const caret0 = Math.max(0, column - 1);
+    const firstBar = lineContent.lastIndexOf("|", caret0);
+    const secondBar = firstBar >= 0 ? lineContent.indexOf("|", caret0 + 1) : -1;
+    const inPipeLabel = firstBar >= 0 && secondBar > firstBar && firstBar < caret0 && secondBar > caret0;
     if (inLinkRule || hasLinkBefore) {
+      if (tokType === "QuotedString" && inPipeLabel) {
+        return {
+          line,
+          column,
+          severity: "error",
+          code: "FL-EDGE-LABEL-QUOTE-IN-PIPES",
+          message: "Quotes are not supported inside pipe-delimited edge labels.",
+          hint: "Use &quot; inside |...|, e.g., --|e.g. &quot;navigate to example.com&quot;|-->",
+          length: len
+        };
+      }
       if (tokType === "DiamondOpen" || tokType === "DiamondClose") {
         return {
           line,
@@ -64220,17 +64235,6 @@ ${br.example}`,
   if (inRule("arrow") && err.name === "NoViableAltException") {
     return { line, column, severity: "error", code: "SE-ARROW-INVALID", message: `Invalid sequence arrow near '${found}'.`, hint: "Use ->, -->, ->>, -->>, -x, --x, -), --), <<->>, or <<-->>", length: len };
   }
-  if ((err.name === "NoViableAltException" || err.name === "MismatchedTokenException") && tokType === "Minus") {
-    return {
-      line,
-      column,
-      severity: "error",
-      code: "SE-BULLET-LINE-UNSUPPORTED",
-      message: "Bullet list lines starting with '-' are not supported in sequence diagrams.",
-      hint: "Wrap free\u2011form text in a note block instead, for example:\nNote over A : Item 1\nNote over A\n  - Item 1\n  - Item 2\nend note",
-      length: len
-    };
-  }
   if (inRule("noteStmt")) {
     if (err.name === "MismatchedTokenException" && exp("Colon")) {
       return { line, column, severity: "error", code: "SE-NOTE-MALFORMED", message: "Malformed note: missing colon before the note text.", hint: "Example: Note right of Alice: Hello", length: len };
@@ -64239,6 +64243,22 @@ ${br.example}`,
       return { line, column, severity: "error", code: "SE-NOTE-MALFORMED", message: "Malformed note statement. Use left|right of X or over X[,Y]: text", hint: "Examples: Note over A,B: hi", length: len };
     }
   }
+  if ((err.name === "NoViableAltException" || err.name === "MismatchedTokenException") && tokType === "Minus") {
+    const nonWs = ltxt.search(/\S/);
+    const minusAtLineStart = nonWs >= 0 && ltxt[nonWs] === "-" && column === nonWs + 1;
+    if (!minusAtLineStart) {
+    } else {
+      return {
+        line,
+        column,
+        severity: "error",
+        code: "SE-BULLET-LINE-UNSUPPORTED",
+        message: "Bullet list lines starting with '-' are not supported in sequence diagrams.",
+        hint: "Wrap free\u2011form text in a note block instead, for example:\nNote over A : Item 1\nNote over A\n  - Item 1\n  - Item 2\nend note",
+        length: len
+      };
+    }
+  }
   if (tokType === "ElseKeyword" && isInRule(err, "criticalBlock")) {
     return {
       line,
@@ -65038,7 +65058,7 @@ var Identifier2, NumberLiteral3, SequenceKeyword, ParticipantKeyword, ActorKeywo
 var init_lexer4 = __esm({
   "node_modules/@probelabs/maid/out/diagrams/sequence/lexer.js"() {
     init_api6();
-    Identifier2 = createToken({ name: "Identifier", pattern: /[A-Za-z_][A-Za-z0-9_]*/ });
+    Identifier2 = createToken({ name: "Identifier", pattern: /[A-Za-z_][A-Za-z0-9_.]*(?:-(?![>x)\s])[A-Za-z0-9_.]+)*/ });
     NumberLiteral3 = createToken({ name: "NumberLiteral", pattern: /[0-9]+/ });
     SequenceKeyword = createToken({ name: "SequenceKeyword", pattern: /sequenceDiagram/, longer_alt: Identifier2 });
     ParticipantKeyword = createToken({ name: "ParticipantKeyword", pattern: /participant/i, longer_alt: Identifier2 });
@@ -66858,6 +66878,135 @@ var init_validate5 = __esm({
   }
 });
 
+// node_modules/@probelabs/maid/out/core/frontmatter.js
+function parseFrontmatter(input) {
+  const text = input.startsWith("\uFEFF") ? input.slice(1) : input;
+  const lines = text.split(/\r?\n/);
+  if (lines.length < 3 || lines[0].trim() !== "---")
+    return null;
+  let i = 1;
+  const block = [];
+  while (i < lines.length && lines[i].trim() !== "---") {
+    block.push(lines[i]);
+    i++;
+  }
+  if (i >= lines.length)
+    return null;
+  const body = lines.slice(i + 1).join("\n");
+  const bodyStartLine = i + 2;
+  const raw = block.join("\n");
+  const config2 = {};
+  const themeVars = {};
+  let themeUnderConfig = false;
+  let ctx = "root";
+  for (const line of block) {
+    if (!line.trim())
+      continue;
+    const indent = line.match(/^\s*/)?.[0].length ?? 0;
+    const mKey = line.match(/^\s*([A-Za-z0-9_\-]+):\s*(.*)$/);
+    if (!mKey)
+      continue;
+    const key = mKey[1];
+    let value = mKey[2] || "";
+    if (indent === 0) {
+      if (key === "config") {
+        ctx = "config";
+        continue;
+      }
+      if (key === "themeVariables") {
+        ctx = "theme";
+        continue;
+      }
+      ctx = "root";
+      continue;
+    }
+    if (ctx === "config") {
+      if (indent <= 2 && key !== "pie" && key !== "themeVariables")
+        continue;
+      if (key === "pie") {
+        ctx = "config.pie";
+        ensure(config2, "pie", {});
+        continue;
+      }
+      if (key === "themeVariables") {
+        ctx = "theme";
+        themeUnderConfig = true;
+        continue;
+      }
+      continue;
+    }
+    if (ctx === "config.pie") {
+      if (indent < 4) {
+        if (key === "pie") {
+          ctx = "config.pie";
+          ensure(config2, "pie", {});
+          continue;
+        }
+        if (key === "themeVariables") {
+          ctx = "theme";
+          themeUnderConfig = true;
+          continue;
+        }
+        ctx = "config";
+        continue;
+      }
+      setKV(config2.pie, key, value);
+      continue;
+    }
+    if (ctx === "theme") {
+      if (indent < 2) {
+        ctx = "root";
+        continue;
+      }
+      setKV(themeVars, key, value);
+      continue;
+    }
+  }
+  if (themeUnderConfig && Object.keys(themeVars).length) {
+    ensure(config2, "themeVariables", {});
+    Object.assign(config2.themeVariables, themeVars);
+  }
+  return {
+    raw,
+    body,
+    bodyStartLine,
+    config: Object.keys(config2).length ? config2 : void 0,
+    themeVariables: Object.keys(themeVars).length ? themeVars : void 0
+  };
+}
+function ensure(obj, key, def) {
+  if (obj[key] == null)
+    obj[key] = def;
+}
+function unquote(val) {
+  const v = val.trim();
+  if (v.startsWith('"') && v.endsWith('"') || v.startsWith("'") && v.endsWith("'")) {
+    return v.slice(1, -1);
+  }
+  return v;
+}
+function setKV(target, key, rawValue) {
+  const v = unquote(rawValue);
+  if (v === "") {
+    target[key] = "";
+    return;
+  }
+  const num = Number(v);
+  if (!Number.isNaN(num) && /^-?[0-9]+(\.[0-9]+)?$/.test(v)) {
+    target[key] = num;
+    return;
+  }
+  if (/^(true|false)$/i.test(v)) {
+    target[key] = /^true$/i.test(v);
+    return;
+  }
+  target[key] = v;
+}
+var init_frontmatter = __esm({
+  "node_modules/@probelabs/maid/out/core/frontmatter.js"() {
+  }
+});
+
 // node_modules/@probelabs/maid/out/core/router.js
 function firstNonCommentLine(text) {
   const lines = text.split(/\r?\n/);
@@ -66872,7 +67021,8 @@ function firstNonCommentLine(text) {
   return void 0;
 }
 function detectDiagramType(text) {
-  const header = firstNonCommentLine(text);
+  const { content } = stripFrontmatter(text);
+  const header = firstNonCommentLine(content);
   if (!header)
     return "unknown";
   if (/^(flowchart|graph)\b/i.test(header))
@@ -66924,20 +67074,26 @@ function isOtherMermaidDiagram(headerLine) {
   return OTHER.has(t);
 }
 function validate(text, options = {}) {
-  const type = detectDiagramType(text);
+  const { content, lineOffset } = stripFrontmatter(text);
+  const type = detectDiagramType(content);
+  const withOffset = (errors) => {
+    if (lineOffset === 0)
+      return errors;
+    return errors.map((e) => ({ ...e, line: Math.max(1, (e.line || 1) + lineOffset) }));
+  };
   switch (type) {
     case "flowchart":
-      return { type, errors: validateFlowchart(text, options) };
+      return { type, errors: withOffset(validateFlowchart(content, options)) };
     case "pie":
-      return { type, errors: validatePie(text, options) };
+      return { type, errors: withOffset(validatePie(content, options)) };
     case "sequence":
-      return { type, errors: validateSequence(text, options) };
+      return { type, errors: withOffset(validateSequence(content, options)) };
     case "class":
-      return { type, errors: validateClass(text, options) };
+      return { type, errors: withOffset(validateClass(content, options)) };
     case "state":
-      return { type, errors: validateState(text, options) };
+      return { type, errors: withOffset(validateState(content, options)) };
     default:
-      const header = firstNonCommentLine(text);
+      const header = firstNonCommentLine(content);
       if (isOtherMermaidDiagram(header)) {
         return { type, errors: [] };
       }
@@ -66945,7 +67101,7 @@ function validate(text, options = {}) {
         type,
         errors: [
           {
-            line: 1,
+            line: lineOffset + 1,
             column: 1,
             message: 'Diagram must start with "graph", "flowchart", "pie", "sequenceDiagram", "classDiagram" or "stateDiagram[-v2]"',
             severity: "error",
@@ -66956,6 +67112,12 @@ function validate(text, options = {}) {
       };
   }
 }
+function stripFrontmatter(text) {
+  const fm = parseFrontmatter(text);
+  if (!fm)
+    return { content: text, lineOffset: 0 };
+  return { content: fm.body, lineOffset: fm.bodyStartLine - 1 };
+}
 var init_router = __esm({
   "node_modules/@probelabs/maid/out/core/router.js"() {
     init_validate();
@@ -66963,6 +67125,7 @@ var init_router = __esm({
     init_validate3();
     init_validate4();
     init_validate5();
+    init_frontmatter();
   }
 });
 
@@ -67174,16 +67337,18 @@ function computeFixes(text, errors, level = "safe") {
       }
       continue;
     }
-    if (is("FL-EDGE-LABEL-BRACKET", e) || is("FL-EDGE-LABEL-CURLY-IN-PIPES", e)) {
+    if (is("FL-EDGE-LABEL-BRACKET", e) || is("FL-EDGE-LABEL-CURLY-IN-PIPES", e) || is("FL-EDGE-LABEL-QUOTE-IN-PIPES", e)) {
       const lineText = lineTextAt(text, e.line);
-      const firstBar = lineText.indexOf("|");
-      const secondBar = firstBar >= 0 ? lineText.indexOf("|", firstBar + 1) : -1;
+      const col = Math.max(0, e.column - 1);
+      const firstBar = lineText.lastIndexOf("|", col);
+      const secondBar = firstBar >= 0 ? lineText.indexOf("|", col + 1) : -1;
       if (firstBar >= 0 && secondBar > firstBar) {
         const before = lineText.slice(0, firstBar + 1);
         const label = lineText.slice(firstBar + 1, secondBar);
         const after = lineText.slice(secondBar);
         let fixedLabel = label.replace(/\[/g, "&#91;").replace(/\]/g, "&#93;");
         fixedLabel = fixedLabel.replace(/\{/g, "&#123;").replace(/\}/g, "&#125;");
+        fixedLabel = fixedLabel.replace(/\\"/g, "&quot;").replace(/"/g, "&quot;");
         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 });
@@ -78573,7 +78738,7 @@ ${overlay}</g>`;
 });
 
 // node_modules/@probelabs/maid/out/renderer/pie-builder.js
-function unquote(s) {
+function unquote2(s) {
   if (!s)
     return s;
   const first2 = s.charAt(0);
@@ -78623,7 +78788,7 @@ function buildPieModel(text) {
       const collect = (k) => {
         const arr = tnode.children?.[k] ?? [];
         for (const tok of arr)
-          parts.push(unquote(tok.image));
+          parts.push(unquote2(tok.image));
       };
       collect("QuotedString");
       collect("Text");
@@ -78636,7 +78801,7 @@ function buildPieModel(text) {
       const labelTok = snode.children?.sliceLabel?.[0]?.children?.QuotedString?.[0];
       const numTok = snode.children?.NumberLiteral?.[0];
       if (labelTok && numTok) {
-        const label = unquote(labelTok.image).trim();
+        const label = unquote2(labelTok.image).trim();
         const value = Number(numTok.image);
         if (!Number.isNaN(value)) {
           model.slices.push({ label, value });
@@ -79792,128 +79957,6 @@ var init_state_renderer = __esm({
   }
 });
 
-// node_modules/@probelabs/maid/out/core/frontmatter.js
-function parseFrontmatter(input) {
-  const text = input.startsWith("\uFEFF") ? input.slice(1) : input;
-  const lines = text.split(/\r?\n/);
-  if (lines.length < 3 || lines[0].trim() !== "---")
-    return null;
-  let i = 1;
-  const block = [];
-  while (i < lines.length && lines[i].trim() !== "---") {
-    block.push(lines[i]);
-    i++;
-  }
-  if (i >= lines.length)
-    return null;
-  const body = lines.slice(i + 1).join("\n");
-  const raw = block.join("\n");
-  const config2 = {};
-  const themeVars = {};
-  let themeUnderConfig = false;
-  let ctx = "root";
-  for (const line of block) {
-    if (!line.trim())
-      continue;
-    const indent = line.match(/^\s*/)?.[0].length ?? 0;
-    const mKey = line.match(/^\s*([A-Za-z0-9_\-]+):\s*(.*)$/);
-    if (!mKey)
-      continue;
-    const key = mKey[1];
-    let value = mKey[2] || "";
-    if (indent === 0) {
-      if (key === "config") {
-        ctx = "config";
-        continue;
-      }
-      if (key === "themeVariables") {
-        ctx = "theme";
-        continue;
-      }
-      ctx = "root";
-      continue;
-    }
-    if (ctx === "config") {
-      if (indent <= 2 && key !== "pie" && key !== "themeVariables")
-        continue;
-      if (key === "pie") {
-        ctx = "config.pie";
-        ensure(config2, "pie", {});
-        continue;
-      }
-      if (key === "themeVariables") {
-        ctx = "theme";
-        themeUnderConfig = true;
-        continue;
-      }
-      continue;
-    }
-    if (ctx === "config.pie") {
-      if (indent < 4) {
-        if (key === "pie") {
-          ctx = "config.pie";
-          ensure(config2, "pie", {});
-          continue;
-        }
-        if (key === "themeVariables") {
-          ctx = "theme";
-          themeUnderConfig = true;
-          continue;
-        }
-        ctx = "config";
-        continue;
-      }
-      setKV(config2.pie, key, value);
-      continue;
-    }
-    if (ctx === "theme") {
-      if (indent < 2) {
-        ctx = "root";
-        continue;
-      }
-      setKV(themeVars, key, value);
-      continue;
-    }
-  }
-  if (themeUnderConfig && Object.keys(themeVars).length) {
-    ensure(config2, "themeVariables", {});
-    Object.assign(config2.themeVariables, themeVars);
-  }
-  return { raw, body, config: Object.keys(config2).length ? config2 : void 0, themeVariables: Object.keys(themeVars).length ? themeVars : void 0 };
-}
-function ensure(obj, key, def) {
-  if (obj[key] == null)
-    obj[key] = def;
-}
-function unquote2(val) {
-  const v = val.trim();
-  if (v.startsWith('"') && v.endsWith('"') || v.startsWith("'") && v.endsWith("'")) {
-    return v.slice(1, -1);
-  }
-  return v;
-}
-function setKV(target, key, rawValue) {
-  const v = unquote2(rawValue);
-  if (v === "") {
-    target[key] = "";
-    return;
-  }
-  const num = Number(v);
-  if (!Number.isNaN(num) && /^-?[0-9]+(\.[0-9]+)?$/.test(v)) {
-    target[key] = num;
-    return;
-  }
-  if (/^(true|false)$/i.test(v)) {
-    target[key] = /^true$/i.test(v);
-    return;
-  }
-  target[key] = v;
-}
-var init_frontmatter = __esm({
-  "node_modules/@probelabs/maid/out/core/frontmatter.js"() {
-  }
-});
-
 // node_modules/@probelabs/maid/out/renderer/class-builder.js
 function textFromTokens3(tokens) {
   if (!tokens || tokens.length === 0)
@@ -89236,6 +89279,7 @@ var init_client = __esm({
         this.debug = options.debug || process.env.DEBUG_MCP === "1";
         this.config = null;
         this.tracer = options.tracer || null;
+        this.agentEvents = options.agentEvents || null;
       }
       /**
        * Record an MCP telemetry event if tracer is available
@@ -89463,11 +89507,21 @@ var init_client = __esm({
           throw new Error(`Server ${tool6.serverName} not connected`);
         }
         const startTime = Date.now();
+        const toolCallId = `mcp-${toolName}-${startTime}`;
         this.recordMcpEvent("tool.call_started", {
           toolName,
           serverName: tool6.serverName,
           originalToolName: tool6.originalName
         });
+        if (this.agentEvents) {
+          this.agentEvents.emit("toolCall", {
+            toolCallId,
+            name: toolName,
+            args,
+            status: "started",
+            timestamp: (/* @__PURE__ */ new Date()).toISOString()
+          });
+        }
         try {
           if (this.debug) {
             console.error(`[MCP DEBUG] Calling ${toolName} with args:`, JSON.stringify(args, null, 2));
@@ -89497,6 +89551,14 @@ var init_client = __esm({
             originalToolName: tool6.originalName,
             durationMs
           });
+          if (this.agentEvents) {
+            this.agentEvents.emit("toolCall", {
+              toolCallId,
+              name: toolName,
+              status: "completed",
+              timestamp: (/* @__PURE__ */ new Date()).toISOString()
+            });
+          }
           return result;
         } catch (error40) {
           const durationMs = Date.now() - startTime;
@@ -89512,6 +89574,14 @@ var init_client = __esm({
             durationMs,
             isTimeout: error40.message.includes("timeout")
           });
+          if (this.agentEvents) {
+            this.agentEvents.emit("toolCall", {
+              toolCallId,
+              name: toolName,
+              status: "error",
+              timestamp: (/* @__PURE__ */ new Date()).toISOString()
+            });
+          }
           throw error40;
         }
       }
@@ -89697,6 +89767,7 @@ var init_xmlBridge = __esm({
       constructor(options = {}) {
         this.debug = options.debug || false;
         this.tracer = options.tracer || null;
+        this.agentEvents = options.agentEvents || null;
         this.mcpTools = {};
         this.mcpManager = null;
         this.toolDescriptions = {};
@@ -89739,7 +89810,7 @@ var init_xmlBridge = __esm({
           if (this.debug) {
             console.error("[MCP DEBUG] Initializing MCP client manager...");
           }
-          this.mcpManager = new MCPClientManager({ debug: this.debug, tracer: this.tracer });
+          this.mcpManager = new MCPClientManager({ debug: this.debug, tracer: this.tracer, agentEvents: this.agentEvents });
           const result = await this.mcpManager.initialize(mcpConfigs);
           const vercelTools = this.mcpManager.getVercelTools();
           this.mcpTools = vercelTools;
@@ -97186,7 +97257,7 @@ function deriveDescription(rawDescription, body) {
   }
   return truncateDescription(description);
 }
-function stripFrontmatter(content) {
+function stripFrontmatter2(content) {
   const { body } = extractFrontmatter(content);
   return body.trim();
 }
@@ -97327,7 +97398,7 @@ var init_registry = __esm({
         const skill = this.skillsByName.get(name15);
         if (!skill) return null;
         const content = await (0, import_promises3.readFile)(skill.skillFilePath, "utf8");
-        return stripFrontmatter(content);
+        return stripFrontmatter2(content);
       }
       async _resolveRealPath(target) {
         try {
@@ -102027,7 +102098,7 @@ var init_ProbeAgent = __esm({
             }
             mcpConfig = null;
           }
-          this.mcpBridge = new MCPXmlBridge({ debug: this.debug });
+          this.mcpBridge = new MCPXmlBridge({ debug: this.debug, agentEvents: this.events });
           await this.mcpBridge.initialize(mcpConfig);
           const mcpToolNames = this.mcpBridge.getToolNames();
           const mcpToolCount = mcpToolNames.length;
@@ -102903,6 +102974,14 @@ or
                     active_tools_count: activeToolsList.length
                   });
                 }
+                this.events.emit("timeout.extended", {
+                  grantedMs,
+                  reason: decision.reason || "work in progress",
+                  extensionsUsed: negotiatedTimeoutState.extensionsUsed,
+                  extensionsRemaining: negotiatedTimeoutState.maxRequests - negotiatedTimeoutState.extensionsUsed,
+                  totalExtraTimeMs: negotiatedTimeoutState.totalExtraTimeMs,
+                  budgetRemainingMs: remainingBudgetMs - grantedMs
+                });
               } else {
                 if (this.debug) {
                   console.log(`[DEBUG] Timeout observer: declined extension (reason: ${decision.reason}). Initiating graceful stop.`);
@@ -102916,6 +102995,11 @@ or
                     active_tools: activeToolsList.map((t) => t.name)
                   });
                 }
+                this.events.emit("timeout.windingDown", {
+                  reason: decision.reason || "observer declined",
+                  extensionsUsed: negotiatedTimeoutState.extensionsUsed,
+                  totalExtraTimeMs: negotiatedTimeoutState.totalExtraTimeMs
+                });
                 await this._initiateGracefulStop(gracefulTimeoutState, `observer declined: ${decision.reason}`);
               }
             };