@google/adk 1.6.0 → 2.1.0

package/dist/esm/models/interactions_utils.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2026 Google LLC
@@ -21,6 +23,7 @@ function getInteractionMediaType(mimeType) {
       return "document";
   }
 }
+__name(getInteractionMediaType, "getInteractionMediaType");
 function getLatestUserContents(contents) {
   if (!contents || contents.length === 0) {
     return [];
@@ -63,6 +66,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 };
@@ -85,6 +89,7 @@ function convertPartToMediaContent(part) {
   }
   return null;
 }
+__name(convertPartToMediaContent, "convertPartToMediaContent");
 function convertContentToSteps(content) {
   const steps = [];
   const role = content.role || "user";
@@ -170,6 +175,7 @@ function convertContentToSteps(content) {
   }
   return steps;
 }
+__name(convertContentToSteps, "convertContentToSteps");
 function convertMediaContentToPart(content) {
   if (content.type === "text") {
     return { text: content.text || "" };
@@ -194,6 +200,7 @@ function convertMediaContentToPart(content) {
   }
   return null;
 }
+__name(convertMediaContentToPart, "convertMediaContentToPart");
 function convertStepToParts(step) {
   if (!step || !step.type) {
     return [];
@@ -289,6 +296,7 @@ function convertStepToParts(step) {
       return [];
   }
 }
+__name(convertStepToParts, "convertStepToParts");
 function convertToolsConfigToInteractionsFormat(config) {
   if (!config.tools) {
     return [];
@@ -337,6 +345,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])) {
@@ -345,6 +354,7 @@ function findLastPart(parts, predicate) {
   }
   return void 0;
 }
+__name(findLastPart, "findLastPart");
 function getLatestModelParts(steps) {
   if (!steps || steps.length === 0) {
     return [];
@@ -360,6 +370,7 @@ function getLatestModelParts(steps) {
   }
   return latestParts;
 }
+__name(getLatestModelParts, "getLatestModelParts");
 function convertInteractionToLlmResponse(interaction) {
   if (interaction.status === "failed") {
     let errorMsg = "Unknown error";
@@ -401,6 +412,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;
@@ -556,6 +568,7 @@ function convertInteractionEventToLlmResponse(event, aggregatedParts, interactio
   }
   return null;
 }
+__name(convertInteractionEventToLlmResponse, "convertInteractionEventToLlmResponse");
 function buildGenerationConfig(config) {
   const generationConfig = {};
   if (config.temperature !== void 0 && config.temperature !== null) {
@@ -581,6 +594,7 @@ function buildGenerationConfig(config) {
   }
   return generationConfig;
 }
+__name(buildGenerationConfig, "buildGenerationConfig");
 function extractSystemInstruction(config) {
   const systemInstruction = config.systemInstruction;
   if (!systemInstruction) {
@@ -601,6 +615,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;
@@ -613,6 +628,7 @@ function extractStreamInteractionId(event) {
   }
   return void 0;
 }
+__name(extractStreamInteractionId, "extractStreamInteractionId");
 async function* generateContentViaInteractions(apiClient, llmRequest, stream) {
   let contents = llmRequest.contents;
   if (llmRequest.previousInteractionId && contents) {
@@ -646,13 +662,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
       );
@@ -683,6 +698,7 @@ async function* generateContentViaInteractions(apiClient, llmRequest, stream) {
     yield convertInteractionToLlmResponse(interaction);
   }
 }
+__name(generateContentViaInteractions, "generateContentViaInteractions");
 export {
   buildGenerationConfig,
   convertContentToSteps,

package/dist/esm/models/llm_request.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -16,6 +18,7 @@ function appendInstructions(llmRequest, instructions) {
     llmRequest.config.systemInstruction = newInstructions;
   }
 }
+__name(appendInstructions, "appendInstructions");
 function appendTools(llmRequest, tools) {
   if (!(tools == null ? void 0 : tools.length)) {
     return;
@@ -38,6 +41,7 @@ function appendTools(llmRequest, tools) {
     llmRequest.config.tools.push({ functionDeclarations });
   }
 }
+__name(appendTools, "appendTools");
 function setOutputSchema(llmRequest, schema) {
   if (!llmRequest.config) {
     llmRequest.config = {};
@@ -45,6 +49,7 @@ function setOutputSchema(llmRequest, schema) {
   llmRequest.config.responseSchema = schema;
   llmRequest.config.responseMimeType = "application/json";
 }
+__name(setOutputSchema, "setOutputSchema");
 export {
   appendInstructions,
   appendTools,

package/dist/esm/models/llm_response.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -40,6 +42,7 @@ function createLlmResponse(response) {
     usageMetadata
   };
 }
+__name(createLlmResponse, "createLlmResponse");
 export {
   createLlmResponse
 };

package/dist/esm/models/registry.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -7,8 +9,9 @@ const require = topLevelCreateRequire(import.meta.url);
  */
 import { logger } from "../utils/logger.js";
 import { ApigeeLlm } from "./apigee_llm.js";
+import { ChromeBuiltInLlm } from "./chrome_prompt_llm.js";
 import { Gemini } from "./google_llm.js";
-class LRUCache {
+const _LRUCache = class _LRUCache {
   constructor(maxSize) {
     this.maxSize = maxSize;
     this.cache = /* @__PURE__ */ new Map();
@@ -30,7 +33,9 @@ class LRUCache {
     }
     this.cache.set(key, value);
   }
-}
+};
+__name(_LRUCache, "LRUCache");
+let LRUCache = _LRUCache;
 const _LLMRegistry = class _LLMRegistry {
   /**
    * Creates a new LLM instance.
@@ -81,6 +86,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.
@@ -90,6 +96,7 @@ _LLMRegistry.resolveCache = new LRUCache(32);
 let LLMRegistry = _LLMRegistry;
 LLMRegistry.register(Gemini);
 LLMRegistry.register(ApigeeLlm);
+LLMRegistry.register(ChromeBuiltInLlm);
 export {
   LLMRegistry
 };

package/dist/esm/models/routed_llm.js

@@ -1,25 +1,16 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
 var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-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 __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 import { BaseLlm } from "./base_llm.js";
-import { experimental } from "../utils/experimental.js";
 import { runWithRouting } from "../utils/failover_utils.js";
 import { logger } from "../utils/logger.js";
-let RoutedLlm = class extends BaseLlm {
+const _RoutedLlm = class _RoutedLlm extends BaseLlm {
   constructor({
     models,
     router,
@@ -65,9 +56,8 @@ let RoutedLlm = class extends BaseLlm {
     return result.value;
   }
 };
-RoutedLlm = __decorateClass([
-  experimental
-], RoutedLlm);
+__name(_RoutedLlm, "RoutedLlm");
+let RoutedLlm = _RoutedLlm;
 export {
   RoutedLlm
 };

package/dist/esm/plugins/_reflect_retry_utils.js

@@ -0,0 +1,111 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+/**
+ * @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;
+export {
+  GLOBAL_SCOPE_KEY,
+  REFLECT_AND_RETRY_RESPONSE_TYPE,
+  ScopedFailureTracker,
+  TrackingScope,
+  resolveScopeKey
+};

package/dist/esm/plugins/base_plugin.js

@@ -2,6 +2,7 @@ import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 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--)
@@ -22,7 +23,7 @@ var ContextCompactionTrigger = /* @__PURE__ */ ((ContextCompactionTrigger2) => {
   ContextCompactionTrigger2["AgentControlled"] = "AgentControlled";
   return ContextCompactionTrigger2;
 })(ContextCompactionTrigger || {});
-class BasePlugin {
+const _BasePlugin = class _BasePlugin {
   /**
    * Initializes the plugin.
    *
@@ -127,6 +128,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.
    *
@@ -255,16 +287,18 @@ class BasePlugin {
   async onToolErrorCallback(params) {
     return;
   }
-}
+};
+__name(_BasePlugin, "BasePlugin");
 __decorateClass([
   experimental
-], BasePlugin.prototype, "beforeToolSelection", 1);
+], _BasePlugin.prototype, "beforeToolSelection", 1);
 __decorateClass([
   experimental
-], BasePlugin.prototype, "beforeContextCompaction", 1);
+], _BasePlugin.prototype, "beforeContextCompaction", 1);
 __decorateClass([
   experimental
-], BasePlugin.prototype, "afterContextCompaction", 1);
+], _BasePlugin.prototype, "afterContextCompaction", 1);
+let BasePlugin = _BasePlugin;
 export {
   BasePlugin,
   ContextCompactionTrigger

package/dist/esm/plugins/global_instruction_plugin.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2026 Google LLC
@@ -8,7 +10,7 @@ const require = topLevelCreateRequire(import.meta.url);
 import { injectSessionState } from "../agents/instructions.js";
 import { ReadonlyContext } from "../agents/readonly_context.js";
 import { BasePlugin } from "./base_plugin.js";
-class GlobalInstructionPlugin extends BasePlugin {
+const _GlobalInstructionPlugin = class _GlobalInstructionPlugin extends BasePlugin {
   /**
    * Initializes the GlobalInstructionPlugin.
    *
@@ -63,7 +65,9 @@ ${existingInstruction}`;
     }
     return await this.globalInstruction(readonlyContext);
   }
-}
+};
+__name(_GlobalInstructionPlugin, "GlobalInstructionPlugin");
+let GlobalInstructionPlugin = _GlobalInstructionPlugin;
 export {
   GlobalInstructionPlugin
 };

package/dist/esm/plugins/logging_plugin.js

@@ -1,5 +1,7 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -12,7 +14,7 @@ import {
 } from "../events/event.js";
 import { logger } from "../utils/logger.js";
 import { BasePlugin } from "./base_plugin.js";
-class LoggingPlugin extends BasePlugin {
+const _LoggingPlugin = class _LoggingPlugin extends BasePlugin {
   /**
    * Initialize the logging plugin.
    *
@@ -25,13 +27,13 @@ class LoggingPlugin extends 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}`);
@@ -41,10 +43,12 @@ class LoggingPlugin extends 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({
@@ -73,10 +77,10 @@ class LoggingPlugin extends 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({
@@ -229,7 +233,9 @@ class LoggingPlugin extends BasePlugin {
     }
     return formatted;
   }
-}
+};
+__name(_LoggingPlugin, "LoggingPlugin");
+let LoggingPlugin = _LoggingPlugin;
 export {
   LoggingPlugin
 };

package/dist/esm/plugins/plugin_manager.js

@@ -1,12 +1,14 @@
 import {createRequire as topLevelCreateRequire} from 'module';
 const require = topLevelCreateRequire(import.meta.url);
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 /**
  * @license
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 import { logger } from "../utils/logger.js";
-class PluginManager {
+const _PluginManager = class _PluginManager {
   /**
    * Initializes the plugin service.
    *
@@ -21,6 +23,10 @@ class PluginManager {
       }
     }
   }
+  /** Whether any plugin is registered. */
+  get hasPlugins() {
+    return this.plugins.size > 0;
+  }
   /**
    * Registers a new plugin.
    *
@@ -153,6 +159,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.
    */
@@ -276,7 +310,9 @@ class PluginManager {
       "onToolErrorCallback"
     );
   }
-}
+};
+__name(_PluginManager, "PluginManager");
+let PluginManager = _PluginManager;
 export {
   PluginManager
 };