@google/adk 0.2.4 → 0.2.5

package/dist/cjs/plugins/base_plugin.js

@@ -48,13 +48,13 @@ class BasePlugin {
    * This callback helps logging and modifying the user message before the
    * runner starts the invocation.
    *
-   * @param invocationContext The context for the entire invocation.
-   * @param userMessage The message content input by user.
+   * @param params.invocationContext The context for the entire invocation.
+   * @param params.userMessage The message content input by user.
    * @returns An optional `Content` to be returned to the ADK. Returning a
    *     value to replace the user message. Returning `undefined` to proceed
    *     normally.
    */
-  async onUserMessageCallback({ invocationContext, userMessage }) {
+  async onUserMessageCallback(params) {
     return;
   }
   /**
@@ -63,13 +63,13 @@ class BasePlugin {
    * This is the first callback to be called in the lifecycle, ideal for global
    * setup or initialization tasks.
    *
-   * @param invocationContext The context for the entire invocation, containing
+   * @param params.invocationContext The context for the entire invocation, containing
    *     session information, the root agent, etc.
    * @returns An optional `Event` to be returned to the ADK. Returning a value
    *     to halt execution of the runner and ends the runner with that event.
    *     Return `undefined` to proceed normally.
    */
-  async beforeRunCallback({ invocationContext }) {
+  async beforeRunCallback(params) {
     return;
   }
   /**
@@ -78,13 +78,13 @@ class BasePlugin {
    * This is the ideal place to make modification to the event before the event
    * is handled by the underlying agent app.
    *
-   * @param invocationContext The context for the entire invocation.
-   * @param event The event raised by the runner.
+   * @param params.invocationContext The context for the entire invocation.
+   * @param params.event The event raised by the runner.
    * @returns An optional value. A non-`undefined` return may be used by the
    *     framework to modify or replace the response. Returning `undefined`
    *     allows the original response to be used.
    */
-  async onEventCallback({ invocationContext, event }) {
+  async onEventCallback(params) {
     return;
   }
   /**
@@ -93,10 +93,10 @@ class BasePlugin {
    * This is the final callback in the ADK lifecycle, suitable for cleanup,
    * final logging, or reporting tasks.
    *
-   * @param invocationContext The context for the entire invocation.
+   * @param params.invocationContext The context for the entire invocation.
    * @returns undefined
    */
-  async afterRunCallback({ invocationContext }) {
+  async afterRunCallback(params) {
     return;
   }
   /**
@@ -105,13 +105,13 @@ class BasePlugin {
    * This callback can be used for logging, setup, or to short-circuit the
    * agent's execution by returning a value.
    *
-   * @param agent The agent that is about to run.
-   * @param callbackContext The context for the agent invocation.
+   * @param params.agent The agent that is about to run.
+   * @param params.callbackContext The context for the agent invocation.
    * @returns An optional `Content` object. If a value is returned, it will
    *     bypass the agent's callbacks and its execution, and return this value
    *     directly. Returning `undefined` allows the agent to proceed normally.
    */
-  async beforeAgentCallback({ agent, callbackContext }) {
+  async beforeAgentCallback(params) {
     return;
   }
   /**
@@ -120,13 +120,13 @@ class BasePlugin {
    * This callback can be used to inspect, log, or modify the agent's final
    * result before it is returned.
    *
-   * @param agent The agent that has just run.
-   * @param callbackContext The context for the agent invocation.
+   * @param params.agent The agent that has just run.
+   * @param params.callbackContext The context for the agent invocation.
    * @returns An optional `Content` object. If a value is returned, it will
    *     replace the agent's original result. Returning `undefined` uses the
    *     original, unmodified result.
    */
-  async afterAgentCallback({ agent, callbackContext }) {
+  async afterAgentCallback(params) {
     return;
   }
   /**
@@ -136,13 +136,13 @@ class BasePlugin {
    * object. It can also be used to implement caching by returning a cached
    * `LlmResponse`, which would skip the actual model call.
    *
-   * @param callbackContext The context for the current agent call.
-   * @param llmRequest The prepared request object to be sent to the model.
+   * @param params.callbackContext The context for the current agent call.
+   * @param params.llmRequest The prepared request object to be sent to the model.
    * @returns An optional value. The interpretation of a non-`undefined`
    *     trigger an early exit and returns the response immediately. Returning
    *     `undefined` allows the LLM request to proceed normally.
    */
-  async beforeModelCallback({ callbackContext, llmRequest }) {
+  async beforeModelCallback(params) {
     return;
   }
   /**
@@ -151,13 +151,13 @@ class BasePlugin {
    * This is the ideal place to log model responses, collect metrics on token
    * usage, or perform post-processing on the raw `LlmResponse`.
    *
-   * @param callbackContext The context for the current agent call.
-   * @param llmResponse The response object received from the model.
+   * @param params.callbackContext The context for the current agent call.
+   * @param params.llmResponse The response object received from the model.
    * @returns An optional value. A non-`undefined` return may be used by the
    *     framework to modify or replace the response. Returning `undefined`
    *     allows the original response to be used.
    */
-  async afterModelCallback({ callbackContext, llmResponse }) {
+  async afterModelCallback(params) {
     return;
   }
   /**
@@ -166,15 +166,15 @@ class BasePlugin {
    * This callback provides an opportunity to handle model errors gracefully,
    * potentially providing alternative responses or recovery mechanisms.
    *
-   * @param callbackContext The context for the current agent call.
-   * @param llmRequest The request that was sent to the model when the error
+   * @param params.callbackContext The context for the current agent call.
+   * @param params.llmRequest The request that was sent to the model when the error
    *     occurred.
-   * @param error The exception that was raised during model execution.
+   * @param params.error The exception that was raised during model execution.
    * @returns An optional LlmResponse. If an LlmResponse is returned, it will be
    *     used instead of propagating the error. Returning `undefined` allows
    *     the original error to be raised.
    */
-  async onModelErrorCallback({ callbackContext, llmRequest, error }) {
+  async onModelErrorCallback(params) {
     return;
   }
   /**
@@ -183,15 +183,15 @@ class BasePlugin {
    * This callback is useful for logging tool usage, input validation, or
    * modifying the arguments before they are passed to the tool.
    *
-   * @param tool The tool instance that is about to be executed.
-   * @param toolArgs The dictionary of arguments to be used for invoking the
+   * @param params.tool The tool instance that is about to be executed.
+   * @param params.toolArgs The dictionary of arguments to be used for invoking the
    *     tool.
-   * @param toolContext The context specific to the tool execution.
+   * @param params.toolContext The context specific to the tool execution.
    * @returns An optional dictionary. If a dictionary is returned, it will stop
    *     the tool execution and return this response immediately. Returning
    *     `undefined` uses the original, unmodified arguments.
    */
-  async beforeToolCallback({ tool, toolArgs, toolContext }) {
+  async beforeToolCallback(params) {
     return;
   }
   /**
@@ -200,33 +200,43 @@ class BasePlugin {
    * This callback allows for inspecting, logging, or modifying the result
    * returned by a tool.
    *
-   * @param tool The tool instance that has just been executed.
-   * @param toolArgs The original arguments that were passed to the tool.
-   * @param toolContext The context specific to the tool execution.
-   * @param result The dictionary returned by the tool invocation.
+   * @param params.tool The tool instance that has just been executed.
+   * @param params.toolArgs The original arguments that were passed to the tool.
+   * @param params.toolContext The context specific to the tool execution.
+   * @param params.result The dictionary returned by the tool invocation.
    * @returns An optional dictionary. If a dictionary is returned, it will
    *     **replace** the original result from the tool. This allows for
    *     post-processing or altering tool outputs. Returning `undefined` uses
    *     the original, unmodified result.
    */
-  async afterToolCallback({ tool, toolArgs, toolContext, result }) {
+  async afterToolCallback(params) {
     return;
   }
   /**
-   * Callback executed when a tool call encounters an error.
-   *
-   * This callback provides an opportunity to handle tool errors gracefully,
-   * potentially providing alternative responses or recovery mechanisms.
-   *
-   * @param tool The tool instance that encountered an error.
-   * @param toolArgs The arguments that were passed to the tool.
-   * @param toolContext The context specific to the tool execution.
-   * @param error The exception that was raised during tool execution.
-   * @returns An optional dictionary. If a dictionary is returned, it will be
-   *     used as the tool response instead of propagating the error. Returning
-   *     `undefined` allows the original error to be raised.
-   */
-  async onToolErrorCallback({ tool, toolArgs, toolContext, error }) {
+     * Callback executed when a tool call encounters an error.
+      tool: BaseTool;
+      toolArgs: Record<string, unknown>;
+      toolContext: ToolContext;
+      result: Record<string, unknown>;
+    }): Promise<Record<string, unknown> | undefined> {
+      return;
+    }
+  
+    /**
+     * Callback executed when a tool call encounters an error.
+     *
+     * This callback provides an opportunity to handle tool errors gracefully,
+     * potentially providing alternative responses or recovery mechanisms.
+     *
+     * @param params.tool The tool instance that encountered an error.
+     * @param params.toolArgs The arguments that were passed to the tool.
+     * @param params.toolContext The context specific to the tool execution.
+     * @param params.error The exception that was raised during tool execution.
+     * @returns An optional dictionary. If a dictionary is returned, it will be
+     *     used as the tool response instead of propagating the error. Returning
+     *     `undefined` allows the original error to be raised.
+     */
+  async onToolErrorCallback(params) {
     return;
   }
 }

package/dist/cjs/plugins/security_plugin.js

@@ -112,7 +112,10 @@ class SecurityPlugin extends import_base_plugin.BasePlugin {
     toolArgs,
     toolContext
   }) {
-    const policyCheckResult = await this.policyEngine.evaluate({ tool, toolArgs });
+    const policyCheckResult = await this.policyEngine.evaluate({
+      tool,
+      toolArgs
+    });
     this.setToolCallCheckState(toolContext, policyCheckResult.outcome);
     switch (policyCheckResult.outcome) {
       case "DENY" /* DENY */:

package/dist/cjs/runner/runner.js

@@ -58,26 +58,26 @@ class Runner {
    * Runs the agent with the given message, and returns an async generator of
    * events.
    *
-   * @param userId The user ID of the session.
-   * @param sessionId The session ID of the session.
-   * @param newMessage A new message to append to the session.
-   * @param stateDelta An optional state delta to apply to the session.
-   * @param runConfig The run config for the agent.
+   * @param params.userId The user ID of the session.
+   * @param params.sessionId The session ID of the session.
+   * @param params.newMessage A new message to append to the session.
+   * @param params.stateDelta An optional state delta to apply to the session.
+   * @param params.runConfig The run config for the agent.
    * @yields The events generated by the agent.
    */
   // TODO - b/425992518: user, sessionId, and runConfig can be internalized.
-  async *runAsync({
-    userId,
-    sessionId,
-    newMessage,
-    stateDelta,
-    runConfig
-  }) {
+  async *runAsync(params) {
     var _a;
-    runConfig = (0, import_run_config.createRunConfig)(runConfig);
+    const { userId, sessionId, stateDelta } = params;
+    const runConfig = (0, import_run_config.createRunConfig)(params.runConfig);
+    let newMessage = params.newMessage;
     const span = import_api.trace.getTracer("gcp.vertex.agent").startSpan("invocation");
     try {
-      const session = await this.sessionService.getSession({ appName: this.appName, userId, sessionId });
+      const session = await this.sessionService.getSession({
+        appName: this.appName,
+        userId,
+        sessionId
+      });
       if (!session) {
         if (!this.appName) {
           throw new Error(
@@ -86,12 +86,14 @@ class Runner {
         }
         throw new Error(`Session not found: ${sessionId}`);
       }
-      if (runConfig.supportCfc && this.agent instanceof import_llm_agent.LlmAgent) {
+      if (runConfig.supportCfc && (0, import_llm_agent.isLlmAgent)(this.agent)) {
         const modelName = this.agent.canonicalModel.model;
         if (!(0, import_model_name.isGemini2OrAbove)(modelName)) {
-          throw new Error(`CFC is not supported for model: ${modelName} in agent: ${this.agent.name}`);
+          throw new Error(
+            `CFC is not supported for model: ${modelName} in agent: ${this.agent.name}`
+          );
         }
-        if (!(this.agent.codeExecutor instanceof import_built_in_code_executor.BuiltInCodeExecutor)) {
+        if (!(0, import_built_in_code_executor.isBuiltInCodeExecutor)(this.agent.codeExecutor)) {
           this.agent.codeExecutor = new import_built_in_code_executor.BuiltInCodeExecutor();
         }
       }
@@ -136,7 +138,10 @@ class Runner {
           })
         });
       }
-      invocationContext.agent = this.determineAgentForResumption(session, this.agent);
+      invocationContext.agent = this.determineAgentForResumption(
+        session,
+        this.agent
+      );
       const beforeRunCallbackResponse = await this.pluginManager.runBeforeRunCallback({ invocationContext });
       if (beforeRunCallbackResponse) {
         const earlyExitEvent = (0, import_event.createEvent)({
@@ -153,9 +158,10 @@ class Runner {
           if (!event.partial) {
             await this.sessionService.appendEvent({ session, event });
           }
-          const modifiedEvent = await this.pluginManager.runOnEventCallback(
-            { invocationContext, event }
-          );
+          const modifiedEvent = await this.pluginManager.runOnEventCallback({
+            invocationContext,
+            event
+          });
           if (modifiedEvent) {
             yield modifiedEvent;
           } else {
@@ -222,7 +228,9 @@ class Runner {
       }
       const agent = rootAgent.findSubAgent(event2.author);
       if (!agent) {
-        import_logger.logger.warn(`Event from an unknown agent: ${event2.author}, event id: ${event2.id}`);
+        import_logger.logger.warn(
+          `Event from an unknown agent: ${event2.author}, event id: ${event2.id}`
+        );
         continue;
       }
       if (this.isRoutableLlmAgent(agent)) {
@@ -245,7 +253,7 @@ class Runner {
   isRoutableLlmAgent(agentToRun) {
     let agent = agentToRun;
     while (agent) {
-      if (!(agent instanceof import_llm_agent.LlmAgent)) {
+      if (!(0, import_llm_agent.isLlmAgent)(agent)) {
         return false;
       }
       if (agent.disallowTransferToParent) {
@@ -263,7 +271,9 @@ function findEventByLastFunctionResponseId(events) {
     return null;
   }
   const lastEvent = events[events.length - 1];
-  const functionCallId = (_d = (_c = (_b = (_a = lastEvent.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.find((part) => part.functionResponse)) == null ? void 0 : _c.functionResponse) == null ? void 0 : _d.id;
+  const functionCallId = (_d = (_c = (_b = (_a = lastEvent.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.find(
+    (part) => part.functionResponse
+  )) == null ? void 0 : _c.functionResponse) == null ? void 0 : _d.id;
   if (!functionCallId) {
     return null;
   }

package/dist/cjs/tools/agent_tool.js

@@ -24,7 +24,8 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var agent_tool_exports = {};
 __export(agent_tool_exports, {
-  AgentTool: () => AgentTool
+  AgentTool: () => AgentTool,
+  isAgentTool: () => isAgentTool
 });
 module.exports = __toCommonJS(agent_tool_exports);
 var import_genai = require("@google/genai");
@@ -40,17 +41,24 @@ var import_forwarding_artifact_service = require("./forwarding_artifact_service.
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-class AgentTool extends import_base_tool.BaseTool {
+var _a, _b;
+const AGENT_TOOL_SIGNATURE_SYMBOL = Symbol.for("google.adk.agentTool");
+function isAgentTool(obj) {
+  return typeof obj === "object" && obj !== null && AGENT_TOOL_SIGNATURE_SYMBOL in obj && obj[AGENT_TOOL_SIGNATURE_SYMBOL] === true;
+}
+class AgentTool extends (_b = import_base_tool.BaseTool, _a = AGENT_TOOL_SIGNATURE_SYMBOL, _b) {
   constructor(config) {
     super(
       { name: config.agent.name, description: config.agent.description || "" }
     );
+    /** A unique symbol to identify ADK agent tool class. */
+    this[_a] = true;
     this.agent = config.agent;
     this.skipSummarization = config.skipSummarization || false;
   }
   _getDeclaration() {
     let declaration;
-    if (this.agent instanceof import_llm_agent.LlmAgent && this.agent.inputSchema) {
+    if ((0, import_llm_agent.isLlmAgent)(this.agent) && this.agent.inputSchema) {
       declaration = {
         name: this.name,
         description: this.description,
@@ -75,17 +83,17 @@ class AgentTool extends import_base_tool.BaseTool {
       };
     }
     if (this.apiVariant !== import_variant_utils.GoogleLLMVariant.GEMINI_API) {
-      const hasOutputSchema = this.agent instanceof import_llm_agent.LlmAgent && this.agent.outputSchema;
+      const hasOutputSchema = (0, import_llm_agent.isLlmAgent)(this.agent) && this.agent.outputSchema;
       declaration.response = hasOutputSchema ? { type: import_genai.Type.OBJECT } : { type: import_genai.Type.STRING };
     }
     return declaration;
   }
   async runAsync({ args, toolContext }) {
-    var _a, _b;
+    var _a2, _b2;
     if (this.skipSummarization) {
       toolContext.actions.skipSummarization = true;
     }
-    const hasInputSchema = this.agent instanceof import_llm_agent.LlmAgent && this.agent.inputSchema;
+    const hasInputSchema = (0, import_llm_agent.isLlmAgent)(this.agent) && this.agent.inputSchema;
     const content = {
       role: "user",
       parts: [
@@ -120,15 +128,16 @@ class AgentTool extends import_base_tool.BaseTool {
       }
       lastEvent = event;
     }
-    if (!((_b = (_a = lastEvent == null ? void 0 : lastEvent.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.length)) {
+    if (!((_b2 = (_a2 = lastEvent == null ? void 0 : lastEvent.content) == null ? void 0 : _a2.parts) == null ? void 0 : _b2.length)) {
       return "";
     }
-    const hasOutputSchema = this.agent instanceof import_llm_agent.LlmAgent && this.agent.outputSchema;
+    const hasOutputSchema = (0, import_llm_agent.isLlmAgent)(this.agent) && this.agent.outputSchema;
     const mergedText = lastEvent.content.parts.map((part) => part.text).filter((text) => text).join("\n");
     return hasOutputSchema ? JSON.parse(mergedText) : mergedText;
   }
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  AgentTool
+  AgentTool,
+  isAgentTool
 });

package/dist/cjs/tools/base_tool.js

@@ -24,7 +24,8 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var base_tool_exports = {};
 __export(base_tool_exports, {
-  BaseTool: () => BaseTool
+  BaseTool: () => BaseTool,
+  isBaseTool: () => isBaseTool
 });
 module.exports = __toCommonJS(base_tool_exports);
 var import_variant_utils = require("../utils/variant_utils.js");
@@ -33,6 +34,12 @@ var import_variant_utils = require("../utils/variant_utils.js");
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+var _a;
+const BASE_TOOL_SIGNATURE_SYMBOL = Symbol.for("google.adk.baseTool");
+function isBaseTool(obj) {
+  return typeof obj === "object" && obj !== null && BASE_TOOL_SIGNATURE_SYMBOL in obj && obj[BASE_TOOL_SIGNATURE_SYMBOL] === true;
+}
+_a = BASE_TOOL_SIGNATURE_SYMBOL;
 class BaseTool {
   /**
    * Base constructor for a tool.
@@ -40,10 +47,12 @@ class BaseTool {
    * @param params The parameters for `BaseTool`.
    */
   constructor(params) {
-    var _a;
+    /** A unique symbol to identify ADK base tool class. */
+    this[_a] = true;
+    var _a2;
     this.name = params.name;
     this.description = params.description;
-    this.isLongRunning = (_a = params.isLongRunning) != null ? _a : false;
+    this.isLongRunning = (_a2 = params.isLongRunning) != null ? _a2 : false;
   }
   /**
    * Gets the OpenAPI specification of this tool in the form of a
@@ -98,10 +107,11 @@ class BaseTool {
   }
 }
 function findToolWithFunctionDeclarations(llmRequest) {
-  var _a;
-  return (((_a = llmRequest.config) == null ? void 0 : _a.tools) || []).find((tool) => "functionDeclarations" in tool);
+  var _a2;
+  return (((_a2 = llmRequest.config) == null ? void 0 : _a2.tools) || []).find((tool) => "functionDeclarations" in tool);
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  BaseTool
+  BaseTool,
+  isBaseTool
 });

package/dist/cjs/tools/function_tool.js

@@ -24,7 +24,8 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var function_tool_exports = {};
 __export(function_tool_exports, {
-  FunctionTool: () => FunctionTool
+  FunctionTool: () => FunctionTool,
+  isFunctionTool: () => isFunctionTool
 });
 module.exports = __toCommonJS(function_tool_exports);
 var import_genai = require("@google/genai");
@@ -36,6 +37,7 @@ var import_base_tool = require("./base_tool.js");
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+var _a, _b;
 function toSchema(parameters) {
   if (parameters === void 0) {
     return { type: import_genai.Type.OBJECT, properties: {} };
@@ -45,14 +47,18 @@ function toSchema(parameters) {
   }
   return parameters;
 }
-class FunctionTool extends import_base_tool.BaseTool {
+const FUNCTION_TOOL_SIGNATURE_SYMBOL = Symbol.for("google.adk.functionTool");
+function isFunctionTool(obj) {
+  return typeof obj === "object" && obj !== null && FUNCTION_TOOL_SIGNATURE_SYMBOL in obj && obj[FUNCTION_TOOL_SIGNATURE_SYMBOL] === true;
+}
+class FunctionTool extends (_b = import_base_tool.BaseTool, _a = FUNCTION_TOOL_SIGNATURE_SYMBOL, _b) {
   /**
    * The constructor acts as the user-friendly factory.
    * @param options The configuration for the tool.
    */
   constructor(options) {
-    var _a;
-    const name = (_a = options.name) != null ? _a : options.execute.name;
+    var _a2;
+    const name = (_a2 = options.name) != null ? _a2 : options.execute.name;
     if (!name) {
       throw new Error(
         "Tool name cannot be empty. Either name the `execute` function or provide a `name`."
@@ -63,6 +69,8 @@ class FunctionTool extends import_base_tool.BaseTool {
       description: options.description,
       isLongRunning: options.isLongRunning
     });
+    /** A unique symbol to identify ADK function tool class. */
+    this[_a] = true;
     this.execute = options.execute;
     this.parameters = options.parameters;
   }
@@ -97,5 +105,6 @@ class FunctionTool extends import_base_tool.BaseTool {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  FunctionTool
+  FunctionTool,
+  isFunctionTool
 });

package/dist/cjs/tools/google_search_tool.js

@@ -24,7 +24,8 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var google_search_tool_exports = {};
 __export(google_search_tool_exports, {
-  GOOGLE_SEARCH: () => GOOGLE_SEARCH
+  GOOGLE_SEARCH: () => GOOGLE_SEARCH,
+  GoogleSearchTool: () => GoogleSearchTool
 });
 module.exports = __toCommonJS(google_search_tool_exports);
 var import_model_name = require("../utils/model_name.js");
@@ -41,7 +42,10 @@ class GoogleSearchTool extends import_base_tool.BaseTool {
   runAsync(request) {
     return Promise.resolve();
   }
-  async processLlmRequest({ toolContext, llmRequest }) {
+  async processLlmRequest({
+    toolContext,
+    llmRequest
+  }) {
     if (!llmRequest.model) {
       return;
     }
@@ -72,5 +76,6 @@ class GoogleSearchTool extends import_base_tool.BaseTool {
 const GOOGLE_SEARCH = new GoogleSearchTool();
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  GOOGLE_SEARCH
+  GOOGLE_SEARCH,
+  GoogleSearchTool
 });

package/dist/cjs/tools/tool_context.js

@@ -47,15 +47,10 @@ class ToolContext extends import_callback_context.CallbackContext {
    * @param params.toolConfirmation The tool confirmation of the current tool
    *     call.
    */
-  constructor({
-    invocationContext,
-    eventActions,
-    functionCallId,
-    toolConfirmation
-  }) {
-    super({ invocationContext, eventActions });
-    this.functionCallId = functionCallId;
-    this.toolConfirmation = toolConfirmation;
+  constructor(params) {
+    super(params);
+    this.functionCallId = params.functionCallId;
+    this.toolConfirmation = params.toolConfirmation;
   }
   get actions() {
     return this.eventActions;

package/dist/cjs/utils/gemini_schema_util.js

@@ -40,6 +40,7 @@ const MCPToolSchema = import_zod.z.object({
   required: import_zod.z.string().array().optional()
 });
 function toGeminiType(mcpType) {
+  if (!mcpType) return import_genai.Type.TYPE_UNSPECIFIED;
   switch (mcpType.toLowerCase()) {
     case "text":
     case "string":

package/dist/cjs/version.js

@@ -32,7 +32,7 @@ module.exports = __toCommonJS(version_exports);
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-const version = "0.2.4";
+const version = "0.2.5";
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   version

package/dist/esm/agents/base_agent.js

@@ -24,7 +24,9 @@ class BaseAgent {
     this.parentAgent = config.parentAgent;
     this.subAgents = config.subAgents || [];
     this.rootAgent = getRootAgent(this);
-    this.beforeAgentCallback = getCannonicalCallback(config.beforeAgentCallback);
+    this.beforeAgentCallback = getCannonicalCallback(
+      config.beforeAgentCallback
+    );
     this.afterAgentCallback = getCannonicalCallback(config.afterAgentCallback);
     this.setParentAgentForSubAgents();
   }
@@ -186,7 +188,9 @@ class BaseAgent {
   setParentAgentForSubAgents() {
     for (const subAgent of this.subAgents) {
       if (subAgent.parentAgent) {
-        throw new Error(`Agent "${subAgent.name}" already has a parent agent, current parent: "${subAgent.parentAgent.name}", trying to add: "${this.name}"`);
+        throw new Error(
+          `Agent "${subAgent.name}" already has a parent agent, current parent: "${subAgent.parentAgent.name}", trying to add: "${this.name}"`
+        );
       }
       subAgent.parentAgent = this;
     }
@@ -194,7 +198,9 @@ class BaseAgent {
 }
 function validateAgentName(name) {
   if (!isIdentifier(name)) {
-    throw new Error(`Found invalid agent name: "${name}". Agent name must be a valid identifier. It should start with a letter (a-z, A-Z) or an underscore (_), and can only contain letters, digits (0-9), and underscores.`);
+    throw new Error(
+      `Found invalid agent name: "${name}". Agent name must be a valid identifier. It should start with a letter (a-z, A-Z) or an underscore (_), and can only contain letters, digits (0-9), and underscores.`
+    );
   }
   if (name === "user") {
     throw new Error(

package/dist/esm/agents/invocation_context.js

@@ -17,7 +17,9 @@ class InvocationCostManager {
   incrementAndEnforceLlmCallsLimit(runConfig) {
     this.numberOfLlmCalls++;
     if (runConfig && runConfig.maxLlmCalls > 0 && this.numberOfLlmCalls > runConfig.maxLlmCalls) {
-      throw new Error(`Max number of llm calls limit of ${runConfig.maxLlmCalls} exceeded`);
+      throw new Error(
+        `Max number of llm calls limit of ${runConfig.maxLlmCalls} exceeded`
+      );
     }
   }
 }