@probelabs/probe 0.6.0-rc128 → 0.6.0-rc129

package/cjs/index.cjs

@@ -54812,7 +54812,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 || "");
@@ -54822,7 +54822,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".',
@@ -54834,12 +54834,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;
           }
         }
@@ -55240,6 +55240,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,
@@ -55272,6 +55302,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 };
@@ -55293,6 +55341,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 {
@@ -55631,6 +55698,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 };
@@ -55788,6 +55866,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 };
   }
@@ -55989,6 +56078,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.',
@@ -58407,6 +58520,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;
+              }
             }
           }
         }
@@ -58414,6 +58534,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;
@@ -58463,6 +58589,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);
@@ -58799,6 +58931,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);
@@ -58873,7 +59008,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);
@@ -73422,7 +73557,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
@@ -74304,12 +74439,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.";
@@ -75104,6 +75251,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)
        */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc128",
+  "version": "0.6.0-rc129",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",
@@ -80,7 +80,7 @@
     "@opentelemetry/sdk-node": "^0.203.0",
     "@opentelemetry/sdk-trace-base": "^1.30.0",
     "@opentelemetry/semantic-conventions": "^1.36.0",
-    "@probelabs/maid": "^0.0.13",
+    "@probelabs/maid": "^0.0.15",
     "ai": "^5.0.0",
     "axios": "^1.8.3",
     "fs-extra": "^11.1.1",

package/src/agent/ProbeAgent.js

@@ -1152,11 +1152,28 @@ When troubleshooting:
       }
 
       // Initialize conversation with existing history + new user message
-      let currentMessages = [
-        { role: 'system', content: systemMessage },
-        ...this.history, // Include previous conversation history
-        userMessage
-      ];
+      // If history already contains a system message (from session cloning), reuse it for cache efficiency
+      // Otherwise add a fresh system message
+      const hasSystemMessage = this.history.length > 0 && this.history[0].role === 'system';
+      let currentMessages;
+
+      if (hasSystemMessage) {
+        // Reuse existing system message from history for cache efficiency
+        currentMessages = [
+          ...this.history,
+          userMessage
+        ];
+        if (this.debug) {
+          console.log('[DEBUG] Reusing existing system message from history for cache efficiency');
+        }
+      } else {
+        // Add fresh system message (first call or empty history)
+        currentMessages = [
+          { role: 'system', content: systemMessage },
+          ...this.history, // Include previous conversation history
+          userMessage
+        ];
+      }
 
       let currentIteration = 0;
       let completionAttempted = false;
@@ -2147,6 +2164,163 @@ Convert your previous response content into actual JSON data that follows this s
     }
   }
 
+  /**
+   * 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 = randomUUID(),
+      stripInternalMessages = true,
+      keepSystemMessage = true,
+      deepCopy = true,
+      overrides = {}
+    } = options;
+
+    // Clone the history
+    let clonedHistory = deepCopy
+      ? JSON.parse(JSON.stringify(this.history))
+      : [...this.history];
+
+    // Strip internal messages if requested
+    if (stripInternalMessages) {
+      clonedHistory = this._stripInternalMessages(clonedHistory, keepSystemMessage);
+    }
+
+    // Create new agent with same configuration
+    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
+    });
+
+    // Set the cloned history directly (before initialization to avoid overwriting)
+    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 i = 0; i < history.length; i++) {
+      const message = history[i];
+
+      // Handle system message
+      if (message.role === 'system') {
+        if (keepSystemMessage) {
+          filtered.push(message);
+        } else if (this.debug) {
+          console.log(`[DEBUG] Removing system message at index ${i}`);
+        }
+        continue;
+      }
+
+      // Check if this is an internal message that should be stripped
+      if (this._isInternalMessage(message, i, history)) {
+        if (this.debug) {
+          console.log(`[DEBUG] Stripping internal message at index ${i}: ${message.role}`);
+        }
+        continue;
+      }
+
+      // Keep this message
+      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; // Only user messages can be internal reminders
+    }
+
+    // Handle null/undefined content
+    if (!message.content) {
+      return false;
+    }
+
+    const content = typeof message.content === 'string'
+      ? message.content
+      : JSON.stringify(message.content);
+
+    // Schema reminder messages
+    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;
+    }
+
+    // Tool use reminder messages
+    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;
+    }
+
+    // Mermaid fix prompts
+    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;
+    }
+
+    // JSON correction prompts
+    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;
+    }
+
+    // Empty attempt_complete reminders
+    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)
    */