@google/adk 1.6.0 → 2.1.0

package/dist/web/models/llm_request.js

@@ -1,52 +0,0 @@
-import {createRequire as topLevelCreateRequire} from 'module';
-const require = topLevelCreateRequire(import.meta.url);
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-function appendInstructions(llmRequest, instructions) {
-  if (!llmRequest.config) {
-    llmRequest.config = {};
-  }
-  const newInstructions = instructions.join("\n\n");
-  if (llmRequest.config.systemInstruction) {
-    llmRequest.config.systemInstruction += "\n\n" + newInstructions;
-  } else {
-    llmRequest.config.systemInstruction = newInstructions;
-  }
-}
-function appendTools(llmRequest, tools) {
-  if (!(tools == null ? void 0 : tools.length)) {
-    return;
-  }
-  const functionDeclarations = [];
-  for (const tool of tools) {
-    const declaration = tool._getDeclaration();
-    if (declaration) {
-      functionDeclarations.push(declaration);
-      llmRequest.toolsDict[tool.name] = tool;
-    }
-  }
-  if (functionDeclarations.length) {
-    if (!llmRequest.config) {
-      llmRequest.config = {};
-    }
-    if (!llmRequest.config.tools) {
-      llmRequest.config.tools = [];
-    }
-    llmRequest.config.tools.push({ functionDeclarations });
-  }
-}
-function setOutputSchema(llmRequest, schema) {
-  if (!llmRequest.config) {
-    llmRequest.config = {};
-  }
-  llmRequest.config.responseSchema = schema;
-  llmRequest.config.responseMimeType = "application/json";
-}
-export {
-  appendInstructions,
-  appendTools,
-  setOutputSchema
-};

package/dist/web/models/llm_response.js

@@ -1,45 +0,0 @@
-import {createRequire as topLevelCreateRequire} from 'module';
-const require = topLevelCreateRequire(import.meta.url);
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-function createLlmResponse(response) {
-  var _a;
-  const usageMetadata = response.usageMetadata;
-  if (response.candidates && response.candidates.length > 0) {
-    const candidate = response.candidates[0];
-    if (((_a = candidate.content) == null ? void 0 : _a.parts) && candidate.content.parts.length > 0) {
-      return {
-        content: candidate.content,
-        groundingMetadata: candidate.groundingMetadata,
-        citationMetadata: candidate.citationMetadata,
-        usageMetadata,
-        finishReason: candidate.finishReason
-      };
-    }
-    return {
-      errorCode: candidate.finishReason,
-      errorMessage: candidate.finishMessage,
-      usageMetadata,
-      citationMetadata: candidate.citationMetadata,
-      finishReason: candidate.finishReason
-    };
-  }
-  if (response.promptFeedback) {
-    return {
-      errorCode: response.promptFeedback.blockReason,
-      errorMessage: response.promptFeedback.blockReasonMessage,
-      usageMetadata
-    };
-  }
-  return {
-    errorCode: "UNKNOWN_ERROR",
-    errorMessage: "Unknown error.",
-    usageMetadata
-  };
-}
-export {
-  createLlmResponse
-};

package/dist/web/models/registry.js

@@ -1,95 +0,0 @@
-import {createRequire as topLevelCreateRequire} from 'module';
-const require = topLevelCreateRequire(import.meta.url);
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { logger } from "../utils/logger.js";
-import { ApigeeLlm } from "./apigee_llm.js";
-import { Gemini } from "./google_llm.js";
-class LRUCache {
-  constructor(maxSize) {
-    this.maxSize = maxSize;
-    this.cache = /* @__PURE__ */ new Map();
-  }
-  get(key) {
-    const item = this.cache.get(key);
-    if (item) {
-      this.cache.delete(key);
-      this.cache.set(key, item);
-    }
-    return item;
-  }
-  set(key, value) {
-    if (this.cache.size >= this.maxSize && !this.cache.has(key)) {
-      const lruKey = this.cache.keys().next().value;
-      if (lruKey !== void 0) {
-        this.cache.delete(lruKey);
-      }
-    }
-    this.cache.set(key, value);
-  }
-}
-const _LLMRegistry = class _LLMRegistry {
-  /**
-   * Creates a new LLM instance.
-   * @param model The model name.
-   * @returns The LLM instance.
-   */
-  static newLlm(model) {
-    return new (_LLMRegistry.resolve(model))({ model });
-  }
-  static _register(modelNameRegex, llmCls) {
-    if (_LLMRegistry.llmRegistryDict.has(modelNameRegex)) {
-      logger.info(
-        "Updating LLM class for ".concat(modelNameRegex, " from ").concat(_LLMRegistry.llmRegistryDict.get(modelNameRegex), " to ").concat(llmCls)
-      );
-    }
-    _LLMRegistry.llmRegistryDict.set(modelNameRegex, llmCls);
-  }
-  /**
-   * Registers a new LLM class.
-   * @param llmCls The class that implements the model.
-   */
-  static register(llmCls) {
-    for (const regex of llmCls.supportedModels) {
-      _LLMRegistry._register(regex, llmCls);
-    }
-  }
-  /**
-   * Resolves the model to a BaseLlm subclass.
-   * @param model The model name.
-   * @returns The BaseLlm subclass.
-   * @throws If the model is not found.
-   */
-  static resolve(model) {
-    const cachedLlm = _LLMRegistry.resolveCache.get(model);
-    if (cachedLlm) {
-      return cachedLlm;
-    }
-    for (const [regex, llmClass] of _LLMRegistry.llmRegistryDict.entries()) {
-      const pattern = new RegExp(
-        "^".concat(regex instanceof RegExp ? regex.source : regex, "$"),
-        regex instanceof RegExp ? regex.flags : void 0
-      );
-      if (pattern.test(model)) {
-        _LLMRegistry.resolveCache.set(model, llmClass);
-        return llmClass;
-      }
-    }
-    throw new Error("Model ".concat(model, " not found."));
-  }
-};
-/**
- * Key is the regex that matches the model name.
- * Value is the class that implements the model.
- */
-_LLMRegistry.llmRegistryDict = /* @__PURE__ */ new Map();
-_LLMRegistry.resolveCache = new LRUCache(32);
-let LLMRegistry = _LLMRegistry;
-LLMRegistry.register(Gemini);
-LLMRegistry.register(ApigeeLlm);
-export {
-  LLMRegistry
-};

package/dist/web/models/routed_llm.js

@@ -1,122 +0,0 @@
-import {createRequire as topLevelCreateRequire} from 'module';
-const require = topLevelCreateRequire(import.meta.url);
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __knownSymbol = (name, symbol) => (symbol = Symbol[name]) ? symbol : Symbol.for("Symbol." + name);
-var __typeError = (msg) => {
-  throw TypeError(msg);
-};
-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 __await = function(promise, isYieldStar) {
-  this[0] = promise;
-  this[1] = isYieldStar;
-};
-var __asyncGenerator = (__this, __arguments, generator) => {
-  var resume = (k, v, yes, no) => {
-    try {
-      var x = generator[k](v), isAwait = (v = x.value) instanceof __await, done = x.done;
-      Promise.resolve(isAwait ? v[0] : v).then((y) => isAwait ? resume(k === "return" ? k : "next", v[1] ? { done: y.done, value: y.value } : y, yes, no) : yes({ value: y, done })).catch((e) => resume("throw", e, yes, no));
-    } catch (e) {
-      no(e);
-    }
-  }, method = (k) => it[k] = (x) => new Promise((yes, no) => resume(k, x, yes, no)), it = {};
-  return generator = generator.apply(__this, __arguments), it[__knownSymbol("asyncIterator")] = () => it, method("next"), method("throw"), method("return"), it;
-};
-var __yieldStar = (value) => {
-  var obj = value[__knownSymbol("asyncIterator")], isAwait = false, method, it = {};
-  if (obj == null) {
-    obj = value[__knownSymbol("iterator")]();
-    method = (k) => it[k] = (x) => obj[k](x);
-  } else {
-    obj = obj.call(value);
-    method = (k) => it[k] = (v) => {
-      if (isAwait) {
-        isAwait = false;
-        if (k === "throw") throw v;
-        return v;
-      }
-      isAwait = true;
-      return {
-        done: false,
-        value: new __await(new Promise((resolve) => {
-          var x = obj[k](v);
-          if (!(x instanceof Object)) __typeError("Object expected");
-          resolve(x);
-        }), 1)
-      };
-    };
-  }
-  return it[__knownSymbol("iterator")] = () => it, method("next"), "throw" in obj ? method("throw") : it.throw = (x) => {
-    throw x;
-  }, "return" in obj && method("return"), it;
-};
-/**
- * @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 {
-  constructor({
-    models,
-    router,
-    modelName = "routed-llm"
-  }) {
-    const modelsMap = Array.isArray(models) ? Object.fromEntries(models.map((m) => [m.model, m])) : models;
-    const modelNames = Object.entries(modelsMap).map(
-      ([name, model]) => "".concat(name, " (").concat(model.model, ")")
-    );
-    const computedName = "RoutedLlm[".concat(modelNames.join(", "), "]");
-    super({ model: modelName === "routed-llm" ? computedName : modelName });
-    this.models = modelsMap;
-    this.router = router;
-  }
-  /**
-   * Generates content by delegating to the selected model.
-   */
-  generateContentAsync(llmRequest, stream) {
-    return __asyncGenerator(this, null, function* () {
-      logger.info("Routing request via ".concat(this.model));
-      yield* __yieldStar(runWithRouting(
-        this.models,
-        llmRequest,
-        this.router,
-        (model) => model.generateContentAsync(llmRequest, stream)
-      ));
-    });
-  }
-  /**
-   * Creates a live connection to the LLM by delegating to the selected model.
-   * This live connection cannot be switched mid-stream, it is tied to the model
-   * selected at the time of connection.
-   */
-  async connect(llmRequest) {
-    const generator = runWithRouting(
-      this.models,
-      llmRequest,
-      this.router,
-      (model) => model.connect(llmRequest)
-    );
-    const result = await generator.next();
-    if (result.done || result.value === void 0) {
-      throw new Error("Failed to establish connection: No connection yielded.");
-    }
-    return result.value;
-  }
-};
-RoutedLlm = __decorateClass([
-  experimental
-], RoutedLlm);
-export {
-  RoutedLlm
-};

package/dist/web/plugins/base_plugin.js

@@ -1,271 +0,0 @@
-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;
-};
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { experimental } from "../utils/experimental.js";
-var ContextCompactionTrigger = /* @__PURE__ */ ((ContextCompactionTrigger2) => {
-  ContextCompactionTrigger2["Manual"] = "Manual";
-  ContextCompactionTrigger2["Auto"] = "Auto";
-  ContextCompactionTrigger2["AgentControlled"] = "AgentControlled";
-  return ContextCompactionTrigger2;
-})(ContextCompactionTrigger || {});
-class BasePlugin {
-  /**
-   * Initializes the plugin.
-   *
-   * @param name A unique identifier for this plugin instance.
-   */
-  constructor(name) {
-    this.name = name;
-  }
-  /**
-   * Callback executed when a user message is received before an invocation
-   * starts.
-   *
-   * This callback helps logging and modifying the user message before the
-   * runner starts the invocation.
-   *
-   * @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.
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async onUserMessageCallback(params) {
-    return;
-  }
-  /**
-   * Callback executed before the ADK runner runs.
-   *
-   * This is the first callback to be called in the lifecycle, ideal for global
-   * setup or initialization tasks.
-   *
-   * @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.
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async beforeRunCallback(params) {
-    return;
-  }
-  /**
-   * Callback executed after an event is yielded from runner.
-   *
-   * This is the ideal place to make modification to the event before the event
-   * is handled by the underlying agent app.
-   *
-   * @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. Copy `params.event` when
-   *     constructing a replacement to preserve fields that are not being
-   *     modified, such as event actions. Returning `undefined` allows the
-   *     original response to be used.
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async onEventCallback(params) {
-    return;
-  }
-  /**
-   * Callback executed after an ADK runner run has completed.
-   *
-   * This is the final callback in the ADK lifecycle, suitable for cleanup,
-   * final logging, or reporting tasks.
-   *
-   * @param params.invocationContext The context for the entire invocation.
-   * @returns undefined
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async afterRunCallback(params) {
-    return;
-  }
-  /**
-   * Callback executed before an agent's primary logic is invoked.
-   *
-   * This callback can be used for logging, setup, or to short-circuit the
-   * agent's execution by returning a value.
-   *
-   * @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.
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async beforeAgentCallback(params) {
-    return;
-  }
-  /**
-   * Callback executed after an agent's primary logic has completed.
-   *
-   * This callback can be used to inspect, log, or modify the agent's final
-   * result before it is returned.
-   *
-   * @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.
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async afterAgentCallback(params) {
-    return;
-  }
-  /**
-   * Callback executed before a request is sent to the model.
-   *
-   * This provides an opportunity to inspect, log, or modify the `LlmRequest`
-   * object. It can also be used to implement caching by returning a cached
-   * `LlmResponse`, which would skip the actual model call.
-   *
-   * @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.
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async beforeModelCallback(params) {
-    return;
-  }
-  /**
-   * Callback executed after a response is received from the model.
-   *
-   * This is the ideal place to log model responses, collect metrics on token
-   * usage, or perform post-processing on the raw `LlmResponse`.
-   *
-   * @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.
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async afterModelCallback(params) {
-    return;
-  }
-  /**
-   * Callback executed when a model call encounters an error.
-   *
-   * This callback provides an opportunity to handle model errors gracefully,
-   * potentially providing alternative responses or recovery mechanisms.
-   *
-   * @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 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.
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async onModelErrorCallback(params) {
-    return;
-  }
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async beforeToolSelection(params) {
-    return;
-  }
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async beforeContextCompaction(params) {
-    return;
-  }
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async afterContextCompaction(params) {
-    return;
-  }
-  /**
-   * Callback executed before a tool is called.
-   *
-   * This callback is useful for logging tool usage, input validation, or
-   * modifying the arguments before they are passed to the tool.
-   *
-   * @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 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.
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async beforeToolCallback(params) {
-    return;
-  }
-  /**
-   * Callback executed after a tool has been called.
-   *
-   * This callback allows for inspecting, logging, or modifying the result
-   * returned by a tool.
-   *
-   * @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.
-   */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async afterToolCallback(params) {
-    return;
-  }
-  /**
-     * Callback executed when a tool call encounters an error.
-      tool: BaseTool;
-      toolArgs: Record<string, unknown>;
-      toolContext: Context;
-      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.
-     */
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async onToolErrorCallback(params) {
-    return;
-  }
-}
-__decorateClass([
-  experimental
-], BasePlugin.prototype, "beforeToolSelection", 1);
-__decorateClass([
-  experimental
-], BasePlugin.prototype, "beforeContextCompaction", 1);
-__decorateClass([
-  experimental
-], BasePlugin.prototype, "afterContextCompaction", 1);
-export {
-  BasePlugin,
-  ContextCompactionTrigger
-};

package/dist/web/plugins/global_instruction_plugin.js

@@ -1,67 +0,0 @@
-import {createRequire as topLevelCreateRequire} from 'module';
-const require = topLevelCreateRequire(import.meta.url);
-/**
- * @license
- * Copyright 2026 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { injectSessionState } from "../agents/instructions.js";
-import { ReadonlyContext } from "../agents/readonly_context.js";
-import { BasePlugin } from "./base_plugin.js";
-class GlobalInstructionPlugin extends BasePlugin {
-  /**
-   * Initializes the GlobalInstructionPlugin.
-   *
-   * @param globalInstruction The instruction to apply globally. Can be a string
-   *     or an InstructionProvider function that takes ReadonlyContext and
-   *     returns a string (sync or async).
-   * @param name The name of the plugin (defaults to 'global_instruction').
-   */
-  constructor(globalInstruction, name = "global_instruction") {
-    super(name);
-    this.globalInstruction = globalInstruction;
-  }
-  async beforeModelCallback(params) {
-    if (!this.globalInstruction) {
-      return;
-    }
-    const readonlyContext = new ReadonlyContext(
-      params.callbackContext.invocationContext
-    );
-    const finalGlobalInstruction = await this.resolveGlobalInstruction(readonlyContext);
-    if (!finalGlobalInstruction) {
-      return;
-    }
-    if (!params.llmRequest.config) {
-      params.llmRequest.config = {};
-    }
-    const existingInstruction = params.llmRequest.config.systemInstruction;
-    if (!existingInstruction || Array.isArray(existingInstruction) && existingInstruction.length === 0) {
-      params.llmRequest.config.systemInstruction = finalGlobalInstruction;
-      return;
-    }
-    if (typeof existingInstruction === "string") {
-      params.llmRequest.config.systemInstruction = "".concat(finalGlobalInstruction, "\n\n").concat(existingInstruction);
-    } else if (Array.isArray(existingInstruction)) {
-      params.llmRequest.config.systemInstruction = [
-        finalGlobalInstruction,
-        ...existingInstruction
-      ];
-    } else {
-      params.llmRequest.config.systemInstruction = [
-        finalGlobalInstruction,
-        existingInstruction
-      ];
-    }
-    return;
-  }
-  async resolveGlobalInstruction(readonlyContext) {
-    if (typeof this.globalInstruction === "string") {
-      return await injectSessionState(this.globalInstruction, readonlyContext);
-    }
-    return await this.globalInstruction(readonlyContext);
-  }
-}
-export {
-  GlobalInstructionPlugin
-};