@iqai/adk 0.0.6 → 0.0.8

package/dist/index.js

@@ -1,4 +1,4 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class; var _class2; var _class3; var _class4; var _class5; var _class6; var _class7; var _class8; var _class9; var _class10; var _class11; var _class12; var _class13; var _class14; var _class15;var __defProp = Object.defineProperty;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class; var _class2; var _class3; var _class4; var _class5; var _class6; var _class7; var _class8; var _class9; var _class10; var _class11; var _class12; var _class13; var _class14; var _class15; var _class16; var _class17; var _class18; var _class19; var _class20; var _class21; var _class22; var _class23; var _class24; var _class25; var _class26; var _class27; var _class28; var _class29; var _class30;var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
@@ -19,28 +19,48 @@ var __copyProps = (to, from, except, desc) => {
 };
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
-// src/helpers/debug.ts
-var isDebugEnabled, debugLog;
-var init_debug = __esm({
-  "src/helpers/debug.ts"() {
-    isDebugEnabled = () => {
-      return process.env.NODE_ENV === "development" || process.env.DEBUG === "true";
-    };
-    debugLog = (message, ...args) => {
-      const time = (/* @__PURE__ */ new Date()).toLocaleTimeString();
-      if (isDebugEnabled()) {
-        console.log(`[DEBUG] ${time}: ${message}`, ...args);
+// src/helpers/logger.ts
+function isDebugEnabled() {
+  return process.env.NODE_ENV === "development" || process.env.DEBUG === "true";
+}
+var Logger;
+var init_logger = __esm({
+  "src/helpers/logger.ts"() {
+    Logger = (_class = class {
+      
+      __init2() {this.isDebugEnabled = isDebugEnabled()}
+      constructor({ name }) {;_class.prototype.__init2.call(this);
+        this.name = name;
+      }
+      debug(message, ...args) {
+        const time = (/* @__PURE__ */ new Date()).toISOString();
+        if (this.isDebugEnabled) {
+          console.log(`[${time}] \u{1F41B} [DEBUG] \u2728 [${this.name}] ${message}`, ...args);
+        }
       }
-    };
+      info(message, ...args) {
+        const time = (/* @__PURE__ */ new Date()).toISOString();
+        console.info(`[${time}] \u2139\uFE0F [INFO] \u2728 [${this.name}] ${message}`, ...args);
+      }
+      warn(message, ...args) {
+        const time = (/* @__PURE__ */ new Date()).toISOString();
+        console.warn(`[${time}] \u{1F6A7} [WARN] \u2728 [${this.name}] ${message}`, ...args);
+      }
+      error(message, ...args) {
+        const time = (/* @__PURE__ */ new Date()).toISOString();
+        console.error(`[${time}] \u274C [ERROR] \u2728 [${this.name}] ${message}`, ...args);
+      }
+    }, _class);
   }
 });
 
 // src/tools/base/base-tool.ts
-var BaseTool;
+var logger2, BaseTool;
 var init_base_tool = __esm({
   "src/tools/base/base-tool.ts"() {
-    init_debug();
-    BaseTool = exports.BaseTool = (_class = class {
+    init_logger();
+    logger2 = new Logger({ name: "BaseTool" });
+    BaseTool = exports.BaseTool = (_class2 = class {
       /**
        * Name of the tool
        */
@@ -64,15 +84,15 @@ var init_base_tool = __esm({
       /**
        * Base delay for retry in ms (will be used with exponential backoff)
        */
-      __init2() {this.baseRetryDelay = 1e3}
+      __init3() {this.baseRetryDelay = 1e3}
       /**
        * Maximum delay for retry in ms
        */
-      __init3() {this.maxRetryDelay = 1e4}
+      __init4() {this.maxRetryDelay = 1e4}
       /**
        * Constructor for BaseTool
        */
-      constructor(config) {;_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);
+      constructor(config) {;_class2.prototype.__init3.call(this);_class2.prototype.__init4.call(this);
         this.name = config.name;
         this.description = config.description;
         this.isLongRunning = config.isLongRunning || false;
@@ -129,8 +149,8 @@ var init_base_tool = __esm({
         while (attempts <= (this.shouldRetryOnFailure ? this.maxRetryAttempts : 0)) {
           try {
             if (attempts > 0) {
-              debugLog(
-                `[BaseTool] Retrying tool ${this.name} (attempt ${attempts} of ${this.maxRetryAttempts})...`
+              logger2.debug(
+                `Retrying tool ${this.name} (attempt ${attempts} of ${this.maxRetryAttempts})...`
               );
               const delay = Math.min(
                 this.baseRetryDelay * 2 ** (attempts - 1) + Math.random() * 1e3,
@@ -152,7 +172,7 @@ var init_base_tool = __esm({
           tool: this.name
         };
       }
-    }, _class);
+    }, _class2);
   }
 });
 
@@ -278,9 +298,9 @@ var init_function_tool = __esm({
   "src/tools/function/function-tool.ts"() {
     init_base_tool();
     init_function_utils();
-    FunctionTool = exports.FunctionTool = (_class2 = class extends BaseTool {
+    FunctionTool = exports.FunctionTool = (_class3 = class extends BaseTool {
       
-      __init4() {this.mandatoryArgs = []}
+      __init5() {this.mandatoryArgs = []}
       /**
        * Creates a new FunctionTool wrapping the provided function.
        *
@@ -296,7 +316,7 @@ var init_function_tool = __esm({
           isLongRunning: _optionalChain([options, 'optionalAccess', _8 => _8.isLongRunning]) || false,
           shouldRetryOnFailure: _optionalChain([options, 'optionalAccess', _9 => _9.shouldRetryOnFailure]) || false,
           maxRetryAttempts: _optionalChain([options, 'optionalAccess', _10 => _10.maxRetryAttempts]) || 3
-        });_class2.prototype.__init4.call(this);;
+        });_class3.prototype.__init5.call(this);;
         this.func = func;
         this.mandatoryArgs = this.getMandatoryArgs(func);
       }
@@ -372,7 +392,7 @@ You could retry calling this tool, but it is IMPORTANT for you to provide all th
       getMissingMandatoryArgs(args) {
         return this.mandatoryArgs.filter((arg) => !(arg in args));
       }
-    }, _class2);
+    }, _class3);
   }
 });
 
@@ -475,11 +495,12 @@ var BaseAgent = class {
 };
 
 // src/agents/llm-agent.ts
-init_debug();
+init_logger();
 
 // src/models/llm-registry.ts
-init_debug();
-var LLMRegistry = (_class3 = class _LLMRegistry {
+init_logger();
+var logger = new Logger({ name: "LLMRegistry" });
+var LLMRegistry = (_class4 = class _LLMRegistry {
   /**
    * Map of model name regex to LLM class
    */
@@ -535,12 +556,12 @@ var LLMRegistry = (_class3 = class _LLMRegistry {
    * Logs all registered models for debugging
    */
   static logRegisteredModels() {
-    debugLog(
+    logger.debug(
       "Registered LLM models:",
       [..._LLMRegistry.llmRegistry.entries()].map(([regex]) => regex.toString())
     );
   }
-}, _class3.__initStatic(), _class3);
+}, _class4.__initStatic(), _class4);
 
 // src/models/llm-request.ts
 var LLMRequest = class {
@@ -559,7 +580,7 @@ var LLMRequest = class {
 };
 
 // src/tools/tool-context.ts
-var ToolContext = (_class4 = class {
+var ToolContext = (_class5 = class {
   /**
    * The parent invocation context
    */
@@ -575,11 +596,11 @@ var ToolContext = (_class4 = class {
   /**
    * Tool name
    */
-  __init5() {this.toolName = ""}
+  __init6() {this.toolName = ""}
   /**
    * Tool ID
    */
-  __init6() {this.toolId = ""}
+  __init7() {this.toolId = ""}
   /**
    * Variables stored in the context
    */
@@ -587,7 +608,7 @@ var ToolContext = (_class4 = class {
   /**
    * Constructor for ToolContext
    */
-  constructor(options) {;_class4.prototype.__init5.call(this);_class4.prototype.__init6.call(this);
+  constructor(options) {;_class5.prototype.__init6.call(this);_class5.prototype.__init7.call(this);
     this.invocationContext = options.invocationContext;
     this.auth = options.auth;
     this.parameters = options.parameters || {};
@@ -653,7 +674,7 @@ var ToolContext = (_class4 = class {
   async searchMemory(query, options) {
     return this.invocationContext.searchMemory(query, options);
   }
-}, _class4);
+}, _class5);
 
 // src/agents/run-config.ts
 var StreamingMode = /* @__PURE__ */ ((StreamingMode2) => {
@@ -705,7 +726,7 @@ var RunConfig = class {
 };
 
 // src/agents/invocation-context.ts
-var InvocationContext = (_class5 = class _InvocationContext {
+var InvocationContext = (_class6 = class _InvocationContext {
   /**
    * Unique session ID for the current conversation
    */
@@ -745,11 +766,11 @@ var InvocationContext = (_class5 = class _InvocationContext {
   /**
    * In-memory storage for node execution results
    */
-  __init7() {this.memory = /* @__PURE__ */ new Map()}
+  __init8() {this.memory = /* @__PURE__ */ new Map()}
   /**
    * Constructor for InvocationContext
    */
-  constructor(options = {}) {;_class5.prototype.__init7.call(this);
+  constructor(options = {}) {;_class6.prototype.__init8.call(this);
     this.sessionId = options.sessionId || this.generateSessionId();
     this.messages = options.messages || [];
     this.config = options.config || new RunConfig();
@@ -852,10 +873,10 @@ var InvocationContext = (_class5 = class _InvocationContext {
     };
     return await this.memoryService.searchMemory(query, searchOptions);
   }
-}, _class5);
+}, _class6);
 
 // src/agents/llm-agent.ts
-var Agent = class extends BaseAgent {
+var Agent = (_class7 = class extends BaseAgent {
   /**
    * The LLM model to use
    */
@@ -904,6 +925,7 @@ var Agent = class extends BaseAgent {
    * The minimum relevance score for memory augmentation (0-1)
    */
   
+  __init9() {this.logger = new Logger({ name: "LlmAgent" })}
   /**
    * Constructor for Agent
    */
@@ -911,7 +933,7 @@ var Agent = class extends BaseAgent {
     super({
       name: config.name,
       description: config.description
-    });
+    });_class7.prototype.__init9.call(this);;
     this.model = config.model;
     this.instructions = config.instructions;
     this.tools = config.tools || [];
@@ -936,7 +958,7 @@ var Agent = class extends BaseAgent {
    */
   async executeTool(toolCall, context) {
     const { name, arguments: argsString } = toolCall.function;
-    debugLog(`Executing tool: ${name}`);
+    this.logger.debug(`Executing tool: ${name}`);
     const tool = this.findTool(name);
     if (!tool) {
       console.warn(`Tool '${name}' not found`);
@@ -954,7 +976,7 @@ var Agent = class extends BaseAgent {
       toolContext.toolName = name;
       toolContext.toolId = toolCall.id;
       const result = await tool.runAsync(args, toolContext);
-      debugLog(`Tool ${name} execution complete`);
+      this.logger.debug(`Tool ${name} execution complete`);
       return {
         name,
         result: typeof result === "string" ? result : JSON.stringify(result)
@@ -1123,7 +1145,7 @@ ${relevantInfo.join("\n\n")}`
       let stepCount = 0;
       while (stepCount < this.maxToolExecutionSteps) {
         stepCount++;
-        debugLog(`Step ${stepCount}: Thinking...`);
+        this.logger.debug(`Step ${stepCount}: Thinking...`);
         const llmRequest = new LLMRequest({
           messages: context.messages,
           config: {
@@ -1140,7 +1162,9 @@ ${relevantInfo.join("\n\n")}`
           throw new Error("No response from LLM");
         }
         if (currentResponse.tool_calls && currentResponse.tool_calls.length > 0) {
-          debugLog(`Tool calls: ${JSON.stringify(currentResponse.tool_calls)}`);
+          this.logger.debug(
+            `Tool calls: ${JSON.stringify(currentResponse.tool_calls)}`
+          );
           context.addMessage({
             role: "assistant",
             content: currentResponse.content || "",
@@ -1159,7 +1183,7 @@ ${relevantInfo.join("\n\n")}`
             });
           }
         } else {
-          debugLog("[Agent] No tool calls, finishing...");
+          this.logger.debug("No tool calls, finishing...");
           context.addMessage({
             role: "assistant",
             content: currentResponse.content || ""
@@ -1197,7 +1221,7 @@ ${relevantInfo.join("\n\n")}`
     let stepCount = 0;
     let hadToolCalls = false;
     while (stepCount < this.maxToolExecutionSteps) {
-      debugLog(`[Agent] Step ${stepCount}: Thinking...`);
+      this.logger.debug(`Step ${stepCount}: Thinking...`);
       const toolDeclarations = this.tools.map((tool) => tool.getDeclaration()).filter((declaration) => declaration !== null);
       const request = {
         messages: context.messages,
@@ -1226,10 +1250,10 @@ ${relevantInfo.join("\n\n")}`
         function_call: finalResponse.function_call
       });
       if (!hadToolCalls) {
-        debugLog("[Agent] No tool calls, finishing...");
+        this.logger.debug("No tool calls, finishing...");
         break;
       }
-      debugLog(`[Agent] Step ${stepCount + 1}: Executing tools...`);
+      this.logger.debug(`Step ${stepCount + 1}: Executing tools...`);
       stepCount++;
       if (finalResponse.function_call) {
         const toolCall = {
@@ -1246,8 +1270,8 @@ ${relevantInfo.join("\n\n")}`
           content: JSON.stringify(result.result)
         });
       } else if (finalResponse.tool_calls && finalResponse.tool_calls.length > 0) {
-        debugLog(
-          `[Agent] Step ${stepCount + 1}: Executing ${finalResponse.tool_calls.length} tool(s)...`
+        this.logger.debug(
+          `Step ${stepCount + 1}: Executing ${finalResponse.tool_calls.length} tool(s)...`
         );
         context.messages.pop();
         context.addMessage({
@@ -1270,11 +1294,12 @@ ${relevantInfo.join("\n\n")}`
     }
     await this.saveToMemory(context);
   }
-};
+}, _class7);
 
 // src/agents/sequential-agent.ts
-init_debug();
-var SequentialAgent = class extends BaseAgent {
+init_logger();
+var SequentialAgent = (_class8 = class extends BaseAgent {
+  __init10() {this.logger = new Logger({ name: "SequentialAgent" })}
   /**
    * Constructor for SequentialAgent
    */
@@ -1282,7 +1307,7 @@ var SequentialAgent = class extends BaseAgent {
     super({
       name: config.name,
       description: config.description
-    });
+    });_class8.prototype.__init10.call(this);;
     if (config.agents && config.agents.length > 0) {
       for (const agent of config.agents) {
         this.addSubAgent(agent);
@@ -1294,8 +1319,8 @@ var SequentialAgent = class extends BaseAgent {
    * Executes sub-agents sequentially, passing output from one to the next
    */
   async run(options) {
-    debugLog(
-      `[SequentialAgent] Running ${this.subAgents.length} sub-agents in sequence`
+    this.logger.debug(
+      `Running ${this.subAgents.length} sub-agents in sequence`
     );
     if (this.subAgents.length === 0) {
       return {
@@ -1312,8 +1337,8 @@ var SequentialAgent = class extends BaseAgent {
     let finalResponse = null;
     for (let i = 0; i < this.subAgents.length; i++) {
       const agent = this.subAgents[i];
-      debugLog(
-        `[SequentialAgent] Running sub-agent ${i + 1}/${this.subAgents.length}: ${agent.name}`
+      this.logger.debug(
+        `Running sub-agent ${i + 1}/${this.subAgents.length}: ${agent.name}`
       );
       try {
         const response = await agent.run({
@@ -1329,10 +1354,7 @@ var SequentialAgent = class extends BaseAgent {
           });
         }
       } catch (error) {
-        console.error(
-          `[SequentialAgent] Error in sub-agent ${agent.name}:`,
-          error
-        );
+        console.error(`Error in sub-agent ${agent.name}:`, error);
         return {
           content: `Error in sub-agent ${agent.name}: ${error instanceof Error ? error.message : String(error)}`,
           role: "assistant",
@@ -1370,8 +1392,8 @@ var SequentialAgent = class extends BaseAgent {
    * Streams responses from each sub-agent in sequence
    */
   async *runStreaming(options) {
-    debugLog(
-      `[SequentialAgent] Streaming ${this.subAgents.length} sub-agents in sequence`
+    this.logger.debug(
+      `Streaming ${this.subAgents.length} sub-agents in sequence`
     );
     if (this.subAgents.length === 0) {
       yield {
@@ -1388,8 +1410,8 @@ var SequentialAgent = class extends BaseAgent {
     const currentMessages = [...options.messages];
     for (let i = 0; i < this.subAgents.length; i++) {
       const agent = this.subAgents[i];
-      debugLog(
-        `[SequentialAgent] Streaming sub-agent ${i + 1}/${this.subAgents.length}: ${agent.name}`
+      this.logger.debug(
+        `Streaming sub-agent ${i + 1}/${this.subAgents.length}: ${agent.name}`
       );
       try {
         const streamGenerator = agent.runStreaming({
@@ -1422,10 +1444,7 @@ var SequentialAgent = class extends BaseAgent {
           });
         }
       } catch (error) {
-        console.error(
-          `[SequentialAgent] Error in streaming sub-agent ${agent.name}:`,
-          error
-        );
+        console.error(`Error in streaming sub-agent ${agent.name}:`, error);
         yield {
           content: `Error in sub-agent ${agent.name}: ${error instanceof Error ? error.message : String(error)}`,
           role: "assistant",
@@ -1440,11 +1459,12 @@ var SequentialAgent = class extends BaseAgent {
       }
     }
   }
-};
+}, _class8);
 
 // src/agents/parallel-agent.ts
-init_debug();
-var ParallelAgent = class extends BaseAgent {
+init_logger();
+var ParallelAgent = (_class9 = class extends BaseAgent {
+  __init11() {this.logger = new Logger({ name: "ParallelAgent" })}
   /**
    * Constructor for ParallelAgent
    */
@@ -1452,7 +1472,7 @@ var ParallelAgent = class extends BaseAgent {
     super({
       name: config.name,
       description: config.description
-    });
+    });_class9.prototype.__init11.call(this);;
     if (config.agents && config.agents.length > 0) {
       for (const agent of config.agents) {
         this.addSubAgent(agent);
@@ -1464,8 +1484,8 @@ var ParallelAgent = class extends BaseAgent {
    * Executes all sub-agents in parallel
    */
   async run(options) {
-    debugLog(
-      `[ParallelAgent] Running ${this.subAgents.length} sub-agents in parallel`
+    this.logger.debug(
+      `Running ${this.subAgents.length} sub-agents in parallel`
     );
     if (this.subAgents.length === 0) {
       return {
@@ -1478,10 +1498,7 @@ var ParallelAgent = class extends BaseAgent {
         messages: options.messages,
         config: options.config
       }).catch((error) => {
-        console.error(
-          `[ParallelAgent] Error in sub-agent ${agent.name}:`,
-          error
-        );
+        console.error(`Error in sub-agent ${agent.name}:`, error);
         return {
           content: `Error in sub-agent ${agent.name}: ${error instanceof Error ? error.message : String(error)}`,
           role: "assistant"
@@ -1509,8 +1526,8 @@ ${result.content || "No content"}
    * Collects streaming responses from all sub-agents
    */
   async *runStreaming(options) {
-    debugLog(
-      `[ParallelAgent] Streaming ${this.subAgents.length} sub-agents in parallel`
+    this.logger.debug(
+      `Streaming ${this.subAgents.length} sub-agents in parallel`
     );
     if (this.subAgents.length === 0) {
       yield {
@@ -1524,10 +1541,7 @@ ${result.content || "No content"}
         messages: options.messages,
         config: options.config
       }).catch((error) => {
-        console.error(
-          `[ParallelAgent] Error in sub-agent ${agent.name}:`,
-          error
-        );
+        console.error(`Error in sub-agent ${agent.name}:`, error);
         return {
           content: `Error in sub-agent ${agent.name}: ${error instanceof Error ? error.message : String(error)}`,
           role: "assistant"
@@ -1574,11 +1588,11 @@ ${response.content || "No content"}
       };
     }
   }
-};
+}, _class9);
 
 // src/agents/loop-agent.ts
-init_debug();
-var LoopAgent = class extends BaseAgent {
+init_logger();
+var LoopAgent = (_class10 = class extends BaseAgent {
   /**
    * Maximum number of iterations to prevent infinite loops
    */
@@ -1591,6 +1605,7 @@ var LoopAgent = class extends BaseAgent {
    * Custom condition check function
    */
   
+  __init12() {this.logger = new Logger({ name: "LoopAgent" })}
   /**
    * Constructor for LoopAgent
    */
@@ -1598,7 +1613,7 @@ var LoopAgent = class extends BaseAgent {
     super({
       name: config.name,
       description: config.description
-    });
+    });_class10.prototype.__init12.call(this);;
     this.maxIterations = config.maxIterations || 10;
     this.conditionAgent = config.conditionAgent;
     this.conditionCheck = config.conditionCheck;
@@ -1618,19 +1633,19 @@ var LoopAgent = class extends BaseAgent {
    */
   async shouldContinue(response, iterationCount, messages, config) {
     if (iterationCount >= this.maxIterations) {
-      debugLog(
-        `[LoopAgent] Maximum iterations (${this.maxIterations}) reached. Stopping loop.`
+      this.logger.debug(
+        `Maximum iterations (${this.maxIterations}) reached. Stopping loop.`
       );
       return false;
     }
     if (this.conditionCheck) {
       const shouldContinue = await this.conditionCheck(response);
-      debugLog(`[LoopAgent] Custom condition check result: ${shouldContinue}`);
+      this.logger.debug(`Custom condition check result: ${shouldContinue}`);
       return shouldContinue;
     }
     if (this.conditionAgent) {
-      debugLog(
-        `[LoopAgent] Using condition agent ${this.conditionAgent.name} to check loop condition`
+      this.logger.debug(
+        `Using condition agent ${this.conditionAgent.name} to check loop condition`
       );
       const conditionMessages = [
         ...messages,
@@ -1650,12 +1665,12 @@ var LoopAgent = class extends BaseAgent {
         });
         const content = _optionalChain([conditionResponse, 'access', _21 => _21.content, 'optionalAccess', _22 => _22.toLowerCase, 'call', _23 => _23()]) || "";
         const shouldContinue = content.includes("yes") && !content.includes("no");
-        debugLog(
-          `[LoopAgent] Condition agent result: ${shouldContinue ? "Continue loop" : "Stop loop"}`
+        this.logger.debug(
+          `Condition agent result: ${shouldContinue ? "Continue loop" : "Stop loop"}`
         );
         return shouldContinue;
       } catch (error) {
-        console.error("[LoopAgent] Error in condition agent:", error);
+        console.error("Error in condition agent:", error);
         return false;
       }
     }
@@ -1666,8 +1681,8 @@ var LoopAgent = class extends BaseAgent {
    * Executes the sub-agent in a loop until the condition is met
    */
   async run(options) {
-    debugLog(
-      `[LoopAgent] Starting loop with max ${this.maxIterations} iterations`
+    this.logger.debug(
+      `Starting loop with max ${this.maxIterations} iterations`
     );
     if (this.subAgents.length === 0) {
       return {
@@ -1682,8 +1697,8 @@ var LoopAgent = class extends BaseAgent {
     let shouldContinueLoop = true;
     while (shouldContinueLoop && iterationCount < this.maxIterations) {
       iterationCount++;
-      debugLog(
-        `[LoopAgent] Running iteration ${iterationCount}/${this.maxIterations}`
+      this.logger.debug(
+        `Running iteration ${iterationCount}/${this.maxIterations}`
       );
       try {
         const response = await subAgent.run({
@@ -1708,10 +1723,7 @@ var LoopAgent = class extends BaseAgent {
           });
         }
       } catch (error) {
-        console.error(
-          `[LoopAgent] Error in loop iteration ${iterationCount}:`,
-          error
-        );
+        console.error(`Error in loop iteration ${iterationCount}:`, error);
         break;
       }
     }
@@ -1732,8 +1744,8 @@ ${lastResponse.content || ""}`,
    * Runs the agent with streaming support
    */
   async *runStreaming(options) {
-    debugLog(
-      `[LoopAgent] Starting loop with max ${this.maxIterations} iterations (streaming)`
+    this.logger.debug(
+      `Starting loop with max ${this.maxIterations} iterations (streaming)`
     );
     if (this.subAgents.length === 0) {
       yield {
@@ -1753,8 +1765,8 @@ ${lastResponse.content || ""}`,
     };
     while (shouldContinueLoop && iterationCount < this.maxIterations) {
       iterationCount++;
-      debugLog(
-        `[LoopAgent] Running iteration ${iterationCount}/${this.maxIterations} (streaming)`
+      this.logger.debug(
+        `Running iteration ${iterationCount}/${this.maxIterations} (streaming)`
       );
       yield {
         content: `Running iteration ${iterationCount}/${this.maxIterations}...`,
@@ -1779,8 +1791,8 @@ ${lastResponse.content || ""}`,
           }
         }
         if (!lastChunk) {
-          debugLog(
-            `[LoopAgent] No complete chunk received from iteration ${iterationCount}`
+          this.logger.debug(
+            `No complete chunk received from iteration ${iterationCount}`
           );
           shouldContinueLoop = false;
           continue;
@@ -1807,8 +1819,8 @@ ${lastResponse.content || ""}`,
           };
         }
       } catch (error) {
-        debugLog(
-          `[LoopAgent] Error in loop iteration ${iterationCount}: ${error instanceof Error ? error.message : String(error)}`
+        this.logger.debug(
+          `Error in loop iteration ${iterationCount}: ${error instanceof Error ? error.message : String(error)}`
         );
         yield {
           content: `Error in loop iteration ${iterationCount}: ${error instanceof Error ? error.message : String(error)}`,
@@ -1822,11 +1834,11 @@ ${lastResponse.content || ""}`,
       role: "assistant"
     };
   }
-};
+}, _class10);
 
 // src/agents/lang-graph-agent.ts
-init_debug();
-var LangGraphAgent = (_class6 = class extends BaseAgent {
+init_logger();
+var LangGraphAgent = (_class11 = class extends BaseAgent {
   /**
    * Graph nodes (agents and their connections)
    */
@@ -1842,7 +1854,8 @@ var LangGraphAgent = (_class6 = class extends BaseAgent {
   /**
    * Results from node executions
    */
-  __init8() {this.results = []}
+  __init13() {this.results = []}
+  __init14() {this.logger = new Logger({ name: "LangGraphAgent" })}
   /**
    * Constructor for LangGraphAgent
    */
@@ -1850,7 +1863,7 @@ var LangGraphAgent = (_class6 = class extends BaseAgent {
     super({
       name: config.name,
       description: config.description
-    });_class6.prototype.__init8.call(this);;
+    });_class11.prototype.__init13.call(this);_class11.prototype.__init14.call(this);;
     this.nodes = /* @__PURE__ */ new Map();
     for (const node of config.nodes) {
       if (this.nodes.has(node.name)) {
@@ -1917,15 +1930,13 @@ var LangGraphAgent = (_class6 = class extends BaseAgent {
     for (const targetName of currentNode.targets) {
       const targetNode = this.nodes.get(targetName);
       if (!targetNode) {
-        console.error(`[LangGraphAgent] Target node "${targetName}" not found`);
+        console.error(`Target node "${targetName}" not found`);
         continue;
       }
       if (targetNode.condition) {
         const shouldExecute = await targetNode.condition(result, context);
         if (!shouldExecute) {
-          debugLog(
-            `[LangGraphAgent] Skipping node "${targetName}" due to condition`
-          );
+          this.logger.debug(`Skipping node "${targetName}" due to condition`);
           continue;
         }
       }
@@ -1950,9 +1961,7 @@ var LangGraphAgent = (_class6 = class extends BaseAgent {
     };
     const shouldExecute = await node.condition(mockResponse, mockContext);
     if (!shouldExecute) {
-      debugLog(
-        `[LangGraphAgent] Skipping node "${targetName}" due to condition`
-      );
+      this.logger.debug(`Skipping node "${targetName}" due to condition`);
     }
     return { shouldExecute };
   }
@@ -1965,8 +1974,8 @@ var LangGraphAgent = (_class6 = class extends BaseAgent {
       messages: options.messages,
       config: options.config
     });
-    debugLog(
-      `[LangGraphAgent] Starting graph execution from root node "${this.rootNode}"`
+    this.logger.debug(
+      `Starting graph execution from root node "${this.rootNode}"`
     );
     if (this.nodes.size === 0) {
       return {
@@ -1987,9 +1996,7 @@ var LangGraphAgent = (_class6 = class extends BaseAgent {
     while (nodesToExecute.length > 0 && stepCount < this.maxSteps) {
       stepCount++;
       const { node, messages } = nodesToExecute.shift();
-      debugLog(
-        `[LangGraphAgent] Step ${stepCount}: Executing node "${node.name}"`
-      );
+      this.logger.debug(`Step ${stepCount}: Executing node "${node.name}"`);
       executedNodes.push(node.name);
       try {
         const result = await node.agent.run({
@@ -2037,7 +2044,7 @@ var LangGraphAgent = (_class6 = class extends BaseAgent {
           });
         }
       } catch (error) {
-        console.error(`[LangGraphAgent] Error in node "${node.name}":`, error);
+        console.error(`Error in node "${node.name}":`, error);
         return {
           content: `Error in node "${node.name}": ${error instanceof Error ? error.message : String(error)}`,
           role: "assistant"
@@ -2061,8 +2068,8 @@ var LangGraphAgent = (_class6 = class extends BaseAgent {
       messages: options.messages,
       config: options.config
     });
-    debugLog(
-      `[LangGraphAgent] Starting graph execution from root node "${this.rootNode}" (streaming)`
+    this.logger.debug(
+      `Starting graph execution from root node "${this.rootNode}" (streaming)`
     );
     if (this.nodes.size === 0) {
       yield {
@@ -2090,8 +2097,8 @@ var LangGraphAgent = (_class6 = class extends BaseAgent {
     while (nodesToExecute.length > 0 && stepCount < this.maxSteps) {
       stepCount++;
       const { node, messages } = nodesToExecute.shift();
-      debugLog(
-        `[LangGraphAgent] Step ${stepCount}: Executing node "${node.name}" (streaming)`
+      this.logger.debug(
+        `Step ${stepCount}: Executing node "${node.name}" (streaming)`
       );
       executedNodes.push(node.name);
       try {
@@ -2146,7 +2153,7 @@ Node output: ${this.extractTextContent(result.content)}` : ""}`,
           });
         }
       } catch (error) {
-        console.error(`[LangGraphAgent] Error in node "${node.name}":`, error);
+        console.error(`Error in node "${node.name}":`, error);
         yield {
           content: `Error in node "${node.name}": ${error instanceof Error ? error.message : String(error)}`,
           role: "assistant"
@@ -2164,7 +2171,7 @@ Node output: ${this.extractTextContent(result.content)}` : ""}`,
       };
     }
   }
-}, _class6);
+}, _class11);
 
 // src/tools/index.ts
 var tools_exports = {};
@@ -2206,9 +2213,10 @@ function createFunctionTool(func, options) {
 init_function_utils();
 
 // src/tools/common/google-search.ts
-init_debug();
+init_logger();
 init_base_tool();
-var GoogleSearch = class extends BaseTool {
+var GoogleSearch = (_class12 = class extends BaseTool {
+  __init15() {this.logger = new Logger({ name: "GoogleSearch" })}
   /**
    * Constructor for GoogleSearch
    */
@@ -2216,7 +2224,7 @@ var GoogleSearch = class extends BaseTool {
     super({
       name: "google_search",
       description: "Search the web using Google"
-    });
+    });_class12.prototype.__init15.call(this);;
   }
   /**
    * Get the function declaration for the tool
@@ -2247,7 +2255,9 @@ var GoogleSearch = class extends BaseTool {
    * This is a simplified implementation that doesn't actually search, just returns mock results
    */
   async runAsync(args, _context) {
-    debugLog(`[GoogleSearch] Executing Google search for: ${args.query}`);
+    this.logger.debug(
+      `[GoogleSearch] Executing Google search for: ${args.query}`
+    );
     return {
       results: [
         {
@@ -2263,7 +2273,7 @@ var GoogleSearch = class extends BaseTool {
       ]
     };
   }
-};
+}, _class12);
 
 // src/tools/common/http-request-tool.ts
 init_base_tool();
@@ -2710,9 +2720,10 @@ var UserInteractionTool = class extends BaseTool {
 };
 
 // src/tools/common/exit-loop-tool.ts
-init_debug();
+init_logger();
 init_base_tool();
-var ExitLoopTool = class extends BaseTool {
+var ExitLoopTool = (_class13 = class extends BaseTool {
+  __init16() {this.logger = new Logger({ name: "ExitLoopTool" })}
   /**
    * Constructor for ExitLoopTool
    */
@@ -2720,7 +2731,7 @@ var ExitLoopTool = class extends BaseTool {
     super({
       name: "exit_loop",
       description: "Exits the loop. Call this function only when you are instructed to do so."
-    });
+    });_class13.prototype.__init16.call(this);;
   }
   /**
    * Get the function declaration for the tool
@@ -2740,7 +2751,7 @@ var ExitLoopTool = class extends BaseTool {
    * Execute the exit loop action
    */
   async runAsync(_args, context) {
-    debugLog("[ExitLoopTool] Executing exit loop tool");
+    this.logger.debug("Executing exit loop tool");
     if (context.actions) {
       context.actions.escalate = true;
     } else {
@@ -2752,12 +2763,13 @@ var ExitLoopTool = class extends BaseTool {
       message: "Loop exited successfully"
     };
   }
-};
+}, _class13);
 
 // src/tools/common/get-user-choice-tool.ts
-init_debug();
+init_logger();
 init_base_tool();
-var GetUserChoiceTool = class extends BaseTool {
+var GetUserChoiceTool = (_class14 = class extends BaseTool {
+  __init17() {this.logger = new Logger({ name: "GetUserChoiceTool" })}
   /**
    * Constructor for GetUserChoiceTool
    */
@@ -2766,7 +2778,7 @@ var GetUserChoiceTool = class extends BaseTool {
       name: "get_user_choice",
       description: "This tool provides the options to the user and asks them to choose one. Use this tool when you need the user to make a selection between multiple options. Do not list options in your response - use this tool instead.",
       isLongRunning: true
-    });
+    });_class14.prototype.__init17.call(this);;
   }
   /**
    * Get the function declaration for the tool
@@ -2800,13 +2812,11 @@ var GetUserChoiceTool = class extends BaseTool {
    * and the actual choice will be provided asynchronously
    */
   async runAsync(args, context) {
-    debugLog(
-      `[GetUserChoiceTool] Executing get_user_choice with options: ${args.options.join(
-        ", "
-      )}`
+    this.logger.debug(
+      `Executing get_user_choice with options: ${args.options.join(", ")}`
     );
     if (args.question) {
-      debugLog(`[GetUserChoiceTool] Question: ${args.question}`);
+      this.logger.debug(`Question: ${args.question}`);
     }
     if (context.actions) {
       context.actions.skip_summarization = true;
@@ -2817,12 +2827,13 @@ var GetUserChoiceTool = class extends BaseTool {
     }
     return null;
   }
-};
+}, _class14);
 
 // src/tools/common/transfer-to-agent-tool.ts
-init_debug();
+init_logger();
 init_base_tool();
-var TransferToAgentTool = class extends BaseTool {
+var TransferToAgentTool = (_class15 = class extends BaseTool {
+  __init18() {this.logger = new Logger({ name: "TransferToAgentTool" })}
   /**
    * Constructor for TransferToAgentTool
    */
@@ -2830,7 +2841,7 @@ var TransferToAgentTool = class extends BaseTool {
     super({
       name: "transfer_to_agent",
       description: "Transfer the question to another agent."
-    });
+    });_class15.prototype.__init18.call(this);;
   }
   /**
    * Get the function declaration for the tool
@@ -2855,9 +2866,7 @@ var TransferToAgentTool = class extends BaseTool {
    * Execute the transfer to agent action
    */
   async runAsync(args, context) {
-    debugLog(
-      `[TransferToAgentTool] Executing transfer to agent: ${args.agent_name}`
-    );
+    this.logger.debug(`Executing transfer to agent: ${args.agent_name}`);
     if (context.actions) {
       context.actions.transfer_to_agent = args.agent_name;
     } else {
@@ -2869,12 +2878,13 @@ var TransferToAgentTool = class extends BaseTool {
       message: `Transferred to agent: ${args.agent_name}`
     };
   }
-};
+}, _class15);
 
 // src/tools/common/load-memory-tool.ts
-init_debug();
+init_logger();
 init_base_tool();
-var LoadMemoryTool = class extends BaseTool {
+var LoadMemoryTool = (_class16 = class extends BaseTool {
+  __init19() {this.logger = new Logger({ name: "LoadMemoryTool" })}
   /**
    * Constructor for LoadMemoryTool
    */
@@ -2882,7 +2892,7 @@ var LoadMemoryTool = class extends BaseTool {
     super({
       name: "load_memory",
       description: "Loads the memory for the current user based on a query."
-    });
+    });_class16.prototype.__init19.call(this);;
   }
   /**
    * Get the function declaration for the tool
@@ -2907,9 +2917,7 @@ var LoadMemoryTool = class extends BaseTool {
    * Execute the memory loading action
    */
   async runAsync(args, context) {
-    debugLog(
-      `[LoadMemoryTool] Executing load_memory with query: ${args.query}`
-    );
+    this.logger.debug(`Executing load_memory with query: ${args.query}`);
     if (!context.memoryService) {
       return {
         error: "Memory service is not available",
@@ -2930,12 +2938,21 @@ var LoadMemoryTool = class extends BaseTool {
       };
     }
   }
-};
+}, _class16);
 
 // src/tools/mcp/client.ts
+init_logger();
 var _indexjs = require('@modelcontextprotocol/sdk/client/index.js');
 var _ssejs = require('@modelcontextprotocol/sdk/client/sse.js');
 var _stdiojs = require('@modelcontextprotocol/sdk/client/stdio.js');
+var _typesjs = require('@modelcontextprotocol/sdk/types.js');
+
+// src/tools/mcp/sampling-handler.ts
+init_logger();
+
+
+
+
 
 // src/tools/mcp/types.ts
 var McpErrorType = /* @__PURE__ */ ((McpErrorType2) => {
@@ -2944,6 +2961,8 @@ var McpErrorType = /* @__PURE__ */ ((McpErrorType2) => {
   McpErrorType2["RESOURCE_CLOSED_ERROR"] = "resource_closed_error";
   McpErrorType2["TIMEOUT_ERROR"] = "timeout_error";
   McpErrorType2["INVALID_SCHEMA_ERROR"] = "invalid_schema_error";
+  McpErrorType2["SAMPLING_ERROR"] = "SAMPLING_ERROR";
+  McpErrorType2["INVALID_REQUEST_ERROR"] = "INVALID_REQUEST_ERROR";
   return McpErrorType2;
 })(McpErrorType || {});
 var McpError = class extends Error {
@@ -2957,6 +2976,154 @@ var McpError = class extends Error {
   }
 };
 
+// src/tools/mcp/sampling-handler.ts
+var McpSamplingHandler = (_class17 = class {
+  __init20() {this.logger = new Logger({ name: "McpSamplingHandler" })}
+  
+  constructor(adkHandler) {;_class17.prototype.__init20.call(this);
+    this.adkHandler = adkHandler;
+  }
+  /**
+   * Handle MCP sampling request and convert between formats
+   */
+  async handleSamplingRequest(request) {
+    try {
+      const validationResult = _typesjs.CreateMessageRequestSchema.safeParse(request);
+      if (!validationResult.success) {
+        this.logger.error(
+          "Invalid MCP sampling request:",
+          validationResult.error
+        );
+        throw new McpError(
+          `Invalid sampling request: ${validationResult.error.message}`,
+          "INVALID_REQUEST_ERROR" /* INVALID_REQUEST_ERROR */
+        );
+      }
+      const mcpParams = request.params;
+      if (!mcpParams.messages || !Array.isArray(mcpParams.messages)) {
+        throw new McpError(
+          "Invalid sampling request: messages array is required",
+          "INVALID_REQUEST_ERROR" /* INVALID_REQUEST_ERROR */
+        );
+      }
+      if (!mcpParams.maxTokens || mcpParams.maxTokens <= 0) {
+        throw new McpError(
+          "Invalid sampling request: maxTokens must be a positive number",
+          "INVALID_REQUEST_ERROR" /* INVALID_REQUEST_ERROR */
+        );
+      }
+      this.logger.debug("Converting MCP request to ADK format");
+      const adkMessages = this.convertMcpMessagesToADK(mcpParams.messages);
+      const adkRequest = {
+        messages: adkMessages,
+        systemPrompt: mcpParams.systemPrompt,
+        modelPreferences: mcpParams.modelPreferences,
+        includeContext: mcpParams.includeContext,
+        temperature: mcpParams.temperature,
+        maxTokens: mcpParams.maxTokens,
+        stopSequences: mcpParams.stopSequences,
+        metadata: mcpParams.metadata
+      };
+      this.logger.debug("Calling ADK sampling handler");
+      const adkResponse = await this.adkHandler(adkRequest);
+      this.logger.debug("Converting ADK response to MCP format");
+      const mcpResponse = this.convertADKResponseToMcp(adkResponse);
+      const responseValidation = _typesjs.CreateMessageResultSchema.safeParse(mcpResponse);
+      if (!responseValidation.success) {
+        this.logger.error(
+          "Invalid MCP response generated:",
+          responseValidation.error
+        );
+        throw new McpError(
+          `Invalid response generated: ${responseValidation.error.message}`,
+          "SAMPLING_ERROR" /* SAMPLING_ERROR */
+        );
+      }
+      return mcpResponse;
+    } catch (error) {
+      this.logger.error("Error handling sampling request:", error);
+      if (error instanceof McpError) {
+        throw error;
+      }
+      throw new McpError(
+        `Sampling request failed: ${error instanceof Error ? error.message : String(error)}`,
+        "SAMPLING_ERROR" /* SAMPLING_ERROR */,
+        error instanceof Error ? error : void 0
+      );
+    }
+  }
+  /**
+   * Convert MCP messages to ADK message format
+   */
+  convertMcpMessagesToADK(mcpMessages) {
+    return mcpMessages.map((mcpMessage) => {
+      const adkRole = mcpMessage.role === "assistant" ? "assistant" : "user";
+      let adkContent;
+      if (mcpMessage.content.type === "text") {
+        adkContent = mcpMessage.content.text || "";
+      } else if (mcpMessage.content.type === "image") {
+        const contentParts = [];
+        if (mcpMessage.content.text) {
+          contentParts.push({
+            type: "text",
+            text: mcpMessage.content.text
+          });
+        }
+        if (mcpMessage.content.data) {
+          const mimeType = mcpMessage.content.mimeType || "image/jpeg";
+          const dataUrl = `data:${mimeType};base64,${mcpMessage.content.data}`;
+          contentParts.push({
+            type: "image",
+            image_url: {
+              url: dataUrl
+            }
+          });
+        }
+        adkContent = contentParts.length > 0 ? contentParts : "";
+      } else {
+        this.logger.warn(
+          `Unknown MCP content type: ${mcpMessage.content.type}`
+        );
+        adkContent = mcpMessage.content.text || "";
+      }
+      const adkMessage = {
+        role: adkRole,
+        content: adkContent
+      };
+      this.logger.debug(
+        `Converted MCP message - role: ${mcpMessage.role} -> ${adkRole}, content type: ${mcpMessage.content.type}`
+      );
+      return adkMessage;
+    });
+  }
+  /**
+   * Convert ADK response to MCP response format
+   */
+  convertADKResponseToMcp(adkResponse) {
+    const mcpResponse = {
+      model: adkResponse.model,
+      stopReason: adkResponse.stopReason,
+      role: "assistant",
+      // ADK responses are always from assistant
+      content: {
+        type: "text",
+        text: adkResponse.content || ""
+      }
+    };
+    this.logger.debug(
+      `Converted ADK response - model: ${adkResponse.model}, content length: ${_optionalChain([adkResponse, 'access', _39 => _39.content, 'optionalAccess', _40 => _40.length]) || 0}`
+    );
+    return mcpResponse;
+  }
+  /**
+   * Update the ADK handler
+   */
+  updateHandler(handler) {
+    this.adkHandler = handler;
+    this.logger.debug("ADK sampling handler updated");
+  }
+}, _class17);
+
 // src/tools/mcp/utils.ts
 function withRetry(fn, instance, reinitMethod, maxRetries = 1) {
   return async (...args) => {
@@ -2986,12 +3153,14 @@ function withRetry(fn, instance, reinitMethod, maxRetries = 1) {
 }
 
 // src/tools/mcp/client.ts
-var McpClientService = (_class7 = class {
-  
-  __init9() {this.client = null}
-  __init10() {this.transport = null}
-  __init11() {this.isClosing = false}
-  constructor(config) {;_class7.prototype.__init9.call(this);_class7.prototype.__init10.call(this);_class7.prototype.__init11.call(this);
+var McpClientService = (_class18 = class {
+  
+  __init21() {this.client = null}
+  __init22() {this.transport = null}
+  __init23() {this.isClosing = false}
+  __init24() {this.mcpSamplingHandler = null}
+  __init25() {this.logger = new Logger({ name: "McpClientService" })}
+  constructor(config) {;_class18.prototype.__init21.call(this);_class18.prototype.__init22.call(this);_class18.prototype.__init23.call(this);_class18.prototype.__init24.call(this);_class18.prototype.__init25.call(this);
     this.config = config;
   }
   /**
@@ -3041,6 +3210,7 @@ var McpClientService = (_class7 = class {
       } else {
         await connectPromise;
       }
+      await this.setupSamplingHandler(client);
       if (this.config.debug) {
         console.log("\u2705 MCP client connected successfully");
       }
@@ -3161,7 +3331,7 @@ var McpClientService = (_class7 = class {
         },
         this,
         async (instance) => await instance.reinitialize(),
-        _optionalChain([this, 'access', _39 => _39.config, 'access', _40 => _40.retryOptions, 'optionalAccess', _41 => _41.maxRetries]) || 2
+        _optionalChain([this, 'access', _41 => _41.config, 'access', _42 => _42.retryOptions, 'optionalAccess', _43 => _43.maxRetries]) || 2
       );
       return await wrappedCall();
     } catch (error) {
@@ -3192,10 +3362,67 @@ var McpClientService = (_class7 = class {
   isConnected() {
     return !!this.client && !this.isClosing;
   }
-}, _class7);
+  async setupSamplingHandler(client) {
+    if (!this.mcpSamplingHandler) {
+      if (this.config.debug) {
+        console.log(
+          "\u26A0\uFE0F No sampling handler provided - sampling requests will be rejected"
+        );
+      }
+      return;
+    }
+    client.setRequestHandler(_typesjs.CreateMessageRequestSchema, async (request) => {
+      try {
+        this.logger.debug("Received sampling request:", request);
+        const response = await this.mcpSamplingHandler.handleSamplingRequest(request);
+        if (this.config.debug) {
+          console.log("\u2705 Sampling request completed successfully");
+        }
+        return response;
+      } catch (error) {
+        console.error("\u274C Error handling sampling request:", error);
+        if (error instanceof McpError) {
+          throw error;
+        }
+        throw new McpError(
+          `Sampling request failed: ${error instanceof Error ? error.message : String(error)}`,
+          "SAMPLING_ERROR" /* SAMPLING_ERROR */,
+          error instanceof Error ? error : void 0
+        );
+      }
+    });
+    if (this.config.debug) {
+      console.log("\u{1F3AF} Sampling handler registered successfully");
+    }
+  }
+  /**
+   * Set an ADK sampling handler
+   */
+  setSamplingHandler(handler) {
+    this.mcpSamplingHandler = new McpSamplingHandler(handler);
+    if (this.client) {
+      this.setupSamplingHandler(this.client).catch((error) => {
+        console.error("Failed to update sampling handler:", error);
+      });
+    }
+  }
+  /**
+   * Remove the sampling handler
+   */
+  removeSamplingHandler() {
+    this.mcpSamplingHandler = null;
+    if (this.client) {
+      try {
+        _optionalChain([this, 'access', _44 => _44.client, 'access', _45 => _45.removeRequestHandler, 'optionalCall', _46 => _46("sampling/createMessage")]);
+      } catch (error) {
+        console.error("Failed to remove sampling handler:", error);
+      }
+    }
+  }
+}, _class18);
 
 // src/tools/mcp/create-tool.ts
-init_debug();
+init_logger();
 init_base_tool();
 
 // src/tools/mcp/schema-conversion.ts
@@ -3394,10 +3621,11 @@ async function createTool(mcpTool, client) {
     throw error;
   }
 }
-var McpToolAdapter = (_class8 = class extends BaseTool {
+var McpToolAdapter = (_class19 = class extends BaseTool {
   
   
-  __init12() {this.clientService = null}
+  __init26() {this.clientService = null}
+  __init27() {this.logger = new Logger({ name: "McpToolAdapter" })}
   constructor(mcpTool, client) {
     const metadata = mcpTool.metadata || {};
     super({
@@ -3406,7 +3634,7 @@ var McpToolAdapter = (_class8 = class extends BaseTool {
       isLongRunning: _nullishCoalesce(metadata.isLongRunning, () => ( false)),
       shouldRetryOnFailure: _nullishCoalesce(metadata.shouldRetryOnFailure, () => ( false)),
       maxRetryAttempts: _nullishCoalesce(metadata.maxRetryAttempts, () => ( 3))
-    });_class8.prototype.__init12.call(this);;
+    });_class19.prototype.__init26.call(this);_class19.prototype.__init27.call(this);;
     this.mcpTool = mcpTool;
     this.client = client;
     if (client.reinitialize && typeof client.reinitialize === "function") {
@@ -3430,10 +3658,7 @@ var McpToolAdapter = (_class8 = class extends BaseTool {
     }
   }
   async runAsync(args, _context) {
-    debugLog(
-      `[McpToolAdapter] Executing MCP tool ${this.name} with args:`,
-      args
-    );
+    this.logger.debug(`Executing MCP tool ${this.name} with args:`, args);
     try {
       if (typeof this.mcpTool.execute === "function") {
         return await this.mcpTool.execute(args);
@@ -3482,16 +3707,16 @@ var McpToolAdapter = (_class8 = class extends BaseTool {
       throw error;
     }
   }
-}, _class8);
+}, _class19);
 
 // src/tools/mcp/index.ts
-var McpToolset = (_class9 = class {
+var McpToolset = (_class20 = class {
   
-  __init13() {this.clientService = null}
-  __init14() {this.toolFilter = null}
-  __init15() {this.tools = []}
-  __init16() {this.isClosing = false}
-  constructor(config, toolFilter = null) {;_class9.prototype.__init13.call(this);_class9.prototype.__init14.call(this);_class9.prototype.__init15.call(this);_class9.prototype.__init16.call(this);
+  __init28() {this.clientService = null}
+  __init29() {this.toolFilter = null}
+  __init30() {this.tools = []}
+  __init31() {this.isClosing = false}
+  constructor(config, toolFilter = null) {;_class20.prototype.__init28.call(this);_class20.prototype.__init29.call(this);_class20.prototype.__init30.call(this);_class20.prototype.__init31.call(this);
     this.config = config;
     this.toolFilter = toolFilter;
     this.clientService = new McpClientService(config);
@@ -3540,7 +3765,7 @@ var McpToolset = (_class9 = class {
           "resource_closed_error" /* RESOURCE_CLOSED_ERROR */
         );
       }
-      if (this.tools.length > 0 && !_optionalChain([this, 'access', _42 => _42.config, 'access', _43 => _43.cacheConfig, 'optionalAccess', _44 => _44.enabled]) === false) {
+      if (this.tools.length > 0 && !_optionalChain([this, 'access', _47 => _47.config, 'access', _48 => _48.cacheConfig, 'optionalAccess', _49 => _49.enabled]) === false) {
         return this.tools;
       }
       if (!this.clientService) {
@@ -3566,7 +3791,7 @@ var McpToolset = (_class9 = class {
           }
         }
       }
-      if (_optionalChain([this, 'access', _45 => _45.config, 'access', _46 => _46.cacheConfig, 'optionalAccess', _47 => _47.enabled]) !== false) {
+      if (_optionalChain([this, 'access', _50 => _50.config, 'access', _51 => _51.cacheConfig, 'optionalAccess', _52 => _52.enabled]) !== false) {
         this.tools = tools;
       }
       return tools;
@@ -3626,7 +3851,7 @@ var McpToolset = (_class9 = class {
   async dispose() {
     await this.close();
   }
-}, _class9);
+}, _class20);
 async function getMcpTools(config, toolFilter) {
   const toolset = new McpToolset(config, toolFilter);
   try {
@@ -3733,11 +3958,11 @@ var BaseLLM = class {
 };
 
 // src/models/base-llm-connection.ts
-var BaseLLMConnection = (_class10 = class {constructor() { _class10.prototype.__init17.call(this); }
+var BaseLLMConnection = (_class21 = class {constructor() { _class21.prototype.__init32.call(this); }
   /**
    * Whether the connection is active
    */
-  __init17() {this._isActive = true}
+  __init32() {this._isActive = true}
   /**
    * Gets whether the connection is active
    */
@@ -3750,15 +3975,15 @@ var BaseLLMConnection = (_class10 = class {constructor() { _class10.prototype.__
   close() {
     this._isActive = false;
   }
-}, _class10);
+}, _class21);
 
 // src/models/anthropic-llm.ts
-init_debug();
+init_logger();
 var _axios = require('axios'); var _axios2 = _interopRequireDefault(_axios);
 
 // src/models/anthropic-llm-connection.ts
-init_debug();
-var AnthropicLLMConnection = class extends BaseLLMConnection {
+init_logger();
+var AnthropicLLMConnection = (_class22 = class extends BaseLLMConnection {
   /**
    * Axios instance for API calls
    */
@@ -3785,11 +4010,12 @@ var AnthropicLLMConnection = class extends BaseLLMConnection {
   
   
   
+  __init33() {this.logger = new Logger({ name: "AnthropicLlmConnection" })}
   /**
    * Constructor
    */
   constructor(client, model, initialRequest, defaultParams) {
-    super();
+    super();_class22.prototype.__init33.call(this);;
     this.client = client;
     this.model = model;
     this.defaultParams = defaultParams;
@@ -3869,7 +4095,7 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
    * Convert Anthropic tool calls to ADK tool calls
    */
   convertToolCalls(toolUses) {
-    if (!_optionalChain([toolUses, 'optionalAccess', _48 => _48.length])) {
+    if (!_optionalChain([toolUses, 'optionalAccess', _53 => _53.length])) {
       return void 0;
     }
     return toolUses.map((toolUse) => ({
@@ -3884,15 +4110,13 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
    * Extract tool uses from response content
    */
   extractToolUses(content) {
-    if (!_optionalChain([content, 'optionalAccess', _49 => _49.length])) return [];
+    if (!_optionalChain([content, 'optionalAccess', _54 => _54.length])) return [];
     const toolUses = [];
     for (const block of content) {
-      debugLog(
-        `[AnthropicLLMConnection] Processing content block of type: ${block.type}`
-      );
+      this.logger.debug(`Processing content block of type: ${block.type}`);
       if (block.type === "tool_use") {
-        debugLog(
-          "[AnthropicLLMConnection] Found tool_use block:",
+        this.logger.debug(
+          "Found tool_use block:",
           JSON.stringify(block, null, 2)
         );
         toolUses.push({
@@ -3902,12 +4126,10 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
         });
       }
     }
-    debugLog(
-      `[AnthropicLLMConnection] Found ${toolUses.length} tool uses in content`
-    );
+    this.logger.debug(`Found ${toolUses.length} tool uses in content`);
     if (toolUses.length > 0) {
-      debugLog(
-        "[AnthropicLLMConnection] Extracted tool uses:",
+      this.logger.debug(
+        "Extracted tool uses:",
         JSON.stringify(toolUses, null, 2)
       );
     }
@@ -4003,37 +4225,34 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
       }
       const toolUses = this.extractToolUses(apiResponse.content);
       const toolCalls = this.convertToolCalls(toolUses);
-      debugLog(
-        `[AnthropicLLMConnection] - Extracted ${toolUses.length} tool uses in content and converted ${_optionalChain([toolCalls, 'optionalAccess', _50 => _50.length]) || 0} tool calls`
+      this.logger.debug(
+        `- Extracted ${toolUses.length} tool uses in content and converted ${_optionalChain([toolCalls, 'optionalAccess', _55 => _55.length]) || 0} tool calls`
       );
       const llmResponse = new LLMResponse({
         role: "assistant",
         content,
-        tool_calls: _optionalChain([toolCalls, 'optionalAccess', _51 => _51.length]) ? toolCalls : void 0,
+        tool_calls: _optionalChain([toolCalls, 'optionalAccess', _56 => _56.length]) ? toolCalls : void 0,
         raw_response: apiResponse
       });
       const logObject = {
         role: llmResponse.role,
-        content: _optionalChain([llmResponse, 'access', _52 => _52.content, 'optionalAccess', _53 => _53.substring, 'call', _54 => _54(0, 50)]) + (llmResponse.content && llmResponse.content.length > 50 ? "..." : ""),
+        content: _optionalChain([llmResponse, 'access', _57 => _57.content, 'optionalAccess', _58 => _58.substring, 'call', _59 => _59(0, 50)]) + (llmResponse.content && llmResponse.content.length > 50 ? "..." : ""),
         tool_calls: llmResponse.tool_calls ? `[${llmResponse.tool_calls.length} calls]` : "undefined"
       };
-      debugLog(
-        "[AnthropicLLMConnection] Final LLMResponse object:",
+      this.logger.debug(
+        "Final LLMResponse object:",
         JSON.stringify(logObject, null, 2)
       );
       return llmResponse;
     } catch (error) {
-      debugLog(
-        "[AnthropicLLMConnection] Error sending message to Anthropic:",
-        error
-      );
+      this.logger.debug("Error sending message to Anthropic:", error);
       throw error;
     }
   }
-};
+}, _class22);
 
 // src/models/anthropic-llm.ts
-var AnthropicLLM = class extends BaseLLM {
+var AnthropicLLM = (_class23 = class extends BaseLLM {
   /**
    * Anthropic API key
    */
@@ -4046,22 +4265,23 @@ var AnthropicLLM = class extends BaseLLM {
    * Default parameters for requests
    */
   
+  __init34() {this.logger = new Logger({ name: "AnthropicLLM" })}
   /**
    * Constructor for AnthropicLLM
    */
   constructor(model, config) {
-    super(model);
-    this.apiKey = _optionalChain([config, 'optionalAccess', _55 => _55.apiKey]) || process.env.ANTHROPIC_API_KEY || "";
-    this.baseURL = _optionalChain([config, 'optionalAccess', _56 => _56.baseURL]) || "https://api.anthropic.com/v1";
+    super(model);_class23.prototype.__init34.call(this);;
+    this.apiKey = _optionalChain([config, 'optionalAccess', _60 => _60.apiKey]) || process.env.ANTHROPIC_API_KEY || "";
+    this.baseURL = _optionalChain([config, 'optionalAccess', _61 => _61.baseURL]) || "https://api.anthropic.com/v1";
     if (!this.apiKey) {
       throw new Error(
         "Anthropic API key is required. Provide it in config or set ANTHROPIC_API_KEY environment variable."
       );
     }
     this.defaultParams = {
-      temperature: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _57 => _57.defaultParams, 'optionalAccess', _58 => _58.temperature]), () => ( 0.7)),
-      top_p: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _59 => _59.defaultParams, 'optionalAccess', _60 => _60.top_p]), () => ( 1)),
-      max_tokens: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _61 => _61.defaultParams, 'optionalAccess', _62 => _62.max_tokens]), () => ( 1024))
+      temperature: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _62 => _62.defaultParams, 'optionalAccess', _63 => _63.temperature]), () => ( 0.7)),
+      top_p: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _64 => _64.defaultParams, 'optionalAccess', _65 => _65.top_p]), () => ( 1)),
+      max_tokens: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _66 => _66.defaultParams, 'optionalAccess', _67 => _67.max_tokens]), () => ( 1024))
     };
   }
   /**
@@ -4147,7 +4367,7 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
    * Convert ADK function declarations to Anthropic tool format
    */
   convertFunctionsToTools(functions) {
-    if (!_optionalChain([functions, 'optionalAccess', _63 => _63.length])) {
+    if (!_optionalChain([functions, 'optionalAccess', _68 => _68.length])) {
       return [];
     }
     return functions.map((func) => ({
@@ -4160,7 +4380,7 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
    * Convert Anthropic tool calls to ADK tool calls
    */
   convertToolUses(toolUses) {
-    if (!_optionalChain([toolUses, 'optionalAccess', _64 => _64.length])) {
+    if (!_optionalChain([toolUses, 'optionalAccess', _69 => _69.length])) {
       return [];
     }
     return toolUses.map((toolUse) => ({
@@ -4175,15 +4395,13 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
    * Extract tool uses from response content
    */
   extractToolUses(content) {
-    if (!_optionalChain([content, 'optionalAccess', _65 => _65.length])) return [];
+    if (!_optionalChain([content, 'optionalAccess', _70 => _70.length])) return [];
     const toolUses = [];
     for (const block of content) {
-      debugLog(
-        `[AnthropicLLM] Processing content block of type: ${block.type}`
-      );
+      this.logger.debug(`Processing content block of type: ${block.type}`);
       if (block.type === "tool_use") {
-        debugLog(
-          `[AnthropicLLM] Found tool_use block: ${JSON.stringify(block, null, 2)}`
+        this.logger.debug(
+          `Found tool_use block: ${JSON.stringify(block, null, 2)}`
         );
         toolUses.push({
           id: block.id || "unknown-id",
@@ -4192,8 +4410,8 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
         });
       }
     }
-    debugLog(
-      `[AnthropicLLM] Found ${toolUses.length} tool uses in content`,
+    this.logger.debug(
+      `Found ${toolUses.length} tool uses in content`,
       toolUses
     );
     return toolUses;
@@ -4217,12 +4435,12 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
         },
         responseType: stream ? "stream" : "json"
       });
-      debugLog(
-        `[AnthropicLLM] API Response done with ${response.status}:`,
+      this.logger.debug(
+        `API Response done with ${response.status}:`,
         response.data
       );
-      debugLog(
-        "[AnthropicLLM] API Response content:",
+      this.logger.debug(
+        "API Response content:",
         response.data.content.map((block) => ({ type: block.type }))
       );
       return response.data;
@@ -4248,9 +4466,9 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
         temperature: _nullishCoalesce(llmRequest.config.temperature, () => ( this.defaultParams.temperature)),
         max_tokens: _nullishCoalesce(llmRequest.config.max_tokens, () => ( this.defaultParams.max_tokens)),
         top_p: _nullishCoalesce(llmRequest.config.top_p, () => ( this.defaultParams.top_p)),
-        tools: _optionalChain([tools, 'optionalAccess', _66 => _66.length]) ? tools : void 0
+        tools: _optionalChain([tools, 'optionalAccess', _71 => _71.length]) ? tools : void 0
       };
-      debugLog("[AnthropicLLM] API Request:", {
+      this.logger.debug("API Request:", {
         model: params.model,
         messageCount: params.messages.length,
         systemMessage: params.system ? "present" : "none",
@@ -4260,7 +4478,7 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
         throw new Error("Streaming is not supported in this implementation");
       }
       const response = await this.callAnthropicAPI(params);
-      debugLog("[AnthropicLLM] Full Response Content:", response.content);
+      this.logger.debug("Full Response Content:", response.content);
       let content = "";
       for (const block of response.content) {
         if (block.type === "text") {
@@ -4269,8 +4487,8 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
       }
       const toolUses = this.extractToolUses(response.content);
       const toolCalls = this.convertToolUses(toolUses);
-      debugLog("[AnthropicLLM] Extracted Tool Uses:", toolUses);
-      debugLog("[AnthropicLLM] Converted Tool Calls:", toolCalls);
+      this.logger.debug("Extracted Tool Uses:", toolUses);
+      this.logger.debug("Converted Tool Calls:", toolCalls);
       const llmResponse = new LLMResponse({
         role: "assistant",
         content,
@@ -4279,16 +4497,16 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
       });
       const logObject = {
         role: llmResponse.role,
-        content: _optionalChain([llmResponse, 'access', _67 => _67.content, 'optionalAccess', _68 => _68.substring, 'call', _69 => _69(0, 50)]) + (llmResponse.content && llmResponse.content.length > 50 ? "..." : ""),
+        content: _optionalChain([llmResponse, 'access', _72 => _72.content, 'optionalAccess', _73 => _73.substring, 'call', _74 => _74(0, 50)]) + (llmResponse.content && llmResponse.content.length > 50 ? "..." : ""),
         tool_calls: llmResponse.tool_calls ? `[${llmResponse.tool_calls.length} calls]` : "undefined"
       };
-      debugLog(
-        "[AnthropicLLM] Final LLMResponse object:",
+      this.logger.debug(
+        "Final LLMResponse object:",
         JSON.stringify(logObject, null, 2)
       );
       yield llmResponse;
     } catch (error) {
-      debugLog("[AnthropicLLM] Error:", error);
+      this.logger.debug("Error:", error);
       throw error;
     }
   }
@@ -4311,7 +4529,7 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
       this.defaultParams
     );
   }
-};
+}, _class23);
 
 // src/models/google-llm.ts
 
@@ -4332,9 +4550,9 @@ var GoogleLLM = class extends BaseLLM {
   constructor(model, config) {
     super(model);
     const apiKey = process.env.GOOGLE_API_KEY;
-    const projectId = _optionalChain([config, 'optionalAccess', _70 => _70.projectId]) || process.env.GOOGLE_CLOUD_PROJECT;
-    const location = _optionalChain([config, 'optionalAccess', _71 => _71.location]) || process.env.GOOGLE_CLOUD_LOCATION;
-    const useVertexAI = _optionalChain([process, 'access', _72 => _72.env, 'access', _73 => _73.USE_VERTEX_AI, 'optionalAccess', _74 => _74.toLowerCase, 'call', _75 => _75()]) === "true";
+    const projectId = _optionalChain([config, 'optionalAccess', _75 => _75.projectId]) || process.env.GOOGLE_CLOUD_PROJECT;
+    const location = _optionalChain([config, 'optionalAccess', _76 => _76.location]) || process.env.GOOGLE_CLOUD_LOCATION;
+    const useVertexAI = _optionalChain([process, 'access', _77 => _77.env, 'access', _78 => _78.USE_VERTEX_AI, 'optionalAccess', _79 => _79.toLowerCase, 'call', _80 => _80()]) === "true";
     if (!useVertexAI && !apiKey) {
       throw new Error(
         "Google API Key is required. Provide via config or GOOGLE_API_KEY env var."
@@ -4359,9 +4577,9 @@ var GoogleLLM = class extends BaseLLM {
     }
     this.ai = new (0, _genai.GoogleGenAI)(options);
     this.defaultParams = {
-      temperature: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _76 => _76.defaultParams, 'optionalAccess', _77 => _77.temperature]), () => ( 0.7)),
-      topP: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _78 => _78.defaultParams, 'optionalAccess', _79 => _79.top_p]), () => ( 1)),
-      maxOutputTokens: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _80 => _80.defaultParams, 'optionalAccess', _81 => _81.maxOutputTokens]), () => ( 1024))
+      temperature: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _81 => _81.defaultParams, 'optionalAccess', _82 => _82.temperature]), () => ( 0.7)),
+      topP: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _83 => _83.defaultParams, 'optionalAccess', _84 => _84.top_p]), () => ( 1)),
+      maxOutputTokens: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _85 => _85.defaultParams, 'optionalAccess', _86 => _86.maxOutputTokens]), () => ( 1024))
     };
   }
   /**
@@ -4499,7 +4717,7 @@ var GoogleLLM = class extends BaseLLM {
       );
       parts.push({ text: "" });
     }
-    if (googleRole === "function" && (parts.length !== 1 || !_optionalChain([parts, 'access', _82 => _82[0], 'optionalAccess', _83 => _83.functionResponse]))) {
+    if (googleRole === "function" && (parts.length !== 1 || !_optionalChain([parts, 'access', _87 => _87[0], 'optionalAccess', _88 => _88.functionResponse]))) {
       console.error(
         `[GoogleLLM] convertMessage - Invalid parts for 'function' role. Expected 1 functionResponse part. Got:`,
         JSON.stringify(parts),
@@ -4607,13 +4825,13 @@ var GoogleLLM = class extends BaseLLM {
       role: "assistant",
       content: null
     });
-    if (typeof _optionalChain([response, 'optionalAccess', _84 => _84.candidates, 'optionalAccess', _85 => _85[0], 'optionalAccess', _86 => _86.content, 'optionalAccess', _87 => _87.parts, 'optionalAccess', _88 => _88[0], 'optionalAccess', _89 => _89.text]) === "string") {
+    if (typeof _optionalChain([response, 'optionalAccess', _89 => _89.candidates, 'optionalAccess', _90 => _90[0], 'optionalAccess', _91 => _91.content, 'optionalAccess', _92 => _92.parts, 'optionalAccess', _93 => _93[0], 'optionalAccess', _94 => _94.text]) === "string") {
       result.content = response.candidates[0].content.parts[0].text;
     }
-    if (_optionalChain([response, 'optionalAccess', _90 => _90.candidates, 'optionalAccess', _91 => _91[0], 'optionalAccess', _92 => _92.content, 'optionalAccess', _93 => _93.parts, 'optionalAccess', _94 => _94[0], 'optionalAccess', _95 => _95.text])) {
+    if (_optionalChain([response, 'optionalAccess', _95 => _95.candidates, 'optionalAccess', _96 => _96[0], 'optionalAccess', _97 => _97.content, 'optionalAccess', _98 => _98.parts, 'optionalAccess', _99 => _99[0], 'optionalAccess', _100 => _100.text])) {
       result.content = response.candidates[0].content.parts[0].text;
     }
-    if (_optionalChain([response, 'optionalAccess', _96 => _96.candidates, 'optionalAccess', _97 => _97[0], 'optionalAccess', _98 => _98.content, 'optionalAccess', _99 => _99.parts, 'optionalAccess', _100 => _100[0], 'optionalAccess', _101 => _101.functionCall])) {
+    if (_optionalChain([response, 'optionalAccess', _101 => _101.candidates, 'optionalAccess', _102 => _102[0], 'optionalAccess', _103 => _103.content, 'optionalAccess', _104 => _104.parts, 'optionalAccess', _105 => _105[0], 'optionalAccess', _106 => _106.functionCall])) {
       const functionCall = response.candidates[0].content.parts[0].functionCall;
       result.function_call = {
         name: functionCall.name,
@@ -4660,10 +4878,10 @@ var GoogleLLM = class extends BaseLLM {
       if (stream) {
         const streamingResult = await this.ai.models.generateContentStream(requestOptions);
         for await (const chunk of streamingResult) {
-          if (!_optionalChain([chunk, 'access', _102 => _102.candidates, 'optionalAccess', _103 => _103[0], 'optionalAccess', _104 => _104.content, 'optionalAccess', _105 => _105.parts, 'optionalAccess', _106 => _106[0], 'optionalAccess', _107 => _107.text])) {
+          if (!_optionalChain([chunk, 'access', _107 => _107.candidates, 'optionalAccess', _108 => _108[0], 'optionalAccess', _109 => _109.content, 'optionalAccess', _110 => _110.parts, 'optionalAccess', _111 => _111[0], 'optionalAccess', _112 => _112.text])) {
             continue;
           }
-          const partialText = _optionalChain([chunk, 'access', _108 => _108.candidates, 'access', _109 => _109[0], 'optionalAccess', _110 => _110.content, 'optionalAccess', _111 => _111.parts, 'access', _112 => _112[0], 'optionalAccess', _113 => _113.text]) || "";
+          const partialText = _optionalChain([chunk, 'access', _113 => _113.candidates, 'access', _114 => _114[0], 'optionalAccess', _115 => _115.content, 'optionalAccess', _116 => _116.parts, 'access', _117 => _117[0], 'optionalAccess', _118 => _118.text]) || "";
           const partialResponse = new LLMResponse({
             content: partialText,
             role: "assistant",
@@ -4683,11 +4901,11 @@ var GoogleLLM = class extends BaseLLM {
 };
 
 // src/models/openai-llm.ts
-init_debug();
+init_logger();
 var _openai = require('openai'); var _openai2 = _interopRequireDefault(_openai);
 
 // src/models/openai-llm-connection.ts
-var OpenAILLMConnection = (_class11 = class extends BaseLLMConnection {
+var OpenAILLMConnection = (_class24 = class extends BaseLLMConnection {
   /**
    * OpenAI client
    */
@@ -4719,12 +4937,12 @@ var OpenAILLMConnection = (_class11 = class extends BaseLLMConnection {
   /**
    * Ongoing chat history
    */
-  __init18() {this.messages = []}
+  __init35() {this.messages = []}
   /**
    * Constructor for OpenAILLMConnection
    */
   constructor(client, model, initialRequest, defaultParams) {
-    super();_class11.prototype.__init18.call(this);;
+    super();_class24.prototype.__init35.call(this);;
     this.client = client;
     this.model = model;
     this.initialRequest = initialRequest;
@@ -4804,10 +5022,10 @@ var OpenAILLMConnection = (_class11 = class extends BaseLLMConnection {
       for await (const chunk of stream) {
         if (chunk.choices.length === 0) continue;
         const delta = chunk.choices[0].delta;
-        if (_optionalChain([delta, 'optionalAccess', _114 => _114.content])) {
+        if (_optionalChain([delta, 'optionalAccess', _119 => _119.content])) {
           responseContent += delta.content;
         }
-        if (_optionalChain([delta, 'optionalAccess', _115 => _115.function_call])) {
+        if (_optionalChain([delta, 'optionalAccess', _120 => _120.function_call])) {
           if (!functionCall) {
             functionCall = {
               name: delta.function_call.name || "",
@@ -4818,7 +5036,7 @@ var OpenAILLMConnection = (_class11 = class extends BaseLLMConnection {
             functionCall.arguments += delta.function_call.arguments || "";
           }
         }
-        if (_optionalChain([delta, 'optionalAccess', _116 => _116.tool_calls])) {
+        if (_optionalChain([delta, 'optionalAccess', _121 => _121.tool_calls])) {
           for (const toolDelta of delta.tool_calls) {
             const id = toolDelta.id || "";
             let tool = toolCalls.find((t) => t.id === id);
@@ -4826,20 +5044,20 @@ var OpenAILLMConnection = (_class11 = class extends BaseLLMConnection {
               tool = {
                 id,
                 function: {
-                  name: _optionalChain([toolDelta, 'access', _117 => _117.function, 'optionalAccess', _118 => _118.name]) || "",
-                  arguments: _optionalChain([toolDelta, 'access', _119 => _119.function, 'optionalAccess', _120 => _120.arguments]) || ""
+                  name: _optionalChain([toolDelta, 'access', _122 => _122.function, 'optionalAccess', _123 => _123.name]) || "",
+                  arguments: _optionalChain([toolDelta, 'access', _124 => _124.function, 'optionalAccess', _125 => _125.arguments]) || ""
                 }
               };
               toolCalls.push(tool);
             } else {
-              tool.function.name += _optionalChain([toolDelta, 'access', _121 => _121.function, 'optionalAccess', _122 => _122.name]) || "";
-              tool.function.arguments += _optionalChain([toolDelta, 'access', _123 => _123.function, 'optionalAccess', _124 => _124.arguments]) || "";
+              tool.function.name += _optionalChain([toolDelta, 'access', _126 => _126.function, 'optionalAccess', _127 => _127.name]) || "";
+              tool.function.arguments += _optionalChain([toolDelta, 'access', _128 => _128.function, 'optionalAccess', _129 => _129.arguments]) || "";
             }
           }
         }
         if (this.responseCallback) {
           const response = new LLMResponse({
-            content: _optionalChain([delta, 'optionalAccess', _125 => _125.content]) || null,
+            content: _optionalChain([delta, 'optionalAccess', _130 => _130.content]) || null,
             role: "assistant",
             function_call: functionCall,
             tool_calls: toolCalls.length > 0 ? toolCalls : void 0,
@@ -4931,10 +5149,10 @@ var OpenAILLMConnection = (_class11 = class extends BaseLLMConnection {
   onEnd(callback) {
     this.endCallback = callback;
   }
-}, _class11);
+}, _class24);
 
 // src/models/openai-llm.ts
-var OpenAILLM = class extends BaseLLM {
+var OpenAILLM = (_class25 = class extends BaseLLM {
   /**
    * OpenAI client instance
    */
@@ -4943,22 +5161,23 @@ var OpenAILLM = class extends BaseLLM {
    * Default parameters for requests
    */
   
+  __init36() {this.logger = new Logger({ name: "OpenAILLM" })}
   /**
    * Constructor for OpenAILLM
    */
   constructor(model, config) {
-    super(model);
+    super(model);_class25.prototype.__init36.call(this);;
     this.client = new (0, _openai2.default)({
-      apiKey: _optionalChain([config, 'optionalAccess', _126 => _126.apiKey]) || process.env.OPENAI_API_KEY,
-      baseURL: _optionalChain([config, 'optionalAccess', _127 => _127.baseURL]),
-      organization: _optionalChain([config, 'optionalAccess', _128 => _128.organization])
+      apiKey: _optionalChain([config, 'optionalAccess', _131 => _131.apiKey]) || process.env.OPENAI_API_KEY,
+      baseURL: _optionalChain([config, 'optionalAccess', _132 => _132.baseURL]),
+      organization: _optionalChain([config, 'optionalAccess', _133 => _133.organization])
     });
     this.defaultParams = {
-      temperature: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _129 => _129.defaultParams, 'optionalAccess', _130 => _130.temperature]), () => ( 0.7)),
-      top_p: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _131 => _131.defaultParams, 'optionalAccess', _132 => _132.top_p]), () => ( 1)),
-      max_tokens: _optionalChain([config, 'optionalAccess', _133 => _133.defaultParams, 'optionalAccess', _134 => _134.max_tokens]),
-      frequency_penalty: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _135 => _135.defaultParams, 'optionalAccess', _136 => _136.frequency_penalty]), () => ( 0)),
-      presence_penalty: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _137 => _137.defaultParams, 'optionalAccess', _138 => _138.presence_penalty]), () => ( 0))
+      temperature: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _134 => _134.defaultParams, 'optionalAccess', _135 => _135.temperature]), () => ( 0.7)),
+      top_p: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _136 => _136.defaultParams, 'optionalAccess', _137 => _137.top_p]), () => ( 1)),
+      max_tokens: _optionalChain([config, 'optionalAccess', _138 => _138.defaultParams, 'optionalAccess', _139 => _139.max_tokens]),
+      frequency_penalty: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _140 => _140.defaultParams, 'optionalAccess', _141 => _141.frequency_penalty]), () => ( 0)),
+      presence_penalty: _nullishCoalesce(_optionalChain([config, 'optionalAccess', _142 => _142.defaultParams, 'optionalAccess', _143 => _143.presence_penalty]), () => ( 0))
     };
   }
   /**
@@ -5068,16 +5287,16 @@ var OpenAILLM = class extends BaseLLM {
    */
   convertResponse(response) {
     const result = new LLMResponse({
-      content: _optionalChain([response, 'access', _139 => _139.message, 'optionalAccess', _140 => _140.content]) || null,
-      role: _optionalChain([response, 'access', _141 => _141.message, 'optionalAccess', _142 => _142.role]) || "assistant"
+      content: _optionalChain([response, 'access', _144 => _144.message, 'optionalAccess', _145 => _145.content]) || null,
+      role: _optionalChain([response, 'access', _146 => _146.message, 'optionalAccess', _147 => _147.role]) || "assistant"
     });
-    if (_optionalChain([response, 'access', _143 => _143.message, 'optionalAccess', _144 => _144.function_call])) {
+    if (_optionalChain([response, 'access', _148 => _148.message, 'optionalAccess', _149 => _149.function_call])) {
       result.function_call = {
         name: response.message.function_call.name,
         arguments: response.message.function_call.arguments
       };
     }
-    if (_optionalChain([response, 'access', _145 => _145.message, 'optionalAccess', _146 => _146.tool_calls])) {
+    if (_optionalChain([response, 'access', _150 => _150.message, 'optionalAccess', _151 => _151.tool_calls])) {
       result.tool_calls = response.message.tool_calls.map((tool) => ({
         id: tool.id,
         function: {
@@ -5092,27 +5311,27 @@ var OpenAILLM = class extends BaseLLM {
    * Convert OpenAI streaming chunk to LLMResponse
    */
   convertChunk(chunk) {
-    debugLog(
-      `[OpenAILLM]: Converting chunk - delta: ${JSON.stringify(chunk.delta || {})}`
+    this.logger.debug(
+      `Converting chunk - delta: ${JSON.stringify(chunk.delta || {})}`
     );
-    const content = _optionalChain([chunk, 'access', _147 => _147.delta, 'optionalAccess', _148 => _148.content]);
+    const content = _optionalChain([chunk, 'access', _152 => _152.delta, 'optionalAccess', _153 => _153.content]);
     const result = new LLMResponse({
       content: content !== void 0 ? content : null,
-      role: _optionalChain([chunk, 'access', _149 => _149.delta, 'optionalAccess', _150 => _150.role]) || "assistant",
+      role: _optionalChain([chunk, 'access', _154 => _154.delta, 'optionalAccess', _155 => _155.role]) || "assistant",
       is_partial: true
     });
-    if (_optionalChain([chunk, 'access', _151 => _151.delta, 'optionalAccess', _152 => _152.function_call])) {
+    if (_optionalChain([chunk, 'access', _156 => _156.delta, 'optionalAccess', _157 => _157.function_call])) {
       result.function_call = {
         name: chunk.delta.function_call.name || "",
         arguments: chunk.delta.function_call.arguments || ""
       };
     }
-    if (_optionalChain([chunk, 'access', _153 => _153.delta, 'optionalAccess', _154 => _154.tool_calls])) {
+    if (_optionalChain([chunk, 'access', _158 => _158.delta, 'optionalAccess', _159 => _159.tool_calls])) {
       result.tool_calls = chunk.delta.tool_calls.map((tool) => ({
         id: tool.id || "",
         function: {
-          name: _optionalChain([tool, 'access', _155 => _155.function, 'optionalAccess', _156 => _156.name]) || "",
-          arguments: _optionalChain([tool, 'access', _157 => _157.function, 'optionalAccess', _158 => _158.arguments]) || ""
+          name: _optionalChain([tool, 'access', _160 => _160.function, 'optionalAccess', _161 => _161.name]) || "",
+          arguments: _optionalChain([tool, 'access', _162 => _162.function, 'optionalAccess', _163 => _163.arguments]) || ""
         }
       }));
     }
@@ -5135,24 +5354,24 @@ var OpenAILLM = class extends BaseLLM {
       presence_penalty: _nullishCoalesce(llmRequest.config.presence_penalty, () => ( this.defaultParams.presence_penalty)),
       stream: shouldStream
     };
-    debugLog(
-      `[OpenAILLM] Request parameters - model: ${params.model}, messages: ${params.messages.length}, functions: ${params.tools ? params.tools.length : 0}, streaming: ${shouldStream}`
+    this.logger.debug(
+      `Request parameters - model: ${params.model}, messages: ${params.messages.length}, functions: ${params.tools ? params.tools.length : 0}, streaming: ${shouldStream}`
     );
     if (tools && tools.length > 0) {
       params.tools = tools;
     }
     try {
       if (shouldStream) {
-        debugLog("[OpenAILLM] Starting streaming request");
+        this.logger.debug("Starting streaming request");
         const streamResponse = await this.client.chat.completions.create(params);
         let partialFunctionCall;
         const partialToolCalls = /* @__PURE__ */ new Map();
         let accumulatedContent = "";
         const asyncIterable = streamResponse;
-        debugLog("[OpenAILLM] Stream response received, processing chunks");
+        this.logger.debug("Stream response received, processing chunks");
         for await (const chunk of asyncIterable) {
           if (!chunk.choices || chunk.choices.length === 0) {
-            debugLog("[OpenAILLM] Empty chunk received, skipping");
+            this.logger.debug("Empty chunk received, skipping");
             continue;
           }
           const choice = chunk.choices[0];
@@ -5160,8 +5379,8 @@ var OpenAILLM = class extends BaseLLM {
           if (responseChunk.content !== null) {
             accumulatedContent += responseChunk.content;
           }
-          debugLog(
-            `[OpenAILLM] Chunk received - delta: "${_optionalChain([choice, 'access', _159 => _159.delta, 'optionalAccess', _160 => _160.content]) || ""}"`,
+          this.logger.debug(
+            `Chunk received - delta: "${_optionalChain([choice, 'access', _164 => _164.delta, 'optionalAccess', _165 => _165.content]) || ""}"`,
             `responseChunk content: "${responseChunk.content || ""}"`,
             `is_partial: ${responseChunk.is_partial}`,
             `accumulated: "${accumulatedContent.substring(0, 30)}${accumulatedContent.length > 30 ? "..." : ""}"`
@@ -5190,12 +5409,12 @@ var OpenAILLM = class extends BaseLLM {
             }
             responseChunk.tool_calls = Array.from(partialToolCalls.values());
           }
-          debugLog("[OpenAILLM] Yielding chunk to caller");
+          this.logger.debug("Yielding chunk to caller");
           yield responseChunk;
         }
         if (accumulatedContent.length > 0) {
-          debugLog(
-            `[OpenAILLM] Yielding final accumulated content: "${accumulatedContent.substring(0, 30)}${accumulatedContent.length > 30 ? "..." : ""}"`
+          this.logger.debug(
+            `Yielding final accumulated content: "${accumulatedContent.substring(0, 30)}${accumulatedContent.length > 30 ? "..." : ""}"`
           );
           yield new LLMResponse({
             content: accumulatedContent,
@@ -5203,14 +5422,14 @@ var OpenAILLM = class extends BaseLLM {
             is_partial: false
           });
         }
-        debugLog("[OpenAILLM] Finished processing all stream chunks");
+        this.logger.debug("Finished processing all stream chunks");
       } else {
-        debugLog("[OpenAILLM] Making non-streaming request");
+        this.logger.debug("Making non-streaming request");
         const response = await this.client.chat.completions.create(params);
         if (!response.choices || response.choices.length === 0) {
           throw new Error("No response from OpenAI");
         }
-        debugLog("[OpenAILLM] Non-streaming response received");
+        this.logger.debug("Non-streaming response received");
         yield this.convertResponse(response.choices[0]);
       }
     } catch (error) {
@@ -5229,7 +5448,7 @@ var OpenAILLM = class extends BaseLLM {
       this.defaultParams
     );
   }
-};
+}, _class25);
 
 // src/models/registry.ts
 function registerProviders() {
@@ -5427,7 +5646,7 @@ var OAuth2Credential = class extends AuthCredential {
         "Cannot refresh token: no refresh token or refresh function"
       );
     }
-    const result = await _optionalChain([this, 'access', _161 => _161.refreshFunction, 'optionalCall', _162 => _162(this.refreshToken)]);
+    const result = await _optionalChain([this, 'access', _166 => _166.refreshFunction, 'optionalCall', _167 => _167(this.refreshToken)]);
     if (!result) {
       throw new Error("Failed to refresh token");
     }
@@ -5481,7 +5700,7 @@ var AuthHandler = class {
    * Gets the authentication token
    */
   getToken() {
-    return _optionalChain([this, 'access', _163 => _163.credential, 'optionalAccess', _164 => _164.getToken, 'call', _165 => _165()]);
+    return _optionalChain([this, 'access', _168 => _168.credential, 'optionalAccess', _169 => _169.getToken, 'call', _170 => _170()]);
   }
   /**
    * Gets headers for HTTP requests
@@ -5496,7 +5715,7 @@ var AuthHandler = class {
    * Refreshes the token if necessary
    */
   async refreshToken() {
-    if (_optionalChain([this, 'access', _166 => _166.credential, 'optionalAccess', _167 => _167.canRefresh, 'call', _168 => _168()])) {
+    if (_optionalChain([this, 'access', _171 => _171.credential, 'optionalAccess', _172 => _172.canRefresh, 'call', _173 => _173()])) {
       await this.credential.refresh();
     }
   }
@@ -5722,7 +5941,7 @@ var InMemoryMemoryService = class {
     };
     const normalizedQuery = query.toLowerCase().trim();
     const queryTerms = normalizedQuery.split(/\s+/);
-    const sessionsToSearch = _optionalChain([options, 'optionalAccess', _169 => _169.sessionId]) ? this.sessions.has(options.sessionId) ? [this.sessions.get(options.sessionId)] : [] : Array.from(this.sessions.values());
+    const sessionsToSearch = _optionalChain([options, 'optionalAccess', _174 => _174.sessionId]) ? this.sessions.has(options.sessionId) ? [this.sessions.get(options.sessionId)] : [] : Array.from(this.sessions.values());
     for (const session of sessionsToSearch) {
       const matchedEvents = [];
       const scores = [];
@@ -5748,7 +5967,7 @@ var InMemoryMemoryService = class {
           }
         }
         const score = queryTerms.length > 0 ? termMatches / queryTerms.length : 0;
-        if (_optionalChain([options, 'optionalAccess', _170 => _170.threshold]) !== void 0 && score < options.threshold) {
+        if (_optionalChain([options, 'optionalAccess', _175 => _175.threshold]) !== void 0 && score < options.threshold) {
           continue;
         }
         if (score > 0) {
@@ -5768,7 +5987,7 @@ var InMemoryMemoryService = class {
     response.memories.sort(
       (a, b) => (_nullishCoalesce(b.relevanceScore, () => ( 0))) - (_nullishCoalesce(a.relevanceScore, () => ( 0)))
     );
-    if (_optionalChain([options, 'optionalAccess', _171 => _171.limit]) !== void 0 && options.limit > 0) {
+    if (_optionalChain([options, 'optionalAccess', _176 => _176.limit]) !== void 0 && options.limit > 0) {
       response.memories = response.memories.slice(0, options.limit);
     }
     return response;
@@ -5797,10 +6016,10 @@ var InMemoryMemoryService = class {
 };
 
 // src/memory/persistent-memory-service.ts
-init_debug();
+init_logger();
 var _fs = require('fs'); var fs3 = _interopRequireWildcard(_fs);
 
-var PersistentMemoryService = class {
+var PersistentMemoryService = (_class26 = class {
   /**
    * In-memory service used for search operations
    */
@@ -5813,10 +6032,11 @@ var PersistentMemoryService = class {
    * File prefix for memory files
    */
   
+  __init37() {this.logger = new Logger({ name: "PersistentMemoryService" })}
   /**
    * Constructor for PersistentMemoryService
    */
-  constructor(config) {
+  constructor(config) {;_class26.prototype.__init37.call(this);
     this.inMemoryService = new InMemoryMemoryService();
     this.storageDir = config.storageDir;
     this.filePrefix = config.filePrefix || "memory";
@@ -5901,7 +6121,7 @@ var PersistentMemoryService = class {
           }
         }
       }
-      debugLog(
+      this.logger.debug(
         `Loaded ${this.inMemoryService.getAllSessions().length} sessions from persistent storage`
       );
     } catch (error) {
@@ -5958,7 +6178,7 @@ var PersistentMemoryService = class {
       );
     }
   }
-};
+}, _class26);
 
 // src/sessions/index.ts
 var sessions_exports = {};
@@ -6033,17 +6253,17 @@ var InMemorySessionService = class {
     let sessions = Array.from(this.sessions.values()).filter(
       (session) => session.userId === userId
     );
-    if (_optionalChain([options, 'optionalAccess', _172 => _172.createdAfter])) {
+    if (_optionalChain([options, 'optionalAccess', _177 => _177.createdAfter])) {
       sessions = sessions.filter(
         (session) => session.createdAt >= options.createdAfter
       );
     }
-    if (_optionalChain([options, 'optionalAccess', _173 => _173.updatedAfter])) {
+    if (_optionalChain([options, 'optionalAccess', _178 => _178.updatedAfter])) {
       sessions = sessions.filter(
         (session) => session.updatedAt >= options.updatedAfter
       );
     }
-    if (_optionalChain([options, 'optionalAccess', _174 => _174.metadataFilter])) {
+    if (_optionalChain([options, 'optionalAccess', _179 => _179.metadataFilter])) {
       sessions = sessions.filter((session) => {
         for (const [key, value] of Object.entries(options.metadataFilter)) {
           if (session.metadata[key] !== value) {
@@ -6054,7 +6274,7 @@ var InMemorySessionService = class {
       });
     }
     sessions.sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
-    if (_optionalChain([options, 'optionalAccess', _175 => _175.limit]) !== void 0 && options.limit > 0) {
+    if (_optionalChain([options, 'optionalAccess', _180 => _180.limit]) !== void 0 && options.limit > 0) {
       sessions = sessions.slice(0, options.limit);
     }
     return sessions;
@@ -6089,7 +6309,7 @@ var InMemorySessionService = class {
     if (event.is_partial) {
       return event;
     }
-    if (_optionalChain([event, 'access', _176 => _176.actions, 'optionalAccess', _177 => _177.stateDelta])) {
+    if (_optionalChain([event, 'access', _181 => _181.actions, 'optionalAccess', _182 => _182.stateDelta])) {
       for (const [key, value] of Object.entries(event.actions.stateDelta)) {
         if (key.startsWith("_temp_")) {
           continue;
@@ -6195,7 +6415,7 @@ var PostgresSessionService = class {
   }
   async listSessions(userId, options) {
     let query = this.db.select().from(this.sessionsTable).where(_drizzleorm.eq.call(void 0, this.sessionsTable.userId, userId));
-    if (_optionalChain([options, 'optionalAccess', _178 => _178.limit]) !== void 0 && options.limit > 0) {
+    if (_optionalChain([options, 'optionalAccess', _183 => _183.limit]) !== void 0 && options.limit > 0) {
       query = query.limit(options.limit);
     }
     const results = await query;
@@ -6222,12 +6442,12 @@ var PostgresSessionService = class {
     if (event.is_partial) {
       return event;
     }
-    if (_optionalChain([event, 'access', _179 => _179.actions, 'optionalAccess', _180 => _180.stateDelta])) {
+    if (_optionalChain([event, 'access', _184 => _184.actions, 'optionalAccess', _185 => _185.stateDelta])) {
       for (const [key, value] of Object.entries(event.actions.stateDelta)) {
         if (key.startsWith("_temp_")) {
           continue;
         }
-        _optionalChain([session, 'access', _181 => _181.state, 'optionalAccess', _182 => _182.set, 'call', _183 => _183(key, value)]);
+        _optionalChain([session, 'access', _186 => _186.state, 'optionalAccess', _187 => _187.set, 'call', _188 => _188(key, value)]);
       }
     }
     if (!session.events) {
@@ -6258,11 +6478,11 @@ var sessionsSchema2 = _pgcore.pgTable.call(void 0, "sessions", {
   updatedAt: _pgcore.timestamp.call(void 0, "updated_at", { withTimezone: true }).defaultNow().notNull(),
   state: _pgcore.jsonb.call(void 0, "state").default("{}").$type()
 });
-var PgLiteSessionService = (_class12 = class {
+var PgLiteSessionService = (_class27 = class {
   
   
-  __init19() {this.initialized = false}
-  constructor(config) {;_class12.prototype.__init19.call(this);
+  __init38() {this.initialized = false}
+  constructor(config) {;_class27.prototype.__init38.call(this);
     this.db = _pglite.drizzle.call(void 0, config.pglite, {
       schema: { sessions: sessionsSchema2 }
     });
@@ -6371,7 +6591,7 @@ var PgLiteSessionService = (_class12 = class {
   async listSessions(userId, options) {
     await this.ensureInitialized();
     let query = this.db.select().from(this.sessionsTable).where(_drizzleorm.eq.call(void 0, this.sessionsTable.userId, userId));
-    if (_optionalChain([options, 'optionalAccess', _184 => _184.limit]) !== void 0 && options.limit > 0) {
+    if (_optionalChain([options, 'optionalAccess', _189 => _189.limit]) !== void 0 && options.limit > 0) {
       query = query.limit(options.limit);
     }
     const results = await query;
@@ -6394,12 +6614,12 @@ var PgLiteSessionService = (_class12 = class {
     if (event.is_partial) {
       return event;
     }
-    if (_optionalChain([event, 'access', _185 => _185.actions, 'optionalAccess', _186 => _186.stateDelta])) {
+    if (_optionalChain([event, 'access', _190 => _190.actions, 'optionalAccess', _191 => _191.stateDelta])) {
       for (const [key, value] of Object.entries(event.actions.stateDelta)) {
         if (key.startsWith("_temp_")) {
           continue;
         }
-        _optionalChain([session, 'access', _187 => _187.state, 'optionalAccess', _188 => _188.set, 'call', _189 => _189(key, value)]);
+        _optionalChain([session, 'access', _192 => _192.state, 'optionalAccess', _193 => _193.set, 'call', _194 => _194(key, value)]);
       }
     }
     if (!session.events) {
@@ -6410,7 +6630,7 @@ var PgLiteSessionService = (_class12 = class {
     await this.updateSession(session);
     return event;
   }
-}, _class12);
+}, _class27);
 
 // src/sessions/sqlite-session-service.ts
 
@@ -6430,12 +6650,12 @@ var sessionsSchema3 = _sqlitecore.sqliteTable.call(void 0, "sessions", {
   updatedAt: _sqlitecore.integer.call(void 0, "updated_at", { mode: "timestamp" }).notNull(),
   state: _sqlitecore.text.call(void 0, "state", { mode: "json" }).default("{}").$type()
 });
-var SqliteSessionService = (_class13 = class {
+var SqliteSessionService = (_class28 = class {
   
   
-  __init20() {this.initialized = false}
+  __init39() {this.initialized = false}
   
-  constructor(config) {;_class13.prototype.__init20.call(this);
+  constructor(config) {;_class28.prototype.__init39.call(this);
     this.sqliteInstance = config.sqlite;
     const dbPath = this.sqliteInstance.name;
     if (dbPath && dbPath !== ":memory:") {
@@ -6556,7 +6776,7 @@ var SqliteSessionService = (_class13 = class {
   async listSessions(userId, options) {
     await this.ensureInitialized();
     let query = this.db.select().from(this.sessionsTable).where(_drizzleorm.eq.call(void 0, this.sessionsTable.userId, userId));
-    if (_optionalChain([options, 'optionalAccess', _190 => _190.limit]) !== void 0 && options.limit > 0) {
+    if (_optionalChain([options, 'optionalAccess', _195 => _195.limit]) !== void 0 && options.limit > 0) {
       query = query.limit(options.limit);
     }
     const results = await query;
@@ -6579,12 +6799,12 @@ var SqliteSessionService = (_class13 = class {
     if (event.is_partial) {
       return event;
     }
-    if (_optionalChain([event, 'access', _191 => _191.actions, 'optionalAccess', _192 => _192.stateDelta])) {
+    if (_optionalChain([event, 'access', _196 => _196.actions, 'optionalAccess', _197 => _197.stateDelta])) {
       for (const [key, value] of Object.entries(event.actions.stateDelta)) {
         if (key.startsWith("_temp_")) {
           continue;
         }
-        _optionalChain([session, 'access', _193 => _193.state, 'optionalAccess', _194 => _194.set, 'call', _195 => _195(key, value)]);
+        _optionalChain([session, 'access', _198 => _198.state, 'optionalAccess', _199 => _199.set, 'call', _200 => _200(key, value)]);
       }
     }
     if (!session.events) {
@@ -6595,7 +6815,7 @@ var SqliteSessionService = (_class13 = class {
     await this.updateSession(session);
     return event;
   }
-}, _class13);
+}, _class28);
 
 // src/sessions/session-util.ts
 function generateSessionId() {
@@ -6631,7 +6851,7 @@ function cloneSession(session) {
 var _uuid = require('uuid');
 
 // src/events/event-actions.ts
-var EventActions = (_class14 = class {
+var EventActions = (_class29 = class {
   /**
    * If true, it won't call model to summarize function response.
    * Only used for function_response event.
@@ -6640,12 +6860,12 @@ var EventActions = (_class14 = class {
   /**
    * Indicates that the event is updating the state with the given delta.
    */
-  __init21() {this.stateDelta = {}}
+  __init40() {this.stateDelta = {}}
   /**
    * Indicates that the event is updating an artifact. key is the filename,
    * value is the version.
    */
-  __init22() {this.artifactDelta = {}}
+  __init41() {this.artifactDelta = {}}
   /**
    * If set, the event transfers to the specified agent.
    */
@@ -6657,21 +6877,21 @@ var EventActions = (_class14 = class {
   /**
    * Constructor for EventActions
    */
-  constructor(options = {}) {;_class14.prototype.__init21.call(this);_class14.prototype.__init22.call(this);
+  constructor(options = {}) {;_class29.prototype.__init40.call(this);_class29.prototype.__init41.call(this);
     this.skipSummarization = options.skipSummarization;
     this.stateDelta = options.stateDelta || {};
     this.artifactDelta = options.artifactDelta || {};
     this.transferToAgent = options.transferToAgent;
     this.escalate = options.escalate;
   }
-}, _class14);
+}, _class29);
 
 // src/events/event.ts
-var Event = (_class15 = class _Event extends LLMResponse {
+var Event = (_class30 = class _Event extends LLMResponse {
   /**
    * The invocation ID of the event.
    */
-  __init23() {this.invocationId = ""}
+  __init42() {this.invocationId = ""}
   /**
    * 'user' or the name of the agent, indicating who appended the event to the session.
    */
@@ -6679,7 +6899,7 @@ var Event = (_class15 = class _Event extends LLMResponse {
   /**
    * The actions taken by the agent.
    */
-  __init24() {this.actions = new EventActions()}
+  __init43() {this.actions = new EventActions()}
   /**
    * Set of ids of the long running function calls.
    * Agent client will know from this field about which function call is long running.
@@ -6697,7 +6917,7 @@ var Event = (_class15 = class _Event extends LLMResponse {
   /**
    * The unique identifier of the event.
    */
-  __init25() {this.id = ""}
+  __init44() {this.id = ""}
   /**
    * The timestamp of the event.
    */
@@ -6727,7 +6947,7 @@ var Event = (_class15 = class _Event extends LLMResponse {
       role,
       is_partial: partial,
       raw_response
-    });_class15.prototype.__init23.call(this);_class15.prototype.__init24.call(this);_class15.prototype.__init25.call(this);;
+    });_class30.prototype.__init42.call(this);_class30.prototype.__init43.call(this);_class30.prototype.__init44.call(this);;
     this.invocationId = invocationId;
     this.author = author;
     this.actions = actions;
@@ -6761,7 +6981,7 @@ var Event = (_class15 = class _Event extends LLMResponse {
   static newId() {
     return _uuid.v4.call(void 0, ).substring(0, 8);
   }
-}, _class15);
+}, _class30);
 
 // src/runners.ts
 var Runner = class {