@probelabs/probe 0.6.0-rc125 → 0.6.0-rc126

package/cjs/index.cjs

@@ -24340,7 +24340,7 @@ var require_package = __commonJS({
     module2.exports = {
       name: "@aws-sdk/client-bedrock-runtime",
       description: "AWS SDK for JavaScript Bedrock Runtime Client for Node.js, Browser and React Native",
-      version: "3.906.0",
+      version: "3.907.0",
       scripts: {
         build: "concurrently 'yarn:build:cjs' 'yarn:build:es' 'yarn:build:types'",
         "build:cjs": "node ../../scripts/compilation/inline client-bedrock-runtime",
@@ -24359,21 +24359,21 @@ var require_package = __commonJS({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.906.0",
-        "@aws-sdk/credential-provider-node": "3.906.0",
+        "@aws-sdk/core": "3.907.0",
+        "@aws-sdk/credential-provider-node": "3.907.0",
         "@aws-sdk/eventstream-handler-node": "3.901.0",
         "@aws-sdk/middleware-eventstream": "3.901.0",
         "@aws-sdk/middleware-host-header": "3.901.0",
         "@aws-sdk/middleware-logger": "3.901.0",
         "@aws-sdk/middleware-recursion-detection": "3.901.0",
-        "@aws-sdk/middleware-user-agent": "3.906.0",
+        "@aws-sdk/middleware-user-agent": "3.907.0",
         "@aws-sdk/middleware-websocket": "3.901.0",
         "@aws-sdk/region-config-resolver": "3.901.0",
-        "@aws-sdk/token-providers": "3.906.0",
+        "@aws-sdk/token-providers": "3.907.0",
         "@aws-sdk/types": "3.901.0",
         "@aws-sdk/util-endpoints": "3.901.0",
-        "@aws-sdk/util-user-agent-browser": "3.901.0",
-        "@aws-sdk/util-user-agent-node": "3.906.0",
+        "@aws-sdk/util-user-agent-browser": "3.907.0",
+        "@aws-sdk/util-user-agent-node": "3.907.0",
         "@smithy/config-resolver": "^4.3.0",
         "@smithy/core": "^3.14.0",
         "@smithy/eventstream-serde-browser": "^4.2.0",
@@ -25121,7 +25121,7 @@ var init_package = __esm({
   "node_modules/@aws-sdk/nested-clients/package.json"() {
     package_default = {
       name: "@aws-sdk/nested-clients",
-      version: "3.906.0",
+      version: "3.907.0",
       description: "Nested clients for AWS SDK packages.",
       main: "./dist-cjs/index.js",
       module: "./dist-es/index.js",
@@ -25150,16 +25150,16 @@ var init_package = __esm({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.906.0",
+        "@aws-sdk/core": "3.907.0",
         "@aws-sdk/middleware-host-header": "3.901.0",
         "@aws-sdk/middleware-logger": "3.901.0",
         "@aws-sdk/middleware-recursion-detection": "3.901.0",
-        "@aws-sdk/middleware-user-agent": "3.906.0",
+        "@aws-sdk/middleware-user-agent": "3.907.0",
         "@aws-sdk/region-config-resolver": "3.901.0",
         "@aws-sdk/types": "3.901.0",
         "@aws-sdk/util-endpoints": "3.901.0",
-        "@aws-sdk/util-user-agent-browser": "3.901.0",
-        "@aws-sdk/util-user-agent-node": "3.906.0",
+        "@aws-sdk/util-user-agent-browser": "3.907.0",
+        "@aws-sdk/util-user-agent-node": "3.907.0",
         "@smithy/config-resolver": "^4.3.0",
         "@smithy/core": "^3.14.0",
         "@smithy/fetch-http-handler": "^5.3.0",
@@ -26596,7 +26596,7 @@ var require_package2 = __commonJS({
     module2.exports = {
       name: "@aws-sdk/client-sso",
       description: "AWS SDK for JavaScript Sso Client for Node.js, Browser and React Native",
-      version: "3.906.0",
+      version: "3.907.0",
       scripts: {
         build: "concurrently 'yarn:build:cjs' 'yarn:build:es' 'yarn:build:types'",
         "build:cjs": "node ../../scripts/compilation/inline client-sso",
@@ -26615,16 +26615,16 @@ var require_package2 = __commonJS({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.906.0",
+        "@aws-sdk/core": "3.907.0",
         "@aws-sdk/middleware-host-header": "3.901.0",
         "@aws-sdk/middleware-logger": "3.901.0",
         "@aws-sdk/middleware-recursion-detection": "3.901.0",
-        "@aws-sdk/middleware-user-agent": "3.906.0",
+        "@aws-sdk/middleware-user-agent": "3.907.0",
         "@aws-sdk/region-config-resolver": "3.901.0",
         "@aws-sdk/types": "3.901.0",
         "@aws-sdk/util-endpoints": "3.901.0",
-        "@aws-sdk/util-user-agent-browser": "3.901.0",
-        "@aws-sdk/util-user-agent-node": "3.906.0",
+        "@aws-sdk/util-user-agent-browser": "3.907.0",
+        "@aws-sdk/util-user-agent-node": "3.907.0",
         "@smithy/config-resolver": "^4.3.0",
         "@smithy/core": "^3.14.0",
         "@smithy/fetch-http-handler": "^5.3.0",
@@ -33872,7 +33872,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"];
   }
 });
 
@@ -34268,8 +34268,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.");
         }
@@ -34295,12 +34296,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;
         }
@@ -34311,15 +34312,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;
@@ -34331,6 +34337,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) {
@@ -34344,7 +34352,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;
@@ -34782,6 +34790,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 (data2) => {
+          this.off(hookName, wrappedCallback);
+          await callback(data2);
+        };
+        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, data2) {
+        const callbacks = this.hooks.get(hookName);
+        if (!callbacks || callbacks.size === 0) return;
+        const promises = Array.from(callbacks).map((callback) => {
+          try {
+            return Promise.resolve(callback(data2));
+          } catch (error2) {
+            return Promise.reject(error2);
+          }
+        });
+        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/agent/xmlParsingUtils.js
 function removeThinkingTags(xmlString) {
   let result = xmlString;
@@ -72856,6 +73084,8 @@ var init_ProbeAgent = __esm({
     import_promises2 = require("fs/promises");
     import_path9 = require("path");
     init_tokenCounter();
+    init_InMemoryStorageAdapter();
+    init_HookManager();
     init_tools2();
     init_common();
     init_probeTool();
@@ -72888,6 +73118,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 || (0, import_crypto5.randomUUID)();
@@ -72901,6 +73133,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)) {
@@ -72934,9 +73173,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 (error2) {
+          console.error(`[ERROR] Failed to load history from storage:`, error2);
+          this.history = [];
+        }
         if (this.enableMcp && !this._mcpInitialized) {
           this._mcpInitialized = true;
           try {
@@ -72969,6 +73222,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
@@ -73020,7 +73277,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;
@@ -73068,7 +73325,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.");
         }
       }
       /**
@@ -73706,6 +73963,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) {
@@ -73791,6 +74054,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) {
@@ -73855,6 +74121,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 i3 = 0; i3 < finalResult.length; i3 += chunkSize) {
+                        const chunk = finalResult.slice(i3, Math.min(i3 + 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);
@@ -73929,7 +74202,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") {
@@ -74113,6 +74388,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 (error2) {
+            console.error(`[ERROR] Failed to save messages to storage:`, error2);
+          }
           const reachedMaxIterations = currentIteration >= maxIterations && !completionAttempted;
           if (options.schema && !options._schemaFormatted && !completionAttempted && !reachedMaxIterations) {
             if (this.debug) {
@@ -74477,10 +74764,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 (error2) {
+          console.error(`[ERROR] Failed to clear history in storage:`, error2);
+        }
         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}`);
         }
@@ -74739,13 +75034,34 @@ var init_simpleTelemetry = __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 index_exports = {};
 __export(index_exports, {
   DEFAULT_SYSTEM_MESSAGE: () => DEFAULT_SYSTEM_MESSAGE,
+  HOOK_TYPES: () => HOOK_TYPES,
+  HookManager: () => HookManager,
+  InMemoryStorageAdapter: () => InMemoryStorageAdapter,
   ProbeAgent: () => ProbeAgent,
   SimpleAppTracer: () => SimpleAppTracer,
   SimpleTelemetry: () => SimpleTelemetry,
+  StorageAdapter: () => StorageAdapter,
   attemptCompletionSchema: () => attemptCompletionSchema,
   attemptCompletionToolDefinition: () => attemptCompletionToolDefinition,
   bashSchema: () => bashSchema,
@@ -74795,15 +75111,21 @@ var init_index = __esm({
     init_ProbeAgent();
     init_simpleTelemetry();
     init_probeTool();
+    init_storage();
+    init_hooks();
   }
 });
 init_index();
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   DEFAULT_SYSTEM_MESSAGE,
+  HOOK_TYPES,
+  HookManager,
+  InMemoryStorageAdapter,
   ProbeAgent,
   SimpleAppTracer,
   SimpleTelemetry,
+  StorageAdapter,
   attemptCompletionSchema,
   attemptCompletionToolDefinition,
   bashSchema,