@iqai/adk 0.0.5 → 0.0.7

package/dist/index.mjs

@@ -19,10 +19,47 @@ var __copyProps = (to, from, except, desc) => {
 };
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
+// 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 {
+      name;
+      isDebugEnabled = isDebugEnabled();
+      constructor({ name }) {
+        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);
+      }
+    };
+  }
+});
+
 // src/tools/base/base-tool.ts
-var BaseTool;
+var logger2, BaseTool;
 var init_base_tool = __esm({
   "src/tools/base/base-tool.ts"() {
+    init_logger();
+    logger2 = new Logger({ name: "BaseTool" });
     BaseTool = class {
       /**
        * Name of the tool
@@ -112,11 +149,9 @@ var init_base_tool = __esm({
         while (attempts <= (this.shouldRetryOnFailure ? this.maxRetryAttempts : 0)) {
           try {
             if (attempts > 0) {
-              if (process.env.DEBUG === "true") {
-                console.log(
-                  `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,
                 this.maxRetryDelay
@@ -459,7 +494,12 @@ var BaseAgent = class {
   }
 };
 
+// src/agents/llm-agent.ts
+init_logger();
+
 // src/models/llm-registry.ts
+init_logger();
+var logger = new Logger({ name: "LLMRegistry" });
 var LLMRegistry = class _LLMRegistry {
   /**
    * Map of model name regex to LLM class
@@ -516,12 +556,10 @@ var LLMRegistry = class _LLMRegistry {
    * Logs all registered models for debugging
    */
   static logRegisteredModels() {
-    if (process.env.DEBUG === "true") {
-      console.log("Registered LLM models:");
-      for (const [regex, llmClass] of _LLMRegistry.llmRegistry.entries()) {
-        console.log(`  - Pattern: ${regex.toString()}`);
-      }
-    }
+    logger.debug(
+      "Registered LLM models:",
+      [..._LLMRegistry.llmRegistry.entries()].map(([regex]) => regex.toString())
+    );
   }
 };
 
@@ -887,6 +925,7 @@ var Agent = class extends BaseAgent {
    * The minimum relevance score for memory augmentation (0-1)
    */
   memoryRelevanceThreshold;
+  logger = new Logger({ name: "LlmAgent" });
   /**
    * Constructor for Agent
    */
@@ -919,9 +958,7 @@ var Agent = class extends BaseAgent {
    */
   async executeTool(toolCall, context) {
     const { name, arguments: argsString } = toolCall.function;
-    if (process.env.DEBUG === "true") {
-      console.log(`Executing tool: ${name}`);
-    }
+    this.logger.debug(`Executing tool: ${name}`);
     const tool = this.findTool(name);
     if (!tool) {
       console.warn(`Tool '${name}' not found`);
@@ -939,9 +976,7 @@ var Agent = class extends BaseAgent {
       toolContext.toolName = name;
       toolContext.toolId = toolCall.id;
       const result = await tool.runAsync(args, toolContext);
-      if (process.env.DEBUG === "true") {
-        console.log(`Tool ${name} execution complete`);
-      }
+      this.logger.debug(`Tool ${name} execution complete`);
       return {
         name,
         result: typeof result === "string" ? result : JSON.stringify(result)
@@ -1110,10 +1145,7 @@ ${relevantInfo.join("\n\n")}`
       let stepCount = 0;
       while (stepCount < this.maxToolExecutionSteps) {
         stepCount++;
-        if (process.env.DEBUG === "true") {
-          console.log(`
-[Agent] Step ${stepCount}: Thinking...`);
-        }
+        this.logger.debug(`Step ${stepCount}: Thinking...`);
         const llmRequest = new LLMRequest({
           messages: context.messages,
           config: {
@@ -1130,9 +1162,9 @@ ${relevantInfo.join("\n\n")}`
           throw new Error("No response from LLM");
         }
         if (currentResponse.tool_calls && currentResponse.tool_calls.length > 0) {
-          if (process.env.DEBUG === "true") {
-            console.log("[Agent] Executing tools...");
-          }
+          this.logger.debug(
+            `Tool calls: ${JSON.stringify(currentResponse.tool_calls)}`
+          );
           context.addMessage({
             role: "assistant",
             content: currentResponse.content || "",
@@ -1151,9 +1183,7 @@ ${relevantInfo.join("\n\n")}`
             });
           }
         } else {
-          if (process.env.DEBUG === "true") {
-            console.log("[Agent] No tool calls, finishing...");
-          }
+          this.logger.debug("No tool calls, finishing...");
           context.addMessage({
             role: "assistant",
             content: currentResponse.content || ""
@@ -1191,10 +1221,7 @@ ${relevantInfo.join("\n\n")}`
     let stepCount = 0;
     let hadToolCalls = false;
     while (stepCount < this.maxToolExecutionSteps) {
-      if (process.env.DEBUG === "true") {
-        console.log(`
-[Agent] Step ${stepCount + 1}: Thinking...`);
-      }
+      this.logger.debug(`Step ${stepCount}: Thinking...`);
       const toolDeclarations = this.tools.map((tool) => tool.getDeclaration()).filter((declaration) => declaration !== null);
       const request = {
         messages: context.messages,
@@ -1223,14 +1250,10 @@ ${relevantInfo.join("\n\n")}`
         function_call: finalResponse.function_call
       });
       if (!hadToolCalls) {
-        if (process.env.DEBUG === "true") {
-          console.log("[Agent] No tool calls, finishing...");
-        }
+        this.logger.debug("No tool calls, finishing...");
         break;
       }
-      if (process.env.DEBUG === "true") {
-        console.log("[Agent] Executing tools...");
-      }
+      this.logger.debug(`Step ${stepCount + 1}: Executing tools...`);
       stepCount++;
       if (finalResponse.function_call) {
         const toolCall = {
@@ -1247,11 +1270,9 @@ ${relevantInfo.join("\n\n")}`
           content: JSON.stringify(result.result)
         });
       } else if (finalResponse.tool_calls && finalResponse.tool_calls.length > 0) {
-        if (process.env.DEBUG === "true") {
-          console.log(
-            `[Agent] 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({
           role: "assistant",
@@ -1276,7 +1297,9 @@ ${relevantInfo.join("\n\n")}`
 };
 
 // src/agents/sequential-agent.ts
+init_logger();
 var SequentialAgent = class extends BaseAgent {
+  logger = new Logger({ name: "SequentialAgent" });
   /**
    * Constructor for SequentialAgent
    */
@@ -1296,11 +1319,9 @@ var SequentialAgent = class extends BaseAgent {
    * Executes sub-agents sequentially, passing output from one to the next
    */
   async run(options) {
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `[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 {
         content: "No sub-agents defined for sequential execution.",
@@ -1316,11 +1337,9 @@ var SequentialAgent = class extends BaseAgent {
     let finalResponse = null;
     for (let i = 0; i < this.subAgents.length; i++) {
       const agent = this.subAgents[i];
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `[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({
           messages: currentMessages,
@@ -1335,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",
@@ -1376,11 +1392,9 @@ var SequentialAgent = class extends BaseAgent {
    * Streams responses from each sub-agent in sequence
    */
   async *runStreaming(options) {
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `[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 {
         content: "No sub-agents defined for sequential execution.",
@@ -1396,11 +1410,9 @@ var SequentialAgent = class extends BaseAgent {
     const currentMessages = [...options.messages];
     for (let i = 0; i < this.subAgents.length; i++) {
       const agent = this.subAgents[i];
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `[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({
           messages: currentMessages,
@@ -1432,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",
@@ -1453,7 +1462,9 @@ var SequentialAgent = class extends BaseAgent {
 };
 
 // src/agents/parallel-agent.ts
+init_logger();
 var ParallelAgent = class extends BaseAgent {
+  logger = new Logger({ name: "ParallelAgent" });
   /**
    * Constructor for ParallelAgent
    */
@@ -1473,11 +1484,9 @@ var ParallelAgent = class extends BaseAgent {
    * Executes all sub-agents in parallel
    */
   async run(options) {
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `[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 {
         content: "No sub-agents defined for parallel execution.",
@@ -1489,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"
@@ -1520,11 +1526,9 @@ ${result.content || "No content"}
    * Collects streaming responses from all sub-agents
    */
   async *runStreaming(options) {
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `[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 {
         content: "No sub-agents defined for parallel execution.",
@@ -1537,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"
@@ -1590,6 +1591,7 @@ ${response.content || "No content"}
 };
 
 // src/agents/loop-agent.ts
+init_logger();
 var LoopAgent = class extends BaseAgent {
   /**
    * Maximum number of iterations to prevent infinite loops
@@ -1603,6 +1605,7 @@ var LoopAgent = class extends BaseAgent {
    * Custom condition check function
    */
   conditionCheck;
+  logger = new Logger({ name: "LoopAgent" });
   /**
    * Constructor for LoopAgent
    */
@@ -1630,28 +1633,20 @@ var LoopAgent = class extends BaseAgent {
    */
   async shouldContinue(response, iterationCount, messages, config) {
     if (iterationCount >= this.maxIterations) {
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `[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);
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `[LoopAgent] Custom condition check result: ${shouldContinue}`
-        );
-      }
+      this.logger.debug(`Custom condition check result: ${shouldContinue}`);
       return shouldContinue;
     }
     if (this.conditionAgent) {
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `[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,
         {
@@ -1670,14 +1665,12 @@ var LoopAgent = class extends BaseAgent {
         });
         const content = conditionResponse.content?.toLowerCase() || "";
         const shouldContinue = content.includes("yes") && !content.includes("no");
-        if (process.env.DEBUG === "true") {
-          console.log(
-            `[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;
       }
     }
@@ -1688,11 +1681,9 @@ var LoopAgent = class extends BaseAgent {
    * Executes the sub-agent in a loop until the condition is met
    */
   async run(options) {
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `[LoopAgent] Starting loop with max ${this.maxIterations} iterations`
-      );
-    }
+    this.logger.debug(
+      `Starting loop with max ${this.maxIterations} iterations`
+    );
     if (this.subAgents.length === 0) {
       return {
         content: "No sub-agent defined for loop execution.",
@@ -1706,11 +1697,9 @@ var LoopAgent = class extends BaseAgent {
     let shouldContinueLoop = true;
     while (shouldContinueLoop && iterationCount < this.maxIterations) {
       iterationCount++;
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `[LoopAgent] Running iteration ${iterationCount}/${this.maxIterations}`
-        );
-      }
+      this.logger.debug(
+        `Running iteration ${iterationCount}/${this.maxIterations}`
+      );
       try {
         const response = await subAgent.run({
           messages: currentMessages,
@@ -1734,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;
       }
     }
@@ -1758,11 +1744,9 @@ ${lastResponse.content || ""}`,
    * Runs the agent with streaming support
    */
   async *runStreaming(options) {
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `[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 {
         content: "No sub-agent defined for loop execution.",
@@ -1781,11 +1765,9 @@ ${lastResponse.content || ""}`,
     };
     while (shouldContinueLoop && iterationCount < this.maxIterations) {
       iterationCount++;
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `[LoopAgent] Running iteration ${iterationCount}/${this.maxIterations} (streaming)`
-        );
-      }
+      this.logger.debug(
+        `Running iteration ${iterationCount}/${this.maxIterations} (streaming)`
+      );
       yield {
         content: `Running iteration ${iterationCount}/${this.maxIterations}...`,
         role: "assistant",
@@ -1809,11 +1791,9 @@ ${lastResponse.content || ""}`,
           }
         }
         if (!lastChunk) {
-          if (process.env.DEBUG === "true") {
-            console.warn(
-              `[LoopAgent] No complete chunk received from iteration ${iterationCount}`
-            );
-          }
+          this.logger.debug(
+            `No complete chunk received from iteration ${iterationCount}`
+          );
           shouldContinueLoop = false;
           continue;
         }
@@ -1839,9 +1819,8 @@ ${lastResponse.content || ""}`,
           };
         }
       } catch (error) {
-        console.error(
-          `[LoopAgent] Error in loop iteration ${iterationCount}:`,
-          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)}`,
@@ -1858,6 +1837,7 @@ ${lastResponse.content || ""}`,
 };
 
 // src/agents/lang-graph-agent.ts
+init_logger();
 var LangGraphAgent = class extends BaseAgent {
   /**
    * Graph nodes (agents and their connections)
@@ -1875,6 +1855,7 @@ var LangGraphAgent = class extends BaseAgent {
    * Results from node executions
    */
   results = [];
+  logger = new Logger({ name: "LangGraphAgent" });
   /**
    * Constructor for LangGraphAgent
    */
@@ -1949,17 +1930,13 @@ var LangGraphAgent = 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) {
-          if (process.env.DEBUG === "true") {
-            console.log(
-              `[LangGraphAgent] Skipping node "${targetName}" due to condition`
-            );
-          }
+          this.logger.debug(`Skipping node "${targetName}" due to condition`);
           continue;
         }
       }
@@ -1984,11 +1961,7 @@ var LangGraphAgent = class extends BaseAgent {
     };
     const shouldExecute = await node.condition(mockResponse, mockContext);
     if (!shouldExecute) {
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `[LangGraphAgent] Skipping node "${targetName}" due to condition`
-        );
-      }
+      this.logger.debug(`Skipping node "${targetName}" due to condition`);
     }
     return { shouldExecute };
   }
@@ -2001,11 +1974,9 @@ var LangGraphAgent = class extends BaseAgent {
       messages: options.messages,
       config: options.config
     });
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `[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 {
         content: "No nodes defined in the graph.",
@@ -2025,11 +1996,7 @@ var LangGraphAgent = class extends BaseAgent {
     while (nodesToExecute.length > 0 && stepCount < this.maxSteps) {
       stepCount++;
       const { node, messages } = nodesToExecute.shift();
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `[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({
@@ -2077,7 +2044,7 @@ var LangGraphAgent = 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"
@@ -2101,11 +2068,9 @@ var LangGraphAgent = class extends BaseAgent {
       messages: options.messages,
       config: options.config
     });
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `[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 {
         content: "No nodes defined in the graph.",
@@ -2132,11 +2097,9 @@ var LangGraphAgent = class extends BaseAgent {
     while (nodesToExecute.length > 0 && stepCount < this.maxSteps) {
       stepCount++;
       const { node, messages } = nodesToExecute.shift();
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `[LangGraphAgent] Step ${stepCount}: Executing node "${node.name}" (streaming)`
-        );
-      }
+      this.logger.debug(
+        `Step ${stepCount}: Executing node "${node.name}" (streaming)`
+      );
       executedNodes.push(node.name);
       try {
         const result = await node.agent.run({
@@ -2190,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"
@@ -2250,8 +2213,10 @@ function createFunctionTool(func, options) {
 init_function_utils();
 
 // src/tools/common/google-search.ts
+init_logger();
 init_base_tool();
 var GoogleSearch = class extends BaseTool {
+  logger = new Logger({ name: "GoogleSearch" });
   /**
    * Constructor for GoogleSearch
    */
@@ -2290,9 +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) {
-    if (process.env.DEBUG === "true") {
-      console.log(`Executing Google search for: ${args.query}`);
-    }
+    this.logger.debug(
+      `[GoogleSearch] Executing Google search for: ${args.query}`
+    );
     return {
       results: [
         {
@@ -2755,8 +2720,10 @@ var UserInteractionTool = class extends BaseTool {
 };
 
 // src/tools/common/exit-loop-tool.ts
+init_logger();
 init_base_tool();
 var ExitLoopTool = class extends BaseTool {
+  logger = new Logger({ name: "ExitLoopTool" });
   /**
    * Constructor for ExitLoopTool
    */
@@ -2784,9 +2751,7 @@ var ExitLoopTool = class extends BaseTool {
    * Execute the exit loop action
    */
   async runAsync(_args, context) {
-    if (process.env.DEBUG === "true") {
-      console.log("Executing exit loop tool");
-    }
+    this.logger.debug("Executing exit loop tool");
     if (context.actions) {
       context.actions.escalate = true;
     } else {
@@ -2801,8 +2766,10 @@ var ExitLoopTool = class extends BaseTool {
 };
 
 // src/tools/common/get-user-choice-tool.ts
+init_logger();
 init_base_tool();
 var GetUserChoiceTool = class extends BaseTool {
+  logger = new Logger({ name: "GetUserChoiceTool" });
   /**
    * Constructor for GetUserChoiceTool
    */
@@ -2845,13 +2812,11 @@ var GetUserChoiceTool = class extends BaseTool {
    * and the actual choice will be provided asynchronously
    */
   async runAsync(args, context) {
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `Executing get_user_choice with options: ${args.options.join(", ")}`
-      );
-      if (args.question) {
-        console.log(`Question: ${args.question}`);
-      }
+    this.logger.debug(
+      `Executing get_user_choice with options: ${args.options.join(", ")}`
+    );
+    if (args.question) {
+      this.logger.debug(`Question: ${args.question}`);
     }
     if (context.actions) {
       context.actions.skip_summarization = true;
@@ -2865,8 +2830,10 @@ var GetUserChoiceTool = class extends BaseTool {
 };
 
 // src/tools/common/transfer-to-agent-tool.ts
+init_logger();
 init_base_tool();
 var TransferToAgentTool = class extends BaseTool {
+  logger = new Logger({ name: "TransferToAgentTool" });
   /**
    * Constructor for TransferToAgentTool
    */
@@ -2899,9 +2866,7 @@ var TransferToAgentTool = class extends BaseTool {
    * Execute the transfer to agent action
    */
   async runAsync(args, context) {
-    if (process.env.DEBUG === "true") {
-      console.log(`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 {
@@ -2916,8 +2881,10 @@ var TransferToAgentTool = class extends BaseTool {
 };
 
 // src/tools/common/load-memory-tool.ts
+init_logger();
 init_base_tool();
 var LoadMemoryTool = class extends BaseTool {
+  logger = new Logger({ name: "LoadMemoryTool" });
   /**
    * Constructor for LoadMemoryTool
    */
@@ -2950,9 +2917,7 @@ var LoadMemoryTool = class extends BaseTool {
    * Execute the memory loading action
    */
   async runAsync(args, context) {
-    if (process.env.DEBUG === "true") {
-      console.log(`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",
@@ -2976,9 +2941,11 @@ var LoadMemoryTool = class extends BaseTool {
 };
 
 // src/tools/mcp/client.ts
+init_logger();
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
 import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import { CreateMessageRequestSchema } from "@modelcontextprotocol/sdk/types.js";
 
 // src/tools/mcp/types.ts
 var McpErrorType = /* @__PURE__ */ ((McpErrorType2) => {
@@ -2987,6 +2954,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 {
@@ -3034,8 +3003,11 @@ var McpClientService = class {
   client = null;
   transport = null;
   isClosing = false;
+  samplingHandler = null;
+  logger = new Logger({ name: "McpClientService" });
   constructor(config) {
     this.config = config;
+    this.samplingHandler = config.samplingHandler || null;
   }
   /**
    * Initializes and returns an MCP client based on configuration.
@@ -3084,6 +3056,7 @@ var McpClientService = class {
       } else {
         await connectPromise;
       }
+      await this.setupSamplingHandler(client);
       if (this.config.debug) {
         console.log("\u2705 MCP client connected successfully");
       }
@@ -3235,9 +3208,77 @@ var McpClientService = class {
   isConnected() {
     return !!this.client && !this.isClosing;
   }
+  async setupSamplingHandler(client) {
+    if (!this.samplingHandler) {
+      if (this.config.debug) {
+        console.log(
+          "\u26A0\uFE0F No sampling handler provided - sampling requests will be rejected"
+        );
+      }
+      return;
+    }
+    client.setRequestHandler(CreateMessageRequestSchema, async (request) => {
+      try {
+        this.logger.debug("Received sampling request:", request);
+        const samplingRequest = request.params;
+        if (!samplingRequest.messages || !Array.isArray(samplingRequest.messages)) {
+          throw new McpError(
+            "Invalid sampling request: messages array is required",
+            "INVALID_REQUEST_ERROR" /* INVALID_REQUEST_ERROR */
+          );
+        }
+        if (!samplingRequest.maxTokens || samplingRequest.maxTokens <= 0) {
+          throw new McpError(
+            "Invalid sampling request: maxTokens must be a positive number",
+            "INVALID_REQUEST_ERROR" /* INVALID_REQUEST_ERROR */
+          );
+        }
+        const response = await this.samplingHandler({
+          method: request.method,
+          params: request.params
+        });
+        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");
+    }
+  }
+  setSamplingHandler(handler) {
+    this.samplingHandler = handler;
+    if (this.client) {
+      this.setupSamplingHandler(this.client).catch((error) => {
+        console.error("Failed to update sampling handler:", error);
+      });
+    }
+  }
+  removeSamplingHandler() {
+    this.samplingHandler = null;
+    if (this.client) {
+      try {
+        this.client.removeRequestHandler?.("sampling/createMessage");
+      } catch (error) {
+        console.error("Failed to remove sampling handler:", error);
+      }
+    }
+  }
 };
 
 // src/tools/mcp/create-tool.ts
+init_logger();
 init_base_tool();
 
 // src/tools/mcp/schema-conversion.ts
@@ -3440,6 +3481,7 @@ var McpToolAdapter = class extends BaseTool {
   mcpTool;
   client;
   clientService = null;
+  logger = new Logger({ name: "McpToolAdapter" });
   constructor(mcpTool, client) {
     const metadata = mcpTool.metadata || {};
     super({
@@ -3472,9 +3514,7 @@ var McpToolAdapter = class extends BaseTool {
     }
   }
   async runAsync(args, _context) {
-    if (process.env.DEBUG === "true") {
-      console.log(`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);
@@ -3794,9 +3834,11 @@ var BaseLLMConnection = class {
 };
 
 // src/models/anthropic-llm.ts
+init_logger();
 import axios from "axios";
 
 // src/models/anthropic-llm-connection.ts
+init_logger();
 var AnthropicLLMConnection = class extends BaseLLMConnection {
   /**
    * Axios instance for API calls
@@ -3824,6 +3866,7 @@ var AnthropicLLMConnection = class extends BaseLLMConnection {
   responseCallback;
   errorCallback;
   endCallback;
+  logger = new Logger({ name: "AnthropicLlmConnection" });
   /**
    * Constructor
    */
@@ -3926,19 +3969,12 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
     if (!content?.length) return [];
     const toolUses = [];
     for (const block of content) {
-      if (process.env.DEBUG === "true") {
-        console.log(
-          "Connection - Processing content block of type:",
-          block.type
-        );
-      }
+      this.logger.debug(`Processing content block of type: ${block.type}`);
       if (block.type === "tool_use") {
-        if (process.env.DEBUG === "true") {
-          console.log(
-            "Connection - 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",
           name: block.name || "unknown-name",
@@ -3946,14 +3982,12 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
         });
       }
     }
-    if (process.env.DEBUG === "true") {
-      console.log(`Connection - Found ${toolUses.length} tool uses in content`);
-      if (toolUses.length > 0) {
-        console.log(
-          "Connection - Extracted tool uses:",
-          JSON.stringify(toolUses, null, 2)
-        );
-      }
+    this.logger.debug(`Found ${toolUses.length} tool uses in content`);
+    if (toolUses.length > 0) {
+      this.logger.debug(
+        "Extracted tool uses:",
+        JSON.stringify(toolUses, null, 2)
+      );
     }
     return toolUses;
   }
@@ -4047,43 +4081,27 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
       }
       const toolUses = this.extractToolUses(apiResponse.content);
       const toolCalls = this.convertToolCalls(toolUses);
-      if (process.env.DEBUG === "true") {
-        if (toolUses.length > 0) {
-          console.log(
-            "Connection - Extracted Tool Uses:",
-            JSON.stringify(toolUses, null, 2)
-          );
-          console.log(
-            "Connection - Converted Tool Calls:",
-            JSON.stringify(toolCalls, null, 2)
-          );
-        }
-      }
+      this.logger.debug(
+        `- Extracted ${toolUses.length} tool uses in content and converted ${toolCalls?.length || 0} tool calls`
+      );
       const llmResponse = new LLMResponse({
         role: "assistant",
         content,
         tool_calls: toolCalls?.length ? toolCalls : void 0,
         raw_response: apiResponse
       });
-      if (process.env.DEBUG === "true") {
-        console.log(
-          "Connection - Final LLMResponse object:",
-          JSON.stringify(
-            {
-              role: llmResponse.role,
-              content: llmResponse.content?.substring(0, 50) + (llmResponse.content && llmResponse.content.length > 50 ? "..." : ""),
-              tool_calls: llmResponse.tool_calls ? `[${llmResponse.tool_calls.length} calls]` : "undefined"
-            },
-            null,
-            2
-          )
-        );
-      }
+      const logObject = {
+        role: llmResponse.role,
+        content: llmResponse.content?.substring(0, 50) + (llmResponse.content && llmResponse.content.length > 50 ? "..." : ""),
+        tool_calls: llmResponse.tool_calls ? `[${llmResponse.tool_calls.length} calls]` : "undefined"
+      };
+      this.logger.debug(
+        "Final LLMResponse object:",
+        JSON.stringify(logObject, null, 2)
+      );
       return llmResponse;
     } catch (error) {
-      if (process.env.DEBUG === "true") {
-        console.error("Error sending message to Anthropic:", error);
-      }
+      this.logger.debug("Error sending message to Anthropic:", error);
       throw error;
     }
   }
@@ -4103,6 +4121,7 @@ var AnthropicLLM = class extends BaseLLM {
    * Default parameters for requests
    */
   defaultParams;
+  logger = new Logger({ name: "AnthropicLLM" });
   /**
    * Constructor for AnthropicLLM
    */
@@ -4235,13 +4254,11 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
     if (!content?.length) return [];
     const toolUses = [];
     for (const block of content) {
-      if (process.env.DEBUG === "true") {
-        console.log("Processing content block of type:", block.type);
-      }
+      this.logger.debug(`Processing content block of type: ${block.type}`);
       if (block.type === "tool_use") {
-        if (process.env.DEBUG === "true") {
-          console.log("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",
           name: block.name || "unknown-name",
@@ -4249,12 +4266,10 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
         });
       }
     }
-    if (process.env.DEBUG === "true") {
-      console.log(`Found ${toolUses.length} tool uses in content`);
-      if (toolUses.length > 0) {
-        console.log("Extracted tool uses:", JSON.stringify(toolUses, null, 2));
-      }
-    }
+    this.logger.debug(
+      `Found ${toolUses.length} tool uses in content`,
+      toolUses
+    );
     return toolUses;
   }
   /**
@@ -4276,16 +4291,14 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
         },
         responseType: stream ? "stream" : "json"
       });
-      if (process.env.DEBUG === "true") {
-        console.log("Anthropic API Response Status:", response.status);
-        if (!stream) {
-          console.log("Response Data Structure:", Object.keys(response.data));
-          console.log(
-            "Response Content Structure:",
-            response.data.content.map((block) => ({ type: block.type }))
-          );
-        }
-      }
+      this.logger.debug(
+        `API Response done with ${response.status}:`,
+        response.data
+      );
+      this.logger.debug(
+        "API Response content:",
+        response.data.content.map((block) => ({ type: block.type }))
+      );
       return response.data;
     } catch (error) {
       console.error("Error calling Anthropic API:", error);
@@ -4311,24 +4324,17 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
         top_p: llmRequest.config.top_p ?? this.defaultParams.top_p,
         tools: tools?.length ? tools : void 0
       };
-      if (process.env.DEBUG === "true") {
-        console.log("Anthropic API Request:", {
-          model: params.model,
-          messageCount: params.messages.length,
-          systemMessage: params.system ? "present" : "none",
-          tools: params.tools ? params.tools.map((t) => t.name) : "none"
-        });
-      }
+      this.logger.debug("API Request:", {
+        model: params.model,
+        messageCount: params.messages.length,
+        systemMessage: params.system ? "present" : "none",
+        tools: params.tools ? params.tools.map((t) => t.name) : "none"
+      });
       if (stream) {
         throw new Error("Streaming is not supported in this implementation");
       }
       const response = await this.callAnthropicAPI(params);
-      if (process.env.DEBUG === "true") {
-        console.log(
-          "Full Response Content:",
-          JSON.stringify(response.content, null, 2)
-        );
-      }
+      this.logger.debug("Full Response Content:", response.content);
       let content = "";
       for (const block of response.content) {
         if (block.type === "text") {
@@ -4337,43 +4343,26 @@ ${typeof message.content === "string" ? message.content : JSON.stringify(message
       }
       const toolUses = this.extractToolUses(response.content);
       const toolCalls = this.convertToolUses(toolUses);
-      if (process.env.DEBUG === "true") {
-        if (toolUses.length > 0) {
-          console.log(
-            "Extracted Tool Uses:",
-            JSON.stringify(toolUses, null, 2)
-          );
-          console.log(
-            "Converted Tool Calls:",
-            JSON.stringify(toolCalls, null, 2)
-          );
-        }
-      }
+      this.logger.debug("Extracted Tool Uses:", toolUses);
+      this.logger.debug("Converted Tool Calls:", toolCalls);
       const llmResponse = new LLMResponse({
         role: "assistant",
         content,
         tool_calls: toolCalls.length > 0 ? toolCalls : void 0,
         raw_response: response
       });
-      if (process.env.DEBUG === "true") {
-        console.log(
-          "Final LLMResponse object:",
-          JSON.stringify(
-            {
-              role: llmResponse.role,
-              content: llmResponse.content?.substring(0, 50) + (llmResponse.content && llmResponse.content.length > 50 ? "..." : ""),
-              tool_calls: llmResponse.tool_calls ? `[${llmResponse.tool_calls.length} calls]` : "undefined"
-            },
-            null,
-            2
-          )
-        );
-      }
+      const logObject = {
+        role: llmResponse.role,
+        content: llmResponse.content?.substring(0, 50) + (llmResponse.content && llmResponse.content.length > 50 ? "..." : ""),
+        tool_calls: llmResponse.tool_calls ? `[${llmResponse.tool_calls.length} calls]` : "undefined"
+      };
+      this.logger.debug(
+        "Final LLMResponse object:",
+        JSON.stringify(logObject, null, 2)
+      );
       yield llmResponse;
     } catch (error) {
-      if (process.env.DEBUG === "true") {
-        console.error("Error calling Anthropic:", error);
-      }
+      this.logger.debug("Error:", error);
       throw error;
     }
   }
@@ -4768,6 +4757,7 @@ var GoogleLLM = class extends BaseLLM {
 };
 
 // src/models/openai-llm.ts
+init_logger();
 import OpenAI from "openai";
 
 // src/models/openai-llm-connection.ts
@@ -5027,6 +5017,7 @@ var OpenAILLM = class extends BaseLLM {
    * Default parameters for requests
    */
   defaultParams;
+  logger = new Logger({ name: "OpenAILLM" });
   /**
    * Constructor for OpenAILLM
    */
@@ -5176,11 +5167,9 @@ var OpenAILLM = class extends BaseLLM {
    * Convert OpenAI streaming chunk to LLMResponse
    */
   convertChunk(chunk) {
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `OpenAI: Converting chunk - delta: ${JSON.stringify(chunk.delta || {})}`
-      );
-    }
+    this.logger.debug(
+      `Converting chunk - delta: ${JSON.stringify(chunk.delta || {})}`
+    );
     const content = chunk.delta?.content;
     const result = new LLMResponse({
       content: content !== void 0 ? content : null,
@@ -5221,32 +5210,24 @@ var OpenAILLM = class extends BaseLLM {
       presence_penalty: llmRequest.config.presence_penalty ?? this.defaultParams.presence_penalty,
       stream: shouldStream
     };
-    if (process.env.DEBUG === "true") {
-      console.log(
-        `OpenAI: Streaming mode ${shouldStream ? "enabled" : "disabled"}`
-      );
-    }
+    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) {
-        if (process.env.DEBUG === "true") {
-          console.log("OpenAI: 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;
-        if (process.env.DEBUG === "true") {
-          console.log("OpenAI: 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) {
-            if (process.env.DEBUG === "true") {
-              console.log("OpenAI: Empty chunk received, skipping");
-            }
+            this.logger.debug("Empty chunk received, skipping");
             continue;
           }
           const choice = chunk.choices[0];
@@ -5254,14 +5235,12 @@ var OpenAILLM = class extends BaseLLM {
           if (responseChunk.content !== null) {
             accumulatedContent += responseChunk.content;
           }
-          if (process.env.DEBUG === "true") {
-            console.log(
-              `OpenAI: Chunk received - delta: "${choice.delta?.content || ""}"`,
-              `responseChunk content: "${responseChunk.content || ""}"`,
-              `is_partial: ${responseChunk.is_partial}`,
-              `accumulated: "${accumulatedContent.substring(0, 30)}${accumulatedContent.length > 30 ? "..." : ""}"`
-            );
-          }
+          this.logger.debug(
+            `Chunk received - delta: "${choice.delta?.content || ""}"`,
+            `responseChunk content: "${responseChunk.content || ""}"`,
+            `is_partial: ${responseChunk.is_partial}`,
+            `accumulated: "${accumulatedContent.substring(0, 30)}${accumulatedContent.length > 30 ? "..." : ""}"`
+          );
           if (responseChunk.function_call) {
             if (!partialFunctionCall) {
               partialFunctionCall = {
@@ -5286,37 +5265,27 @@ var OpenAILLM = class extends BaseLLM {
             }
             responseChunk.tool_calls = Array.from(partialToolCalls.values());
           }
-          if (process.env.DEBUG === "true") {
-            console.log("OpenAI: Yielding chunk to caller");
-          }
+          this.logger.debug("Yielding chunk to caller");
           yield responseChunk;
         }
         if (accumulatedContent.length > 0) {
-          if (process.env.DEBUG === "true") {
-            console.log(
-              `OpenAI: 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,
             role: "assistant",
             is_partial: false
           });
         }
-        if (process.env.DEBUG === "true") {
-          console.log("OpenAI: Finished processing all stream chunks");
-        }
+        this.logger.debug("Finished processing all stream chunks");
       } else {
-        if (process.env.DEBUG === "true") {
-          console.log("OpenAI: 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");
         }
-        if (process.env.DEBUG === "true") {
-          console.log("OpenAI: Non-streaming response received");
-        }
+        this.logger.debug("Non-streaming response received");
         yield this.convertResponse(response.choices[0]);
       }
     } catch (error) {
@@ -5903,6 +5872,7 @@ var InMemoryMemoryService = class {
 };
 
 // src/memory/persistent-memory-service.ts
+init_logger();
 import fs2 from "fs";
 import path2 from "path";
 var PersistentMemoryService = class {
@@ -5918,6 +5888,7 @@ var PersistentMemoryService = class {
    * File prefix for memory files
    */
   filePrefix;
+  logger = new Logger({ name: "PersistentMemoryService" });
   /**
    * Constructor for PersistentMemoryService
    */
@@ -6006,11 +5977,9 @@ var PersistentMemoryService = class {
           }
         }
       }
-      if (process.env.DEBUG === "true") {
-        console.log(
-          `Loaded ${this.inMemoryService.getAllSessions().length} sessions from persistent storage`
-        );
-      }
+      this.logger.debug(
+        `Loaded ${this.inMemoryService.getAllSessions().length} sessions from persistent storage`
+      );
     } catch (error) {
       console.error("Error loading memory files:", error);
     }
@@ -6074,9 +6043,9 @@ __export(sessions_exports, {
   PgLiteSessionService: () => PgLiteSessionService,
   PostgresSessionService: () => PostgresSessionService,
   SessionState: () => SessionState,
+  SqliteSessionService: () => SqliteSessionService,
   cloneSession: () => cloneSession,
   generateSessionId: () => generateSessionId,
-  sessionsSchema: () => sessionsSchema2,
   validateSession: () => validateSession
 });
 
@@ -6519,6 +6488,191 @@ var PgLiteSessionService = class {
   }
 };
 
+// src/sessions/sqlite-session-service.ts
+import * as fs3 from "fs";
+import * as path3 from "path";
+import { eq as eq3 } from "drizzle-orm";
+import {
+  drizzle as drizzle2
+} from "drizzle-orm/better-sqlite3";
+import { integer, text } from "drizzle-orm/sqlite-core";
+import { sqliteTable } from "drizzle-orm/sqlite-core";
+var sessionsSchema3 = sqliteTable("sessions", {
+  id: text("id").primaryKey(),
+  userId: text("user_id").notNull(),
+  messages: text("messages", { mode: "json" }).default("[]").$type(),
+  metadata: text("metadata", { mode: "json" }).default("{}").$type(),
+  createdAt: integer("created_at", { mode: "timestamp" }).notNull(),
+  updatedAt: integer("updated_at", { mode: "timestamp" }).notNull(),
+  state: text("state", { mode: "json" }).default("{}").$type()
+});
+var SqliteSessionService = class {
+  db;
+  sessionsTable;
+  initialized = false;
+  sqliteInstance;
+  constructor(config) {
+    this.sqliteInstance = config.sqlite;
+    const dbPath = this.sqliteInstance.name;
+    if (dbPath && dbPath !== ":memory:") {
+      const dbDir = path3.dirname(dbPath);
+      if (!fs3.existsSync(dbDir)) {
+        fs3.mkdirSync(dbDir, { recursive: true });
+      }
+    }
+    this.db = drizzle2(config.sqlite, {
+      schema: { sessions: sessionsSchema3 }
+    });
+    this.sessionsTable = sessionsSchema3;
+    if (!config.skipTableCreation) {
+      this.initializeDatabase().catch((error) => {
+        console.error("Failed to initialize SQLite database:", error);
+      });
+    }
+  }
+  /**
+   * Initialize the database by creating required tables if they don't exist
+   */
+  async initializeDatabase() {
+    if (this.initialized) {
+      return;
+    }
+    try {
+      this.sqliteInstance.pragma("journal_mode = WAL");
+      this.sqliteInstance.exec(`
+				CREATE TABLE IF NOT EXISTS sessions (
+					id TEXT PRIMARY KEY,
+					user_id TEXT NOT NULL,
+					messages TEXT DEFAULT '[]',
+					metadata TEXT DEFAULT '{}',
+					created_at INTEGER NOT NULL,
+					updated_at INTEGER NOT NULL,
+					state TEXT DEFAULT '{}'
+				);
+			`);
+      this.sqliteInstance.exec(`
+				CREATE INDEX IF NOT EXISTS idx_sessions_user_id ON sessions(user_id);
+			`);
+      this.initialized = true;
+    } catch (error) {
+      console.error("Error initializing SQLite database:", error);
+      throw error;
+    }
+  }
+  /**
+   * Ensure database is initialized before any operation
+   */
+  async ensureInitialized() {
+    if (!this.initialized) {
+      await this.initializeDatabase();
+    }
+  }
+  generateSessionId() {
+    return `session-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`;
+  }
+  async createSession(userId, metadata = {}) {
+    await this.ensureInitialized();
+    const sessionId = this.generateSessionId();
+    const now = /* @__PURE__ */ new Date();
+    const sessionState = new SessionState();
+    const newSessionData = {
+      id: sessionId,
+      userId,
+      messages: [],
+      metadata,
+      createdAt: now,
+      updatedAt: now,
+      state: sessionState.toObject()
+    };
+    const results = await this.db.insert(this.sessionsTable).values(newSessionData).returning();
+    const result = results[0];
+    if (!result) {
+      throw new Error(
+        "Failed to create session, no data returned from insert."
+      );
+    }
+    return {
+      id: result.id,
+      userId: result.userId,
+      messages: Array.isArray(result.messages) ? result.messages : [],
+      metadata: result.metadata || {},
+      state: SessionState.fromObject(result.state || {}),
+      createdAt: result.createdAt,
+      updatedAt: result.updatedAt
+    };
+  }
+  async getSession(sessionId) {
+    await this.ensureInitialized();
+    const results = await this.db.select().from(this.sessionsTable).where(eq3(this.sessionsTable.id, sessionId)).limit(1);
+    const sessionData = results[0];
+    if (!sessionData) {
+      return void 0;
+    }
+    return {
+      id: sessionData.id,
+      userId: sessionData.userId,
+      messages: Array.isArray(sessionData.messages) ? sessionData.messages : [],
+      metadata: sessionData.metadata || {},
+      state: SessionState.fromObject(sessionData.state || {}),
+      createdAt: sessionData.createdAt,
+      updatedAt: sessionData.updatedAt
+    };
+  }
+  async updateSession(session) {
+    await this.ensureInitialized();
+    const updateData = {
+      userId: session.userId,
+      messages: session.messages,
+      metadata: session.metadata,
+      updatedAt: /* @__PURE__ */ new Date(),
+      state: session.state.toObject()
+    };
+    await this.db.update(this.sessionsTable).set(updateData).where(eq3(this.sessionsTable.id, session.id));
+  }
+  async listSessions(userId, options) {
+    await this.ensureInitialized();
+    let query = this.db.select().from(this.sessionsTable).where(eq3(this.sessionsTable.userId, userId));
+    if (options?.limit !== void 0 && options.limit > 0) {
+      query = query.limit(options.limit);
+    }
+    const results = await query;
+    return results.map((sessionData) => ({
+      id: sessionData.id,
+      userId: sessionData.userId,
+      messages: Array.isArray(sessionData.messages) ? sessionData.messages : [],
+      metadata: sessionData.metadata || {},
+      state: SessionState.fromObject(sessionData.state || {}),
+      createdAt: sessionData.createdAt,
+      updatedAt: sessionData.updatedAt
+    }));
+  }
+  async deleteSession(sessionId) {
+    await this.ensureInitialized();
+    await this.db.delete(this.sessionsTable).where(eq3(this.sessionsTable.id, sessionId));
+  }
+  async appendEvent(session, event) {
+    await this.ensureInitialized();
+    if (event.is_partial) {
+      return event;
+    }
+    if (event.actions?.stateDelta) {
+      for (const [key, value] of Object.entries(event.actions.stateDelta)) {
+        if (key.startsWith("_temp_")) {
+          continue;
+        }
+        session.state?.set(key, value);
+      }
+    }
+    if (!session.events) {
+      session.events = [];
+    }
+    session.events.push(event);
+    session.updatedAt = /* @__PURE__ */ new Date();
+    await this.updateSession(session);
+    return event;
+  }
+};
+
 // src/sessions/session-util.ts
 function generateSessionId() {
   return `session-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`;
@@ -6886,6 +7040,7 @@ export {
   SequentialAgent,
   SessionState,
   sessions_exports as Sessions,
+  SqliteSessionService,
   StreamingMode,
   ToolContext,
   tools_exports as Tools,
@@ -6902,6 +7057,5 @@ export {
   mcpSchemaToParameters,
   normalizeJsonSchema,
   registerProviders,
-  sessionsSchema2 as sessionsSchema,
   validateSession
 };