@google/adk 0.2.5 → 0.4.0

package/dist/cjs/models/gemini_llm_connection.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -116,6 +116,7 @@ class GeminiLlmConnection {
     };
   }
   // TODO(b/425992518): GenAI SDK inconsistent API, missing methods.
+  // eslint-disable-next-line require-yield
   async *receive() {
     throw new Error("Not Implemented.");
   }

package/dist/cjs/models/google_llm.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -24,10 +24,12 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var google_llm_exports = {};
 __export(google_llm_exports, {
-  Gemini: () => Gemini
+  Gemini: () => Gemini,
+  geminiInitParams: () => geminiInitParams
 });
 module.exports = __toCommonJS(google_llm_exports);
 var import_genai = require("@google/genai");
+var import_env_aware_utils = require("../utils/env_aware_utils.js");
 var import_logger = require("../utils/logger.js");
 var import_variant_utils = require("../utils/variant_utils.js");
 var import_base_llm = require("./base_llm.js");
@@ -38,8 +40,6 @@ var import_llm_response = require("./llm_response.js");
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-const AGENT_ENGINE_TELEMETRY_TAG = "remote_reasoning_engine";
-const AGENT_ENGINE_TELEMETRY_ENV_VARIABLE_NAME = "GOOGLE_CLOUD_AGENT_ENGINE_ID";
 class Gemini extends import_base_llm.BaseLlm {
   /**
    * @param params The parameters for creating a Gemini instance.
@@ -56,45 +56,23 @@ class Gemini extends import_base_llm.BaseLlm {
       model = "gemini-2.5-flash";
     }
     super({ model });
-    this.project = project;
-    this.location = location;
-    this.apiKey = apiKey;
-    this.headers = headers;
-    const canReadEnv = typeof process === "object";
-    this.vertexai = !!vertexai;
-    if (!this.vertexai && canReadEnv) {
-      const vertexAIfromEnv = process.env["GOOGLE_GENAI_USE_VERTEXAI"];
-      if (vertexAIfromEnv) {
-        this.vertexai = vertexAIfromEnv.toLowerCase() === "true" || vertexAIfromEnv === "1";
-      }
-    }
-    if (this.vertexai) {
-      if (canReadEnv && !this.project) {
-        this.project = process.env["GOOGLE_CLOUD_PROJECT"];
-      }
-      if (canReadEnv && !this.location) {
-        this.location = process.env["GOOGLE_CLOUD_LOCATION"];
-      }
-      if (!this.project) {
-        throw new Error(
-          "VertexAI project must be provided via constructor or GOOGLE_CLOUD_PROJECT environment variable."
-        );
-      }
-      if (!this.location) {
-        throw new Error(
-          "VertexAI location must be provided via constructor or GOOGLE_CLOUD_LOCATION environment variable."
-        );
-      }
-    } else {
-      if (!this.apiKey && canReadEnv) {
-        this.apiKey = process.env["GOOGLE_GENAI_API_KEY"] || process.env["GEMINI_API_KEY"];
-      }
-      if (!this.apiKey) {
-        throw new Error(
-          "API key must be provided via constructor or GOOGLE_GENAI_API_KEY or GEMINI_API_KEY environment variable."
-        );
-      }
+    const params = geminiInitParams({
+      model,
+      vertexai,
+      project,
+      location,
+      apiKey
+    });
+    if (!params.vertexai && !params.apiKey) {
+      throw new Error(
+        "API key must be provided via constructor or GOOGLE_GENAI_API_KEY or GEMINI_API_KEY environment variable."
+      );
     }
+    this.project = params.project;
+    this.location = params.location;
+    this.apiKey = params.apiKey;
+    this.headers = headers;
+    this.vertexai = !!params.vertexai;
   }
   /**
    * Sends a request to the Gemini model.
@@ -183,25 +161,24 @@ class Gemini extends import_base_llm.BaseLlm {
       yield (0, import_llm_response.createLlmResponse)(response);
     }
   }
+  getHttpOptions() {
+    return { headers: { ...this.trackingHeaders, ...this.headers } };
+  }
   get apiClient() {
     if (this._apiClient) {
       return this._apiClient;
     }
-    const combinedHeaders = {
-      ...this.trackingHeaders,
-      ...this.headers
-    };
     if (this.vertexai) {
       this._apiClient = new import_genai.GoogleGenAI({
         vertexai: this.vertexai,
         project: this.project,
         location: this.location,
-        httpOptions: { headers: combinedHeaders }
+        httpOptions: this.getHttpOptions()
       });
     } else {
       this._apiClient = new import_genai.GoogleGenAI({
         apiKey: this.apiKey,
-        httpOptions: { headers: combinedHeaders }
+        httpOptions: this.getHttpOptions()
       });
     }
     return this._apiClient;
@@ -218,14 +195,17 @@ class Gemini extends import_base_llm.BaseLlm {
     }
     return this._liveApiVersion;
   }
+  getLiveHttpOptions() {
+    return {
+      headers: this.trackingHeaders,
+      apiVersion: this.liveApiVersion
+    };
+  }
   get liveApiClient() {
     if (!this._liveApiClient) {
       this._liveApiClient = new import_genai.GoogleGenAI({
         apiKey: this.apiKey,
-        httpOptions: {
-          headers: this.trackingHeaders,
-          apiVersion: this.liveApiVersion
-        }
+        httpOptions: this.getLiveHttpOptions()
       });
     }
     return this._liveApiClient;
@@ -252,7 +232,9 @@ class Gemini extends import_base_llm.BaseLlm {
       llmRequest.liveConnectConfig.systemInstruction = {
         role: "system",
         // TODO - b/425992518: validate type casting works well.
-        parts: [(0, import_genai.createPartFromText)(llmRequest.config.systemInstruction)]
+        parts: [
+          (0, import_genai.createPartFromText)(llmRequest.config.systemInstruction)
+        ]
       };
     }
     llmRequest.liveConnectConfig.tools = (_c = llmRequest.config) == null ? void 0 : _c.tools;
@@ -301,7 +283,44 @@ function removeDisplayNameIfPresent(dataObj) {
     dataObj.displayName = void 0;
   }
 }
+function geminiInitParams({
+  model,
+  vertexai,
+  project,
+  location,
+  apiKey
+}) {
+  const params = { model, vertexai, project, location, apiKey };
+  params.vertexai = !!vertexai;
+  if (!params.vertexai && !(0, import_env_aware_utils.isBrowser)()) {
+    params.vertexai = (0, import_env_aware_utils.getBooleanEnvVar)("GOOGLE_GENAI_USE_VERTEXAI");
+  }
+  if (params.vertexai) {
+    if (!(0, import_env_aware_utils.isBrowser)() && !params.project) {
+      params.project = process.env["GOOGLE_CLOUD_PROJECT"];
+    }
+    if (!(0, import_env_aware_utils.isBrowser)() && !params.location) {
+      params.location = process.env["GOOGLE_CLOUD_LOCATION"];
+    }
+    if (!params.project) {
+      throw new Error(
+        "VertexAI project must be provided via constructor or GOOGLE_CLOUD_PROJECT environment variable."
+      );
+    }
+    if (!params.location) {
+      throw new Error(
+        "VertexAI location must be provided via constructor or GOOGLE_CLOUD_LOCATION environment variable."
+      );
+    }
+  } else {
+    if (!params.apiKey && !(0, import_env_aware_utils.isBrowser)()) {
+      params.apiKey = process.env["GOOGLE_GENAI_API_KEY"] || process.env["GEMINI_API_KEY"];
+    }
+  }
+  return params;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  Gemini
+  Gemini,
+  geminiInitParams
 });

package/dist/cjs/models/llm_request.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/models/llm_response.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/models/registry.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -28,6 +28,7 @@ __export(registry_exports, {
 });
 module.exports = __toCommonJS(registry_exports);
 var import_logger = require("../utils/logger.js");
+var import_apigee_llm = require("./apigee_llm.js");
 var import_google_llm = require("./google_llm.js");
 /**
  * @license
@@ -115,6 +116,7 @@ _LLMRegistry.llmRegistryDict = /* @__PURE__ */ new Map();
 _LLMRegistry.resolveCache = new LRUCache(32);
 let LLMRegistry = _LLMRegistry;
 LLMRegistry.register(import_google_llm.Gemini);
+LLMRegistry.register(import_apigee_llm.ApigeeLlm);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   LLMRegistry

package/dist/cjs/plugins/base_plugin.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -54,6 +54,7 @@ class BasePlugin {
    *     value to replace the user message. Returning `undefined` to proceed
    *     normally.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async onUserMessageCallback(params) {
     return;
   }
@@ -69,6 +70,7 @@ class BasePlugin {
    *     to halt execution of the runner and ends the runner with that event.
    *     Return `undefined` to proceed normally.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async beforeRunCallback(params) {
     return;
   }
@@ -84,6 +86,7 @@ class BasePlugin {
    *     framework to modify or replace the response. Returning `undefined`
    *     allows the original response to be used.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async onEventCallback(params) {
     return;
   }
@@ -96,6 +99,7 @@ class BasePlugin {
    * @param params.invocationContext The context for the entire invocation.
    * @returns undefined
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async afterRunCallback(params) {
     return;
   }
@@ -111,6 +115,7 @@ class BasePlugin {
    *     bypass the agent's callbacks and its execution, and return this value
    *     directly. Returning `undefined` allows the agent to proceed normally.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async beforeAgentCallback(params) {
     return;
   }
@@ -126,6 +131,7 @@ class BasePlugin {
    *     replace the agent's original result. Returning `undefined` uses the
    *     original, unmodified result.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async afterAgentCallback(params) {
     return;
   }
@@ -142,6 +148,7 @@ class BasePlugin {
    *     trigger an early exit and returns the response immediately. Returning
    *     `undefined` allows the LLM request to proceed normally.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async beforeModelCallback(params) {
     return;
   }
@@ -157,6 +164,7 @@ class BasePlugin {
    *     framework to modify or replace the response. Returning `undefined`
    *     allows the original response to be used.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async afterModelCallback(params) {
     return;
   }
@@ -174,6 +182,7 @@ class BasePlugin {
    *     used instead of propagating the error. Returning `undefined` allows
    *     the original error to be raised.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async onModelErrorCallback(params) {
     return;
   }
@@ -191,6 +200,7 @@ class BasePlugin {
    *     the tool execution and return this response immediately. Returning
    *     `undefined` uses the original, unmodified arguments.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async beforeToolCallback(params) {
     return;
   }
@@ -209,6 +219,7 @@ class BasePlugin {
    *     post-processing or altering tool outputs. Returning `undefined` uses
    *     the original, unmodified result.
    */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async afterToolCallback(params) {
     return;
   }
@@ -236,6 +247,7 @@ class BasePlugin {
      *     used as the tool response instead of propagating the error. Returning
      *     `undefined` allows the original error to be raised.
      */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async onToolErrorCallback(params) {
     return;
   }

package/dist/cjs/plugins/logging_plugin.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -44,7 +44,10 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
   constructor(name = "logging_plugin") {
     super(name);
   }
-  async onUserMessageCallback({ invocationContext, userMessage }) {
+  async onUserMessageCallback({
+    invocationContext,
+    userMessage
+  }) {
     var _a;
     this.log("\u{1F680} USER MESSAGE RECEIVED");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
@@ -58,14 +61,18 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
     }
     return void 0;
   }
-  async beforeRunCallback({ invocationContext }) {
+  async beforeRunCallback({
+    invocationContext
+  }) {
     var _a;
     this.log("\u{1F3C3} INVOCATION STARTING");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
     this.log(`   Starting Agent: ${(_a = invocationContext.agent.name) != null ? _a : "Unknown"}`);
     return void 0;
   }
-  async onEventCallback({ invocationContext, event }) {
+  async onEventCallback({
+    event
+  }) {
     this.log("\u{1F4E2} EVENT YIELDED");
     this.log(`   Event ID: ${event.id}`);
     this.log(`   Author: ${event.author}`);
@@ -86,14 +93,18 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
     }
     return void 0;
   }
-  async afterRunCallback({ invocationContext }) {
+  async afterRunCallback({
+    invocationContext
+  }) {
     var _a;
     this.log("\u2705 INVOCATION COMPLETED");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
     this.log(`   Final Agent: ${(_a = invocationContext.agent.name) != null ? _a : "Unknown"}`);
     return void 0;
   }
-  async beforeAgentCallback({ agent, callbackContext }) {
+  async beforeAgentCallback({
+    callbackContext
+  }) {
     this.log("\u{1F916} AGENT STARTING");
     this.log(`   Agent Name: ${callbackContext.agentName}`);
     this.log(`   Invocation ID: ${callbackContext.invocationId}`);
@@ -102,13 +113,18 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
     }
     return void 0;
   }
-  async afterAgentCallback({ agent, callbackContext }) {
+  async afterAgentCallback({
+    callbackContext
+  }) {
     this.log("\u{1F916} AGENT COMPLETED");
     this.log(`   Agent Name: ${callbackContext.agentName}`);
     this.log(`   Invocation ID: ${callbackContext.invocationId}`);
     return void 0;
   }
-  async beforeModelCallback({ callbackContext, llmRequest }) {
+  async beforeModelCallback({
+    callbackContext,
+    llmRequest
+  }) {
     var _a;
     this.log("\u{1F9E0} LLM REQUEST");
     this.log(`   Model: ${(_a = llmRequest.model) != null ? _a : "default"}`);
@@ -126,7 +142,10 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
     }
     return void 0;
   }
-  async afterModelCallback({ callbackContext, llmResponse }) {
+  async afterModelCallback({
+    callbackContext,
+    llmResponse
+  }) {
     this.log("\u{1F9E0} LLM RESPONSE");
     this.log(`   Agent: ${callbackContext.agentName}`);
     if (llmResponse.errorCode) {
@@ -142,11 +161,17 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
       }
     }
     if (llmResponse.usageMetadata) {
-      this.log(`   Token Usage - Input: ${llmResponse.usageMetadata.promptTokenCount}, Output: ${llmResponse.usageMetadata.candidatesTokenCount}`);
+      this.log(
+        `   Token Usage - Input: ${llmResponse.usageMetadata.promptTokenCount}, Output: ${llmResponse.usageMetadata.candidatesTokenCount}`
+      );
     }
     return void 0;
   }
-  async beforeToolCallback({ tool, toolArgs, toolContext }) {
+  async beforeToolCallback({
+    tool,
+    toolArgs,
+    toolContext
+  }) {
     this.log("\u{1F527} TOOL STARTING");
     this.log(`   Tool Name: ${tool.name}`);
     this.log(`   Agent: ${toolContext.agentName}`);
@@ -154,7 +179,11 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
     this.log(`   Arguments: ${this.formatArgs(toolArgs)}`);
     return void 0;
   }
-  async afterToolCallback({ tool, toolArgs, toolContext, result }) {
+  async afterToolCallback({
+    tool,
+    toolContext,
+    result
+  }) {
     this.log("\u{1F527} TOOL COMPLETED");
     this.log(`   Tool Name: ${tool.name}`);
     this.log(`   Agent: ${toolContext.agentName}`);
@@ -162,13 +191,21 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
     this.log(`   Result: ${this.formatArgs(result)}`);
     return void 0;
   }
-  async onModelErrorCallback({ callbackContext, llmRequest, error }) {
+  async onModelErrorCallback({
+    callbackContext,
+    error
+  }) {
     this.log("\u{1F9E0} LLM ERROR");
     this.log(`   Agent: ${callbackContext.agentName}`);
     this.log(`   Error: ${error}`);
     return void 0;
   }
-  async onToolErrorCallback({ tool, toolArgs, toolContext, error }) {
+  async onToolErrorCallback({
+    tool,
+    toolArgs,
+    toolContext,
+    error
+  }) {
     this.log("\u{1F527} TOOL ERROR");
     this.log(`   Tool Name: ${tool.name}`);
     this.log(`   Agent: ${toolContext.agentName}`);

package/dist/cjs/plugins/plugin_manager.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -107,19 +107,22 @@ class PluginManager {
   /**
    * Runs the `onUserMessageCallback` for all plugins.
    */
-  async runOnUserMessageCallback({ userMessage, invocationContext }) {
+  async runOnUserMessageCallback({
+    userMessage,
+    invocationContext
+  }) {
     return await this.runCallbacks(
       this.plugins,
-      (plugin) => plugin.onUserMessageCallback(
-        { userMessage, invocationContext }
-      ),
+      (plugin) => plugin.onUserMessageCallback({ userMessage, invocationContext }),
       "onUserMessageCallback"
     );
   }
   /**
    * Runs the `beforeRunCallback` for all plugins.
    */
-  async runBeforeRunCallback({ invocationContext }) {
+  async runBeforeRunCallback({
+    invocationContext
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.beforeRunCallback({ invocationContext }),
@@ -129,7 +132,9 @@ class PluginManager {
   /**
    * Runs the `afterRunCallback` for all plugins.
    */
-  async runAfterRunCallback({ invocationContext }) {
+  async runAfterRunCallback({
+    invocationContext
+  }) {
     await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.afterRunCallback({ invocationContext }),
@@ -139,7 +144,10 @@ class PluginManager {
   /**
    * Runs the `onEventCallback` for all plugins.
    */
-  async runOnEventCallback({ invocationContext, event }) {
+  async runOnEventCallback({
+    invocationContext,
+    event
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.onEventCallback({ invocationContext, event }),
@@ -149,7 +157,10 @@ class PluginManager {
   /**
    * Runs the `beforeAgentCallback` for all plugins.
    */
-  async runBeforeAgentCallback({ agent, callbackContext }) {
+  async runBeforeAgentCallback({
+    agent,
+    callbackContext
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.beforeAgentCallback({ agent, callbackContext }),
@@ -159,7 +170,10 @@ class PluginManager {
   /**
    * Runs the `afterAgentCallback` for all plugins.
    */
-  async runAfterAgentCallback({ agent, callbackContext }) {
+  async runAfterAgentCallback({
+    agent,
+    callbackContext
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.afterAgentCallback({ agent, callbackContext }),
@@ -169,7 +183,11 @@ class PluginManager {
   /**
    * Runs the `beforeToolCallback` for all plugins.
    */
-  async runBeforeToolCallback({ tool, toolArgs, toolContext }) {
+  async runBeforeToolCallback({
+    tool,
+    toolArgs,
+    toolContext
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.beforeToolCallback({ tool, toolArgs, toolContext }),
@@ -179,31 +197,39 @@ class PluginManager {
   /**
    * Runs the `afterToolCallback` for all plugins.
    */
-  async runAfterToolCallback({ tool, toolArgs, toolContext, result }) {
+  async runAfterToolCallback({
+    tool,
+    toolArgs,
+    toolContext,
+    result
+  }) {
     return await this.runCallbacks(
       this.plugins,
-      (plugin) => plugin.afterToolCallback(
-        { tool, toolArgs, toolContext, result }
-      ),
+      (plugin) => plugin.afterToolCallback({ tool, toolArgs, toolContext, result }),
       "afterToolCallback"
     );
   }
   /**
    * Runs the `onModelErrorCallback` for all plugins.
    */
-  async runOnModelErrorCallback({ callbackContext, llmRequest, error }) {
+  async runOnModelErrorCallback({
+    callbackContext,
+    llmRequest,
+    error
+  }) {
     return await this.runCallbacks(
       this.plugins,
-      (plugin) => plugin.onModelErrorCallback(
-        { callbackContext, llmRequest, error }
-      ),
+      (plugin) => plugin.onModelErrorCallback({ callbackContext, llmRequest, error }),
       "onModelErrorCallback"
     );
   }
   /**
    * Runs the `beforeModelCallback` for all plugins.
    */
-  async runBeforeModelCallback({ callbackContext, llmRequest }) {
+  async runBeforeModelCallback({
+    callbackContext,
+    llmRequest
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.beforeModelCallback({ callbackContext, llmRequest }),
@@ -213,7 +239,10 @@ class PluginManager {
   /**
    * Runs the `afterModelCallback` for all plugins.
    */
-  async runAfterModelCallback({ callbackContext, llmResponse }) {
+  async runAfterModelCallback({
+    callbackContext,
+    llmResponse
+  }) {
     return await this.runCallbacks(
       this.plugins,
       (plugin) => plugin.afterModelCallback({ callbackContext, llmResponse }),
@@ -223,12 +252,15 @@ class PluginManager {
   /**
    * Runs the `onToolErrorCallback` for all plugins.
    */
-  async runOnToolErrorCallback({ tool, toolArgs, toolContext, error }) {
+  async runOnToolErrorCallback({
+    tool,
+    toolArgs,
+    toolContext,
+    error
+  }) {
     return await this.runCallbacks(
       this.plugins,
-      (plugin) => plugin.onToolErrorCallback(
-        { tool, toolArgs, toolContext, error }
-      ),
+      (plugin) => plugin.onToolErrorCallback({ tool, toolArgs, toolContext, error }),
       "onToolErrorCallback"
     );
   }