@probelabs/probe 0.6.0-rc125 → 0.6.0-rc127

package/build/agent/index.js

@@ -556,7 +556,7 @@ var init_BytePairEncodingCore = __esm({
 var chatEnabledModels;
 var init_modelsChatEnabled_gen = __esm({
   "node_modules/gpt-tokenizer/esm/modelsChatEnabled.gen.js"() {
-    chatEnabledModels = ["chatgpt-4o-latest", "codex-mini-latest", "computer-use-preview", "computer-use-preview-2025-03-11", "gpt-3.5", "gpt-3.5-0301", "gpt-3.5-turbo", "gpt-3.5-turbo-0125", "gpt-3.5-turbo-0613", "gpt-3.5-turbo-1106", "gpt-3.5-turbo-16k-0613", "gpt-3.5-turbo-instruct", "gpt-4", "gpt-4-0125-preview", "gpt-4-0314", "gpt-4-0613", "gpt-4-1106-preview", "gpt-4-1106-vision-preview", "gpt-4-32k", "gpt-4-turbo", "gpt-4-turbo-2024-04-09", "gpt-4-turbo-preview", "gpt-4.1", "gpt-4.1-2025-04-14", "gpt-4.1-mini", "gpt-4.1-mini-2025-04-14", "gpt-4.1-nano", "gpt-4.1-nano-2025-04-14", "gpt-4.5-preview", "gpt-4.5-preview-2025-02-27", "gpt-4o", "gpt-4o-2024-05-13", "gpt-4o-2024-08-06", "gpt-4o-2024-11-20", "gpt-4o-audio-preview", "gpt-4o-audio-preview-2024-10-01", "gpt-4o-audio-preview-2024-12-17", "gpt-4o-audio-preview-2025-06-03", "gpt-4o-mini", "gpt-4o-mini-2024-07-18", "gpt-4o-mini-audio-preview", "gpt-4o-mini-audio-preview-2024-12-17", "gpt-4o-mini-search-preview", "gpt-4o-mini-search-preview-2025-03-11", "gpt-4o-search-preview", "gpt-4o-search-preview-2025-03-11", "o1", "o1-2024-12-17", "o1-mini", "o1-mini-2024-09-12", "o1-preview", "o1-preview-2024-09-12", "o1-pro", "o1-pro-2025-03-19", "o3", "o3-2025-04-16", "o3-mini", "o3-mini-2025-01-31", "o3-pro", "o3-pro-2025-06-10", "o4-mini", "o4-mini-2025-04-16"];
+    chatEnabledModels = ["chatgpt-4o-latest", "codex-mini-latest", "computer-use-preview", "computer-use-preview-2025-03-11", "gpt-3.5", "gpt-3.5-0301", "gpt-3.5-turbo", "gpt-3.5-turbo-0125", "gpt-3.5-turbo-0613", "gpt-3.5-turbo-1106", "gpt-3.5-turbo-16k-0613", "gpt-3.5-turbo-instruct", "gpt-4", "gpt-4-0125-preview", "gpt-4-0314", "gpt-4-0613", "gpt-4-1106-preview", "gpt-4-1106-vision-preview", "gpt-4-32k", "gpt-4-turbo", "gpt-4-turbo-2024-04-09", "gpt-4-turbo-preview", "gpt-4.1", "gpt-4.1-2025-04-14", "gpt-4.1-mini", "gpt-4.1-mini-2025-04-14", "gpt-4.1-nano", "gpt-4.1-nano-2025-04-14", "gpt-4.5-preview", "gpt-4.5-preview-2025-02-27", "gpt-4o", "gpt-4o-2024-05-13", "gpt-4o-2024-08-06", "gpt-4o-2024-11-20", "gpt-4o-audio-preview", "gpt-4o-audio-preview-2024-10-01", "gpt-4o-audio-preview-2024-12-17", "gpt-4o-audio-preview-2025-06-03", "gpt-4o-mini", "gpt-4o-mini-2024-07-18", "gpt-4o-mini-audio-preview", "gpt-4o-mini-audio-preview-2024-12-17", "gpt-4o-mini-search-preview", "gpt-4o-mini-search-preview-2025-03-11", "gpt-4o-search-preview", "gpt-4o-search-preview-2025-03-11", "gpt-5", "gpt-5-2025-08-07", "gpt-5-chat-latest", "gpt-5-codex", "gpt-5-mini", "gpt-5-mini-2025-08-07", "gpt-5-nano", "gpt-5-nano-2025-08-07", "gpt-5-pro", "gpt-5-pro-2025-10-06", "gpt-audio", "gpt-audio-2025-08-28", "gpt-audio-mini", "gpt-audio-mini-2025-10-06", "gpt-oss-120b", "gpt-oss-20b", "o1", "o1-2024-12-17", "o1-mini", "o1-mini-2024-09-12", "o1-preview", "o1-preview-2024-09-12", "o1-pro", "o1-pro-2025-03-19", "o3", "o3-2025-04-16", "o3-deep-research", "o3-deep-research-2025-06-26", "o3-mini", "o3-mini-2025-01-31", "o3-pro", "o3-pro-2025-06-10", "o4-mini", "o4-mini-2025-04-16", "o4-mini-deep-research", "o4-mini-deep-research-2025-06-26"];
   }
 });
 
@@ -952,8 +952,9 @@ var init_GptEncoding = __esm({
        * Returns tokens assuming the 'gpt-3.5-turbo-0301' / 'gpt-4-0314' format.
        * Based on OpenAI's guidelines: https://github.com/openai/openai-python/blob/main/chatml.md
        * Also mentioned in section 6 of this document: https://github.com/openai/openai-cookbook/blob/main/examples/How_to_count_tokens_with_tiktoken.ipynb
+       * @param encodeOptions Options controlling how special tokens are handled.
        */
-      *encodeChatGenerator(chat, model = this.modelName) {
+      *encodeChatGenerator(chat, model = this.modelName, encodeOptions) {
         if (!model) {
           throw new Error("Model name must be provided either during initialization or passed in to the method.");
         }
@@ -979,12 +980,12 @@ var init_GptEncoding = __esm({
           if (encodedRoleSeparator.length > 0) {
             yield encodedRoleSeparator;
           }
-          yield* this.encodeGenerator(content);
+          yield* this.encodeGenerator(content, encodeOptions);
           yield [chatEndToken];
           yield encodedMessageSeparator;
         }
         yield [chatStartToken];
-        yield* this.encodeGenerator("assistant");
+        yield* this.encodeGenerator("assistant", encodeOptions);
         if (encodedRoleSeparator.length > 0) {
           yield encodedRoleSeparator;
         }
@@ -995,15 +996,20 @@ var init_GptEncoding = __esm({
        * Returns tokens assuming the 'gpt-3.5-turbo-0301' / 'gpt-4-0314' format.
        * Based on OpenAI's guidelines: https://github.com/openai/openai-python/blob/main/chatml.md
        * Also mentioned in section 6 of this document: https://github.com/openai/openai-cookbook/blob/main/examples/How_to_count_tokens_with_tiktoken.ipynb
+       * @param encodeOptions Options controlling how special tokens are handled.
        */
-      encodeChat(chat, model = this.modelName) {
-        return [...this.encodeChatGenerator(chat, model)].flat();
+      encodeChat(chat, model = this.modelName, encodeOptions) {
+        return [...this.encodeChatGenerator(chat, model, encodeOptions)].flat();
       }
       /**
+       * Checks whether the provided input stays within the provided token limit.
+       * @param input The string or chat messages to evaluate.
+       * @param tokenLimit The maximum allowed number of tokens.
+       * @param encodeOptions Options controlling how special tokens are handled.
        * @returns {false | number} false if token limit is exceeded, otherwise the number of tokens
        */
-      isWithinTokenLimit(input, tokenLimit) {
-        const tokenGenerator = typeof input === "string" ? this.encodeGenerator(input) : this.encodeChatGenerator(input);
+      isWithinTokenLimit(input, tokenLimit, encodeOptions) {
+        const tokenGenerator = typeof input === "string" ? this.encodeGenerator(input, encodeOptions) : this.encodeChatGenerator(input, void 0, encodeOptions);
         let count = 0;
         for (const tokens of tokenGenerator) {
           count += tokens.length;
@@ -1015,6 +1021,8 @@ var init_GptEncoding = __esm({
       }
       /**
        * Counts the number of tokens in the input.
+       * @param input The string or chat messages to evaluate.
+       * @param encodeOptions Options controlling how special tokens are handled.
        * @returns {number} The number of tokens.
        */
       countTokens(input, encodeOptions) {
@@ -1028,7 +1036,7 @@ var init_GptEncoding = __esm({
           }
           return this.bytePairEncodingCoreProcessor.countNative(input, specialTokenConfig.allowedSpecial);
         }
-        const tokenGenerator = this.encodeChatGenerator(input);
+        const tokenGenerator = this.encodeChatGenerator(input, void 0, encodeOptions);
         let count = 0;
         for (const tokens of tokenGenerator) {
           count += tokens.length;
@@ -1466,6 +1474,226 @@ var init_tokenCounter = __esm({
   }
 });
 
+// src/agent/storage/StorageAdapter.js
+var StorageAdapter;
+var init_StorageAdapter = __esm({
+  "src/agent/storage/StorageAdapter.js"() {
+    "use strict";
+    StorageAdapter = class {
+      /**
+       * Load conversation history for a session
+       * @param {string} sessionId - Session identifier
+       * @returns {Promise<Array<Object>>} Array of message objects with {role, content, ...}
+       */
+      async loadHistory(sessionId) {
+        throw new Error("StorageAdapter.loadHistory() must be implemented by subclass");
+      }
+      /**
+       * Save a message to storage
+       * @param {string} sessionId - Session identifier
+       * @param {Object} message - Message object { role, content, ... }
+       * @returns {Promise<void>}
+       */
+      async saveMessage(sessionId, message) {
+        throw new Error("StorageAdapter.saveMessage() must be implemented by subclass");
+      }
+      /**
+       * Clear history for a session
+       * @param {string} sessionId - Session identifier
+       * @returns {Promise<void>}
+       */
+      async clearHistory(sessionId) {
+        throw new Error("StorageAdapter.clearHistory() must be implemented by subclass");
+      }
+      /**
+       * Get session metadata (optional)
+       * @param {string} sessionId - Session identifier
+       * @returns {Promise<Object|null>} Session metadata or null
+       */
+      async getSessionMetadata(sessionId) {
+        return null;
+      }
+      /**
+       * Update session activity timestamp (optional)
+       * @param {string} sessionId - Session identifier
+       * @returns {Promise<void>}
+       */
+      async updateSessionActivity(sessionId) {
+      }
+    };
+  }
+});
+
+// src/agent/storage/InMemoryStorageAdapter.js
+var InMemoryStorageAdapter;
+var init_InMemoryStorageAdapter = __esm({
+  "src/agent/storage/InMemoryStorageAdapter.js"() {
+    "use strict";
+    init_StorageAdapter();
+    InMemoryStorageAdapter = class extends StorageAdapter {
+      constructor() {
+        super();
+        this.sessions = /* @__PURE__ */ new Map();
+      }
+      async loadHistory(sessionId) {
+        const session = this.sessions.get(sessionId);
+        return session ? session.messages : [];
+      }
+      async saveMessage(sessionId, message) {
+        if (!this.sessions.has(sessionId)) {
+          this.sessions.set(sessionId, {
+            messages: [],
+            metadata: {
+              createdAt: (/* @__PURE__ */ new Date()).toISOString(),
+              lastActivity: (/* @__PURE__ */ new Date()).toISOString()
+            }
+          });
+        }
+        const session = this.sessions.get(sessionId);
+        session.messages.push(message);
+        session.metadata.lastActivity = (/* @__PURE__ */ new Date()).toISOString();
+      }
+      async clearHistory(sessionId) {
+        this.sessions.delete(sessionId);
+      }
+      async getSessionMetadata(sessionId) {
+        const session = this.sessions.get(sessionId);
+        return session ? session.metadata : null;
+      }
+      async updateSessionActivity(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (session) {
+          session.metadata.lastActivity = (/* @__PURE__ */ new Date()).toISOString();
+        }
+      }
+    };
+  }
+});
+
+// src/agent/hooks/HookManager.js
+var HookManager, HOOK_TYPES;
+var init_HookManager = __esm({
+  "src/agent/hooks/HookManager.js"() {
+    "use strict";
+    HookManager = class {
+      constructor() {
+        this.hooks = /* @__PURE__ */ new Map();
+      }
+      /**
+       * Register a hook callback
+       * @param {string} hookName - Name of the hook
+       * @param {Function} callback - Callback function
+       * @returns {Function} Unregister function
+       */
+      on(hookName, callback) {
+        if (!this.hooks.has(hookName)) {
+          this.hooks.set(hookName, /* @__PURE__ */ new Set());
+        }
+        this.hooks.get(hookName).add(callback);
+        return () => this.off(hookName, callback);
+      }
+      /**
+       * Register a one-time hook callback
+       * @param {string} hookName - Name of the hook
+       * @param {Function} callback - Callback function
+       * @returns {Function} Unregister function
+       */
+      once(hookName, callback) {
+        const wrappedCallback = async (data) => {
+          this.off(hookName, wrappedCallback);
+          await callback(data);
+        };
+        return this.on(hookName, wrappedCallback);
+      }
+      /**
+       * Unregister a hook callback
+       * @param {string} hookName - Name of the hook
+       * @param {Function} callback - Callback function
+       */
+      off(hookName, callback) {
+        const callbacks = this.hooks.get(hookName);
+        if (callbacks) {
+          callbacks.delete(callback);
+        }
+      }
+      /**
+       * Emit a hook event
+       * @param {string} hookName - Name of the hook
+       * @param {any} data - Data to pass to callbacks
+       * @returns {Promise<void>}
+       */
+      async emit(hookName, data) {
+        const callbacks = this.hooks.get(hookName);
+        if (!callbacks || callbacks.size === 0) return;
+        const promises = Array.from(callbacks).map((callback) => {
+          try {
+            return Promise.resolve(callback(data));
+          } catch (error) {
+            return Promise.reject(error);
+          }
+        });
+        const results = await Promise.allSettled(promises);
+        results.forEach((result, index) => {
+          if (result.status === "rejected") {
+            console.error(`[HookManager] Error in hook "${hookName}" (callback ${index + 1}):`, result.reason);
+          }
+        });
+      }
+      /**
+       * Clear all hooks or hooks for a specific event
+       * @param {string} [hookName] - Optional hook name to clear
+       */
+      clear(hookName) {
+        if (hookName) {
+          this.hooks.delete(hookName);
+        } else {
+          this.hooks.clear();
+        }
+      }
+      /**
+       * Get list of registered hook names
+       * @returns {string[]} Array of hook names
+       */
+      getHookNames() {
+        return Array.from(this.hooks.keys());
+      }
+      /**
+       * Get number of callbacks for a hook
+       * @param {string} hookName - Name of the hook
+       * @returns {number} Number of callbacks
+       */
+      getCallbackCount(hookName) {
+        const callbacks = this.hooks.get(hookName);
+        return callbacks ? callbacks.size : 0;
+      }
+    };
+    HOOK_TYPES = {
+      // Lifecycle hooks
+      AGENT_INITIALIZED: "agent:initialized",
+      AGENT_CLEANUP: "agent:cleanup",
+      // Message hooks
+      MESSAGE_USER: "message:user",
+      MESSAGE_ASSISTANT: "message:assistant",
+      MESSAGE_SYSTEM: "message:system",
+      // Tool execution hooks
+      TOOL_START: "tool:start",
+      TOOL_END: "tool:end",
+      TOOL_ERROR: "tool:error",
+      // AI streaming hooks
+      AI_STREAM_START: "ai:stream:start",
+      AI_STREAM_DELTA: "ai:stream:delta",
+      AI_STREAM_END: "ai:stream:end",
+      // Storage hooks
+      STORAGE_LOAD: "storage:load",
+      STORAGE_SAVE: "storage:save",
+      STORAGE_CLEAR: "storage:clear",
+      // Iteration hooks
+      ITERATION_START: "iteration:start",
+      ITERATION_END: "iteration:end"
+    };
+  }
+});
+
 // src/directory-resolver.js
 import path from "path";
 import os from "os";
@@ -16121,6 +16349,23 @@ var init_probeTool = __esm({
   }
 });
 
+// src/agent/storage/index.js
+var init_storage = __esm({
+  "src/agent/storage/index.js"() {
+    "use strict";
+    init_StorageAdapter();
+    init_InMemoryStorageAdapter();
+  }
+});
+
+// src/agent/hooks/index.js
+var init_hooks = __esm({
+  "src/agent/hooks/index.js"() {
+    "use strict";
+    init_HookManager();
+  }
+});
+
 // src/index.js
 var init_index = __esm({
   "src/index.js"() {
@@ -16140,6 +16385,8 @@ var init_index = __esm({
     init_ProbeAgent();
     init_simpleTelemetry();
     init_probeTool();
+    init_storage();
+    init_hooks();
   }
 });
 
@@ -28104,7 +28351,6 @@ var init_parser2 = __esm({
         this.linkStyleStatement = this.RULE("linkStyleStatement", () => {
           this.CONSUME(LinkStyleKeyword);
           this.SUBRULE(this.linkStyleIndexList);
-          this.OPTION1(() => this.CONSUME1(Newline));
           this.SUBRULE(this.linkStylePairs);
           this.OPTION2(() => this.CONSUME2(Newline));
         });
@@ -28568,7 +28814,7 @@ var init_semantics = __esm({
           }
           const fnTok = ch.fn?.[0];
           if (!fnTok) {
-            this.ctx.errors.push({ line: modeTok2?.startLine ?? 1, column: modeTok2?.startColumn ?? 1, severity: "error", code: "FL-CLICK-CALL-NAME-MISSING", message: "'click \u2026 call' requires a function name.", hint: 'Example: click A call doThing() "Tooltip"' });
+            this.ctx.errors.push({ line: modeTok2?.startLine ?? 1, column: modeTok2?.startColumn ?? 1, severity: "error", code: "FL-CLICK-CALL-NAME-MISSING", message: "'click \u2026 call' requires a function name.", hint: "Example: click A call doThing()" });
           }
           const tipTok = ch.tooltip?.[0];
           if (tipTok) {
@@ -28625,7 +28871,7 @@ var init_semantics = __esm({
               severity: "error",
               code: "FL-CLICK-CALL-NAME-MISSING",
               message: "'click \u2026 call' requires a function name.",
-              hint: 'Example: click A call doThing() "Tooltip"'
+              hint: "Example: click A call doThing()"
             });
           }
           return;
@@ -28706,6 +28952,8 @@ var init_semantics = __esm({
         const linksHere = Array.isArray(ctx.link) ? ctx.link.length : 0;
         if (linksHere > 0)
           this.edgeCount += linksHere;
+        if (ctx.link)
+          ctx.link.forEach((ln) => this.visit(ln));
       }
       // Edge attribute object statements must target a known edge id
       // Edge attribute statements are parsed as nodeStatements with a typed attrObject and no links.
@@ -28765,7 +29013,7 @@ var init_semantics = __esm({
                   severity: "error",
                   code: "FL-TYPED-SHAPE-UNKNOWN",
                   message: `Unknown shape '${v}' in '@{ shape: \u2026 }'.`,
-                  hint: "Use one of: rect, round, stadium, subroutine, circle, cylinder, diamond, trapezoid, parallelogram, hexagon, lean-l, lean-r, icon, image"
+                  hint: 'Use one of: rect, rounded, stadium, subroutine, circle, cylinder, diamond, trapezoid, parallelogram, hexagon, "lean-l", "lean-r", icon, image'
                 });
               } else {
                 const supportedByCli = /* @__PURE__ */ new Set(["rect", "round", "rounded", "diamond", "circle", "cylinder", "stadium", "subroutine", "lean-l", "lean-r"]);
@@ -28967,8 +29215,8 @@ var init_semantics = __esm({
               column: t.startColumn ?? 1,
               severity: "error",
               code: "FL-LABEL-PARENS-UNQUOTED",
-              message: "Parentheses inside an unquoted label are not supported by Mermaid. Wrap the label in quotes.",
-              hint: 'Example: A["Calls func(arg)"]'
+              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: &#40; and &#41;'
             });
           }
         }
@@ -29016,6 +29264,29 @@ var init_semantics = __esm({
           }
         }
       }
+      // Validate link specifics that Mermaid CLI enforces but our grammar may parse loosely
+      link(ctx) {
+        const inline = ctx.linkTextInline?.[0];
+        if (inline) {
+          const ch = inline.children || {};
+          const parts = [].concat(ch.Identifier || []).concat(ch.Text || []).concat(ch.NumberLiteral || []).concat(ch.Pipe || []);
+          if (parts.length) {
+            const raw = parts.map((t) => String(t.image || "")).join("").trim();
+            if (raw === "x" || raw === "o") {
+              const p = parts[0];
+              this.ctx.errors.push({
+                line: p.startLine ?? 1,
+                column: p.startColumn ?? 1,
+                severity: "error",
+                code: "FL-LINK-UNSUPPORTED-MARKER",
+                message: `Unsupported one-sided link marker '${raw}'. Use symmetric '${raw}--${raw}' or a plain arrow with a label.`,
+                hint: raw === "x" ? "Example: A x--x B  (or)  A --> B  and label it: A --|Skipped|--> B" : "Example: A o--o B  (or)  A --> B",
+                length: p.image?.length ?? 1
+              });
+            }
+          }
+        }
+      }
       checkBackticksInContent(contentNodes) {
         if (!contentNodes)
           return;
@@ -29367,6 +29638,31 @@ function mapFlowchartParserError(err, text) {
       }
     }
   }
+  if (isInRule(err, "linkStylePairs") && tokType === "Newline") {
+    const nextLine = Math.min(allLines.length, line + 1);
+    const nxt = allLines[nextLine - 1] || "";
+    const first2 = (nxt.match(/\S/) || { index: 0 }).index || 0;
+    return {
+      line: nextLine,
+      column: Math.max(1, first2 + 1),
+      severity: "error",
+      code: "FL-LINKSTYLE-MULTILINE",
+      message: "'linkStyle' styles must be on the same line as the indices.",
+      hint: "Example: linkStyle 0,1 stroke:#f00,stroke-width:2px",
+      length: 1
+    };
+  }
+  if ((isInRule(err, "linkStyleIndexList") || isInRule(err, "linkStyleStatement")) && tokType === "Colon") {
+    return {
+      line,
+      column,
+      severity: "error",
+      code: "FL-LINKSTYLE-RANGE-NOT-SUPPORTED",
+      message: "Ranges in 'linkStyle' indices are not supported. Use comma-separated indices.",
+      hint: "Example: linkStyle 0,1 stroke:#f00,stroke-width:2px",
+      length: len
+    };
+  }
   if (tokType === "QuotedString") {
     const context = err?.context;
     const inLinkRule = context?.ruleStack?.includes("linkTextInline") || context?.ruleStack?.includes("link") || false;
@@ -29413,6 +29709,17 @@ function mapFlowchartParserError(err, text) {
   }
   if (isInRule(err, "nodeShape") && err.name === "MismatchedTokenException") {
     if (expecting(err, "SquareClose")) {
+      if (tokType === "RoundOpen" || tokType === "RoundClose") {
+        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: &#40; and &#41;.',
+          length: len
+        };
+      }
       if (tokType === "QuotedString") {
         return {
           line,
@@ -29908,6 +30215,32 @@ ${br.example}`,
           break;
         }
       }
+      if (blk.label === "box" && openIdx !== -1) {
+        let endIdx = -1;
+        for (let i = openIdx + 1; i < lines2.length; i++) {
+          const raw = lines2[i] || "";
+          const ind = raw.match(/^(\s*)/)?.[1] || "";
+          if (/^\s*end\s*$/.test(raw) && ind.length <= openIndent.length) {
+            endIdx = i;
+            break;
+          }
+        }
+        if (endIdx !== -1) {
+          const body = lines2.slice(openIdx + 1, endIdx).map((s) => (s || "").trim());
+          const hasMsgWithActivation = body.some((s) => /->/.test(s) && /[+-]/.test(s));
+          if (hasMsgWithActivation) {
+            return {
+              line: openIdx + 1,
+              column: 1,
+              severity: "error",
+              code: "SE-BOX-EMPTY",
+              message: "Box block has no participant/actor declarations. Use 'rect' to group messages visually.",
+              hint: "Replace 'box' with 'rect' if you want to group messages:\nrect rgb(240, 240, 255)\n  A->>B: Message\n  Note over A: Info\nend",
+              length: 3
+            };
+          }
+        }
+      }
       let caretLine = line;
       if (openIdx !== -1) {
         caretLine = lines2.length;
@@ -31247,7 +31580,7 @@ var init_validate3 = __esm({
 function tokenize4(text) {
   return ClassLexer.tokenize(text);
 }
-var Identifier3, NumberLiteral4, ClassDiagramKeyword, DirectionKw, Direction2, ClassKw, AsKw, NoteKw, ForKw, RelCompToAgg, RelAggToComp, RelCompBoth, RelAggBoth, LollipopLeft, LollipopRight, RelExtends, RelComposition, RelAggregation, RelDependency, RelRealization, RelAssociation, RelDependencyLeft, RelRealizationLeft, RelExtendsRight, InvalidRelArrow, LCurly, RCurly, LParen2, RParen2, Colon4, Comma3, Visibility, LTlt, GTgt, GenericAngle, QuotedString4, BacktickName, SquareOpen2, SquareClose2, Comment4, WhiteSpace4, Newline4, allTokens4, ClassLexer;
+var Identifier3, NumberLiteral4, ClassDiagramKeyword, DirectionKw, Direction2, TitleKw, NamespaceKw, ClassKw, InterfaceKw, AsKw, NoteKw, ForKw, RelCompToAgg, RelAggToComp, RelCompBoth, RelAggBoth, LollipopLeft, LollipopRight, RelExtends, RelComposition, RelAggregation, RelDependency, RelRealization, RelAssociation, RelDependencyLeft, RelRealizationLeft, RelExtendsRight, InvalidRelArrow, LCurly, RCurly, LParen2, RParen2, Colon4, Comma3, Visibility, LTlt, GTgt, GenericAngle, QuotedString4, BacktickName, SquareOpen2, SquareClose2, Comment4, WhiteSpace4, Newline4, allTokens4, ClassLexer;
 var init_lexer5 = __esm({
   "node_modules/@probelabs/maid/out/diagrams/class/lexer.js"() {
     init_api5();
@@ -31256,7 +31589,10 @@ var init_lexer5 = __esm({
     ClassDiagramKeyword = createToken({ name: "ClassDiagramKeyword", pattern: /classDiagram/, longer_alt: Identifier3 });
     DirectionKw = createToken({ name: "DirectionKw", pattern: /direction/, longer_alt: Identifier3 });
     Direction2 = createToken({ name: "Direction", pattern: /LR|RL|TB|BT|TD/, longer_alt: Identifier3 });
+    TitleKw = createToken({ name: "TitleKw", pattern: /title/, longer_alt: Identifier3 });
+    NamespaceKw = createToken({ name: "NamespaceKw", pattern: /namespace/, longer_alt: Identifier3 });
     ClassKw = createToken({ name: "ClassKw", pattern: /class/, longer_alt: Identifier3 });
+    InterfaceKw = createToken({ name: "InterfaceKw", pattern: /interface\b/, longer_alt: Identifier3 });
     AsKw = createToken({ name: "AsKw", pattern: /as/, longer_alt: Identifier3 });
     NoteKw = createToken({ name: "NoteKw", pattern: /note/, longer_alt: Identifier3 });
     ForKw = createToken({ name: "ForKw", pattern: /for/, longer_alt: Identifier3 });
@@ -31299,6 +31635,9 @@ var init_lexer5 = __esm({
       // Keywords
       ClassDiagramKeyword,
       DirectionKw,
+      TitleKw,
+      NamespaceKw,
+      InterfaceKw,
       ClassKw,
       AsKw,
       NoteKw,
@@ -31367,8 +31706,11 @@ var init_parser5 = __esm({
         });
         this.statement = this.RULE("statement", () => {
           this.OR([
+            { ALT: () => this.SUBRULE(this.titleStmt) },
             { ALT: () => this.SUBRULE(this.directionStmt) },
+            { ALT: () => this.SUBRULE(this.namespaceStmt) },
             { ALT: () => this.SUBRULE(this.classLine) },
+            { ALT: () => this.SUBRULE(this.interfaceLine) },
             { ALT: () => this.SUBRULE(this.relationStmt) },
             { ALT: () => this.SUBRULE(this.noteStmt) },
             { ALT: () => this.SUBRULE(this.memberAssignStmt) },
@@ -31380,6 +31722,43 @@ var init_parser5 = __esm({
           this.CONSUME(Direction2);
           this.OPTION(() => this.CONSUME(Newline4));
         });
+        this.titleStmt = this.RULE("titleStmt", () => {
+          this.CONSUME(TitleKw);
+          this.OR([
+            { ALT: () => this.CONSUME(QuotedString4) },
+            {
+              ALT: () => {
+                this.AT_LEAST_ONE(() => {
+                  this.OR2([
+                    { ALT: () => this.CONSUME(Identifier3) },
+                    { ALT: () => this.CONSUME(NumberLiteral4) }
+                  ]);
+                });
+              }
+            }
+          ]);
+          this.OPTION(() => this.CONSUME(Newline4));
+        });
+        this.namespaceStmt = this.RULE("namespaceStmt", () => {
+          this.CONSUME(NamespaceKw);
+          this.OR([
+            { ALT: () => this.CONSUME(QuotedString4) },
+            { ALT: () => this.CONSUME(Identifier3) }
+          ]);
+          this.CONSUME(LCurly);
+          this.MANY(() => {
+            this.OR2([
+              { ALT: () => this.CONSUME(Newline4) },
+              { ALT: () => this.SUBRULE(this.classLine) },
+              { ALT: () => this.SUBRULE(this.interfaceLine) },
+              { ALT: () => this.SUBRULE(this.relationStmt) },
+              { ALT: () => this.SUBRULE(this.noteStmt) },
+              { ALT: () => this.SUBRULE(this.memberAssignStmt) }
+            ]);
+          });
+          this.CONSUME(RCurly);
+          this.OPTION(() => this.CONSUME2(Newline4));
+        });
         this.classLine = this.RULE("classLine", () => {
           this.CONSUME(ClassKw);
           this.SUBRULE(this.classRef);
@@ -31418,6 +31797,44 @@ var init_parser5 = __esm({
             }
           ]);
         });
+        this.interfaceLine = this.RULE("interfaceLine", () => {
+          this.CONSUME(InterfaceKw);
+          this.SUBRULE(this.classRef);
+          this.OR([
+            {
+              ALT: () => {
+                this.CONSUME(LCurly);
+                this.MANY(() => {
+                  this.OR2([
+                    { ALT: () => this.CONSUME3(Newline4) },
+                    { ALT: () => this.SUBRULE(this.memberLineStmt) }
+                  ]);
+                });
+                this.CONSUME(RCurly);
+                this.OPTION(() => this.CONSUME(Newline4));
+              }
+            },
+            {
+              ALT: () => {
+                this.OPTION1(() => {
+                  this.CONSUME(SquareOpen2);
+                  this.CONSUME(QuotedString4);
+                  this.CONSUME(SquareClose2);
+                });
+                this.OPTION2(() => {
+                  this.CONSUME(LTlt);
+                  this.CONSUME2(Identifier3);
+                  this.CONSUME(GTgt);
+                });
+                this.OPTION3(() => {
+                  this.CONSUME(AsKw);
+                  this.CONSUME3(Identifier3);
+                });
+                this.OPTION4(() => this.CONSUME2(Newline4));
+              }
+            }
+          ]);
+        });
         this.memberAssignStmt = this.RULE("memberAssignStmt", () => {
           this.SUBRULE(this.classRef);
           this.CONSUME(Colon4);
@@ -31587,6 +32004,31 @@ function validateClass(text, _options = {}) {
             length: b.image?.length ?? 1
           });
         }
+        if (a.tokenType === NamespaceKw && b.tokenType === QuotedString4) {
+          errs.push({
+            line: b.startLine ?? 1,
+            column: b.startColumn ?? 1,
+            severity: "error",
+            code: "CL-NAMESPACE-NAME-QUOTED",
+            message: "Quoted namespace names are not supported by mermaid.js. Use an unquoted identifier.",
+            hint: 'Change: namespace "ProbeAgent Core" { ... } \u2192 namespace ProbeAgentCore { ... }',
+            length: b.image?.length ?? 1
+          });
+        }
+        if (a.tokenType === InterfaceKw) {
+          const prevToken = i > 0 ? tokList[i - 1] : null;
+          if (!prevToken || prevToken.tokenType !== LTlt) {
+            errs.push({
+              line: a.startLine ?? 1,
+              column: a.startColumn ?? 1,
+              severity: "error",
+              code: "CL-INTERFACE-KEYWORD-UNSUPPORTED",
+              message: 'The "interface" keyword is not supported by mermaid.js. Use the <<interface>> annotation instead.',
+              hint: 'Auto-fix converts to "class". Then manually add: <<interface>> ClassName after the class definition',
+              length: a.image?.length ?? 9
+            });
+          }
+        }
       }
       for (const tk of tokList) {
         if (tk.tokenType === InvalidRelArrow) {
@@ -31643,10 +32085,13 @@ function validateClass(text, _options = {}) {
       const has2 = (code, line) => (prev || []).some((e) => e.code === code && e.line === line && e.severity === "error");
       const lines = src.split(/\r?\n/);
       const classDeclOpen = [];
+      const namespaceDeclOpen = [];
       for (let i = 0; i < lines.length; i++) {
         const raw = lines[i] || "";
         if (/^\s*class\b.*\{\s*$/.test(raw))
           classDeclOpen.push(i + 1);
+        if (/^\s*namespace\b.*\{\s*$/.test(raw))
+          namespaceDeclOpen.push(i + 1);
       }
       if (classDeclOpen.length > 0) {
         const hasClose = lines.some((l) => /\}/.test(l));
@@ -31655,6 +32100,13 @@ function validateClass(text, _options = {}) {
           errors.push({ line: Math.max(1, lines.length), column: 1, severity: "error", code: "CL-BLOCK-MISSING-RBRACE", message: "Missing '}' to close class block.", hint: "Close the block: class Foo { ... }" });
         }
       }
+      if (namespaceDeclOpen.length > 0) {
+        const hasClose = lines.some((l) => /\}/.test(l));
+        if (!hasClose && !has2("CL-NAMESPACE-MISSING-RBRACE", Math.max(1, lines.length))) {
+          const last2 = namespaceDeclOpen[namespaceDeclOpen.length - 1];
+          errors.push({ line: Math.max(1, lines.length), column: 1, severity: "error", code: "CL-NAMESPACE-MISSING-RBRACE", message: "Missing '}' to close namespace block.", hint: 'Close the block: namespace "Name" { ... }' });
+        }
+      }
       return errors;
     }
   });
@@ -32309,6 +32761,10 @@ function computeFixes(text, errors, level = "safe") {
       edits.push(replaceRange(text, at(e), e.length ?? 2, "-->"));
       continue;
     }
+    if (is("FL-LINK-UNSUPPORTED-MARKER", e)) {
+      edits.push(replaceRange(text, at(e), e.length ?? 1, ""));
+      continue;
+    }
     if (is("FL-EDGE-LABEL-QUOTED", e)) {
       const lineText = lineTextAt(text, e.line);
       const col = Math.max(0, e.column - 1);
@@ -32381,6 +32837,29 @@ function computeFixes(text, errors, level = "safe") {
       }
       continue;
     }
+    if (is("CL-NAMESPACE-NAME-QUOTED", e)) {
+      const lineText = lineTextAt(text, e.line);
+      const nsIdx = lineText.indexOf("namespace");
+      const startSearch = nsIdx >= 0 ? nsIdx + 9 : 0;
+      const q1 = lineText.indexOf('"', startSearch);
+      if (q1 !== -1) {
+        const q2 = lineText.indexOf('"', q1 + 1);
+        if (q2 > q1) {
+          const namespaceName = lineText.slice(q1 + 1, q2);
+          const validIdentifier = namespaceName.replace(/[^A-Za-z0-9_]/g, "");
+          edits.push(replaceRange(text, { line: e.line, column: q1 + 1 }, q2 - q1 + 1, validIdentifier));
+        }
+      }
+      continue;
+    }
+    if (is("CL-INTERFACE-KEYWORD-UNSUPPORTED", e)) {
+      const lineText = lineTextAt(text, e.line);
+      const ifIdx = lineText.indexOf("interface");
+      if (ifIdx !== -1) {
+        edits.push(replaceRange(text, { line: e.line, column: ifIdx + 1 }, 9, "class"));
+      }
+      continue;
+    }
     if (is("FL-LABEL-ESCAPED-QUOTE", e)) {
       const lineText = lineTextAt(text, e.line);
       const caret0 = Math.max(0, e.column - 1);
@@ -32528,6 +33007,42 @@ function computeFixes(text, errors, level = "safe") {
       const msg = e.message || "";
       const lineText = lineTextAt(text, e.line);
       const caret0 = Math.max(0, e.column - 1);
+      {
+        const openIdx = lineText.indexOf("([");
+        if (openIdx !== -1) {
+          const badClose = lineText.indexOf("})", openIdx + 2);
+          if (badClose !== -1) {
+            edits.push({ start: { line: e.line, column: badClose + 1 }, end: { line: e.line, column: badClose + 2 }, newText: "]" });
+            patchedLines.add(e.line);
+            continue;
+          }
+        }
+      }
+      {
+        const openIdx = lineText.indexOf("[(");
+        if (openIdx !== -1) {
+          const closePair = lineText.indexOf("))", openIdx + 2);
+          if (closePair !== -1) {
+            edits.push({ start: { line: e.line, column: closePair + 2 }, end: { line: e.line, column: closePair + 3 }, newText: "]" });
+            patchedLines.add(e.line);
+            continue;
+          }
+        }
+      }
+      {
+        const openPair = lineText.indexOf("([");
+        if (openPair !== -1) {
+          const badClose = lineText.indexOf("})", openPair + 2);
+          if (badClose !== -1) {
+            edits.push({ start: { line: e.line, column: badClose + 1 }, end: { line: e.line, column: badClose + 2 }, newText: "]" });
+            continue;
+          }
+        }
+      }
+      if (/\(\[/.test(lineText) && lineText.indexOf("})", Math.max(0, caret0 - 1)) !== -1) {
+        edits.push(replaceRange(text, at(e), e.length ?? 1, "]"));
+        continue;
+      }
       if (msg.includes("opened '('") && msg.includes("closed with ']'")) {
         const openIdx = lineText.lastIndexOf("(", caret0);
         if (openIdx !== -1) {
@@ -32600,7 +33115,36 @@ function computeFixes(text, errors, level = "safe") {
         const caret0 = Math.max(0, e.column - 1);
         const msg = e.message || "";
         const bracketMatch = msg.match(/Unclosed '(.+?)'/);
-        const expectedOpener = bracketMatch ? bracketMatch[1] : null;
+        const expectedOpener = bracketMatch ? (bracketMatch[1] || "").trim() : null;
+        if (expectedOpener === "((") {
+          if (level === "all") {
+            const openIdx = lineText.lastIndexOf("((", caret0);
+            if (openIdx !== -1) {
+              const contentStart = openIdx + 2;
+              const picks = [];
+              const pushIdx2 = (i) => {
+                if (i >= 0)
+                  picks.push(i);
+              };
+              pushIdx2(lineText.indexOf("-", contentStart));
+              pushIdx2(lineText.indexOf("=", contentStart));
+              pushIdx2(lineText.indexOf(".", contentStart));
+              pushIdx2(lineText.indexOf("|", contentStart));
+              let insertIdx = picks.length ? Math.min(...picks) : lineText.length;
+              const before = lineText.slice(0, openIdx);
+              const m = before.match(/([A-Za-z0-9_]+)\s*$/);
+              const inferred = m ? m[1] : "";
+              if (inferred) {
+                edits.push({ start: { line: e.line, column: contentStart + 1 }, end: { line: e.line, column: insertIdx + 1 }, newText: inferred + "))" });
+                patchedLines.add(e.line);
+                continue;
+              }
+              patchedLines.add(e.line);
+              continue;
+            }
+          }
+          continue;
+        }
         const bracketMap = {
           "[": "]",
           "{": "}",
@@ -32686,6 +33230,14 @@ function computeFixes(text, errors, level = "safe") {
               patchedLines.add(e.line);
               continue;
             }
+            if (innerSeg.includes("(") || innerSeg.includes(")")) {
+              const replaced = innerSeg.replace(/\(/g, "&#40;").replace(/\)/g, "&#41;");
+              if (replaced !== innerSeg) {
+                edits.push({ start: { line: e.line, column: opened.idx + opened.len + 1 }, end: { line: e.line, column: closerIdx + 1 }, newText: replaced });
+                patchedLines.add(e.line);
+                continue;
+              }
+            }
           }
         }
         if (patchedLines.has(e.line)) {
@@ -32694,9 +33246,35 @@ function computeFixes(text, errors, level = "safe") {
         let closer = "]";
         if (opened)
           closer = opened.close;
-        const avail = lineText.slice(caret0);
-        const replaceLen = Math.min(closer.length, Math.max(1, avail.length));
-        edits.push({ start: { line: e.line, column: caret0 + 1 }, end: { line: e.line, column: caret0 + 1 + replaceLen }, newText: closer });
+        if (closer === "]") {
+          const openIdxSq = lineText.lastIndexOf("[", caret0);
+          if (openIdxSq !== -1) {
+            const picks = [];
+            const pushIdx2 = (i) => {
+              if (i >= 0)
+                picks.push(i);
+            };
+            pushIdx2(lineText.indexOf("-", openIdxSq + 1));
+            pushIdx2(lineText.indexOf("=", openIdxSq + 1));
+            pushIdx2(lineText.indexOf(".", openIdxSq + 1));
+            pushIdx2(lineText.indexOf("|", openIdxSq + 1));
+            let ins = picks.length ? Math.min(...picks) : lineText.length;
+            let tl = ins - 1;
+            while (tl >= 0 && /\s/.test(lineText[tl]))
+              tl--;
+            const startCol2 = tl + 1 + 1;
+            const endCol2 = ins + 1;
+            edits.push({ start: { line: e.line, column: startCol2 }, end: { line: e.line, column: endCol2 }, newText: closer });
+          } else {
+            const avail = lineText.slice(caret0);
+            const replaceLen = Math.min(closer.length, Math.max(1, avail.length));
+            edits.push({ start: { line: e.line, column: caret0 + 1 }, end: { line: e.line, column: caret0 + 1 + replaceLen }, newText: closer });
+          }
+        } else {
+          const avail = lineText.slice(caret0);
+          const replaceLen = Math.min(closer.length, Math.max(1, avail.length));
+          edits.push({ start: { line: e.line, column: caret0 + 1 }, end: { line: e.line, column: caret0 + 1 + replaceLen }, newText: closer });
+        }
       }
       continue;
     }
@@ -32828,9 +33406,11 @@ function computeFixes(text, errors, level = "safe") {
               if (core.length >= 2 && isSlashPair(left, right)) {
                 break;
               }
-              const newInner = '"' + inner + '"';
-              edits.push({ start: { line: e.line, column: contentStart + 1 }, end: { line: e.line, column: closeIdx + 1 }, newText: newInner });
-              patchedLines.add(e.line);
+              const replaced = inner.replace(/\(/g, "&#40;").replace(/\)/g, "&#41;");
+              if (replaced !== inner) {
+                edits.push({ start: { line: e.line, column: contentStart + 1 }, end: { line: e.line, column: closeIdx + 1 }, newText: replaced });
+                patchedLines.add(e.line);
+              }
               break;
             }
             searchStart = openIdx + 1;
@@ -32987,14 +33567,33 @@ function computeFixes(text, errors, level = "safe") {
       continue;
     }
     if (is("SE-BOX-EMPTY", e)) {
-      const lines = text.split(/\r?\n/);
-      const boxIdx = Math.max(0, e.line - 1);
-      const boxLine = lines[boxIdx] || "";
-      const labelMatch = /^\s*box\s+(.+)$/.exec(boxLine);
-      if (labelMatch) {
-        const indent = boxLine.match(/^\s*/)?.[0] || "";
-        const newLine = `${indent}rect rgb(240, 240, 255)`;
-        edits.push({ start: { line: e.line, column: 1 }, end: { line: e.line, column: boxLine.length + 1 }, newText: newLine });
+      if (level === "all") {
+        const lines = text.split(/\r?\n/);
+        const boxIdx = Math.max(0, e.line - 1);
+        const boxLine = lines[boxIdx] || "";
+        const openIndent = (boxLine.match(/^(\s*)/)?.[1] || "").length;
+        let endIdx = -1;
+        for (let i = boxIdx + 1; i < lines.length; i++) {
+          const raw = lines[i] || "";
+          const ind = (raw.match(/^(\s*)/)?.[1] || "").length;
+          if (/^\s*end\s*$/.test(raw) && ind <= openIndent) {
+            endIdx = i;
+            break;
+          }
+        }
+        let hasMsgWithAct = false;
+        if (endIdx !== -1) {
+          const body = lines.slice(boxIdx + 1, endIdx).map((s) => (s || "").trim());
+          hasMsgWithAct = body.some((s) => /->/.test(s) && /[+-]/.test(s));
+        }
+        if (!hasMsgWithAct) {
+          const labelMatch = /^\s*box\s+(.+)$/.exec(boxLine);
+          if (labelMatch) {
+            const indent = boxLine.match(/^\s*/)?.[0] || "";
+            const newLine = `${indent}rect rgb(240, 240, 255)`;
+            edits.push({ start: { line: e.line, column: 1 }, end: { line: e.line, column: boxLine.length + 1 }, newText: newLine });
+          }
+        }
       }
       continue;
     }
@@ -33270,6 +33869,41 @@ ${indent}${tail}` });
         }
       }
       edits.push(insertAt(text, { line: insIdx + 1, column: 1 }, `${openIndent}}
+`));
+      continue;
+    }
+    if (is("CL-NAMESPACE-MISSING-RBRACE", e)) {
+      const lines = text.split(/\r?\n/);
+      const curIdx = Math.max(0, e.line - 1);
+      const openerRe = /^(\s*)namespace\b.*\{\s*$/;
+      let openIdx = -1;
+      let openIndent = "";
+      for (let i = curIdx; i >= 0; i--) {
+        const m = openerRe.exec(lines[i] || "");
+        if (m) {
+          openIdx = i;
+          openIndent = m[1] || "";
+          break;
+        }
+      }
+      if (openIdx === -1) {
+        const indent = inferIndentFromLine(lines[curIdx] || "");
+        edits.push(insertAt(text, { line: curIdx + 1, column: 1 }, `${indent}}
+`));
+        continue;
+      }
+      let insIdx = lines.length;
+      for (let i = openIdx + 1; i < lines.length; i++) {
+        const raw = lines[i] || "";
+        if (raw.trim() === "")
+          continue;
+        const ind = inferIndentFromLine(raw);
+        if (ind.length <= openIndent.length) {
+          insIdx = i;
+          break;
+        }
+      }
+      edits.push(insertAt(text, { line: insIdx + 1, column: 1 }, `${openIndent}}
 `));
       continue;
     }
@@ -47313,6 +47947,8 @@ var init_ProbeAgent = __esm({
   "src/agent/ProbeAgent.js"() {
     "use strict";
     init_tokenCounter();
+    init_InMemoryStorageAdapter();
+    init_HookManager();
     init_tools2();
     init_common();
     init_probeTool();
@@ -47345,6 +47981,8 @@ var init_ProbeAgent = __esm({
        * @param {string} [options.mcpConfigPath] - Path to MCP configuration file
        * @param {Object} [options.mcpConfig] - MCP configuration object (overrides mcpConfigPath)
        * @param {Array} [options.mcpServers] - Deprecated, use mcpConfig instead
+       * @param {Object} [options.storageAdapter] - Custom storage adapter for history management
+       * @param {Object} [options.hooks] - Hook callbacks for events (e.g., {'tool:start': callback})
        */
       constructor(options = {}) {
         this.sessionId = options.sessionId || randomUUID4();
@@ -47358,6 +47996,13 @@ var init_ProbeAgent = __esm({
         this.maxResponseTokens = options.maxResponseTokens || parseInt(process.env.MAX_RESPONSE_TOKENS || "0", 10) || null;
         this.maxIterations = options.maxIterations || null;
         this.disableMermaidValidation = !!options.disableMermaidValidation;
+        this.storageAdapter = options.storageAdapter || new InMemoryStorageAdapter();
+        this.hooks = new HookManager();
+        if (options.hooks) {
+          for (const [hookName, callback] of Object.entries(options.hooks)) {
+            this.hooks.on(hookName, callback);
+          }
+        }
         this.enableBash = !!options.enableBash;
         this.bashConfig = options.bashConfig || {};
         if (options.allowedFolders && Array.isArray(options.allowedFolders)) {
@@ -47391,9 +48036,23 @@ var init_ProbeAgent = __esm({
       }
       /**
        * Initialize the agent asynchronously (must be called after constructor)
-       * This method initializes MCP and merges MCP tools into the tool list
+       * This method initializes MCP and merges MCP tools into the tool list, and loads history from storage
        */
       async initialize() {
+        try {
+          const history = await this.storageAdapter.loadHistory(this.sessionId);
+          this.history = history;
+          if (this.debug && history.length > 0) {
+            console.log(`[DEBUG] Loaded ${history.length} messages from storage for session ${this.sessionId}`);
+          }
+          await this.hooks.emit(HOOK_TYPES.STORAGE_LOAD, {
+            sessionId: this.sessionId,
+            messages: history
+          });
+        } catch (error) {
+          console.error(`[ERROR] Failed to load history from storage:`, error);
+          this.history = [];
+        }
         if (this.enableMcp && !this._mcpInitialized) {
           this._mcpInitialized = true;
           try {
@@ -47426,6 +48085,10 @@ var init_ProbeAgent = __esm({
             this.mcpBridge = null;
           }
         }
+        await this.hooks.emit(HOOK_TYPES.AGENT_INITIALIZED, {
+          sessionId: this.sessionId,
+          agent: this
+        });
       }
       /**
        * Initialize tools with configuration
@@ -47477,7 +48140,7 @@ var init_ProbeAgent = __esm({
         }
         const anthropicApiKey = process.env.ANTHROPIC_API_KEY;
         const openaiApiKey = process.env.OPENAI_API_KEY;
-        const googleApiKey = process.env.GOOGLE_API_KEY;
+        const googleApiKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY || process.env.GOOGLE_API_KEY;
         const awsAccessKeyId = process.env.AWS_ACCESS_KEY_ID;
         const awsSecretAccessKey = process.env.AWS_SECRET_ACCESS_KEY;
         const awsRegion = process.env.AWS_REGION;
@@ -47525,7 +48188,7 @@ var init_ProbeAgent = __esm({
         } else if (awsAccessKeyId && awsSecretAccessKey && awsRegion || awsApiKey) {
           this.initializeBedrockModel(awsAccessKeyId, awsSecretAccessKey, awsRegion, awsSessionToken, awsApiKey, awsBedrockBaseUrl, modelName);
         } else {
-          throw new Error("No API key provided. Please set ANTHROPIC_API_KEY, OPENAI_API_KEY, GOOGLE_API_KEY, AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.");
+          throw new Error("No API key provided. Please set ANTHROPIC_API_KEY, OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.");
         }
       }
       /**
@@ -48163,6 +48826,12 @@ You are working with a repository located at: ${searchDirectory}
           options = schemaOrOptions || {};
         }
         try {
+          const oldHistoryLength = this.history.length;
+          await this.hooks.emit(HOOK_TYPES.MESSAGE_USER, {
+            sessionId: this.sessionId,
+            message,
+            images
+          });
           const systemMessage = await this.getSystemMessage();
           let userMessage = { role: "user", content: message.trim() };
           if (images && images.length > 0) {
@@ -48248,6 +48917,9 @@ You are working with a repository located at: ${searchDirectory}
                 });
                 for await (const delta of result.textStream) {
                   assistantResponseContent += delta;
+                  if (options.onStream) {
+                    options.onStream(delta);
+                  }
                 }
                 const usage = await result.usage;
                 if (usage) {
@@ -48312,6 +48984,13 @@ You are working with a repository located at: ${searchDirectory}
                   const validation = attemptCompletionSchema.safeParse(params);
                   if (validation.success) {
                     finalResult = validation.data.result;
+                    if (options.onStream && finalResult) {
+                      const chunkSize = 50;
+                      for (let i = 0; i < finalResult.length; i += chunkSize) {
+                        const chunk = finalResult.slice(i, Math.min(i + chunkSize, finalResult.length));
+                        options.onStream(chunk);
+                      }
+                    }
                     if (this.debug) console.log(`[DEBUG] Task completed successfully with result: ${finalResult.substring(0, 100)}...`);
                   } else {
                     console.error(`[ERROR] Invalid attempt_completion parameters:`, validation.error);
@@ -48386,7 +49065,9 @@ ${toolResultContent}
                       timestamp: (/* @__PURE__ */ new Date()).toISOString(),
                       name: toolName,
                       args: toolParams,
-                      status: "started"
+                      status: "started",
+                      pauseStream: true
+                      // Signal to pause text streaming
                     });
                     const executeToolCall = async () => {
                       if (toolName === "delegate") {
@@ -48570,6 +49251,18 @@ IMPORTANT: When using <attempt_complete>, this must be the ONLY content in your
             }
           }
           this.tokenCounter.updateHistory(this.history);
+          try {
+            const messagesToSave = currentMessages.slice(oldHistoryLength);
+            for (const message2 of messagesToSave) {
+              await this.storageAdapter.saveMessage(this.sessionId, message2);
+              await this.hooks.emit(HOOK_TYPES.STORAGE_SAVE, {
+                sessionId: this.sessionId,
+                message: message2
+              });
+            }
+          } catch (error) {
+            console.error(`[ERROR] Failed to save messages to storage:`, error);
+          }
           const reachedMaxIterations = currentIteration >= maxIterations && !completionAttempted;
           if (options.schema && !options._schemaFormatted && !completionAttempted && !reachedMaxIterations) {
             if (this.debug) {
@@ -48934,10 +49627,18 @@ Convert your previous response content into actual JSON data that follows this s
       /**
        * Clear conversation history and reset counters
        */
-      clearHistory() {
+      async clearHistory() {
+        try {
+          await this.storageAdapter.clearHistory(this.sessionId);
+        } catch (error) {
+          console.error(`[ERROR] Failed to clear history in storage:`, error);
+        }
         this.history = [];
         this.tokenCounter.clear();
         clearToolExecutionData(this.sessionId);
+        await this.hooks.emit(HOOK_TYPES.STORAGE_CLEAR, {
+          sessionId: this.sessionId
+        });
         if (this.debug) {
           console.log(`[DEBUG] Cleared conversation history and reset counters for session ${this.sessionId}`);
         }