@probelabs/probe 0.6.0-rc125 → 0.6.0-rc126

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();
   }
 });
 
@@ -47313,6 +47560,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 +47594,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 +47609,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 +47649,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 +47698,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 +47753,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 +47801,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 +48439,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 +48530,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 +48597,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 +48678,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 +48864,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 +49240,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}`);
         }

package/build/agent/storage/InMemoryStorageAdapter.js

@@ -0,0 +1,49 @@
+import { StorageAdapter } from './StorageAdapter.js';
+
+/**
+ * Default in-memory storage adapter
+ * This is the default behavior - stores history in a Map in memory
+ */
+export class InMemoryStorageAdapter extends StorageAdapter {
+  constructor() {
+    super();
+    this.sessions = new Map(); // sessionId -> {messages: [], metadata: {}}
+  }
+
+  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: new Date().toISOString(),
+          lastActivity: new Date().toISOString()
+        }
+      });
+    }
+
+    const session = this.sessions.get(sessionId);
+    session.messages.push(message);
+    session.metadata.lastActivity = 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 = new Date().toISOString();
+    }
+  }
+}

package/build/agent/storage/StorageAdapter.js

@@ -0,0 +1,51 @@
+/**
+ * Base class for storage adapters
+ * Implement this interface to provide custom storage backends for ProbeAgent history
+ */
+export class StorageAdapter {
+  /**
+   * 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) {
+    // Optional - implement if you want to track session activity
+  }
+}

package/build/agent/storage/index.js

@@ -0,0 +1,2 @@
+export { StorageAdapter } from './StorageAdapter.js';
+export { InMemoryStorageAdapter } from './InMemoryStorageAdapter.js';

package/build/index.js

@@ -36,6 +36,8 @@ import { bashTool } from './tools/bash.js';
 import { ProbeAgent } from './agent/ProbeAgent.js';
 import { SimpleTelemetry, SimpleAppTracer, initializeSimpleTelemetryFromOptions } from './agent/simpleTelemetry.js';
 import { listFilesToolInstance, searchFilesToolInstance } from './agent/probeTool.js';
+import { StorageAdapter, InMemoryStorageAdapter } from './agent/storage/index.js';
+import { HookManager, HOOK_TYPES } from './agent/hooks/index.js';
 
 export {
 	search,
@@ -50,6 +52,12 @@ export {
 	DEFAULT_SYSTEM_MESSAGE,
 	// Export AI Agent (NEW!)
 	ProbeAgent,
+	// Export storage adapters
+	StorageAdapter,
+	InMemoryStorageAdapter,
+	// Export hooks
+	HookManager,
+	HOOK_TYPES,
 	// Export telemetry classes
 	SimpleTelemetry,
 	SimpleAppTracer,