@iqai/adk 0.1.13 → 0.1.15

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 _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 _class31; var _class32;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 _class31; var _class32; var _class33;var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
@@ -323,15 +323,17 @@ var init_base_tool = __esm({
         if (!llmRequest.config || !llmRequest.config.tools) {
           return null;
         }
-        return llmRequest.config.tools.find(
-          (tool) => tool.functionDeclarations && tool.functionDeclarations.length > 0
+        const toolWithFunctionDeclaration = llmRequest.config.tools.find(
+          (tool) => "functionDeclarations" in tool && tool.functionDeclarations && tool.functionDeclarations.length > 0
         ) || null;
+        return toolWithFunctionDeclaration;
       }
     }, _class2);
   }
 });
 
 // src/tools/function/function-utils.ts
+var _genai = require('@google/genai');
 function buildFunctionDeclaration(func, options = {}) {
   const funcStr = func.toString();
   const name = options.name || func.name;
@@ -352,10 +354,10 @@ function buildFunctionDeclaration(func, options = {}) {
 function extractParametersSchema(func, ignoreParams = []) {
   const funcStr = func.toString();
   const paramMatch = funcStr.match(/\(([^)]*)\)/);
-  if (!paramMatch) return { type: "object", properties: {} };
+  if (!paramMatch) return { type: _genai.Type.OBJECT, properties: {} };
   const paramList = paramMatch[1].split(",").map((param) => param.trim()).filter((param) => param !== "");
   if (paramList.length === 0 || paramList.length === 1 && paramList[0] === "") {
-    return { type: "object", properties: {} };
+    return { type: _genai.Type.OBJECT, properties: {} };
   }
   const jsDocParams = extractJSDocParams(funcStr);
   const jsDocTypes = extractJSDocTypes(funcStr);
@@ -395,7 +397,7 @@ function extractParametersSchema(func, ignoreParams = []) {
     }
   }
   const schema = {
-    type: "object",
+    type: _genai.Type.OBJECT,
     properties
   };
   if (required.length > 0) {
@@ -1364,7 +1366,7 @@ var BaseLLMConnection = class {
 
 
 
-var _genai = require('@google/genai');
+
 var AGENT_ENGINE_TELEMETRY_TAG = "remote_reasoning_engine";
 var AGENT_ENGINE_TELEMETRY_ENV_VARIABLE_NAME = "GOOGLE_CLOUD_AGENT_ENGINE_ID";
 var GoogleLlm = class extends BaseLlm {
@@ -1397,7 +1399,7 @@ var GoogleLlm = class extends BaseLlm {
     this.preprocessRequest(llmRequest);
     const model = llmRequest.model || this.model;
     const contents = this.convertContents(llmRequest.contents || []);
-    const config = this.convertConfig(llmRequest.config);
+    const config = llmRequest.config;
     if (stream) {
       const responses = await this.apiClient.models.generateContentStream({
         model,
@@ -1491,19 +1493,6 @@ var GoogleLlm = class extends BaseLlm {
       parts: content.parts || [{ text: content.content || "" }]
     }));
   }
-  /**
-   * Convert LlmRequest config to GoogleGenAI format
-   */
-  convertConfig(config) {
-    if (!config) return {};
-    return {
-      temperature: config.temperature,
-      topP: config.top_p,
-      maxOutputTokens: config.max_tokens,
-      tools: config.tools,
-      systemInstruction: config.system_instruction
-    };
-  }
   /**
    * Preprocesses the request based on the API backend.
    */
@@ -3295,7 +3284,7 @@ var CallbackContext = class extends ReadonlyContext {
   constructor(invocationContext, options = {}) {
     super(invocationContext);
     this._eventActions = options.eventActions || new EventActions();
-    this._state = new State(
+    this._state = State.create(
       invocationContext.session.state,
       this._eventActions.stateDelta
     );
@@ -3821,6 +3810,7 @@ ${argsStr}
 // src/tools/index.ts
 var tools_exports = {};
 __export(tools_exports, {
+  AgentTool: () => AgentTool,
   BaseTool: () => BaseTool,
   ExitLoopTool: () => ExitLoopTool,
   FileOperationsTool: () => FileOperationsTool,
@@ -3925,6 +3915,283 @@ function createTool(config) {
   return new CreatedTool(config);
 }
 
+// src/tools/common/agent-tool.ts
+init_logger();
+
+
+
+// src/agents/invocation-context.ts
+var LlmCallsLimitExceededError = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "LlmCallsLimitExceededError";
+  }
+};
+var InvocationCostManager = (_class13 = class {constructor() { _class13.prototype.__init22.call(this); }
+  /**
+   * A counter that keeps track of number of llm calls made.
+   */
+  __init22() {this._numberOfLlmCalls = 0}
+  /**
+   * Increments _numberOfLlmCalls and enforces the limit.
+   */
+  incrementAndEnforceLlmCallsLimit(runConfig) {
+    this._numberOfLlmCalls += 1;
+    if (runConfig && runConfig.maxLlmCalls > 0 && this._numberOfLlmCalls > runConfig.maxLlmCalls) {
+      throw new LlmCallsLimitExceededError(
+        `Max number of llm calls limit of \`${runConfig.maxLlmCalls}\` exceeded`
+      );
+    }
+  }
+}, _class13);
+function newInvocationContextId() {
+  return `e-${crypto.randomUUID()}`;
+}
+var InvocationContext = (_class14 = class _InvocationContext {
+  
+  
+  
+  /**
+   * The id of this invocation context. Readonly.
+   */
+  
+  /**
+   * The branch of the invocation context.
+   *
+   * The format is like agent_1.agent_2.agent_3, where agent_1 is the parent of
+   * agent_2, and agent_2 is the parent of agent_3.
+   *
+   * Branch is used when multiple sub-agents shouldn't see their peer agents'
+   * conversation history.
+   */
+  
+  /**
+   * The current agent of this invocation context. Readonly.
+   */
+  
+  /**
+   * The user content that started this invocation. Readonly.
+   */
+  
+  /**
+   * The current session of this invocation context. Readonly.
+   */
+  
+  /**
+   * Whether to end this invocation.
+   *
+   * Set to True in callbacks or tools to terminate this invocation.
+   */
+  __init23() {this.endInvocation = false}
+  /**
+   * The queue to receive live requests.
+   */
+  
+  /**
+   * The running streaming tools of this invocation.
+   */
+  
+  /**
+   * Caches necessary, data audio or contents, that are needed by transcription.
+   */
+  
+  /**
+   * Configurations for live agents under this invocation.
+   */
+  
+  /**
+   * A container to keep track of different kinds of costs incurred as a part
+   * of this invocation.
+   */
+  __init24() {this._invocationCostManager = new InvocationCostManager()}
+  /**
+   * Constructor for InvocationContext
+   */
+  constructor(options) {;_class14.prototype.__init23.call(this);_class14.prototype.__init24.call(this);
+    this.artifactService = options.artifactService;
+    this.sessionService = options.sessionService;
+    this.memoryService = options.memoryService;
+    this.invocationId = options.invocationId || newInvocationContextId();
+    this.branch = options.branch;
+    this.agent = options.agent;
+    this.userContent = options.userContent;
+    this.session = options.session;
+    this.endInvocation = options.endInvocation || false;
+    this.liveRequestQueue = options.liveRequestQueue;
+    this.activeStreamingTools = options.activeStreamingTools;
+    this.transcriptionCache = options.transcriptionCache;
+    this.runConfig = options.runConfig;
+  }
+  /**
+   * App name from the session
+   */
+  get appName() {
+    return this.session.appName;
+  }
+  /**
+   * User ID from the session
+   */
+  get userId() {
+    return this.session.userId;
+  }
+  /**
+   * Tracks number of llm calls made.
+   *
+   * @throws {LlmCallsLimitExceededError} If number of llm calls made exceed the set threshold.
+   */
+  incrementLlmCallCount() {
+    this._invocationCostManager.incrementAndEnforceLlmCallsLimit(
+      this.runConfig
+    );
+  }
+  /**
+   * Creates a child invocation context for a sub-agent
+   */
+  createChildContext(agent) {
+    return new _InvocationContext({
+      artifactService: this.artifactService,
+      sessionService: this.sessionService,
+      memoryService: this.memoryService,
+      invocationId: this.invocationId,
+      // Keep same invocation ID
+      branch: this.branch ? `${this.branch}.${agent.name}` : agent.name,
+      // Update branch
+      agent,
+      // Update to the new agent
+      userContent: this.userContent,
+      session: this.session,
+      endInvocation: this.endInvocation,
+      liveRequestQueue: this.liveRequestQueue,
+      activeStreamingTools: this.activeStreamingTools,
+      transcriptionCache: this.transcriptionCache,
+      runConfig: this.runConfig
+    });
+  }
+}, _class14);
+
+// src/tools/common/agent-tool.ts
+init_base_tool();
+function isLlmAgent(agent) {
+  return true;
+}
+var AgentTool = (_class15 = class extends BaseTool {
+  /**
+   * The agent used by this tool
+   */
+  
+  /**
+   * The function declaration schema
+   */
+  
+  /**
+   * The key to store the tool output in the state
+   */
+  
+  /**
+   * Whether to skip summarization of the agent's response
+   */
+  
+  __init25() {this.logger = new Logger({ name: "AgentTool" })}
+  /**
+   * Create a new agent tool
+   */
+  constructor(config) {
+    super({
+      name: config.name,
+      description: config.description || config.agent.description,
+      isLongRunning: config.isLongRunning || false,
+      shouldRetryOnFailure: config.shouldRetryOnFailure || false,
+      maxRetryAttempts: config.maxRetryAttempts || 3
+    });_class15.prototype.__init25.call(this);;
+    this.agent = config.agent;
+    this.functionDeclaration = config.functionDeclaration;
+    this.outputKey = config.outputKey;
+    this.skipSummarization = config.skipSummarization || false;
+  }
+  /**
+   * Get the function declaration for the tool
+   */
+  getDeclaration() {
+    if (this.functionDeclaration) {
+      return this.functionDeclaration;
+    }
+    const description = isLlmAgent(this.agent) ? typeof this.agent.instruction === "string" ? this.agent.instruction : this.description : this.description;
+    return {
+      name: this.name,
+      description,
+      parameters: {
+        type: _genai.Type.OBJECT,
+        properties: {
+          input: {
+            type: _genai.Type.STRING,
+            description: "The input to provide to the agent"
+          }
+        },
+        required: ["input"]
+      }
+    };
+  }
+  /**
+   * Execute the tool by running the agent with the provided input
+   */
+  async runAsync(params, context) {
+    try {
+      const input = params.input || Object.values(params)[0];
+      if (!isLlmAgent(this.agent)) {
+        throw new Error(
+          `Agent ${this.name} does not support running as a tool`
+        );
+      }
+      const parentInvocation = context._invocationContext;
+      const childInvocationContext = new InvocationContext({
+        invocationId: _uuid.v4.call(void 0, ),
+        agent: this.agent,
+        session: parentInvocation.session,
+        artifactService: parentInvocation.artifactService,
+        sessionService: parentInvocation.sessionService,
+        memoryService: parentInvocation.memoryService,
+        runConfig: parentInvocation.runConfig,
+        userContent: {
+          role: "user",
+          parts: [{ text: String(input) }]
+        },
+        branch: parentInvocation.branch ? `${parentInvocation.branch}.${this.agent.name}` : this.agent.name
+      });
+      let lastEvent = null;
+      for await (const event of this.agent.runAsync(childInvocationContext)) {
+        if (!event.partial) {
+          await childInvocationContext.sessionService.appendEvent(
+            childInvocationContext.session,
+            event
+          );
+        }
+        if (event.content && event.author === this.agent.name) {
+          lastEvent = event;
+        }
+      }
+      if (!lastEvent || !lastEvent.content || !lastEvent.content.parts) {
+        return "";
+      }
+      const mergedText = lastEvent.content.parts.filter((part) => part.text !== void 0 && part.text !== null).map((part) => part.text).join("\n");
+      let toolResult;
+      try {
+        toolResult = JSON.parse(mergedText);
+      } catch (e2) {
+        toolResult = mergedText;
+      }
+      if (this.outputKey && _optionalChain([context, 'optionalAccess', _144 => _144.state])) {
+        context.state[this.outputKey] = toolResult;
+      }
+      return toolResult;
+    } catch (error) {
+      this.logger.error(`Error executing agent tool ${this.name}:`, error);
+      throw new Error(
+        `Agent tool execution failed: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+}, _class15);
+
 // src/tools/tool-context.ts
 var ToolContext = class extends CallbackContext {
   /**
@@ -3992,8 +4259,9 @@ init_function_utils();
 // src/tools/common/google-search.ts
 init_logger();
 init_base_tool();
-var GoogleSearch = (_class13 = class extends BaseTool {
-  __init22() {this.logger = new Logger({ name: "GoogleSearch" })}
+
+var GoogleSearch = (_class16 = class extends BaseTool {
+  __init26() {this.logger = new Logger({ name: "GoogleSearch" })}
   /**
    * Constructor for GoogleSearch
    */
@@ -4001,7 +4269,7 @@ var GoogleSearch = (_class13 = class extends BaseTool {
     super({
       name: "google_search",
       description: "Search the web using Google"
-    });_class13.prototype.__init22.call(this);;
+    });_class16.prototype.__init26.call(this);;
   }
   /**
    * Get the function declaration for the tool
@@ -4011,14 +4279,14 @@ var GoogleSearch = (_class13 = class extends BaseTool {
       name: this.name,
       description: this.description,
       parameters: {
-        type: "object",
+        type: _genai.Type.OBJECT,
         properties: {
           query: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "The search query to execute"
           },
           num_results: {
-            type: "integer",
+            type: _genai.Type.INTEGER,
             description: "Number of results to return (max 10)",
             default: 5
           }
@@ -4050,10 +4318,11 @@ var GoogleSearch = (_class13 = class extends BaseTool {
       ]
     };
   }
-}, _class13);
+}, _class16);
 
 // src/tools/common/http-request-tool.ts
 init_base_tool();
+
 var HttpRequestTool = class extends BaseTool {
   constructor() {
     super({
@@ -4069,38 +4338,32 @@ var HttpRequestTool = class extends BaseTool {
       name: this.name,
       description: this.description,
       parameters: {
-        type: "object",
+        type: _genai.Type.OBJECT,
         properties: {
           url: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "The URL to send the request to"
           },
           method: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "The HTTP method to use (GET, POST, PUT, DELETE, etc.)",
             enum: ["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD", "OPTIONS"],
             default: "GET"
           },
           headers: {
-            type: "object",
-            description: "Request headers to include",
-            additionalProperties: {
-              type: "string"
-            }
+            type: _genai.Type.OBJECT,
+            description: "Request headers to include"
           },
           body: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "Request body content (as string, typically JSON)"
           },
           params: {
-            type: "object",
-            description: "URL query parameters to include",
-            additionalProperties: {
-              type: "string"
-            }
+            type: _genai.Type.OBJECT,
+            description: "URL query parameters to include"
           },
           timeout: {
-            type: "integer",
+            type: _genai.Type.INTEGER,
             description: "Request timeout in milliseconds",
             default: 1e4
           }
@@ -4175,6 +4438,7 @@ var HttpRequestTool = class extends BaseTool {
 init_base_tool();
 var _promises = require('fs/promises'); var _promises2 = _interopRequireDefault(_promises);
 var _path = require('path'); var _path2 = _interopRequireDefault(_path);
+
 var FileOperationsTool = class extends BaseTool {
   
   constructor(options) {
@@ -4182,7 +4446,7 @@ var FileOperationsTool = class extends BaseTool {
       name: "file_operations",
       description: "Perform file system operations like reading, writing, and managing files"
     });
-    this.basePath = _optionalChain([options, 'optionalAccess', _144 => _144.basePath]) || process.cwd();
+    this.basePath = _optionalChain([options, 'optionalAccess', _145 => _145.basePath]) || process.cwd();
   }
   /**
    * Get the function declaration for the tool
@@ -4192,10 +4456,10 @@ var FileOperationsTool = class extends BaseTool {
       name: this.name,
       description: this.description,
       parameters: {
-        type: "object",
+        type: _genai.Type.OBJECT,
         properties: {
           operation: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "The file operation to perform",
             enum: [
               "read",
@@ -4208,15 +4472,15 @@ var FileOperationsTool = class extends BaseTool {
             ]
           },
           filepath: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "Path to the file or directory (relative to the base path)"
           },
           content: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "Content to write to the file (for write and append operations)"
           },
           encoding: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "File encoding to use",
             default: "utf8"
           }
@@ -4363,7 +4627,7 @@ var FileOperationsTool = class extends BaseTool {
         success: true,
         data: true
       };
-    } catch (e2) {
+    } catch (e3) {
       return {
         success: true,
         data: false
@@ -4422,6 +4686,7 @@ var FileOperationsTool = class extends BaseTool {
 
 // src/tools/common/user-interaction-tool.ts
 init_base_tool();
+
 var UserInteractionTool = class extends BaseTool {
   constructor() {
     super({
@@ -4438,21 +4703,21 @@ var UserInteractionTool = class extends BaseTool {
       name: this.name,
       description: this.description,
       parameters: {
-        type: "object",
+        type: _genai.Type.OBJECT,
         properties: {
           prompt: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "The prompt message to display to the user"
           },
           options: {
-            type: "array",
+            type: _genai.Type.ARRAY,
             description: "Optional array of choices to present to the user",
             items: {
-              type: "string"
+              type: _genai.Type.STRING
             }
           },
           defaultValue: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "Optional default value for the input field"
           }
         },
@@ -4499,8 +4764,8 @@ var UserInteractionTool = class extends BaseTool {
 // src/tools/common/exit-loop-tool.ts
 init_logger();
 init_base_tool();
-var ExitLoopTool = (_class14 = class extends BaseTool {
-  __init23() {this.logger = new Logger({ name: "ExitLoopTool" })}
+var ExitLoopTool = (_class17 = class extends BaseTool {
+  __init27() {this.logger = new Logger({ name: "ExitLoopTool" })}
   /**
    * Constructor for ExitLoopTool
    */
@@ -4508,7 +4773,7 @@ var ExitLoopTool = (_class14 = class extends BaseTool {
     super({
       name: "exit_loop",
       description: "Exits the loop. Call this function only when you are instructed to do so."
-    });_class14.prototype.__init23.call(this);;
+    });_class17.prototype.__init27.call(this);;
   }
   /**
    * Execute the exit loop action
@@ -4517,13 +4782,14 @@ var ExitLoopTool = (_class14 = class extends BaseTool {
     this.logger.debug("Executing exit loop tool");
     context.actions.escalate = true;
   }
-}, _class14);
+}, _class17);
 
 // src/tools/common/get-user-choice-tool.ts
 init_logger();
 init_base_tool();
-var GetUserChoiceTool = (_class15 = class extends BaseTool {
-  __init24() {this.logger = new Logger({ name: "GetUserChoiceTool" })}
+
+var GetUserChoiceTool = (_class18 = class extends BaseTool {
+  __init28() {this.logger = new Logger({ name: "GetUserChoiceTool" })}
   /**
    * Constructor for GetUserChoiceTool
    */
@@ -4532,7 +4798,7 @@ var GetUserChoiceTool = (_class15 = 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
-    });_class15.prototype.__init24.call(this);;
+    });_class18.prototype.__init28.call(this);;
   }
   /**
    * Get the function declaration for the tool
@@ -4542,17 +4808,17 @@ var GetUserChoiceTool = (_class15 = class extends BaseTool {
       name: this.name,
       description: this.description,
       parameters: {
-        type: "object",
+        type: _genai.Type.OBJECT,
         properties: {
           options: {
-            type: "array",
+            type: _genai.Type.ARRAY,
             description: "List of options for the user to choose from",
             items: {
-              type: "string"
+              type: _genai.Type.STRING
             }
           },
           question: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "The question or prompt to show the user before presenting options"
           }
         },
@@ -4575,21 +4841,41 @@ var GetUserChoiceTool = (_class15 = class extends BaseTool {
     context.actions.skipSummarization = true;
     return null;
   }
-}, _class15);
+}, _class18);
 
 // src/tools/common/transfer-to-agent-tool.ts
 init_logger();
 init_base_tool();
-var TransferToAgentTool = (_class16 = class extends BaseTool {
-  __init25() {this.logger = new Logger({ name: "TransferToAgentTool" })}
+
+var TransferToAgentTool = (_class19 = class extends BaseTool {
+  __init29() {this.logger = new Logger({ name: "TransferToAgentTool" })}
   /**
    * Constructor for TransferToAgentTool
    */
   constructor() {
     super({
       name: "transfer_to_agent",
-      description: "Transfer the question to another agent."
-    });_class16.prototype.__init25.call(this);;
+      description: "Transfer the question to another agent when it's more suitable to answer the user's question according to the agent's description. Use this function when you determine that another agent in the system would be better equipped to handle the user's request based on their specialized capabilities and expertise areas."
+    });_class19.prototype.__init29.call(this);;
+  }
+  /**
+   * Get the function declaration for the tool
+   */
+  getDeclaration() {
+    return {
+      name: this.name,
+      description: this.description,
+      parameters: {
+        type: _genai.Type.OBJECT,
+        properties: {
+          agent_name: {
+            type: _genai.Type.STRING,
+            description: "The name of the agent to transfer control to"
+          }
+        },
+        required: ["agent_name"]
+      }
+    };
   }
   /**
    * Execute the transfer to agent action
@@ -4598,13 +4884,14 @@ var TransferToAgentTool = (_class16 = class extends BaseTool {
     this.logger.debug(`Executing transfer to agent: ${args.agent_name}`);
     context.actions.transferToAgent = args.agent_name;
   }
-}, _class16);
+}, _class19);
 
 // src/tools/common/load-memory-tool.ts
 init_logger();
 init_base_tool();
-var LoadMemoryTool = (_class17 = class extends BaseTool {
-  __init26() {this.logger = new Logger({ name: "LoadMemoryTool" })}
+
+var LoadMemoryTool = (_class20 = class extends BaseTool {
+  __init30() {this.logger = new Logger({ name: "LoadMemoryTool" })}
   /**
    * Constructor for LoadMemoryTool
    */
@@ -4612,7 +4899,7 @@ var LoadMemoryTool = (_class17 = class extends BaseTool {
     super({
       name: "load_memory",
       description: "Loads the memory for the current user based on a query."
-    });_class17.prototype.__init26.call(this);;
+    });_class20.prototype.__init30.call(this);;
   }
   /**
    * Get the function declaration for the tool
@@ -4622,10 +4909,10 @@ var LoadMemoryTool = (_class17 = class extends BaseTool {
       name: this.name,
       description: this.description,
       parameters: {
-        type: "object",
+        type: _genai.Type.OBJECT,
         properties: {
           query: {
-            type: "string",
+            type: _genai.Type.STRING,
             description: "The query to load memories for"
           }
         },
@@ -4642,7 +4929,7 @@ var LoadMemoryTool = (_class17 = class extends BaseTool {
       const searchResult = await context.searchMemory(args.query);
       return {
         memories: searchResult.memories || [],
-        count: _optionalChain([searchResult, 'access', _145 => _145.memories, 'optionalAccess', _146 => _146.length]) || 0
+        count: _optionalChain([searchResult, 'access', _146 => _146.memories, 'optionalAccess', _147 => _147.length]) || 0
       };
     } catch (error) {
       console.error("Error searching memory:", error);
@@ -4652,10 +4939,11 @@ var LoadMemoryTool = (_class17 = class extends BaseTool {
       };
     }
   }
-}, _class17);
+}, _class20);
 
 // src/tools/common/load-artifacts-tool.ts
 init_base_tool();
+
 var LoadArtifactsTool = class extends BaseTool {
   constructor() {
     super({
@@ -4671,12 +4959,12 @@ var LoadArtifactsTool = class extends BaseTool {
       name: this.name,
       description: this.description,
       parameters: {
-        type: "object",
+        type: _genai.Type.OBJECT,
         properties: {
           artifact_names: {
-            type: "array",
+            type: _genai.Type.ARRAY,
             items: {
-              type: "string"
+              type: _genai.Type.STRING
             },
             description: "List of artifact names to load"
           }
@@ -4803,10 +5091,10 @@ var McpError = class extends Error {
 };
 
 // src/tools/mcp/sampling-handler.ts
-var McpSamplingHandler = (_class18 = class {
-  __init27() {this.logger = new Logger({ name: "McpSamplingHandler" })}
+var McpSamplingHandler = (_class21 = class {
+  __init31() {this.logger = new Logger({ name: "McpSamplingHandler" })}
   
-  constructor(samplingHandler) {;_class18.prototype.__init27.call(this);
+  constructor(samplingHandler) {;_class21.prototype.__init31.call(this);
     this.samplingHandler = samplingHandler;
   }
   /**
@@ -4991,7 +5279,7 @@ var McpSamplingHandler = (_class18 = class {
     this.samplingHandler = handler;
     this.logger.debug("ADK sampling handler updated");
   }
-}, _class18);
+}, _class21);
 function createSamplingHandler(handler) {
   return handler;
 }
@@ -5025,14 +5313,14 @@ function withRetry(fn, instance, reinitMethod, maxRetries = 1) {
 }
 
 // src/tools/mcp/client.ts
-var McpClientService = (_class19 = class {
-  
-  __init28() {this.client = null}
-  __init29() {this.transport = null}
-  __init30() {this.isClosing = false}
-  __init31() {this.mcpSamplingHandler = null}
-  __init32() {this.logger = new Logger({ name: "McpClientService" })}
-  constructor(config) {;_class19.prototype.__init28.call(this);_class19.prototype.__init29.call(this);_class19.prototype.__init30.call(this);_class19.prototype.__init31.call(this);_class19.prototype.__init32.call(this);
+var McpClientService = (_class22 = class {
+  
+  __init32() {this.client = null}
+  __init33() {this.transport = null}
+  __init34() {this.isClosing = false}
+  __init35() {this.mcpSamplingHandler = null}
+  __init36() {this.logger = new Logger({ name: "McpClientService" })}
+  constructor(config) {;_class22.prototype.__init32.call(this);_class22.prototype.__init33.call(this);_class22.prototype.__init34.call(this);_class22.prototype.__init35.call(this);_class22.prototype.__init36.call(this);
     this.config = config;
     if (config.samplingHandler) {
       this.mcpSamplingHandler = new McpSamplingHandler(config.samplingHandler);
@@ -5198,7 +5486,7 @@ var McpClientService = (_class19 = class {
         },
         this,
         async (instance) => await instance.reinitialize(),
-        _optionalChain([this, 'access', _147 => _147.config, 'access', _148 => _148.retryOptions, 'optionalAccess', _149 => _149.maxRetries]) || 2
+        _optionalChain([this, 'access', _148 => _148.config, 'access', _149 => _149.retryOptions, 'optionalAccess', _150 => _150.maxRetries]) || 2
       );
       return await wrappedCall();
     } catch (error) {
@@ -5282,19 +5570,20 @@ var McpClientService = (_class19 = class {
     this.mcpSamplingHandler = null;
     if (this.client) {
       try {
-        _optionalChain([this, 'access', _150 => _150.client, 'access', _151 => _151.removeRequestHandler, 'optionalCall', _152 => _152("sampling/createMessage")]);
+        _optionalChain([this, 'access', _151 => _151.client, 'access', _152 => _152.removeRequestHandler, 'optionalCall', _153 => _153("sampling/createMessage")]);
       } catch (error) {
         this.logger.error("Failed to remove sampling handler:", error);
       }
     }
   }
-}, _class19);
+}, _class22);
 
 // src/tools/mcp/create-tool.ts
 init_logger();
 init_base_tool();
 
 // src/tools/mcp/schema-conversion.ts
+
 function adkToMcpToolType(tool) {
   const declaration = tool.getDeclaration();
   const params = declarationToJsonSchema(declaration);
@@ -5323,13 +5612,13 @@ function jsonSchemaToDeclaration(name, description, schema) {
       parameters = schema;
     } else {
       parameters = {
-        type: "object",
+        type: _genai.Type.OBJECT,
         properties: schema
       };
     }
   } else {
     parameters = {
-      type: "object",
+      type: _genai.Type.OBJECT,
       properties: {}
     };
   }
@@ -5341,7 +5630,7 @@ function jsonSchemaToDeclaration(name, description, schema) {
 }
 function normalizeJsonSchema(schema) {
   if (!schema) {
-    return { type: "object", properties: {} };
+    return { type: _genai.Type.OBJECT, properties: {} };
   }
   const normalizedSchema = { ...schema };
   if (!normalizedSchema.type) {
@@ -5358,39 +5647,39 @@ function normalizeJsonSchema(schema) {
     case "integer":
       return normalizeNumberSchema(normalizedSchema);
     case "boolean":
-      return { type: "boolean" };
+      return { type: _genai.Type.BOOLEAN };
     case "null":
-      return { type: "null" };
+      return { type: _genai.Type.NULL };
     default:
       return normalizedSchema;
   }
 }
 function determineSchemaType(schema) {
   if (schema.properties || schema.required || schema.additionalProperties !== void 0) {
-    return "object";
+    return _genai.Type.OBJECT;
   }
   if (schema.items) {
-    return "array";
+    return _genai.Type.ARRAY;
   }
   if (schema.enum !== void 0) {
-    if (schema.enum.length === 0) return "string";
+    if (schema.enum.length === 0) return _genai.Type.STRING;
     const firstItem = schema.enum[0];
-    if (typeof firstItem === "string") return "string";
-    if (typeof firstItem === "number") return "number";
-    if (typeof firstItem === "boolean") return "boolean";
-    return "string";
+    if (typeof firstItem === "string") return _genai.Type.STRING;
+    if (typeof firstItem === "number") return _genai.Type.NUMBER;
+    if (typeof firstItem === "boolean") return _genai.Type.BOOLEAN;
+    return _genai.Type.STRING;
   }
   if (schema.minLength !== void 0 || schema.maxLength !== void 0 || schema.pattern) {
-    return "string";
+    return _genai.Type.STRING;
   }
   if (schema.minimum !== void 0 || schema.maximum !== void 0 || schema.exclusiveMinimum !== void 0 || schema.exclusiveMaximum !== void 0) {
-    return schema.multipleOf === void 0 || schema.multipleOf % 1 === 0 ? "integer" : "number";
+    return schema.multipleOf === void 0 || schema.multipleOf % 1 === 0 ? _genai.Type.INTEGER : _genai.Type.NUMBER;
   }
-  return "object";
+  return _genai.Type.OBJECT;
 }
 function normalizeObjectSchema(schema) {
   const normalizedSchema = {
-    type: "object",
+    type: _genai.Type.OBJECT,
     properties: {}
   };
   if (schema.properties) {
@@ -5402,15 +5691,13 @@ function normalizeObjectSchema(schema) {
     }
   }
   if (schema.required) normalizedSchema.required = schema.required;
-  if (schema.additionalProperties !== void 0)
-    normalizedSchema.additionalProperties = schema.additionalProperties;
   if (schema.title) normalizedSchema.title = schema.title;
   if (schema.description) normalizedSchema.description = schema.description;
   return normalizedSchema;
 }
 function normalizeArraySchema(schema) {
   const normalizedSchema = {
-    type: "array"
+    type: _genai.Type.ARRAY
   };
   if (schema.items) {
     normalizedSchema.items = normalizeJsonSchema(
@@ -5421,15 +5708,13 @@ function normalizeArraySchema(schema) {
     normalizedSchema.minItems = schema.minItems;
   if (schema.maxItems !== void 0)
     normalizedSchema.maxItems = schema.maxItems;
-  if (schema.uniqueItems !== void 0)
-    normalizedSchema.uniqueItems = schema.uniqueItems;
   if (schema.title) normalizedSchema.title = schema.title;
   if (schema.description) normalizedSchema.description = schema.description;
   return normalizedSchema;
 }
 function normalizeStringSchema(schema) {
   const normalizedSchema = {
-    type: "string"
+    type: _genai.Type.STRING
   };
   if (schema.minLength !== void 0)
     normalizedSchema.minLength = schema.minLength;
@@ -5448,12 +5733,6 @@ function normalizeNumberSchema(schema) {
   };
   if (schema.minimum !== void 0) normalizedSchema.minimum = schema.minimum;
   if (schema.maximum !== void 0) normalizedSchema.maximum = schema.maximum;
-  if (schema.exclusiveMinimum !== void 0)
-    normalizedSchema.exclusiveMinimum = schema.exclusiveMinimum;
-  if (schema.exclusiveMaximum !== void 0)
-    normalizedSchema.exclusiveMaximum = schema.exclusiveMaximum;
-  if (schema.multipleOf !== void 0)
-    normalizedSchema.multipleOf = schema.multipleOf;
   if (schema.enum) normalizedSchema.enum = schema.enum;
   if (schema.title) normalizedSchema.title = schema.title;
   if (schema.description) normalizedSchema.description = schema.description;
@@ -5468,7 +5747,7 @@ function mcpSchemaToParameters(mcpTool) {
   }
   if (!schema) {
     return {
-      type: "object",
+      type: _genai.Type.OBJECT,
       properties: {}
     };
   }
@@ -5490,11 +5769,11 @@ async function createTool2(mcpTool, client) {
     throw error;
   }
 }
-var McpToolAdapter = (_class20 = class extends BaseTool {
+var McpToolAdapter = (_class23 = class extends BaseTool {
   
   
-  __init33() {this.clientService = null}
-  __init34() {this.logger = new Logger({ name: "McpToolAdapter" })}
+  __init37() {this.clientService = null}
+  __init38() {this.logger = new Logger({ name: "McpToolAdapter" })}
   constructor(mcpTool, client) {
     const metadata = mcpTool.metadata || {};
     super({
@@ -5503,7 +5782,7 @@ var McpToolAdapter = (_class20 = class extends BaseTool {
       isLongRunning: _nullishCoalesce(metadata.isLongRunning, () => ( false)),
       shouldRetryOnFailure: _nullishCoalesce(metadata.shouldRetryOnFailure, () => ( false)),
       maxRetryAttempts: _nullishCoalesce(metadata.maxRetryAttempts, () => ( 3))
-    });_class20.prototype.__init33.call(this);_class20.prototype.__init34.call(this);;
+    });_class23.prototype.__init37.call(this);_class23.prototype.__init38.call(this);;
     this.mcpTool = mcpTool;
     this.client = client;
     if (client.reinitialize && typeof client.reinitialize === "function") {
@@ -5576,7 +5855,7 @@ var McpToolAdapter = (_class20 = class extends BaseTool {
       throw error;
     }
   }
-}, _class20);
+}, _class23);
 
 // src/tools/mcp/servers.ts
 function createMcpConfig(name, packageName, config = {}) {
@@ -5713,13 +5992,13 @@ function McpGeneric(packageName, config = {}, name) {
 }
 
 // src/tools/mcp/index.ts
-var McpToolset = (_class21 = class {
+var McpToolset = (_class24 = class {
   
-  __init35() {this.clientService = null}
-  __init36() {this.toolFilter = null}
-  __init37() {this.tools = []}
-  __init38() {this.isClosing = false}
-  constructor(config, toolFilter = null) {;_class21.prototype.__init35.call(this);_class21.prototype.__init36.call(this);_class21.prototype.__init37.call(this);_class21.prototype.__init38.call(this);
+  __init39() {this.clientService = null}
+  __init40() {this.toolFilter = null}
+  __init41() {this.tools = []}
+  __init42() {this.isClosing = false}
+  constructor(config, toolFilter = null) {;_class24.prototype.__init39.call(this);_class24.prototype.__init40.call(this);_class24.prototype.__init41.call(this);_class24.prototype.__init42.call(this);
     this.config = config;
     this.toolFilter = toolFilter;
     this.clientService = new McpClientService(config);
@@ -5794,7 +6073,7 @@ var McpToolset = (_class21 = class {
           "resource_closed_error" /* RESOURCE_CLOSED_ERROR */
         );
       }
-      if (this.tools.length > 0 && !_optionalChain([this, 'access', _153 => _153.config, 'access', _154 => _154.cacheConfig, 'optionalAccess', _155 => _155.enabled]) === false) {
+      if (this.tools.length > 0 && !_optionalChain([this, 'access', _154 => _154.config, 'access', _155 => _155.cacheConfig, 'optionalAccess', _156 => _156.enabled]) === false) {
         return this.tools;
       }
       if (!this.clientService) {
@@ -5820,7 +6099,7 @@ var McpToolset = (_class21 = class {
           }
         }
       }
-      if (_optionalChain([this, 'access', _156 => _156.config, 'access', _157 => _157.cacheConfig, 'optionalAccess', _158 => _158.enabled]) !== false) {
+      if (_optionalChain([this, 'access', _157 => _157.config, 'access', _158 => _158.cacheConfig, 'optionalAccess', _159 => _159.enabled]) !== false) {
         this.tools = tools;
       }
       return tools;
@@ -5880,7 +6159,7 @@ var McpToolset = (_class21 = class {
   async dispose() {
     await this.close();
   }
-}, _class21);
+}, _class24);
 async function getMcpTools(config, toolFilter) {
   const toolset = new McpToolset(config, toolFilter);
   try {
@@ -5908,12 +6187,12 @@ function populateClientFunctionCallId(modelResponseEvent) {
   }
 }
 function removeClientFunctionCallId(content) {
-  if (_optionalChain([content, 'optionalAccess', _159 => _159.parts])) {
+  if (_optionalChain([content, 'optionalAccess', _160 => _160.parts])) {
     for (const part of content.parts) {
-      if (_optionalChain([part, 'access', _160 => _160.functionCall, 'optionalAccess', _161 => _161.id, 'optionalAccess', _162 => _162.startsWith, 'call', _163 => _163(AF_FUNCTION_CALL_ID_PREFIX)])) {
+      if (_optionalChain([part, 'access', _161 => _161.functionCall, 'optionalAccess', _162 => _162.id, 'optionalAccess', _163 => _163.startsWith, 'call', _164 => _164(AF_FUNCTION_CALL_ID_PREFIX)])) {
         part.functionCall.id = void 0;
       }
-      if (_optionalChain([part, 'access', _164 => _164.functionResponse, 'optionalAccess', _165 => _165.id, 'optionalAccess', _166 => _166.startsWith, 'call', _167 => _167(AF_FUNCTION_CALL_ID_PREFIX)])) {
+      if (_optionalChain([part, 'access', _165 => _165.functionResponse, 'optionalAccess', _166 => _166.id, 'optionalAccess', _167 => _167.startsWith, 'call', _168 => _168(AF_FUNCTION_CALL_ID_PREFIX)])) {
         part.functionResponse.id = void 0;
       }
     }
@@ -5961,7 +6240,7 @@ function generateAuthEvent(invocationContext, functionResponseEvent) {
 }
 async function handleFunctionCallsAsync(invocationContext, functionCallEvent, toolsDict, filters) {
   const agent = invocationContext.agent;
-  if (!isLlmAgent(agent)) {
+  if (!isLlmAgent2(agent)) {
     return null;
   }
   const functionCalls = functionCallEvent.getFunctionCalls();
@@ -6058,7 +6337,7 @@ function mergeParallelFunctionResponseEvents(functionResponseEvents) {
   }
   const mergedParts = [];
   for (const event of functionResponseEvents) {
-    if (_optionalChain([event, 'access', _168 => _168.content, 'optionalAccess', _169 => _169.parts])) {
+    if (_optionalChain([event, 'access', _169 => _169.content, 'optionalAccess', _170 => _170.parts])) {
       for (const part of event.content.parts) {
         mergedParts.push(part);
       }
@@ -6085,16 +6364,16 @@ function mergeParallelFunctionResponseEvents(functionResponseEvents) {
   mergedEvent.timestamp = baseEvent.timestamp;
   return mergedEvent;
 }
-function isLlmAgent(agent) {
+function isLlmAgent2(agent) {
   return agent && typeof agent === "object" && "canonicalModel" in agent;
 }
 
 // src/flows/llm-flows/base-llm-flow.ts
 var _ADK_AGENT_NAME_LABEL_KEY = "adk_agent_name";
-var BaseLlmFlow = (_class22 = class {constructor() { _class22.prototype.__init39.call(this);_class22.prototype.__init40.call(this);_class22.prototype.__init41.call(this); }
-  __init39() {this.requestProcessors = []}
-  __init40() {this.responseProcessors = []}
-  __init41() {this.logger = new Logger({ name: "BaseLlmFlow" })}
+var BaseLlmFlow = (_class25 = class {constructor() { _class25.prototype.__init43.call(this);_class25.prototype.__init44.call(this);_class25.prototype.__init45.call(this); }
+  __init43() {this.requestProcessors = []}
+  __init44() {this.responseProcessors = []}
+  __init45() {this.logger = new Logger({ name: "BaseLlmFlow" })}
   async *runAsync(invocationContext) {
     this.logger.info(`Agent '${invocationContext.agent.name}' started.`);
     let stepCount = 0;
@@ -6186,7 +6465,7 @@ var BaseLlmFlow = (_class22 = class {constructor() { _class22.prototype.__init39
     if (tools.length > 0) {
       const toolsData = tools.map((tool) => ({
         Name: tool.name,
-        Description: _optionalChain([tool, 'access', _170 => _170.description, 'optionalAccess', _171 => _171.substring, 'call', _172 => _172(0, 50)]) + (_optionalChain([tool, 'access', _173 => _173.description, 'optionalAccess', _174 => _174.length]) > 50 ? "..." : ""),
+        Description: _optionalChain([tool, 'access', _171 => _171.description, 'optionalAccess', _172 => _172.substring, 'call', _173 => _173(0, 50)]) + (_optionalChain([tool, 'access', _174 => _174.description, 'optionalAccess', _175 => _175.length]) > 50 ? "..." : ""),
         "Long Running": tool.isLongRunning ? "Yes" : "No"
       }));
       this.logger.debugArray("\u{1F6E0}\uFE0F Available Tools", toolsData);
@@ -6249,14 +6528,14 @@ var BaseLlmFlow = (_class22 = class {constructor() { _class22.prototype.__init39
       );
       if (functionResponseEvent) {
         yield functionResponseEvent;
-        const transferToAgent = _optionalChain([functionResponseEvent, 'access', _175 => _175.actions, 'optionalAccess', _176 => _176.transferToAgent]);
+        const transferToAgent = _optionalChain([functionResponseEvent, 'access', _176 => _176.actions, 'optionalAccess', _177 => _177.transferToAgent]);
         if (transferToAgent) {
           this.logger.info(`\u{1F504} Live transfer to agent '${transferToAgent}'`);
           const agentToRun = this._getAgentToRun(
             invocationContext,
             transferToAgent
           );
-          for await (const event of _optionalChain([agentToRun, 'access', _177 => _177.runLive, 'optionalCall', _178 => _178(invocationContext)]) || agentToRun.runAsync(invocationContext)) {
+          for await (const event of _optionalChain([agentToRun, 'access', _178 => _178.runLive, 'optionalCall', _179 => _179(invocationContext)]) || agentToRun.runAsync(invocationContext)) {
             yield event;
           }
         }
@@ -6288,7 +6567,7 @@ var BaseLlmFlow = (_class22 = class {constructor() { _class22.prototype.__init39
         yield authEvent;
       }
       yield functionResponseEvent;
-      const transferToAgent = _optionalChain([functionResponseEvent, 'access', _179 => _179.actions, 'optionalAccess', _180 => _180.transferToAgent]);
+      const transferToAgent = _optionalChain([functionResponseEvent, 'access', _180 => _180.actions, 'optionalAccess', _181 => _181.transferToAgent]);
       if (transferToAgent) {
         this.logger.info(`\u{1F504} Transferring to agent '${transferToAgent}'`);
         const agentToRun = this._getAgentToRun(
@@ -6334,27 +6613,27 @@ var BaseLlmFlow = (_class22 = class {constructor() { _class22.prototype.__init39
     }
     invocationContext.incrementLlmCallCount();
     const isStreaming = invocationContext.runConfig.streamingMode === "sse" /* SSE */;
-    const tools = _optionalChain([llmRequest, 'access', _181 => _181.config, 'optionalAccess', _182 => _182.tools]) || [];
+    const tools = _optionalChain([llmRequest, 'access', _182 => _182.config, 'optionalAccess', _183 => _183.tools]) || [];
     const toolNames = tools.map((tool) => {
       if (tool.functionDeclarations && Array.isArray(tool.functionDeclarations)) {
         return tool.functionDeclarations.map((fn) => fn.name).join(", ");
       }
       if (tool.name) return tool.name;
-      if (_optionalChain([tool, 'access', _183 => _183.function, 'optionalAccess', _184 => _184.name])) return tool.function.name;
-      if (_optionalChain([tool, 'access', _185 => _185.function, 'optionalAccess', _186 => _186.function, 'optionalAccess', _187 => _187.name])) return tool.function.function.name;
+      if (_optionalChain([tool, 'access', _184 => _184.function, 'optionalAccess', _185 => _185.name])) return tool.function.name;
+      if (_optionalChain([tool, 'access', _186 => _186.function, 'optionalAccess', _187 => _187.function, 'optionalAccess', _188 => _188.name])) return tool.function.function.name;
       return "unknown";
     }).join(", ");
     const systemInstruction = llmRequest.getSystemInstructionText() || "";
     const truncatedSystemInstruction = systemInstruction.length > 100 ? `${systemInstruction.substring(0, 100)}...` : systemInstruction;
-    const contentPreview = _optionalChain([llmRequest, 'access', _188 => _188.contents, 'optionalAccess', _189 => _189.length]) > 0 ? LogFormatter.formatContentPreview(llmRequest.contents[0]) : "none";
+    const contentPreview = _optionalChain([llmRequest, 'access', _189 => _189.contents, 'optionalAccess', _190 => _190.length]) > 0 ? LogFormatter.formatContentPreview(llmRequest.contents[0]) : "none";
     this.logger.debugStructured("\u{1F4E4} LLM Request", {
       Model: llm.model,
       Agent: invocationContext.agent.name,
-      "Content Items": _optionalChain([llmRequest, 'access', _190 => _190.contents, 'optionalAccess', _191 => _191.length]) || 0,
+      "Content Items": _optionalChain([llmRequest, 'access', _191 => _191.contents, 'optionalAccess', _192 => _192.length]) || 0,
       "Content Preview": contentPreview,
       "System Instruction": truncatedSystemInstruction || "none",
       "Available Tools": toolNames || "none",
-      "Tool Count": _optionalChain([llmRequest, 'access', _192 => _192.config, 'optionalAccess', _193 => _193.tools, 'optionalAccess', _194 => _194.length]) || 0,
+      "Tool Count": _optionalChain([llmRequest, 'access', _193 => _193.config, 'optionalAccess', _194 => _194.tools, 'optionalAccess', _195 => _195.length]) || 0,
       Streaming: isStreaming ? "Yes" : "No"
     });
     let responseCount = 0;
@@ -6369,8 +6648,8 @@ var BaseLlmFlow = (_class22 = class {constructor() { _class22.prototype.__init39
         llmRequest,
         llmResponse
       );
-      const tokenCount = _optionalChain([llmResponse, 'access', _195 => _195.usageMetadata, 'optionalAccess', _196 => _196.totalTokenCount]) || "unknown";
-      const functionCalls = _optionalChain([llmResponse, 'access', _197 => _197.content, 'optionalAccess', _198 => _198.parts, 'optionalAccess', _199 => _199.filter, 'call', _200 => _200((part) => part.functionCall)]) || [];
+      const tokenCount = _optionalChain([llmResponse, 'access', _196 => _196.usageMetadata, 'optionalAccess', _197 => _197.totalTokenCount]) || "unknown";
+      const functionCalls = _optionalChain([llmResponse, 'access', _198 => _198.content, 'optionalAccess', _199 => _199.parts, 'optionalAccess', _200 => _200.filter, 'call', _201 => _201((part) => part.functionCall)]) || [];
       const functionCallsDisplay = LogFormatter.formatFunctionCalls(functionCalls);
       const responsePreview = LogFormatter.formatResponsePreview(llmResponse);
       this.logger.debugStructured("\u{1F4E5} LLM Response", {
@@ -6466,7 +6745,7 @@ var BaseLlmFlow = (_class22 = class {constructor() { _class22.prototype.__init39
     const llm = invocationContext.agent.canonicalModel;
     return llm;
   }
-}, _class22);
+}, _class25);
 
 // src/flows/llm-flows/base-llm-processor.ts
 var BaseLlmRequestProcessor = class {
@@ -6514,7 +6793,7 @@ var EnhancedAuthConfig = class {
    */
   generateCredentialKey() {
     const schemeKey = this.authScheme.type || "unknown";
-    const credentialKey = _optionalChain([this, 'access', _201 => _201.rawAuthCredential, 'optionalAccess', _202 => _202.type]) || "none";
+    const credentialKey = _optionalChain([this, 'access', _202 => _202.rawAuthCredential, 'optionalAccess', _203 => _203.type]) || "none";
     const timestamp = Date.now();
     return `adk_${schemeKey}_${credentialKey}_${timestamp}`;
   }
@@ -6671,7 +6950,7 @@ var AuthLlmRequestProcessor = class extends BaseLlmRequestProcessor {
    */
   parseAndStoreAuthResponse(authHandler, invocationContext) {
     try {
-      const credentialKey = _optionalChain([authHandler, 'access', _203 => _203.authConfig, 'access', _204 => _204.context, 'optionalAccess', _205 => _205.credentialKey]) || `temp:${Date.now()}`;
+      const credentialKey = _optionalChain([authHandler, 'access', _204 => _204.authConfig, 'access', _205 => _205.context, 'optionalAccess', _206 => _206.credentialKey]) || `temp:${Date.now()}`;
       const fullCredentialKey = credentialKey.startsWith("temp:") ? credentialKey : `temp:${credentialKey}`;
       invocationContext.session.state[fullCredentialKey] = authHandler.credential;
       if (authHandler.authConfig.authScheme.type === "oauth2" || authHandler.authConfig.authScheme.type === "openIdConnect") {
@@ -6779,7 +7058,7 @@ var BuiltInCodeExecutor = class extends BaseCodeExecutor {
    * Pre-process the LLM request for Gemini 2.0+ models to use the code execution tool
    */
   processLlmRequest(llmRequest) {
-    if (!_optionalChain([llmRequest, 'access', _206 => _206.model, 'optionalAccess', _207 => _207.startsWith, 'call', _208 => _208("gemini-2")])) {
+    if (!_optionalChain([llmRequest, 'access', _207 => _207.model, 'optionalAccess', _208 => _208.startsWith, 'call', _209 => _209("gemini-2")])) {
       throw new Error(
         `Gemini code execution tool is not supported for model ${llmRequest.model}`
       );
@@ -6816,7 +7095,7 @@ var CodeExecutionUtils = class _CodeExecutionUtils {
   static isBase64Encoded(str) {
     try {
       return btoa(atob(str)) === str;
-    } catch (e3) {
+    } catch (e4) {
       return false;
     }
   }
@@ -6824,7 +7103,7 @@ var CodeExecutionUtils = class _CodeExecutionUtils {
    * Extracts the first code block from the content and truncates everything after it
    */
   static extractCodeAndTruncateContent(content, codeBlockDelimiters) {
-    if (!_optionalChain([content, 'optionalAccess', _209 => _209.parts, 'optionalAccess', _210 => _210.length])) {
+    if (!_optionalChain([content, 'optionalAccess', _210 => _210.parts, 'optionalAccess', _211 => _211.length])) {
       return null;
     }
     for (let idx = 0; idx < content.parts.length; idx++) {
@@ -6910,7 +7189,7 @@ ${fileNames}`);
    * Converts the code execution parts to text parts in a Content
    */
   static convertCodeExecutionParts(content, codeBlockDelimiter, executionResultDelimiters) {
-    if (!_optionalChain([content, 'access', _211 => _211.parts, 'optionalAccess', _212 => _212.length])) {
+    if (!_optionalChain([content, 'access', _212 => _212.parts, 'optionalAccess', _213 => _213.length])) {
       return;
     }
     const lastPart = content.parts[content.parts.length - 1];
@@ -7303,7 +7582,7 @@ async function* runPostProcessor(invocationContext, llmResponse) {
 function extractAndReplaceInlineFiles(codeExecutorContext, llmRequest) {
   const allInputFiles = codeExecutorContext.getInputFiles();
   const savedFileNames = new Set(allInputFiles.map((f) => f.name));
-  for (let i = 0; i < (_optionalChain([llmRequest, 'access', _213 => _213.contents, 'optionalAccess', _214 => _214.length]) || 0); i++) {
+  for (let i = 0; i < (_optionalChain([llmRequest, 'access', _214 => _214.contents, 'optionalAccess', _215 => _215.length]) || 0); i++) {
     const content = llmRequest.contents[i];
     if (content.role !== "user" || !content.parts) {
       continue;
@@ -7335,7 +7614,7 @@ Available file: \`${fileName}\`
 }
 function getOrSetExecutionId(invocationContext, codeExecutorContext) {
   const agent = invocationContext.agent;
-  if (!hasCodeExecutor(agent) || !_optionalChain([agent, 'access', _215 => _215.codeExecutor, 'optionalAccess', _216 => _216.stateful])) {
+  if (!hasCodeExecutor(agent) || !_optionalChain([agent, 'access', _216 => _216.codeExecutor, 'optionalAccess', _217 => _217.stateful])) {
     return void 0;
   }
   let executionId = codeExecutorContext.getExecutionId();
@@ -7566,7 +7845,7 @@ function rearrangeEventsForLatestFunctionResponse(events) {
       continue;
     }
     const functionResponses2 = event.getFunctionResponses();
-    if (_optionalChain([functionResponses2, 'optionalAccess', _217 => _217.some, 'call', _218 => _218((fr) => fr.id && functionResponsesIds.has(fr.id))])) {
+    if (_optionalChain([functionResponses2, 'optionalAccess', _218 => _218.some, 'call', _219 => _219((fr) => fr.id && functionResponsesIds.has(fr.id))])) {
       functionResponseEvents.push(event);
     }
   }
@@ -7665,7 +7944,7 @@ function mergeFunctionResponseEvents(functionResponseEvents) {
   const partIndicesInMergedEvent = {};
   for (let idx = 0; idx < partsInMergedEvent.length; idx++) {
     const part = partsInMergedEvent[idx];
-    if (_optionalChain([part, 'access', _219 => _219.functionResponse, 'optionalAccess', _220 => _220.id])) {
+    if (_optionalChain([part, 'access', _220 => _220.functionResponse, 'optionalAccess', _221 => _221.id])) {
       partIndicesInMergedEvent[part.functionResponse.id] = idx;
     }
   }
@@ -7674,7 +7953,7 @@ function mergeFunctionResponseEvents(functionResponseEvents) {
       throw new Error("There should be at least one function_response part.");
     }
     for (const part of event.content.parts) {
-      if (_optionalChain([part, 'access', _221 => _221.functionResponse, 'optionalAccess', _222 => _222.id])) {
+      if (_optionalChain([part, 'access', _222 => _222.functionResponse, 'optionalAccess', _223 => _223.id])) {
         const functionCallId = part.functionResponse.id;
         if (functionCallId in partIndicesInMergedEvent) {
           partsInMergedEvent[partIndicesInMergedEvent[functionCallId]] = part;
@@ -7924,7 +8203,7 @@ var PlanReActPlanner = class extends BasePlanner {
     let firstFcPartIndex = -1;
     for (let i = 0; i < responseParts.length; i++) {
       if (responseParts[i].functionCall) {
-        if (!_optionalChain([responseParts, 'access', _223 => _223[i], 'access', _224 => _224.functionCall, 'optionalAccess', _225 => _225.name])) {
+        if (!_optionalChain([responseParts, 'access', _224 => _224[i], 'access', _225 => _225.functionCall, 'optionalAccess', _226 => _226.name])) {
           continue;
         }
         preservedParts.push(responseParts[i]);
@@ -7963,7 +8242,7 @@ var PlanReActPlanner = class extends BasePlanner {
    * Handles non-function-call parts of the response
    */
   _handleNonFunctionCallParts(responsePart, preservedParts) {
-    if (_optionalChain([responsePart, 'access', _226 => _226.text, 'optionalAccess', _227 => _227.includes, 'call', _228 => _228(FINAL_ANSWER_TAG)])) {
+    if (_optionalChain([responsePart, 'access', _227 => _227.text, 'optionalAccess', _228 => _228.includes, 'call', _229 => _229(FINAL_ANSWER_TAG)])) {
       const [reasoningText, finalAnswerText] = this._splitByLastPattern(
         responsePart.text,
         FINAL_ANSWER_TAG
@@ -8133,6 +8412,42 @@ function removeThoughtFromRequest(llmRequest) {
 var requestProcessor7 = new NlPlanningRequestProcessor();
 var responseProcessor2 = new NlPlanningResponseProcessor();
 
+// src/flows/llm-flows/shared-memory.ts
+var SharedMemoryRequestProcessor = class extends BaseLlmRequestProcessor {
+  async *runAsync(invocationContext, llmRequest) {
+    const memoryService = invocationContext.memoryService;
+    if (!memoryService) return;
+    const lastUserEvent = invocationContext.session.events.findLast((e) => e.author === "user" && _optionalChain([e, 'access', _230 => _230.content, 'optionalAccess', _231 => _231.parts, 'optionalAccess', _232 => _232.length]));
+    if (!lastUserEvent) return;
+    const query = (_nullishCoalesce(lastUserEvent.content.parts, () => ( []))).map((p) => p.text || "").join(" ");
+    const results = await memoryService.searchMemory({
+      appName: invocationContext.appName,
+      userId: invocationContext.userId,
+      query
+    });
+    const sessionTexts = new Set(
+      (llmRequest.contents || []).flatMap(
+        (c) => _optionalChain([c, 'access', _233 => _233.parts, 'optionalAccess', _234 => _234.map, 'call', _235 => _235((p) => p.text)]) || []
+      )
+    );
+    for (const memory of results.memories) {
+      const memoryText = (_nullishCoalesce(memory.content.parts, () => ( []))).map((p) => p.text || "").join(" ");
+      if (!sessionTexts.has(memoryText)) {
+        llmRequest.contents = llmRequest.contents || [];
+        llmRequest.contents.push({
+          role: "user",
+          parts: [
+            {
+              text: `[${memory.author}] said: ${memoryText}`
+            }
+          ]
+        });
+      }
+    }
+  }
+};
+var sharedMemoryRequestProcessor = new SharedMemoryRequestProcessor();
+
 // src/flows/llm-flows/single-flow.ts
 var SingleFlow = class extends BaseLlmFlow {
   /**
@@ -8147,6 +8462,7 @@ var SingleFlow = class extends BaseLlmFlow {
       requestProcessor6,
       requestProcessor5,
       requestProcessor4,
+      sharedMemoryRequestProcessor,
       // Some implementations of NL Planning mark planning contents as thoughts
       // in the post processor. Since these need to be unmarked, NL Planning
       // should be after contents.
@@ -8168,12 +8484,12 @@ var SingleFlow = class extends BaseLlmFlow {
 };
 
 // src/flows/llm-flows/agent-transfer.ts
+var _dedent = require('dedent'); var _dedent2 = _interopRequireDefault(_dedent);
 var AgentTransferLlmRequestProcessor = class extends BaseLlmRequestProcessor {
   /**
    * Processes agent transfer by adding transfer instructions and tools
    * if the agent has transfer targets available
    */
-  // eslint-disable-next-line @typescript-eslint/require-yield
   async *runAsync(invocationContext, llmRequest) {
     const agent = invocationContext.agent;
     if (!("subAgents" in agent) || typeof agent.subAgents !== "object") {
@@ -8187,71 +8503,65 @@ var AgentTransferLlmRequestProcessor = class extends BaseLlmRequestProcessor {
       agent,
       transferTargets
     );
-    if (llmRequest.appendInstructions) {
-      llmRequest.appendInstructions([transferInstructions]);
-    } else {
-      const existingInstructions = llmRequest.instructions || "";
-      llmRequest.instructions = `${existingInstructions}
-
-${transferInstructions}`;
-    }
+    llmRequest.appendInstructions([transferInstructions]);
     const transferToAgentTool = new TransferToAgentTool();
     const toolContext = new ToolContext(invocationContext);
     await transferToAgentTool.processLlmRequest(toolContext, llmRequest);
-    for await (const _ of []) {
-      yield _;
+    const shouldYield = false;
+    if (shouldYield) {
+      yield {};
     }
   }
 };
 function buildTargetAgentsInfo(targetAgent) {
-  return `
-Agent name: ${targetAgent.name}
-Agent description: ${targetAgent.description}
-`;
+  return _dedent2.default`
+		Agent name: ${targetAgent.name}
+		Agent description: ${targetAgent.description}
+	`;
 }
 function buildTargetAgentsInstructions(agent, targetAgents) {
   const lineBreak = "\n";
   const transferFunctionName = "transfer_to_agent";
-  let instructions = `
-You have a list of other agents to transfer to:
+  let instructions = _dedent2.default`
+		You have a list of other agents to transfer to:
 
-${targetAgents.map((targetAgent) => buildTargetAgentsInfo(targetAgent)).join(lineBreak)}
+		${targetAgents.map((targetAgent) => buildTargetAgentsInfo(targetAgent)).join(lineBreak)}
 
-If you are the best to answer the question according to your description, you
-can answer it.
+		If you are the best to answer the question according to your description, you
+		can answer it.
 
-If another agent is better for answering the question according to its
-description, call \`${transferFunctionName}\` function to transfer the
-question to that agent. When transferring, do not generate any text other than
-the function call.
+		If another agent is better for answering the question according to its
+		description, call \`${transferFunctionName}\` function to transfer the
+		question to that agent. When transferring, do not generate any text other than
+		the function call.
 `;
   if (agent.parentAgent && !agent.disallowTransferToParent) {
-    instructions += `
-Your parent agent is ${agent.parentAgent.name}. If neither the other agents nor
-you are best for answering the question according to the descriptions, transfer
-to your parent agent.
-`;
+    instructions += _dedent2.default`
+			Your parent agent is ${agent.parentAgent.name}. If neither the other agents nor
+			you are best for answering the question according to the descriptions, transfer
+			to your parent agent.
+		`;
   }
   return instructions;
 }
 function getTransferTargets(agent) {
-  const result = [];
+  const targets = [];
   if (agent.subAgents && Array.isArray(agent.subAgents)) {
-    result.push(...agent.subAgents);
+    targets.push(...agent.subAgents);
   }
   if (!agent.parentAgent || !("subAgents" in agent.parentAgent)) {
-    return result;
+    return targets;
   }
   if (!agent.disallowTransferToParent) {
-    result.push(agent.parentAgent);
+    targets.push(agent.parentAgent);
   }
   if (!agent.disallowTransferToPeers && agent.parentAgent.subAgents) {
     const peerAgents = agent.parentAgent.subAgents.filter(
       (peerAgent) => peerAgent.name !== agent.name
     );
-    result.push(...peerAgents);
+    targets.push(...peerAgents);
   }
-  return result;
+  return targets;
 }
 var requestProcessor8 = new AgentTransferLlmRequestProcessor();
 
@@ -8269,7 +8579,7 @@ var AutoFlow = class extends SingleFlow {
 
 // src/agents/llm-agent.ts
 init_function_tool();
-var LlmAgent = (_class23 = class _LlmAgent extends BaseAgent {
+var LlmAgent = (_class26 = class _LlmAgent extends BaseAgent {
   /**
    * The model to use for the agent
    * When not set, the agent will inherit the model from its ancestor
@@ -8346,15 +8656,18 @@ var LlmAgent = (_class23 = class _LlmAgent extends BaseAgent {
    */
   
   // Schema type - depends on specific implementation
-  __init42() {this.logger = new Logger({ name: "LlmAgent" })}
+  __init46() {this.logger = new Logger({ name: "LlmAgent" })}
   /**
    * Constructor for LlmAgent
    */
   constructor(config) {
     super({
       name: config.name,
-      description: config.description
-    });_class23.prototype.__init42.call(this);;
+      description: config.description,
+      subAgents: config.subAgents,
+      beforeAgentCallback: config.beforeAgentCallback,
+      afterAgentCallback: config.afterAgentCallback
+    });_class26.prototype.__init46.call(this);;
     this.model = config.model || "";
     this.instruction = config.instruction || "";
     this.globalInstruction = config.globalInstruction || "";
@@ -8440,7 +8753,7 @@ var LlmAgent = (_class23 = class _LlmAgent extends BaseAgent {
    * This matches the Python implementation's _llm_flow property
    */
   get llmFlow() {
-    if (this.disallowTransferToParent && this.disallowTransferToPeers && !_optionalChain([this, 'access', _229 => _229.subAgents, 'optionalAccess', _230 => _230.length])) {
+    if (this.disallowTransferToParent && this.disallowTransferToPeers && !_optionalChain([this, 'access', _236 => _236.subAgents, 'optionalAccess', _237 => _237.length])) {
       return new SingleFlow();
     }
     return new AutoFlow();
@@ -8450,7 +8763,7 @@ var LlmAgent = (_class23 = class _LlmAgent extends BaseAgent {
    * This matches the Python implementation's __maybe_save_output_to_state
    */
   maybeSaveOutputToState(event) {
-    if (this.outputKey && event.isFinalResponse() && _optionalChain([event, 'access', _231 => _231.content, 'optionalAccess', _232 => _232.parts])) {
+    if (this.outputKey && event.isFinalResponse() && _optionalChain([event, 'access', _238 => _238.content, 'optionalAccess', _239 => _239.parts])) {
       const result = event.content.parts.map((part) => part.text || "").join("");
       if (result) {
         if (!event.actions.stateDelta) {
@@ -8490,7 +8803,7 @@ var LlmAgent = (_class23 = class _LlmAgent extends BaseAgent {
       yield errorEvent;
     }
   }
-}, _class23);
+}, _class26);
 
 // src/agents/sequential-agent.ts
 var SequentialAgent = class extends BaseAgent {
@@ -8549,155 +8862,6 @@ do not generate any text other than the function call.`;
   }
 };
 
-// src/agents/invocation-context.ts
-var LlmCallsLimitExceededError = class extends Error {
-  constructor(message) {
-    super(message);
-    this.name = "LlmCallsLimitExceededError";
-  }
-};
-var InvocationCostManager = (_class24 = class {constructor() { _class24.prototype.__init43.call(this); }
-  /**
-   * A counter that keeps track of number of llm calls made.
-   */
-  __init43() {this._numberOfLlmCalls = 0}
-  /**
-   * Increments _numberOfLlmCalls and enforces the limit.
-   */
-  incrementAndEnforceLlmCallsLimit(runConfig) {
-    this._numberOfLlmCalls += 1;
-    if (runConfig && runConfig.maxLlmCalls > 0 && this._numberOfLlmCalls > runConfig.maxLlmCalls) {
-      throw new LlmCallsLimitExceededError(
-        `Max number of llm calls limit of \`${runConfig.maxLlmCalls}\` exceeded`
-      );
-    }
-  }
-}, _class24);
-function newInvocationContextId() {
-  return `e-${crypto.randomUUID()}`;
-}
-var InvocationContext = (_class25 = class _InvocationContext {
-  
-  
-  
-  /**
-   * The id of this invocation context. Readonly.
-   */
-  
-  /**
-   * The branch of the invocation context.
-   *
-   * The format is like agent_1.agent_2.agent_3, where agent_1 is the parent of
-   * agent_2, and agent_2 is the parent of agent_3.
-   *
-   * Branch is used when multiple sub-agents shouldn't see their peer agents'
-   * conversation history.
-   */
-  
-  /**
-   * The current agent of this invocation context. Readonly.
-   */
-  
-  /**
-   * The user content that started this invocation. Readonly.
-   */
-  
-  /**
-   * The current session of this invocation context. Readonly.
-   */
-  
-  /**
-   * Whether to end this invocation.
-   *
-   * Set to True in callbacks or tools to terminate this invocation.
-   */
-  __init44() {this.endInvocation = false}
-  /**
-   * The queue to receive live requests.
-   */
-  
-  /**
-   * The running streaming tools of this invocation.
-   */
-  
-  /**
-   * Caches necessary, data audio or contents, that are needed by transcription.
-   */
-  
-  /**
-   * Configurations for live agents under this invocation.
-   */
-  
-  /**
-   * A container to keep track of different kinds of costs incurred as a part
-   * of this invocation.
-   */
-  __init45() {this._invocationCostManager = new InvocationCostManager()}
-  /**
-   * Constructor for InvocationContext
-   */
-  constructor(options) {;_class25.prototype.__init44.call(this);_class25.prototype.__init45.call(this);
-    this.artifactService = options.artifactService;
-    this.sessionService = options.sessionService;
-    this.memoryService = options.memoryService;
-    this.invocationId = options.invocationId || newInvocationContextId();
-    this.branch = options.branch;
-    this.agent = options.agent;
-    this.userContent = options.userContent;
-    this.session = options.session;
-    this.endInvocation = options.endInvocation || false;
-    this.liveRequestQueue = options.liveRequestQueue;
-    this.activeStreamingTools = options.activeStreamingTools;
-    this.transcriptionCache = options.transcriptionCache;
-    this.runConfig = options.runConfig;
-  }
-  /**
-   * App name from the session
-   */
-  get appName() {
-    return this.session.appName;
-  }
-  /**
-   * User ID from the session
-   */
-  get userId() {
-    return this.session.userId;
-  }
-  /**
-   * Tracks number of llm calls made.
-   *
-   * @throws {LlmCallsLimitExceededError} If number of llm calls made exceed the set threshold.
-   */
-  incrementLlmCallCount() {
-    this._invocationCostManager.incrementAndEnforceLlmCallsLimit(
-      this.runConfig
-    );
-  }
-  /**
-   * Creates a child invocation context for a sub-agent
-   */
-  createChildContext(agent) {
-    return new _InvocationContext({
-      artifactService: this.artifactService,
-      sessionService: this.sessionService,
-      memoryService: this.memoryService,
-      invocationId: this.invocationId,
-      // Keep same invocation ID
-      branch: this.branch ? `${this.branch}.${agent.name}` : agent.name,
-      // Update branch
-      agent,
-      // Update to the new agent
-      userContent: this.userContent,
-      session: this.session,
-      endInvocation: this.endInvocation,
-      liveRequestQueue: this.liveRequestQueue,
-      activeStreamingTools: this.activeStreamingTools,
-      transcriptionCache: this.transcriptionCache,
-      runConfig: this.runConfig
-    });
-  }
-}, _class25);
-
 // src/agents/parallel-agent.ts
 function createBranchContextForSubAgent(agent, subAgent, invocationContext) {
   const branchSuffix = `${agent.name}.${subAgent.name}`;
@@ -8813,7 +8977,7 @@ var LoopAgent = class extends BaseAgent {
       for (const subAgent of this.subAgents) {
         for await (const event of subAgent.runAsync(ctx)) {
           yield event;
-          if (_optionalChain([event, 'access', _233 => _233.actions, 'optionalAccess', _234 => _234.escalate])) {
+          if (_optionalChain([event, 'access', _240 => _240.actions, 'optionalAccess', _241 => _241.escalate])) {
             return;
           }
         }
@@ -8831,7 +8995,7 @@ var LoopAgent = class extends BaseAgent {
 
 // src/agents/lang-graph-agent.ts
 init_logger();
-var LangGraphAgent = (_class26 = class extends BaseAgent {
+var LangGraphAgent = (_class27 = class extends BaseAgent {
   /**
    * Graph nodes (agents and their connections)
    */
@@ -8847,8 +9011,8 @@ var LangGraphAgent = (_class26 = class extends BaseAgent {
   /**
    * Results from node executions
    */
-  __init46() {this.results = []}
-  __init47() {this.logger = new Logger({ name: "LangGraphAgent" })}
+  __init47() {this.results = []}
+  __init48() {this.logger = new Logger({ name: "LangGraphAgent" })}
   /**
    * Constructor for LangGraphAgent
    */
@@ -8856,7 +9020,7 @@ var LangGraphAgent = (_class26 = class extends BaseAgent {
     super({
       name: config.name,
       description: config.description
-    });_class26.prototype.__init46.call(this);_class26.prototype.__init47.call(this);;
+    });_class27.prototype.__init47.call(this);_class27.prototype.__init48.call(this);;
     this.nodes = /* @__PURE__ */ new Map();
     for (const node of config.nodes) {
       if (this.nodes.has(node.name)) {
@@ -9052,7 +9216,7 @@ var LangGraphAgent = (_class26 = class extends BaseAgent {
     }
     this.maxSteps = maxSteps;
   }
-}, _class26);
+}, _class27);
 
 // src/agents/agent-builder.ts
 
@@ -9124,17 +9288,17 @@ var RunConfig = class {
    */
   
   constructor(config) {
-    this.speechConfig = _optionalChain([config, 'optionalAccess', _235 => _235.speechConfig]);
-    this.responseModalities = _optionalChain([config, 'optionalAccess', _236 => _236.responseModalities]);
-    this.saveInputBlobsAsArtifacts = _optionalChain([config, 'optionalAccess', _237 => _237.saveInputBlobsAsArtifacts]) || false;
-    this.supportCFC = _optionalChain([config, 'optionalAccess', _238 => _238.supportCFC]) || false;
-    this.streamingMode = _optionalChain([config, 'optionalAccess', _239 => _239.streamingMode]) || "NONE" /* NONE */;
-    this.outputAudioTranscription = _optionalChain([config, 'optionalAccess', _240 => _240.outputAudioTranscription]);
-    this.inputAudioTranscription = _optionalChain([config, 'optionalAccess', _241 => _241.inputAudioTranscription]);
-    this.realtimeInputConfig = _optionalChain([config, 'optionalAccess', _242 => _242.realtimeInputConfig]);
-    this.enableAffectiveDialog = _optionalChain([config, 'optionalAccess', _243 => _243.enableAffectiveDialog]);
-    this.proactivity = _optionalChain([config, 'optionalAccess', _244 => _244.proactivity]);
-    this.maxLlmCalls = _nullishCoalesce(_optionalChain([config, 'optionalAccess', _245 => _245.maxLlmCalls]), () => ( 500));
+    this.speechConfig = _optionalChain([config, 'optionalAccess', _242 => _242.speechConfig]);
+    this.responseModalities = _optionalChain([config, 'optionalAccess', _243 => _243.responseModalities]);
+    this.saveInputBlobsAsArtifacts = _optionalChain([config, 'optionalAccess', _244 => _244.saveInputBlobsAsArtifacts]) || false;
+    this.supportCFC = _optionalChain([config, 'optionalAccess', _245 => _245.supportCFC]) || false;
+    this.streamingMode = _optionalChain([config, 'optionalAccess', _246 => _246.streamingMode]) || "NONE" /* NONE */;
+    this.outputAudioTranscription = _optionalChain([config, 'optionalAccess', _247 => _247.outputAudioTranscription]);
+    this.inputAudioTranscription = _optionalChain([config, 'optionalAccess', _248 => _248.inputAudioTranscription]);
+    this.realtimeInputConfig = _optionalChain([config, 'optionalAccess', _249 => _249.realtimeInputConfig]);
+    this.enableAffectiveDialog = _optionalChain([config, 'optionalAccess', _250 => _250.enableAffectiveDialog]);
+    this.proactivity = _optionalChain([config, 'optionalAccess', _251 => _251.proactivity]);
+    this.maxLlmCalls = _nullishCoalesce(_optionalChain([config, 'optionalAccess', _252 => _252.maxLlmCalls]), () => ( 500));
     this.validateMaxLlmCalls();
   }
   /**
@@ -9155,8 +9319,8 @@ var RunConfig = class {
 };
 
 // src/artifacts/in-memory-artifact-service.ts
-var InMemoryArtifactService = (_class27 = class {constructor() { _class27.prototype.__init48.call(this); }
-  __init48() {this.artifacts = /* @__PURE__ */ new Map()}
+var InMemoryArtifactService = (_class28 = class {constructor() { _class28.prototype.__init49.call(this); }
+  __init49() {this.artifacts = /* @__PURE__ */ new Map()}
   fileHasUserNamespace(filename) {
     return filename.startsWith("user:");
   }
@@ -9229,7 +9393,7 @@ var InMemoryArtifactService = (_class27 = class {constructor() { _class27.protot
     }
     return Array.from({ length: versions.length }, (_, i) => i);
   }
-}, _class27);
+}, _class28);
 
 // src/runners.ts
 init_logger();
@@ -9256,15 +9420,15 @@ function _extractWordsLower(text) {
   const words = text.match(/[A-Za-z]+/g) || [];
   return new Set(words.map((word) => word.toLowerCase()));
 }
-var InMemoryMemoryService = (_class28 = class {
+var InMemoryMemoryService = (_class29 = class {
   /**
    * Keys are app_name/user_id, session_id. Values are session event lists.
    */
-  __init49() {this._sessionEvents = /* @__PURE__ */ new Map()}
+  __init50() {this._sessionEvents = /* @__PURE__ */ new Map()}
   /**
    * Constructor for InMemoryMemoryService
    */
-  constructor() {;_class28.prototype.__init49.call(this);
+  constructor() {;_class29.prototype.__init50.call(this);
     this._sessionEvents = /* @__PURE__ */ new Map();
   }
   /**
@@ -9278,7 +9442,7 @@ var InMemoryMemoryService = (_class28 = class {
     }
     const userSessions = this._sessionEvents.get(userKey);
     const filteredEvents = session.events.filter(
-      (event) => _optionalChain([event, 'access', _246 => _246.content, 'optionalAccess', _247 => _247.parts])
+      (event) => _optionalChain([event, 'access', _253 => _253.content, 'optionalAccess', _254 => _254.parts])
     );
     userSessions.set(session.id, filteredEvents);
   }
@@ -9348,7 +9512,7 @@ var InMemoryMemoryService = (_class28 = class {
   clear() {
     this._sessionEvents.clear();
   }
-}, _class28);
+}, _class29);
 
 // src/sessions/in-memory-session-service.ts
 var _crypto = require('crypto');
@@ -9390,19 +9554,19 @@ var BaseSessionService = class {
 };
 
 // src/sessions/in-memory-session-service.ts
-var InMemorySessionService = (_class29 = class extends BaseSessionService {constructor(...args2) { super(...args2); _class29.prototype.__init50.call(this);_class29.prototype.__init51.call(this);_class29.prototype.__init52.call(this); }
+var InMemorySessionService = (_class30 = class extends BaseSessionService {constructor(...args2) { super(...args2); _class30.prototype.__init51.call(this);_class30.prototype.__init52.call(this);_class30.prototype.__init53.call(this); }
   /**
    * A map from app name to a map from user ID to a map from session ID to session.
    */
-  __init50() {this.sessions = /* @__PURE__ */ new Map()}
+  __init51() {this.sessions = /* @__PURE__ */ new Map()}
   /**
    * A map from app name to a map from user ID to a map from key to the value.
    */
-  __init51() {this.userState = /* @__PURE__ */ new Map()}
+  __init52() {this.userState = /* @__PURE__ */ new Map()}
   /**
    * A map from app name to a map from key to the value.
    */
-  __init52() {this.appState = /* @__PURE__ */ new Map()}
+  __init53() {this.appState = /* @__PURE__ */ new Map()}
   /**
    * Creates a new session.
    */
@@ -9417,7 +9581,7 @@ var InMemorySessionService = (_class29 = class extends BaseSessionService {const
     return this.createSessionImpl(appName, userId, state, sessionId);
   }
   createSessionImpl(appName, userId, state, sessionId) {
-    const finalSessionId = _optionalChain([sessionId, 'optionalAccess', _248 => _248.trim, 'call', _249 => _249()]) || _crypto.randomUUID.call(void 0, );
+    const finalSessionId = _optionalChain([sessionId, 'optionalAccess', _255 => _255.trim, 'call', _256 => _256()]) || _crypto.randomUUID.call(void 0, );
     const session = {
       appName,
       userId,
@@ -9574,7 +9738,7 @@ var InMemorySessionService = (_class29 = class extends BaseSessionService {const
       warning(`sessionId ${sessionId} not in sessions[appName][userId]`);
       return event;
     }
-    if (_optionalChain([event, 'access', _250 => _250.actions, 'optionalAccess', _251 => _251.stateDelta])) {
+    if (_optionalChain([event, 'access', _257 => _257.actions, 'optionalAccess', _258 => _258.stateDelta])) {
       for (const key in event.actions.stateDelta) {
         const value = event.actions.stateDelta[key];
         if (key.startsWith(State.APP_PREFIX)) {
@@ -9599,7 +9763,7 @@ var InMemorySessionService = (_class29 = class extends BaseSessionService {const
     storageSession.lastUpdateTime = event.timestamp;
     return event;
   }
-}, _class29);
+}, _class30);
 
 // src/runners.ts
 function _findFunctionCallEventIfLastEventIsFunctionResponse(session) {
@@ -9608,14 +9772,14 @@ function _findFunctionCallEventIfLastEventIsFunctionResponse(session) {
     return null;
   }
   const lastEvent = events[events.length - 1];
-  if (_optionalChain([lastEvent, 'access', _252 => _252.content, 'optionalAccess', _253 => _253.parts, 'optionalAccess', _254 => _254.some, 'call', _255 => _255((part) => part.functionResponse)])) {
-    const functionCallId = _optionalChain([lastEvent, 'access', _256 => _256.content, 'access', _257 => _257.parts, 'access', _258 => _258.find, 'call', _259 => _259(
+  if (_optionalChain([lastEvent, 'access', _259 => _259.content, 'optionalAccess', _260 => _260.parts, 'optionalAccess', _261 => _261.some, 'call', _262 => _262((part) => part.functionResponse)])) {
+    const functionCallId = _optionalChain([lastEvent, 'access', _263 => _263.content, 'access', _264 => _264.parts, 'access', _265 => _265.find, 'call', _266 => _266(
       (part) => part.functionResponse
-    ), 'optionalAccess', _260 => _260.functionResponse, 'optionalAccess', _261 => _261.id]);
+    ), 'optionalAccess', _267 => _267.functionResponse, 'optionalAccess', _268 => _268.id]);
     if (!functionCallId) return null;
     for (let i = events.length - 2; i >= 0; i--) {
       const event = events[i];
-      const functionCalls = _optionalChain([event, 'access', _262 => _262.getFunctionCalls, 'optionalCall', _263 => _263()]) || [];
+      const functionCalls = _optionalChain([event, 'access', _269 => _269.getFunctionCalls, 'optionalCall', _270 => _270()]) || [];
       for (const functionCall of functionCalls) {
         if (functionCall.id === functionCallId) {
           return event;
@@ -9625,7 +9789,7 @@ function _findFunctionCallEventIfLastEventIsFunctionResponse(session) {
   }
   return null;
 }
-var Runner = (_class30 = class {
+var Runner = (_class31 = class {
   /**
    * The app name of the runner.
    */
@@ -9646,7 +9810,7 @@ var Runner = (_class30 = class {
    * The memory service for the runner.
    */
   
-  __init53() {this.logger = new Logger({ name: "Runner" })}
+  __init54() {this.logger = new Logger({ name: "Runner" })}
   /**
    * Initializes the Runner.
    */
@@ -9656,7 +9820,7 @@ var Runner = (_class30 = class {
     artifactService,
     sessionService,
     memoryService
-  }) {;_class30.prototype.__init53.call(this);
+  }) {;_class31.prototype.__init54.call(this);
     this.appName = appName;
     this.agent = agent;
     this.artifactService = artifactService;
@@ -9745,6 +9909,9 @@ var Runner = (_class30 = class {
       )) {
         if (!event.partial) {
           await this.sessionService.appendEvent(session, event);
+          if (this.memoryService) {
+            await this.memoryService.addSessionToMemory(session);
+          }
         }
         yield event;
       }
@@ -9803,15 +9970,15 @@ var Runner = (_class30 = class {
    */
   _findAgentToRun(session, rootAgent) {
     const event = _findFunctionCallEventIfLastEventIsFunctionResponse(session);
-    if (_optionalChain([event, 'optionalAccess', _264 => _264.author])) {
+    if (_optionalChain([event, 'optionalAccess', _271 => _271.author])) {
       return rootAgent.findAgent(event.author);
     }
-    const nonUserEvents = _optionalChain([session, 'access', _265 => _265.events, 'optionalAccess', _266 => _266.filter, 'call', _267 => _267((e) => e.author !== "user"), 'access', _268 => _268.reverse, 'call', _269 => _269()]) || [];
+    const nonUserEvents = _optionalChain([session, 'access', _272 => _272.events, 'optionalAccess', _273 => _273.filter, 'call', _274 => _274((e) => e.author !== "user"), 'access', _275 => _275.reverse, 'call', _276 => _276()]) || [];
     for (const event2 of nonUserEvents) {
       if (event2.author === rootAgent.name) {
         return rootAgent;
       }
-      const agent = _optionalChain([rootAgent, 'access', _270 => _270.findSubAgent, 'optionalCall', _271 => _271(event2.author)]);
+      const agent = _optionalChain([rootAgent, 'access', _277 => _277.findSubAgent, 'optionalCall', _278 => _278(event2.author)]);
       if (!agent) {
         this.logger.debug(
           `Event from an unknown agent: ${event2.author}, event id: ${event2.id}`
@@ -9860,7 +10027,7 @@ var Runner = (_class30 = class {
       runConfig
     });
   }
-}, _class30);
+}, _class31);
 var InMemoryRunner = class extends Runner {
   /**
    * Deprecated. Please don't use. The in-memory session service for the runner.
@@ -9883,16 +10050,18 @@ var InMemoryRunner = class extends Runner {
 };
 
 // src/agents/agent-builder.ts
-var AgentBuilder = (_class31 = class _AgentBuilder {
+var AgentBuilder = (_class32 = class _AgentBuilder {
+  
+  
   
   
   
+  __init55() {this.agentType = "llm"}
   
-  __init54() {this.agentType = "llm"}
   /**
    * Private constructor - use static create() method
    */
-  constructor(name) {;_class31.prototype.__init54.call(this);
+  constructor(name) {;_class32.prototype.__init55.call(this);
     this.config = { name };
   }
   /**
@@ -9956,6 +10125,51 @@ var AgentBuilder = (_class31 = class _AgentBuilder {
     this.config.planner = planner;
     return this;
   }
+  /**
+   * Set the code executor for the agent
+   * @param codeExecutor The code executor to use for running code
+   * @returns This builder instance for chaining
+   */
+  withCodeExecutor(codeExecutor) {
+    this.config.codeExecutor = codeExecutor;
+    return this;
+  }
+  /**
+   * Set the output key for the agent
+   * @param outputKey The output key in session state to store the output of the agent
+   * @returns This builder instance for chaining
+   */
+  withOutputKey(outputKey) {
+    this.config.outputKey = outputKey;
+    return this;
+  }
+  /**
+   * Add sub-agents to the agent
+   * @param subAgents Sub-agents to add to the agent
+   * @returns This builder instance for chaining
+   */
+  withSubAgents(subAgents) {
+    this.config.subAgents = subAgents;
+    return this;
+  }
+  /**
+   * Set the before agent callback
+   * @param callback Callback to invoke before agent execution
+   * @returns This builder instance for chaining
+   */
+  withBeforeAgentCallback(callback) {
+    this.config.beforeAgentCallback = callback;
+    return this;
+  }
+  /**
+   * Set the after agent callback
+   * @param callback Callback to invoke after agent execution
+   * @returns This builder instance for chaining
+   */
+  withAfterAgentCallback(callback) {
+    this.config.afterAgentCallback = callback;
+    return this;
+  }
   /**
    * Configure as a sequential agent
    * @param subAgents Sub-agents to execute in sequence
@@ -10006,14 +10220,38 @@ var AgentBuilder = (_class31 = class _AgentBuilder {
    * @param options Session configuration options (userId and appName)
    * @returns This builder instance for chaining
    */
-  withSession(service, options = {}) {
-    this.sessionConfig = {
-      service,
+  withSessionService(service, options = {}) {
+    this.sessionService = service;
+    this.sessionOptions = {
       userId: options.userId || this.generateDefaultUserId(),
-      appName: options.appName || this.generateDefaultAppName()
+      appName: options.appName || this.generateDefaultAppName(),
+      state: options.state,
+      sessionId: options.sessionId
     };
     return this;
   }
+  /**
+   * Configure with an existing session instance
+   * @param session Existing session to use
+   * @returns This builder instance for chaining
+   * @throws Error if no session service has been configured via withSessionService()
+   */
+  withSession(session) {
+    if (!this.sessionService) {
+      throw new Error(
+        "Session service must be configured before using withSession(). Call withSessionService() first, or use withQuickSession() for in-memory sessions."
+      );
+    }
+    this.sessionOptions = {
+      ...this.sessionOptions,
+      userId: session.userId,
+      appName: session.appName,
+      sessionId: session.id,
+      state: session.state
+    };
+    this.existingSession = session;
+    return this;
+  }
   /**
    * Configure memory service for the agent
    * @param memoryService Memory service to use for conversation history and context
@@ -10039,7 +10277,7 @@ var AgentBuilder = (_class31 = class _AgentBuilder {
    * @returns This builder instance for chaining
    */
   withQuickSession(options = {}) {
-    return this.withSession(new InMemorySessionService(), options);
+    return this.withSessionService(new InMemorySessionService(), options);
   }
   /**
    * Build the agent and optionally create runner and session
@@ -10049,18 +10287,24 @@ var AgentBuilder = (_class31 = class _AgentBuilder {
     const agent = this.createAgent();
     let runner;
     let session;
-    if (!this.sessionConfig) {
+    if (!this.sessionService) {
       this.withQuickSession();
     }
-    if (this.sessionConfig) {
-      session = await this.sessionConfig.service.createSession(
-        this.sessionConfig.appName,
-        this.sessionConfig.userId
-      );
+    if (this.sessionService && this.sessionOptions) {
+      if (this.existingSession) {
+        session = this.existingSession;
+      } else {
+        session = await this.sessionService.createSession(
+          this.sessionOptions.appName,
+          this.sessionOptions.userId,
+          this.sessionOptions.state,
+          this.sessionOptions.sessionId
+        );
+      }
       const runnerConfig = {
-        appName: this.sessionConfig.appName,
+        appName: this.sessionOptions.appName,
         agent,
-        sessionService: this.sessionConfig.service,
+        sessionService: this.sessionService,
         memoryService: this.memoryService,
         artifactService: this.artifactService
       };
@@ -10095,7 +10339,15 @@ var AgentBuilder = (_class31 = class _AgentBuilder {
           description: this.config.description,
           instruction: this.config.instruction,
           tools: this.config.tools,
-          planner: this.config.planner
+          planner: this.config.planner,
+          codeExecutor: this.config.codeExecutor,
+          subAgents: this.config.subAgents,
+          beforeAgentCallback: this.config.beforeAgentCallback,
+          afterAgentCallback: this.config.afterAgentCallback,
+          memoryService: this.memoryService,
+          artifactService: this.artifactService,
+          outputKey: this.config.outputKey,
+          sessionService: this.sessionService
         });
       }
       case "sequential":
@@ -10160,20 +10412,20 @@ var AgentBuilder = (_class31 = class _AgentBuilder {
    * @returns Enhanced runner with simplified API
    */
   createEnhancedRunner(baseRunner, session) {
-    const sessionConfig = this.sessionConfig;
+    const sessionOptions = this.sessionOptions;
     return {
       async ask(message) {
         const newMessage = typeof message === "string" ? { parts: [{ text: message }] } : typeof message === "object" && "contents" in message ? { parts: message.contents[message.contents.length - 1].parts } : message;
         let response = "";
-        if (!sessionConfig) {
+        if (!_optionalChain([sessionOptions, 'optionalAccess', _279 => _279.userId])) {
           throw new Error("Session configuration is required");
         }
         for await (const event of baseRunner.runAsync({
-          userId: sessionConfig.userId,
+          userId: sessionOptions.userId,
           sessionId: session.id,
           newMessage
         })) {
-          if (_optionalChain([event, 'access', _272 => _272.content, 'optionalAccess', _273 => _273.parts]) && Array.isArray(event.content.parts)) {
+          if (_optionalChain([event, 'access', _280 => _280.content, 'optionalAccess', _281 => _281.parts]) && Array.isArray(event.content.parts)) {
             const content = event.content.parts.map(
               (part) => (part && typeof part === "object" && "text" in part ? part.text : "") || ""
             ).join("");
@@ -10182,14 +10434,14 @@ var AgentBuilder = (_class31 = class _AgentBuilder {
             }
           }
         }
-        return response;
+        return response.trim();
       },
       runAsync(params) {
         return baseRunner.runAsync(params);
       }
     };
   }
-}, _class31);
+}, _class32);
 
 // src/memory/index.ts
 var memory_exports = {};
@@ -10253,7 +10505,7 @@ var VertexAiSessionService = class extends BaseSessionService {
         path: `operations/${operationId}`,
         request_dict: {}
       });
-      if (_optionalChain([lroResponse, 'optionalAccess', _274 => _274.done])) {
+      if (_optionalChain([lroResponse, 'optionalAccess', _282 => _282.done])) {
         break;
       }
       await new Promise((resolve) => setTimeout(resolve, 1e3));
@@ -10523,11 +10775,11 @@ var VertexAiSessionService = class extends BaseSessionService {
 
 // src/sessions/database-session-service.ts
 var _kysely = require('kysely');
-var DatabaseSessionService = (_class32 = class extends BaseSessionService {
+var DatabaseSessionService = (_class33 = class extends BaseSessionService {
   
-  __init55() {this.initialized = false}
+  __init56() {this.initialized = false}
   constructor(config) {
-    super();_class32.prototype.__init55.call(this);;
+    super();_class33.prototype.__init56.call(this);;
     this.db = config.db;
     if (!config.skipTableCreation) {
       this.initializeDatabase().catch((error) => {
@@ -10602,7 +10854,7 @@ var DatabaseSessionService = (_class32 = class extends BaseSessionService {
     if (!jsonString) return defaultValue;
     try {
       return JSON.parse(jsonString);
-    } catch (e4) {
+    } catch (e5) {
       return defaultValue;
     }
   }
@@ -10624,12 +10876,12 @@ var DatabaseSessionService = (_class32 = class extends BaseSessionService {
   }
   async createSession(appName, userId, state, sessionId) {
     await this.ensureInitialized();
-    const id = _optionalChain([sessionId, 'optionalAccess', _275 => _275.trim, 'call', _276 => _276()]) || this.generateSessionId();
+    const id = _optionalChain([sessionId, 'optionalAccess', _283 => _283.trim, 'call', _284 => _284()]) || this.generateSessionId();
     return await this.db.transaction().execute(async (trx) => {
       const appState = await trx.selectFrom("app_states").selectAll().where("app_name", "=", appName).executeTakeFirst();
       const userState = await trx.selectFrom("user_states").selectAll().where("app_name", "=", appName).where("user_id", "=", userId).executeTakeFirst();
-      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _277 => _277.state]), {});
-      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _278 => _278.state]), {});
+      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _285 => _285.state]), {});
+      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _286 => _286.state]), {});
       if (!appState) {
         await trx.insertInto("app_states").values({
           app_name: appName,
@@ -10688,21 +10940,21 @@ var DatabaseSessionService = (_class32 = class extends BaseSessionService {
         return void 0;
       }
       let eventQuery = trx.selectFrom("events").selectAll().where("session_id", "=", sessionId).orderBy("timestamp", "desc");
-      if (_optionalChain([config, 'optionalAccess', _279 => _279.afterTimestamp])) {
+      if (_optionalChain([config, 'optionalAccess', _287 => _287.afterTimestamp])) {
         eventQuery = eventQuery.where(
           "timestamp",
           ">=",
           new Date(config.afterTimestamp * 1e3)
         );
       }
-      if (_optionalChain([config, 'optionalAccess', _280 => _280.numRecentEvents])) {
+      if (_optionalChain([config, 'optionalAccess', _288 => _288.numRecentEvents])) {
         eventQuery = eventQuery.limit(config.numRecentEvents);
       }
       const storageEvents = await eventQuery.execute();
       const appState = await trx.selectFrom("app_states").selectAll().where("app_name", "=", appName).executeTakeFirst();
       const userState = await trx.selectFrom("user_states").selectAll().where("app_name", "=", appName).where("user_id", "=", userId).executeTakeFirst();
-      const currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _281 => _281.state]), {});
-      const currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _282 => _282.state]), {});
+      const currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _289 => _289.state]), {});
+      const currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _290 => _290.state]), {});
       const sessionState = this.parseJsonSafely(storageSession.state, {});
       const mergedState = this.mergeState(
         currentAppState,
@@ -10760,13 +11012,13 @@ var DatabaseSessionService = (_class32 = class extends BaseSessionService {
       }
       const appState = await trx.selectFrom("app_states").selectAll().where("app_name", "=", session.appName).executeTakeFirst();
       const userState = await trx.selectFrom("user_states").selectAll().where("app_name", "=", session.appName).where("user_id", "=", session.userId).executeTakeFirst();
-      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _283 => _283.state]), {});
-      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _284 => _284.state]), {});
+      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _291 => _291.state]), {});
+      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _292 => _292.state]), {});
       let sessionState = this.parseJsonSafely(storageSession.state, {});
       let appStateDelta = {};
       let userStateDelta = {};
       let sessionStateDelta = {};
-      if (_optionalChain([event, 'access', _285 => _285.actions, 'optionalAccess', _286 => _286.stateDelta])) {
+      if (_optionalChain([event, 'access', _293 => _293.actions, 'optionalAccess', _294 => _294.stateDelta])) {
         const deltas = this.extractStateDelta(event.actions.stateDelta);
         appStateDelta = deltas.appStateDelta;
         userStateDelta = deltas.userStateDelta;
@@ -10912,7 +11164,7 @@ var DatabaseSessionService = (_class32 = class extends BaseSessionService {
    * Overrides the base class method to work with plain object state.
    */
   updateSessionState(session, event) {
-    if (!_optionalChain([event, 'access', _287 => _287.actions, 'optionalAccess', _288 => _288.stateDelta])) {
+    if (!_optionalChain([event, 'access', _295 => _295.actions, 'optionalAccess', _296 => _296.stateDelta])) {
       return;
     }
     for (const [key, value] of Object.entries(event.actions.stateDelta)) {
@@ -10921,10 +11173,10 @@ var DatabaseSessionService = (_class32 = class extends BaseSessionService {
       }
     }
   }
-}, _class32);
+}, _class33);
 
 // src/sessions/database-factories.ts
-var _dedent = require('dedent'); var _dedent2 = _interopRequireDefault(_dedent);
+
 
 function createDependencyError(packageName, dbType) {
   return new Error(
@@ -11082,7 +11334,7 @@ var GcsArtifactService = class {
       };
       return part;
     } catch (error) {
-      if (_optionalChain([error, 'optionalAccess', _289 => _289.code]) === 404) {
+      if (_optionalChain([error, 'optionalAccess', _297 => _297.code]) === 404) {
         return null;
       }
       throw error;
@@ -11328,4 +11580,5 @@ var VERSION = "0.1.0";
 
 
 
-exports.AF_FUNCTION_CALL_ID_PREFIX = AF_FUNCTION_CALL_ID_PREFIX; exports.Agent = LlmAgent; exports.AgentBuilder = AgentBuilder; exports.Agents = agents_exports; exports.AiSdkLlm = AiSdkLlm; exports.AnthropicLlm = AnthropicLlm; exports.ApiKeyCredential = ApiKeyCredential; exports.ApiKeyScheme = ApiKeyScheme; exports.AuthConfig = AuthConfig; exports.AuthCredential = AuthCredential; exports.AuthCredentialType = AuthCredentialType; exports.AuthHandler = AuthHandler; exports.AuthScheme = AuthScheme; exports.AuthSchemeType = AuthSchemeType; exports.AuthTool = AuthTool; exports.AutoFlow = AutoFlow; exports.BaseAgent = BaseAgent; exports.BaseCodeExecutor = BaseCodeExecutor; exports.BaseLLMConnection = BaseLLMConnection; exports.BaseLlm = BaseLlm; exports.BaseLlmFlow = BaseLlmFlow; exports.BaseLlmRequestProcessor = BaseLlmRequestProcessor; exports.BaseLlmResponseProcessor = BaseLlmResponseProcessor; exports.BasePlanner = BasePlanner; exports.BaseSessionService = BaseSessionService; exports.BaseTool = BaseTool; exports.BasicAuthCredential = BasicAuthCredential; exports.BearerTokenCredential = BearerTokenCredential; exports.BuiltInCodeExecutor = BuiltInCodeExecutor; exports.BuiltInPlanner = BuiltInPlanner; exports.CallbackContext = CallbackContext; exports.CodeExecutionUtils = CodeExecutionUtils; exports.CodeExecutorContext = CodeExecutorContext; exports.DatabaseSessionService = DatabaseSessionService; exports.EnhancedAuthConfig = EnhancedAuthConfig; exports.Event = Event; exports.EventActions = EventActions; exports.Events = events_exports; exports.ExitLoopTool = ExitLoopTool; exports.FileOperationsTool = FileOperationsTool; exports.Flows = flows_exports; exports.FunctionTool = FunctionTool; exports.GcsArtifactService = GcsArtifactService; exports.GetUserChoiceTool = GetUserChoiceTool; exports.GoogleLlm = GoogleLlm; exports.GoogleSearch = GoogleSearch; exports.HttpRequestTool = HttpRequestTool; exports.HttpScheme = HttpScheme; exports.InMemoryArtifactService = InMemoryArtifactService; exports.InMemoryMemoryService = InMemoryMemoryService; exports.InMemoryRunner = InMemoryRunner; exports.InMemorySessionService = InMemorySessionService; exports.InvocationContext = InvocationContext; exports.LLMRegistry = LLMRegistry; exports.LangGraphAgent = LangGraphAgent; exports.LlmAgent = LlmAgent; exports.LlmCallsLimitExceededError = LlmCallsLimitExceededError; exports.LlmRequest = LlmRequest; exports.LlmResponse = LlmResponse; exports.LoadArtifactsTool = LoadArtifactsTool; exports.LoadMemoryTool = LoadMemoryTool; exports.LoopAgent = LoopAgent; exports.McpAbi = McpAbi; exports.McpAtp = McpAtp; exports.McpBamm = McpBamm; exports.McpCoinGecko = McpCoinGecko; exports.McpDiscord = McpDiscord; exports.McpError = McpError; exports.McpErrorType = McpErrorType; exports.McpFilesystem = McpFilesystem; exports.McpFraxlend = McpFraxlend; exports.McpGeneric = McpGeneric; exports.McpIqWiki = McpIqWiki; exports.McpMemory = McpMemory; exports.McpNearAgent = McpNearAgent; exports.McpNearIntents = McpNearIntents; exports.McpOdos = McpOdos; exports.McpSamplingHandler = McpSamplingHandler; exports.McpTelegram = McpTelegram; exports.McpToolset = McpToolset; exports.Memory = memory_exports; exports.Models = models_exports; exports.OAuth2Credential = OAuth2Credential; exports.OAuth2Scheme = OAuth2Scheme; exports.OpenAiLlm = OpenAiLlm; exports.OpenIdConnectScheme = OpenIdConnectScheme; exports.ParallelAgent = ParallelAgent; exports.PlanReActPlanner = PlanReActPlanner; exports.REQUEST_EUC_FUNCTION_CALL_NAME = REQUEST_EUC_FUNCTION_CALL_NAME; exports.ReadonlyContext = ReadonlyContext; exports.RunConfig = RunConfig; exports.Runner = Runner; exports.SequentialAgent = SequentialAgent; exports.Sessions = sessions_exports; exports.SingleFlow = SingleFlow; exports.State = State; exports.StreamingMode = StreamingMode; exports.TelemetryService = TelemetryService; exports.ToolContext = ToolContext; exports.Tools = tools_exports; exports.TransferToAgentTool = TransferToAgentTool; exports.UserInteractionTool = UserInteractionTool; exports.VERSION = VERSION; exports.VertexAiSessionService = VertexAiSessionService; exports._findFunctionCallEventIfLastEventIsFunctionResponse = _findFunctionCallEventIfLastEventIsFunctionResponse; exports.adkToMcpToolType = adkToMcpToolType; exports.agentTransferRequestProcessor = requestProcessor8; exports.basicRequestProcessor = requestProcessor2; exports.buildFunctionDeclaration = buildFunctionDeclaration; exports.codeExecutionRequestProcessor = requestProcessor3; exports.codeExecutionResponseProcessor = responseProcessor; exports.contentRequestProcessor = requestProcessor4; exports.createAuthToolArguments = createAuthToolArguments; exports.createBranchContextForSubAgent = createBranchContextForSubAgent; exports.createDatabaseSessionService = createDatabaseSessionService; exports.createFunctionTool = createFunctionTool; exports.createMysqlSessionService = createMysqlSessionService; exports.createPostgresSessionService = createPostgresSessionService; exports.createSamplingHandler = createSamplingHandler; exports.createSqliteSessionService = createSqliteSessionService; exports.createTool = createTool; exports.generateAuthEvent = generateAuthEvent; exports.generateClientFunctionCallId = generateClientFunctionCallId; exports.getLongRunningFunctionCalls = getLongRunningFunctionCalls; exports.getMcpTools = getMcpTools; exports.handleFunctionCallsAsync = handleFunctionCallsAsync; exports.handleFunctionCallsLive = handleFunctionCallsLive; exports.identityRequestProcessor = requestProcessor5; exports.initializeTelemetry = initializeTelemetry; exports.injectSessionState = injectSessionState; exports.instructionsRequestProcessor = requestProcessor6; exports.isEnhancedAuthConfig = isEnhancedAuthConfig; exports.jsonSchemaToDeclaration = jsonSchemaToDeclaration; exports.mcpSchemaToParameters = mcpSchemaToParameters; exports.mergeAgentRun = mergeAgentRun; exports.mergeParallelFunctionResponseEvents = mergeParallelFunctionResponseEvents; exports.newInvocationContextId = newInvocationContextId; exports.nlPlanningRequestProcessor = requestProcessor7; exports.nlPlanningResponseProcessor = responseProcessor2; exports.normalizeJsonSchema = normalizeJsonSchema; exports.populateClientFunctionCallId = populateClientFunctionCallId; exports.registerProviders = registerProviders; exports.removeClientFunctionCallId = removeClientFunctionCallId; exports.requestProcessor = requestProcessor; exports.shutdownTelemetry = shutdownTelemetry; exports.telemetryService = telemetryService; exports.traceLlmCall = traceLlmCall; exports.traceToolCall = traceToolCall; exports.tracer = tracer;
+
+exports.AF_FUNCTION_CALL_ID_PREFIX = AF_FUNCTION_CALL_ID_PREFIX; exports.Agent = LlmAgent; exports.AgentBuilder = AgentBuilder; exports.AgentTool = AgentTool; exports.Agents = agents_exports; exports.AiSdkLlm = AiSdkLlm; exports.AnthropicLlm = AnthropicLlm; exports.ApiKeyCredential = ApiKeyCredential; exports.ApiKeyScheme = ApiKeyScheme; exports.AuthConfig = AuthConfig; exports.AuthCredential = AuthCredential; exports.AuthCredentialType = AuthCredentialType; exports.AuthHandler = AuthHandler; exports.AuthScheme = AuthScheme; exports.AuthSchemeType = AuthSchemeType; exports.AuthTool = AuthTool; exports.AutoFlow = AutoFlow; exports.BaseAgent = BaseAgent; exports.BaseCodeExecutor = BaseCodeExecutor; exports.BaseLLMConnection = BaseLLMConnection; exports.BaseLlm = BaseLlm; exports.BaseLlmFlow = BaseLlmFlow; exports.BaseLlmRequestProcessor = BaseLlmRequestProcessor; exports.BaseLlmResponseProcessor = BaseLlmResponseProcessor; exports.BasePlanner = BasePlanner; exports.BaseSessionService = BaseSessionService; exports.BaseTool = BaseTool; exports.BasicAuthCredential = BasicAuthCredential; exports.BearerTokenCredential = BearerTokenCredential; exports.BuiltInCodeExecutor = BuiltInCodeExecutor; exports.BuiltInPlanner = BuiltInPlanner; exports.CallbackContext = CallbackContext; exports.CodeExecutionUtils = CodeExecutionUtils; exports.CodeExecutorContext = CodeExecutorContext; exports.DatabaseSessionService = DatabaseSessionService; exports.EnhancedAuthConfig = EnhancedAuthConfig; exports.Event = Event; exports.EventActions = EventActions; exports.Events = events_exports; exports.ExitLoopTool = ExitLoopTool; exports.FileOperationsTool = FileOperationsTool; exports.Flows = flows_exports; exports.FunctionTool = FunctionTool; exports.GcsArtifactService = GcsArtifactService; exports.GetUserChoiceTool = GetUserChoiceTool; exports.GoogleLlm = GoogleLlm; exports.GoogleSearch = GoogleSearch; exports.HttpRequestTool = HttpRequestTool; exports.HttpScheme = HttpScheme; exports.InMemoryArtifactService = InMemoryArtifactService; exports.InMemoryMemoryService = InMemoryMemoryService; exports.InMemoryRunner = InMemoryRunner; exports.InMemorySessionService = InMemorySessionService; exports.InvocationContext = InvocationContext; exports.LLMRegistry = LLMRegistry; exports.LangGraphAgent = LangGraphAgent; exports.LlmAgent = LlmAgent; exports.LlmCallsLimitExceededError = LlmCallsLimitExceededError; exports.LlmRequest = LlmRequest; exports.LlmResponse = LlmResponse; exports.LoadArtifactsTool = LoadArtifactsTool; exports.LoadMemoryTool = LoadMemoryTool; exports.LoopAgent = LoopAgent; exports.McpAbi = McpAbi; exports.McpAtp = McpAtp; exports.McpBamm = McpBamm; exports.McpCoinGecko = McpCoinGecko; exports.McpDiscord = McpDiscord; exports.McpError = McpError; exports.McpErrorType = McpErrorType; exports.McpFilesystem = McpFilesystem; exports.McpFraxlend = McpFraxlend; exports.McpGeneric = McpGeneric; exports.McpIqWiki = McpIqWiki; exports.McpMemory = McpMemory; exports.McpNearAgent = McpNearAgent; exports.McpNearIntents = McpNearIntents; exports.McpOdos = McpOdos; exports.McpSamplingHandler = McpSamplingHandler; exports.McpTelegram = McpTelegram; exports.McpToolset = McpToolset; exports.Memory = memory_exports; exports.Models = models_exports; exports.OAuth2Credential = OAuth2Credential; exports.OAuth2Scheme = OAuth2Scheme; exports.OpenAiLlm = OpenAiLlm; exports.OpenIdConnectScheme = OpenIdConnectScheme; exports.ParallelAgent = ParallelAgent; exports.PlanReActPlanner = PlanReActPlanner; exports.REQUEST_EUC_FUNCTION_CALL_NAME = REQUEST_EUC_FUNCTION_CALL_NAME; exports.ReadonlyContext = ReadonlyContext; exports.RunConfig = RunConfig; exports.Runner = Runner; exports.SequentialAgent = SequentialAgent; exports.Sessions = sessions_exports; exports.SingleFlow = SingleFlow; exports.State = State; exports.StreamingMode = StreamingMode; exports.TelemetryService = TelemetryService; exports.ToolContext = ToolContext; exports.Tools = tools_exports; exports.TransferToAgentTool = TransferToAgentTool; exports.UserInteractionTool = UserInteractionTool; exports.VERSION = VERSION; exports.VertexAiSessionService = VertexAiSessionService; exports._findFunctionCallEventIfLastEventIsFunctionResponse = _findFunctionCallEventIfLastEventIsFunctionResponse; exports.adkToMcpToolType = adkToMcpToolType; exports.agentTransferRequestProcessor = requestProcessor8; exports.basicRequestProcessor = requestProcessor2; exports.buildFunctionDeclaration = buildFunctionDeclaration; exports.codeExecutionRequestProcessor = requestProcessor3; exports.codeExecutionResponseProcessor = responseProcessor; exports.contentRequestProcessor = requestProcessor4; exports.createAuthToolArguments = createAuthToolArguments; exports.createBranchContextForSubAgent = createBranchContextForSubAgent; exports.createDatabaseSessionService = createDatabaseSessionService; exports.createFunctionTool = createFunctionTool; exports.createMysqlSessionService = createMysqlSessionService; exports.createPostgresSessionService = createPostgresSessionService; exports.createSamplingHandler = createSamplingHandler; exports.createSqliteSessionService = createSqliteSessionService; exports.createTool = createTool; exports.generateAuthEvent = generateAuthEvent; exports.generateClientFunctionCallId = generateClientFunctionCallId; exports.getLongRunningFunctionCalls = getLongRunningFunctionCalls; exports.getMcpTools = getMcpTools; exports.handleFunctionCallsAsync = handleFunctionCallsAsync; exports.handleFunctionCallsLive = handleFunctionCallsLive; exports.identityRequestProcessor = requestProcessor5; exports.initializeTelemetry = initializeTelemetry; exports.injectSessionState = injectSessionState; exports.instructionsRequestProcessor = requestProcessor6; exports.isEnhancedAuthConfig = isEnhancedAuthConfig; exports.jsonSchemaToDeclaration = jsonSchemaToDeclaration; exports.mcpSchemaToParameters = mcpSchemaToParameters; exports.mergeAgentRun = mergeAgentRun; exports.mergeParallelFunctionResponseEvents = mergeParallelFunctionResponseEvents; exports.newInvocationContextId = newInvocationContextId; exports.nlPlanningRequestProcessor = requestProcessor7; exports.nlPlanningResponseProcessor = responseProcessor2; exports.normalizeJsonSchema = normalizeJsonSchema; exports.populateClientFunctionCallId = populateClientFunctionCallId; exports.registerProviders = registerProviders; exports.removeClientFunctionCallId = removeClientFunctionCallId; exports.requestProcessor = requestProcessor; exports.shutdownTelemetry = shutdownTelemetry; exports.telemetryService = telemetryService; exports.traceLlmCall = traceLlmCall; exports.traceToolCall = traceToolCall; exports.tracer = tracer;