@probelabs/probe 0.6.0-rc128 → 0.6.0-rc129

package/cjs/agent/ProbeAgent.cjs

@@ -54632,7 +54632,7 @@ var init_semantics = __esm({
           return;
         for (const cn of contentNodes) {
           const ch = cn.children || {};
-          const inspectTok = (tk) => {
+          const inspectTok = (tk, inQuoted = false) => {
             if (!tk)
               return false;
             const img = String(tk.image || "");
@@ -54642,7 +54642,7 @@ var init_semantics = __esm({
               this.ctx.errors.push({
                 line: tk.startLine ?? 1,
                 column: col,
-                severity: "warning",
+                severity: inQuoted ? "error" : "warning",
                 code: "FL-LABEL-BACKTICK",
                 message: "Backticks (`\u2026`) inside node labels are not supported by Mermaid.",
                 hint: 'Remove the backticks or use quotes instead, e.g., "GITHUB_ACTIONS" and "--cli".',
@@ -54654,12 +54654,12 @@ var init_semantics = __esm({
           };
           const texts = ch.Text || [];
           for (const tk of texts) {
-            if (inspectTok(tk))
+            if (inspectTok(tk, false))
               return;
           }
           const qs = ch.QuotedString || [];
           for (const tk of qs) {
-            if (inspectTok(tk))
+            if (inspectTok(tk, true))
               return;
           }
         }
@@ -55060,6 +55060,36 @@ function mapFlowchartParserError(err, text) {
           length: len
         };
       }
+      {
+        const caret0 = Math.max(0, column - 1);
+        const openIdx = lineStr.lastIndexOf("[", caret0);
+        if (openIdx !== -1) {
+          const closeIdx = lineStr.indexOf("]", openIdx + 1);
+          const seg = closeIdx !== -1 ? lineStr.slice(openIdx + 1, closeIdx) : lineStr.slice(openIdx + 1);
+          if (seg.includes('"')) {
+            return {
+              line,
+              column,
+              severity: "error",
+              code: "FL-LABEL-QUOTE-IN-UNQUOTED",
+              message: "Quotes are not allowed inside unquoted node labels. Use " for quotes or wrap the entire label in quotes.",
+              hint: 'Example: I[Log "processing N items"] or I["Log \\"processing N items\\""]',
+              length: len
+            };
+          }
+          if (seg.includes("(") || seg.includes(")")) {
+            return {
+              line,
+              column,
+              severity: "error",
+              code: "FL-LABEL-PARENS-UNQUOTED",
+              message: "Parentheses inside an unquoted label are not supported by Mermaid.",
+              hint: 'Wrap the label in quotes, e.g., A["Mark (X)"] \u2014 or replace ( and ) with HTML entities: ( and ).',
+              length: len
+            };
+          }
+        }
+      }
       if (tokType === "QuotedString") {
         return {
           line,
@@ -55092,6 +55122,24 @@ function mapFlowchartParserError(err, text) {
           length: len
         };
       }
+      {
+        const caret0 = Math.max(0, column - 1);
+        const openIdx = lineStr.lastIndexOf("(", caret0);
+        if (openIdx !== -1) {
+          const seg = lineStr.slice(openIdx + 1);
+          if (seg.includes("(") || seg.includes(")")) {
+            return {
+              line,
+              column,
+              severity: "error",
+              code: "FL-LABEL-PARENS-UNQUOTED",
+              message: "Parentheses inside an unquoted label are not supported by Mermaid.",
+              hint: 'Wrap the label in quotes, e.g., A["Mark (X)"] \u2014 or replace ( and ) with HTML entities: ( and ).',
+              length: len
+            };
+          }
+        }
+      }
       const q3 = findInnerQuoteIssue("(");
       if (q3?.kind === "escaped") {
         return { line, column: q3.column, severity: "error", code: "FL-LABEL-ESCAPED-QUOTE", message: 'Escaped quotes (\\") in node labels are not supported by Mermaid. Use " instead.', hint: 'Prefer "He said "Hi"".', length: 2 };
@@ -55113,6 +55161,25 @@ function mapFlowchartParserError(err, text) {
           length: len
         };
       }
+      {
+        const caret0 = Math.max(0, column - 1);
+        const openIdx = lineStr.lastIndexOf("{", caret0);
+        if (openIdx !== -1) {
+          const closeIdx = lineStr.indexOf("}", openIdx + 1);
+          const seg = closeIdx !== -1 ? lineStr.slice(openIdx + 1, closeIdx) : lineStr.slice(openIdx + 1);
+          if (seg.includes("(") || seg.includes(")")) {
+            return {
+              line,
+              column,
+              severity: "error",
+              code: "FL-LABEL-PARENS-UNQUOTED",
+              message: "Parentheses inside an unquoted label are not supported by Mermaid.",
+              hint: 'Wrap the label in quotes, e.g., A["Mark (X)"] \u2014 or replace ( and ) with HTML entities: ( and ).',
+              length: len
+            };
+          }
+        }
+      }
       const q3 = findInnerQuoteIssue("{");
       if (q3?.kind === "escaped") {
         return {
@@ -55451,6 +55518,17 @@ ${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 };
@@ -55608,6 +55686,17 @@ ${br.example}`,
       };
     }
   }
+  if ((err.name === "NotAllInputParsedException" || err.name === "NoViableAltException") && found === "-") {
+    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, for example:\nNote over A : Item 1\nNote over A\n  - Item 1\n  - Item 2\nend note",
+      length: len
+    };
+  }
   if ((err.name === "NoViableAltException" || err.name === "NotAllInputParsedException") && tokType === "ElseKeyword") {
     return { line, column, severity: "error", code: "SE-ELSE-OUTSIDE-ALT", message: "'else' is only allowed inside 'alt' blocks.", hint: "Use: alt Condition \u2026 else \u2026 end", length: len };
   }
@@ -55809,6 +55898,30 @@ function validateFlowchart(text, options = {}) {
       return errs;
     },
     postParse: (text2, tokens, _cst, prevErrors) => {
+      {
+        const tks = tokens;
+        const firstByLine = /* @__PURE__ */ new Map();
+        for (const tk of tks) {
+          const ln = tk.startLine ?? 1;
+          const col = tk.startColumn ?? 1;
+          const prev = firstByLine.get(ln);
+          if (!prev || (prev.startColumn ?? Infinity) > col)
+            firstByLine.set(ln, tk);
+        }
+        for (const tk of tks) {
+          if (tk.image === "title" && firstByLine.get(tk.startLine ?? 1) === tk) {
+            prevErrors.push({
+              line: tk.startLine ?? 1,
+              column: tk.startColumn ?? 1,
+              severity: "error",
+              code: "FL-META-UNSUPPORTED",
+              message: "'title' is not supported in flowcharts by the current Mermaid CLI.",
+              hint: 'Use a Markdown heading above the code block, or draw a labeled node at the top (e.g., T["Dependency Relationship"]).',
+              length: tk.image?.length ?? 5
+            });
+          }
+        }
+      }
       const escWarn = detectEscapedQuotes(tokens, {
         code: "FL-LABEL-ESCAPED-QUOTE",
         message: 'Escaped quotes (\\") in node labels are accepted by Mermaid, but using &quot; is preferred for portability.',
@@ -58227,6 +58340,13 @@ function computeFixes(text, errors, level = "safe") {
               const replaced = inner.split('\\"').join("&quot;");
               edits.push({ start: { line: e3.line, column: q1 + 2 }, end: { line: e3.line, column: q22 + 1 }, newText: replaced });
               continue;
+              if (is("FL-META-UNSUPPORTED", e3)) {
+                if (level === "all") {
+                  const lineText2 = lineTextAt(text, e3.line);
+                  edits.push({ start: { line: e3.line, column: 1 }, end: { line: e3.line + 1, column: 1 }, newText: "" });
+                }
+                continue;
+              }
             }
           }
         }
@@ -58234,6 +58354,12 @@ function computeFixes(text, errors, level = "safe") {
       edits.push(replaceRange(text, at(e3), e3.length ?? 2, "&quot;"));
       continue;
     }
+    if (is("FL-META-UNSUPPORTED", e3)) {
+      if (level === "all") {
+        edits.push({ start: { line: e3.line, column: 1 }, end: { line: e3.line + 1, column: 1 }, newText: "" });
+      }
+      continue;
+    }
     if (is("FL-LABEL-BACKTICK", e3)) {
       edits.push(replaceRange(text, at(e3), e3.length ?? 1, ""));
       continue;
@@ -58283,6 +58409,12 @@ function computeFixes(text, errors, level = "safe") {
         edits.push(replaceRange(text, at(e3), e3.length ?? 1, rep));
       continue;
     }
+    if (is("FL-END-WITHOUT-SUBGRAPH", e3)) {
+      if (level === "all") {
+        edits.push({ start: { line: e3.line, column: 1 }, end: { line: e3.line + 1, column: 1 }, newText: "" });
+      }
+      continue;
+    }
     if (is("FL-LABEL-DOUBLE-IN-DOUBLE", e3)) {
       const lineText = lineTextAt(text, e3.line);
       const caret0 = Math.max(0, e3.column - 1);
@@ -58619,6 +58751,9 @@ function computeFixes(text, errors, level = "safe") {
       continue;
     }
     if (is("FL-QUOTE-UNCLOSED", e3)) {
+      if (patchedLines.has(e3.line)) {
+        continue;
+      }
       if (level === "all") {
         const lineText = lineTextAt(text, e3.line);
         const caret0 = Math.max(0, e3.column - 1);
@@ -58693,7 +58828,7 @@ function computeFixes(text, errors, level = "safe") {
               newInner = ltrim + left + replacedMid + right + rtrim;
             } else {
               const replaced = inner.split("&quot;").join("\0").split('"').join("&quot;").split("\0").join("&quot;");
-              newInner = '"' + replaced + '"';
+              newInner = replaced;
             }
             edits.push({ start: { line: e3.line, column: contentStart + 1 }, end: { line: e3.line, column: closeIdx + 1 }, newText: newInner });
             patchedLines.add(e3.line);
@@ -73242,7 +73377,7 @@ var init_ProbeAgent = __esm({
     MAX_HISTORY_MESSAGES = 100;
     SUPPORTED_IMAGE_EXTENSIONS = ["png", "jpg", "jpeg", "webp", "gif", "bmp", "svg"];
     MAX_IMAGE_FILE_SIZE = 20 * 1024 * 1024;
-    ProbeAgent = class {
+    ProbeAgent = class _ProbeAgent {
       /**
        * Create a new ProbeAgent instance
        * @param {Object} options - Configuration options
@@ -74124,12 +74259,24 @@ You are working with a repository located at: ${searchDirectory}
               }))
             ];
           }
-          let currentMessages = [
-            { role: "system", content: systemMessage },
-            ...this.history,
-            // Include previous conversation history
-            userMessage
-          ];
+          const hasSystemMessage = this.history.length > 0 && this.history[0].role === "system";
+          let currentMessages;
+          if (hasSystemMessage) {
+            currentMessages = [
+              ...this.history,
+              userMessage
+            ];
+            if (this.debug) {
+              console.log("[DEBUG] Reusing existing system message from history for cache efficiency");
+            }
+          } else {
+            currentMessages = [
+              { role: "system", content: systemMessage },
+              ...this.history,
+              // Include previous conversation history
+              userMessage
+            ];
+          }
           let currentIteration = 0;
           let completionAttempted = false;
           let finalResult = "I was unable to complete your request due to reaching the maximum number of tool iterations.";
@@ -74924,6 +75071,116 @@ Convert your previous response content into actual JSON data that follows this s
           console.log(`[DEBUG] Cleared conversation history and reset counters for session ${this.sessionId}`);
         }
       }
+      /**
+       * Clone this agent's session to create a new agent with shared conversation history
+       * @param {Object} options - Clone options
+       * @param {string} [options.sessionId] - Session ID for the cloned agent (defaults to new UUID)
+       * @param {boolean} [options.stripInternalMessages=true] - Remove internal messages (schema reminders, mermaid fixes, etc.)
+       * @param {boolean} [options.keepSystemMessage=true] - Keep the system message in cloned history
+       * @param {boolean} [options.deepCopy=true] - Deep copy messages to prevent mutations
+       * @param {Object} [options.overrides] - Override any ProbeAgent constructor options
+       * @returns {ProbeAgent} New agent instance with cloned history
+       */
+      clone(options = {}) {
+        const {
+          sessionId = (0, import_crypto5.randomUUID)(),
+          stripInternalMessages = true,
+          keepSystemMessage = true,
+          deepCopy = true,
+          overrides = {}
+        } = options;
+        let clonedHistory = deepCopy ? JSON.parse(JSON.stringify(this.history)) : [...this.history];
+        if (stripInternalMessages) {
+          clonedHistory = this._stripInternalMessages(clonedHistory, keepSystemMessage);
+        }
+        const clonedAgent = new _ProbeAgent({
+          // Copy current agent's config
+          customPrompt: this.customPrompt,
+          promptType: this.promptType,
+          allowEdit: this.allowEdit,
+          path: this.allowedFolders[0],
+          // Use first allowed folder as primary path
+          allowedFolders: [...this.allowedFolders],
+          provider: this.clientApiProvider,
+          model: this.modelName,
+          debug: this.debug,
+          outline: this.outline,
+          maxResponseTokens: this.maxResponseTokens,
+          maxIterations: this.maxIterations,
+          disableMermaidValidation: this.disableMermaidValidation,
+          enableMcp: !!this.mcpBridge,
+          mcpConfig: this.mcpConfig,
+          enableBash: this.enableBash,
+          bashConfig: this.bashConfig,
+          storageAdapter: this.storageAdapter,
+          // Override with any provided options
+          sessionId,
+          ...overrides
+        });
+        clonedAgent.history = clonedHistory;
+        if (this.debug) {
+          console.log(`[DEBUG] Cloned session ${this.sessionId} -> ${sessionId}`);
+          console.log(`[DEBUG] Cloned ${clonedHistory.length} messages (stripInternal: ${stripInternalMessages})`);
+        }
+        return clonedAgent;
+      }
+      /**
+       * Internal method to strip internal/temporary messages from history
+       * Removes: schema reminders, mermaid fix prompts, tool use reminders, etc.
+       * Keeps: system message, user messages, assistant responses, tool results
+       * @private
+       */
+      _stripInternalMessages(history, keepSystemMessage = true) {
+        const filtered = [];
+        for (let i3 = 0; i3 < history.length; i3++) {
+          const message = history[i3];
+          if (message.role === "system") {
+            if (keepSystemMessage) {
+              filtered.push(message);
+            } else if (this.debug) {
+              console.log(`[DEBUG] Removing system message at index ${i3}`);
+            }
+            continue;
+          }
+          if (this._isInternalMessage(message, i3, history)) {
+            if (this.debug) {
+              console.log(`[DEBUG] Stripping internal message at index ${i3}: ${message.role}`);
+            }
+            continue;
+          }
+          filtered.push(message);
+        }
+        return filtered;
+      }
+      /**
+       * Determine if a message is an internal/temporary message
+       * @private
+       */
+      _isInternalMessage(message, index, history) {
+        if (message.role !== "user") {
+          return false;
+        }
+        if (!message.content) {
+          return false;
+        }
+        const content = typeof message.content === "string" ? message.content : JSON.stringify(message.content);
+        if (content.includes("IMPORTANT: A schema was provided") || content.includes("You MUST respond with data that matches this schema") || content.includes("Your response must conform to this schema:")) {
+          return true;
+        }
+        if (content.includes("Please use one of the available tools") && content.includes("or use attempt_completion") && content.includes("Remember: Use proper XML format")) {
+          return true;
+        }
+        if (content.includes("The mermaid diagram in your response has syntax errors") || content.includes("Please fix the mermaid syntax errors") || content.includes("Here is the corrected version:")) {
+          return true;
+        }
+        if (content.includes("Your response does not match the expected JSON schema") || content.includes("Please provide a valid JSON response") || content.includes("Schema validation error:")) {
+          return true;
+        }
+        if (content.includes("When using <attempt_complete>") && content.includes("this must be the ONLY content in your response")) {
+          return true;
+        }
+        return false;
+      }
       /**
        * Clean up resources (including MCP connections)
        */