@google/adk 1.6.0 → 2.1.0

package/dist/cjs/models/interactions_utils.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -55,6 +56,7 @@ function getInteractionMediaType(mimeType) {
       return "document";
   }
 }
+__name(getInteractionMediaType, "getInteractionMediaType");
 function getLatestUserContents(contents) {
   if (!contents || contents.length === 0) {
     return [];
@@ -97,6 +99,7 @@ function getLatestUserContents(contents) {
   }
   return latestUserContents;
 }
+__name(getLatestUserContents, "getLatestUserContents");
 function convertPartToMediaContent(part) {
   if (part.text !== void 0 && part.text !== null) {
     return { type: "text", text: part.text };
@@ -119,6 +122,7 @@ function convertPartToMediaContent(part) {
   }
   return null;
 }
+__name(convertPartToMediaContent, "convertPartToMediaContent");
 function convertContentToSteps(content) {
   const steps = [];
   const role = content.role || "user";
@@ -204,6 +208,7 @@ function convertContentToSteps(content) {
   }
   return steps;
 }
+__name(convertContentToSteps, "convertContentToSteps");
 function convertMediaContentToPart(content) {
   if (content.type === "text") {
     return { text: content.text || "" };
@@ -228,6 +233,7 @@ function convertMediaContentToPart(content) {
   }
   return null;
 }
+__name(convertMediaContentToPart, "convertMediaContentToPart");
 function convertStepToParts(step) {
   if (!step || !step.type) {
     return [];
@@ -323,6 +329,7 @@ function convertStepToParts(step) {
       return [];
   }
 }
+__name(convertStepToParts, "convertStepToParts");
 function convertToolsConfigToInteractionsFormat(config) {
   if (!config.tools) {
     return [];
@@ -371,6 +378,7 @@ function convertToolsConfigToInteractionsFormat(config) {
   }
   return interactionTools;
 }
+__name(convertToolsConfigToInteractionsFormat, "convertToolsConfigToInteractionsFormat");
 function findLastPart(parts, predicate) {
   for (let i = parts.length - 1; i >= 0; i--) {
     if (predicate(parts[i])) {
@@ -379,6 +387,7 @@ function findLastPart(parts, predicate) {
   }
   return void 0;
 }
+__name(findLastPart, "findLastPart");
 function getLatestModelParts(steps) {
   if (!steps || steps.length === 0) {
     return [];
@@ -394,6 +403,7 @@ function getLatestModelParts(steps) {
   }
   return latestParts;
 }
+__name(getLatestModelParts, "getLatestModelParts");
 function convertInteractionToLlmResponse(interaction) {
   if (interaction.status === "failed") {
     let errorMsg = "Unknown error";
@@ -435,6 +445,7 @@ function convertInteractionToLlmResponse(interaction) {
     interactionId: interaction.id
   };
 }
+__name(convertInteractionToLlmResponse, "convertInteractionToLlmResponse");
 function convertInteractionEventToLlmResponse(event, aggregatedParts, interactionId) {
   var _a, _b;
   const eventType = event.event_type || event.eventType;
@@ -590,6 +601,7 @@ function convertInteractionEventToLlmResponse(event, aggregatedParts, interactio
   }
   return null;
 }
+__name(convertInteractionEventToLlmResponse, "convertInteractionEventToLlmResponse");
 function buildGenerationConfig(config) {
   const generationConfig = {};
   if (config.temperature !== void 0 && config.temperature !== null) {
@@ -615,6 +627,7 @@ function buildGenerationConfig(config) {
   }
   return generationConfig;
 }
+__name(buildGenerationConfig, "buildGenerationConfig");
 function extractSystemInstruction(config) {
   const systemInstruction = config.systemInstruction;
   if (!systemInstruction) {
@@ -635,6 +648,7 @@ function extractSystemInstruction(config) {
   }
   return void 0;
 }
+__name(extractSystemInstruction, "extractSystemInstruction");
 function extractStreamInteractionId(event) {
   if (event.interaction_id || event.interactionId) {
     return event.interaction_id || event.interactionId;
@@ -647,6 +661,7 @@ function extractStreamInteractionId(event) {
   }
   return void 0;
 }
+__name(extractStreamInteractionId, "extractStreamInteractionId");
 async function* generateContentViaInteractions(apiClient, llmRequest, stream) {
   let contents = llmRequest.contents;
   if (llmRequest.previousInteractionId && contents) {
@@ -680,13 +695,12 @@ async function* generateContentViaInteractions(apiClient, llmRequest, stream) {
     });
     const aggregatedParts = [];
     for await (const event of responses) {
-      const sseEvent = event;
-      const interactionId = extractStreamInteractionId(sseEvent);
+      const interactionId = extractStreamInteractionId(event);
       if (interactionId) {
         currentInteractionId = interactionId;
       }
       const llmResponse = convertInteractionEventToLlmResponse(
-        sseEvent,
+        event,
         aggregatedParts,
         currentInteractionId
       );
@@ -717,6 +731,7 @@ async function* generateContentViaInteractions(apiClient, llmRequest, stream) {
     yield convertInteractionToLlmResponse(interaction);
   }
 }
+__name(generateContentViaInteractions, "generateContentViaInteractions");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   buildGenerationConfig,

package/dist/cjs/models/llm_request.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -45,6 +46,7 @@ function appendInstructions(llmRequest, instructions) {
     llmRequest.config.systemInstruction = newInstructions;
   }
 }
+__name(appendInstructions, "appendInstructions");
 function appendTools(llmRequest, tools) {
   if (!(tools == null ? void 0 : tools.length)) {
     return;
@@ -67,6 +69,7 @@ function appendTools(llmRequest, tools) {
     llmRequest.config.tools.push({ functionDeclarations });
   }
 }
+__name(appendTools, "appendTools");
 function setOutputSchema(llmRequest, schema) {
   if (!llmRequest.config) {
     llmRequest.config = {};
@@ -74,6 +77,7 @@ function setOutputSchema(llmRequest, schema) {
   llmRequest.config.responseSchema = schema;
   llmRequest.config.responseMimeType = "application/json";
 }
+__name(setOutputSchema, "setOutputSchema");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   appendInstructions,

package/dist/cjs/models/llm_response.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -67,6 +68,7 @@ function createLlmResponse(response) {
     usageMetadata
   };
 }
+__name(createLlmResponse, "createLlmResponse");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   createLlmResponse

package/dist/cjs/models/registry.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -29,13 +30,14 @@ __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_chrome_prompt_llm = require("./chrome_prompt_llm.js");
 var import_google_llm = require("./google_llm.js");
 /**
  * @license
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-class LRUCache {
+const _LRUCache = class _LRUCache {
   constructor(maxSize) {
     this.maxSize = maxSize;
     this.cache = /* @__PURE__ */ new Map();
@@ -57,7 +59,9 @@ class LRUCache {
     }
     this.cache.set(key, value);
   }
-}
+};
+__name(_LRUCache, "LRUCache");
+let LRUCache = _LRUCache;
 const _LLMRegistry = class _LLMRegistry {
   /**
    * Creates a new LLM instance.
@@ -108,6 +112,7 @@ const _LLMRegistry = class _LLMRegistry {
     throw new Error(`Model ${model} not found.`);
   }
 };
+__name(_LLMRegistry, "LLMRegistry");
 /**
  * Key is the regex that matches the model name.
  * Value is the class that implements the model.
@@ -117,6 +122,7 @@ _LLMRegistry.resolveCache = new LRUCache(32);
 let LLMRegistry = _LLMRegistry;
 LLMRegistry.register(import_google_llm.Gemini);
 LLMRegistry.register(import_apigee_llm.ApigeeLlm);
+LLMRegistry.register(import_chrome_prompt_llm.ChromeBuiltInLlm);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   LLMRegistry

package/dist/cjs/models/routed_llm.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -22,21 +23,12 @@ var __copyProps = (to, from, except, desc) => {
   return to;
 };
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var __decorateClass = (decorators, target, key, kind) => {
-  var result = kind > 1 ? void 0 : kind ? __getOwnPropDesc(target, key) : target;
-  for (var i = decorators.length - 1, decorator; i >= 0; i--)
-    if (decorator = decorators[i])
-      result = (kind ? decorator(target, key, result) : decorator(result)) || result;
-  if (kind && result) __defProp(target, key, result);
-  return result;
-};
 var routed_llm_exports = {};
 __export(routed_llm_exports, {
   RoutedLlm: () => RoutedLlm
 });
 module.exports = __toCommonJS(routed_llm_exports);
 var import_base_llm = require("./base_llm.js");
-var import_experimental = require("../utils/experimental.js");
 var import_failover_utils = require("../utils/failover_utils.js");
 var import_logger = require("../utils/logger.js");
 /**
@@ -44,7 +36,7 @@ var import_logger = require("../utils/logger.js");
  * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-let RoutedLlm = class extends import_base_llm.BaseLlm {
+const _RoutedLlm = class _RoutedLlm extends import_base_llm.BaseLlm {
   constructor({
     models,
     router,
@@ -90,9 +82,8 @@ let RoutedLlm = class extends import_base_llm.BaseLlm {
     return result.value;
   }
 };
-RoutedLlm = __decorateClass([
-  import_experimental.experimental
-], RoutedLlm);
+__name(_RoutedLlm, "RoutedLlm");
+let RoutedLlm = _RoutedLlm;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   RoutedLlm

package/dist/cjs/plugins/_reflect_retry_utils.js

@@ -0,0 +1,142 @@
+/**
+  * @license
+  * Copyright 2026 Google LLC
+  * SPDX-License-Identifier: Apache-2.0
+  */
+
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var reflect_retry_utils_exports = {};
+__export(reflect_retry_utils_exports, {
+  GLOBAL_SCOPE_KEY: () => GLOBAL_SCOPE_KEY,
+  REFLECT_AND_RETRY_RESPONSE_TYPE: () => REFLECT_AND_RETRY_RESPONSE_TYPE,
+  ScopedFailureTracker: () => ScopedFailureTracker,
+  TrackingScope: () => TrackingScope,
+  resolveScopeKey: () => resolveScopeKey
+});
+module.exports = __toCommonJS(reflect_retry_utils_exports);
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const REFLECT_AND_RETRY_RESPONSE_TYPE = "ERROR_HANDLED_BY_REFLECT_AND_RETRY_PLUGIN";
+const GLOBAL_SCOPE_KEY = "__global_reflect_and_retry_scope__";
+var TrackingScope = /* @__PURE__ */ ((TrackingScope2) => {
+  TrackingScope2["INVOCATION"] = "invocation";
+  TrackingScope2["GLOBAL"] = "global";
+  return TrackingScope2;
+})(TrackingScope || {});
+function resolveScopeKey(scope, invocationId) {
+  if (scope === "invocation" /* INVOCATION */) {
+    if (!invocationId) {
+      throw new Error("invocation_id must be provided for INVOCATION scope");
+    }
+    return invocationId;
+  } else if (scope === "global" /* GLOBAL */) {
+    return GLOBAL_SCOPE_KEY;
+  }
+  throw new Error(`Unknown scope: ${scope}`);
+}
+__name(resolveScopeKey, "resolveScopeKey");
+const _ScopedFailureTracker = class _ScopedFailureTracker {
+  constructor() {
+    this.scopedFailureCounters = /* @__PURE__ */ new Map();
+    this.lockPromise = Promise.resolve();
+  }
+  async acquireLock() {
+    let release;
+    const nextLock = new Promise((resolve) => {
+      release = resolve;
+    });
+    const currentLock = this.lockPromise;
+    this.lockPromise = this.lockPromise.then(() => nextLock);
+    await currentLock;
+    return release;
+  }
+  /**
+   * Atomically increments and returns the failure count for an item.
+   *
+   * @param scopeKey - The scope identifier (e.g. invocation ID or global key).
+   * @param itemName - The name of the tool or model.
+   * @returns The updated failure count.
+   */
+  async increment(scopeKey, itemName) {
+    var _a;
+    const release = await this.acquireLock();
+    try {
+      let counter = this.scopedFailureCounters.get(scopeKey);
+      if (!counter) {
+        counter = /* @__PURE__ */ new Map();
+        this.scopedFailureCounters.set(scopeKey, counter);
+      }
+      const current = ((_a = counter.get(itemName)) != null ? _a : 0) + 1;
+      counter.set(itemName, current);
+      return current;
+    } finally {
+      release();
+    }
+  }
+  /**
+   * Atomically resets the failure count for an item and cleans up state.
+   *
+   * @param scopeKey - The scope identifier.
+   * @param itemName - The name of the tool or model.
+   */
+  async reset(scopeKey, itemName) {
+    const release = await this.acquireLock();
+    try {
+      const counter = this.scopedFailureCounters.get(scopeKey);
+      if (counter) {
+        counter.delete(itemName);
+        if (counter.size === 0) {
+          this.scopedFailureCounters.delete(scopeKey);
+        }
+      }
+    } finally {
+      release();
+    }
+  }
+  /**
+   * Gets the current failure count for an item.
+   *
+   * @param scopeKey - The scope identifier.
+   * @param itemName - The name of the tool or model.
+   */
+  async getCount(scopeKey, itemName) {
+    var _a, _b;
+    const release = await this.acquireLock();
+    try {
+      return (_b = (_a = this.scopedFailureCounters.get(scopeKey)) == null ? void 0 : _a.get(itemName)) != null ? _b : 0;
+    } finally {
+      release();
+    }
+  }
+};
+__name(_ScopedFailureTracker, "ScopedFailureTracker");
+let ScopedFailureTracker = _ScopedFailureTracker;
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  GLOBAL_SCOPE_KEY,
+  REFLECT_AND_RETRY_RESPONSE_TYPE,
+  ScopedFailureTracker,
+  TrackingScope,
+  resolveScopeKey
+});

package/dist/cjs/plugins/base_plugin.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -48,7 +49,7 @@ var ContextCompactionTrigger = /* @__PURE__ */ ((ContextCompactionTrigger2) => {
   ContextCompactionTrigger2["AgentControlled"] = "AgentControlled";
   return ContextCompactionTrigger2;
 })(ContextCompactionTrigger || {});
-class BasePlugin {
+const _BasePlugin = class _BasePlugin {
   /**
    * Initializes the plugin.
    *
@@ -153,6 +154,37 @@ class BasePlugin {
   async afterAgentCallback(params) {
     return;
   }
+  /**
+   * Callback executed before a workflow node runs.
+   *
+   * @param params.node The node that is about to run.
+   * @param params.nodeContext The node's execution context.
+   * @param params.input The input the node is about to receive.
+   * @returns An optional value. If anything other than `undefined` is returned,
+   *     the node's body is skipped and the returned value becomes its output,
+   *     which is how a plugin implements a node-level cache or a stub.
+   *     Returning `undefined` lets the node run normally.
+   */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  async beforeNodeCallback(params) {
+    return;
+  }
+  /**
+   * Callback executed after a workflow node has run.
+   *
+   * Not called for a node whose body was skipped by `beforeNodeCallback`, and
+   * not called when the node throws.
+   *
+   * @param params.node The node that has just run.
+   * @param params.nodeContext The node's execution context.
+   * @param params.output The output the node produced.
+   * @returns An optional value replacing the node's output. Returning
+   *     `undefined` keeps the original.
+   */
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  async afterNodeCallback(params) {
+    return;
+  }
   /**
    * Callback executed before a request is sent to the model.
    *
@@ -281,16 +313,18 @@ class BasePlugin {
   async onToolErrorCallback(params) {
     return;
   }
-}
+};
+__name(_BasePlugin, "BasePlugin");
 __decorateClass([
   import_experimental.experimental
-], BasePlugin.prototype, "beforeToolSelection", 1);
+], _BasePlugin.prototype, "beforeToolSelection", 1);
 __decorateClass([
   import_experimental.experimental
-], BasePlugin.prototype, "beforeContextCompaction", 1);
+], _BasePlugin.prototype, "beforeContextCompaction", 1);
 __decorateClass([
   import_experimental.experimental
-], BasePlugin.prototype, "afterContextCompaction", 1);
+], _BasePlugin.prototype, "afterContextCompaction", 1);
+let BasePlugin = _BasePlugin;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   BasePlugin,

package/dist/cjs/plugins/global_instruction_plugin.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -35,7 +36,7 @@ var import_base_plugin = require("./base_plugin.js");
  * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-class GlobalInstructionPlugin extends import_base_plugin.BasePlugin {
+const _GlobalInstructionPlugin = class _GlobalInstructionPlugin extends import_base_plugin.BasePlugin {
   /**
    * Initializes the GlobalInstructionPlugin.
    *
@@ -90,7 +91,9 @@ ${existingInstruction}`;
     }
     return await this.globalInstruction(readonlyContext);
   }
-}
+};
+__name(_GlobalInstructionPlugin, "GlobalInstructionPlugin");
+let GlobalInstructionPlugin = _GlobalInstructionPlugin;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   GlobalInstructionPlugin

package/dist/cjs/plugins/logging_plugin.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -35,7 +36,7 @@ var import_base_plugin = require("./base_plugin.js");
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-class LoggingPlugin extends import_base_plugin.BasePlugin {
+const _LoggingPlugin = class _LoggingPlugin extends import_base_plugin.BasePlugin {
   /**
    * Initialize the logging plugin.
    *
@@ -48,13 +49,13 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
     invocationContext,
     userMessage
   }) {
-    var _a;
+    var _a, _b;
     this.log("\u{1F680} USER MESSAGE RECEIVED");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
     this.log(`   Session ID: ${invocationContext.session.id}`);
     this.log(`   User ID: ${invocationContext.userId}`);
     this.log(`   App Name: ${invocationContext.appName}`);
-    this.log(`   Root Agent: ${(_a = invocationContext.agent.name) != null ? _a : "Unknown"}`);
+    this.log(`   Root Agent: ${(_b = (_a = invocationContext.agent) == null ? void 0 : _a.name) != null ? _b : "Unknown"}`);
     this.log(`   User Content: ${this.formatContent(userMessage)}`);
     if (invocationContext.branch) {
       this.log(`   Branch: ${invocationContext.branch}`);
@@ -64,10 +65,12 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
   async beforeRunCallback({
     invocationContext
   }) {
-    var _a;
+    var _a, _b;
     this.log("\u{1F3C3} INVOCATION STARTING");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
-    this.log(`   Starting Agent: ${(_a = invocationContext.agent.name) != null ? _a : "Unknown"}`);
+    this.log(
+      `   Starting Agent: ${(_b = (_a = invocationContext.agent) == null ? void 0 : _a.name) != null ? _b : "Unknown"}`
+    );
     return void 0;
   }
   async onEventCallback({
@@ -96,10 +99,10 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
   async afterRunCallback({
     invocationContext
   }) {
-    var _a;
+    var _a, _b;
     this.log("\u2705 INVOCATION COMPLETED");
     this.log(`   Invocation ID: ${invocationContext.invocationId}`);
-    this.log(`   Final Agent: ${(_a = invocationContext.agent.name) != null ? _a : "Unknown"}`);
+    this.log(`   Final Agent: ${(_b = (_a = invocationContext.agent) == null ? void 0 : _a.name) != null ? _b : "Unknown"}`);
     return void 0;
   }
   async beforeAgentCallback({
@@ -252,7 +255,9 @@ class LoggingPlugin extends import_base_plugin.BasePlugin {
     }
     return formatted;
   }
-}
+};
+__name(_LoggingPlugin, "LoggingPlugin");
+let LoggingPlugin = _LoggingPlugin;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   LoggingPlugin

package/dist/cjs/plugins/plugin_manager.js

@@ -9,6 +9,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -33,7 +34,7 @@ var import_logger = require("../utils/logger.js");
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-class PluginManager {
+const _PluginManager = class _PluginManager {
   /**
    * Initializes the plugin service.
    *
@@ -48,6 +49,10 @@ class PluginManager {
       }
     }
   }
+  /** Whether any plugin is registered. */
+  get hasPlugins() {
+    return this.plugins.size > 0;
+  }
   /**
    * Registers a new plugin.
    *
@@ -180,6 +185,34 @@ class PluginManager {
       "afterAgentCallback"
     );
   }
+  /**
+   * Runs the `beforeNodeCallback` for all plugins.
+   */
+  async runBeforeNodeCallback({
+    node,
+    nodeContext,
+    input
+  }) {
+    return this.runCallbacks(
+      this.plugins,
+      (plugin) => plugin.beforeNodeCallback({ node, nodeContext, input }),
+      "beforeNodeCallback"
+    );
+  }
+  /**
+   * Runs the `afterNodeCallback` for all plugins.
+   */
+  async runAfterNodeCallback({
+    node,
+    nodeContext,
+    output
+  }) {
+    return this.runCallbacks(
+      this.plugins,
+      (plugin) => plugin.afterNodeCallback({ node, nodeContext, output }),
+      "afterNodeCallback"
+    );
+  }
   /**
    * Runs the `beforeToolSelection` for all plugins.
    */
@@ -303,7 +336,9 @@ class PluginManager {
       "onToolErrorCallback"
     );
   }
-}
+};
+__name(_PluginManager, "PluginManager");
+let PluginManager = _PluginManager;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   PluginManager