@google/adk 2.0.0 → 2.2.0

package/dist/web/plugins/logging_plugin.js

@@ -1,241 +0,0 @@
-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 {
-  getFunctionCalls,
-  getFunctionResponses,
-  isFinalResponse
-} from "../events/event.js";
-import { logger } from "../utils/logger.js";
-import { BasePlugin } from "./base_plugin.js";
-const _LoggingPlugin = class _LoggingPlugin extends BasePlugin {
-  /**
-   * Initialize the logging plugin.
-   *
-   * @param name The name of the plugin instance.
-   */
-  constructor(name = "logging_plugin") {
-    super(name);
-  }
-  async onUserMessageCallback({
-    invocationContext,
-    userMessage
-  }) {
-    var _a, _b;
-    this.log("\u{1F680} USER MESSAGE RECEIVED");
-    this.log("   Invocation ID: ".concat(invocationContext.invocationId));
-    this.log("   Session ID: ".concat(invocationContext.session.id));
-    this.log("   User ID: ".concat(invocationContext.userId));
-    this.log("   App Name: ".concat(invocationContext.appName));
-    this.log("   Root Agent: ".concat((_b = (_a = invocationContext.agent) == null ? void 0 : _a.name) != null ? _b : "Unknown"));
-    this.log("   User Content: ".concat(this.formatContent(userMessage)));
-    if (invocationContext.branch) {
-      this.log("   Branch: ".concat(invocationContext.branch));
-    }
-    return void 0;
-  }
-  async beforeRunCallback({
-    invocationContext
-  }) {
-    var _a, _b;
-    this.log("\u{1F3C3} INVOCATION STARTING");
-    this.log("   Invocation ID: ".concat(invocationContext.invocationId));
-    this.log(
-      "   Starting Agent: ".concat((_b = (_a = invocationContext.agent) == null ? void 0 : _a.name) != null ? _b : "Unknown")
-    );
-    return void 0;
-  }
-  async onEventCallback({
-    event
-  }) {
-    this.log("\u{1F4E2} EVENT YIELDED");
-    this.log("   Event ID: ".concat(event.id));
-    this.log("   Author: ".concat(event.author));
-    this.log("   Content: ".concat(this.formatContent(event.content)));
-    this.log("   Final Response: ".concat(isFinalResponse(event)));
-    const functionCalls = getFunctionCalls(event);
-    if (functionCalls.length > 0) {
-      const funcCalls = functionCalls.map((fc) => fc.name);
-      this.log("   Function Calls: ".concat(funcCalls));
-    }
-    const functionResponses = getFunctionResponses(event);
-    if (functionResponses.length > 0) {
-      const funcResponses = functionResponses.map((fr) => fr.name);
-      this.log("   Function Responses: ".concat(funcResponses));
-    }
-    if (event.longRunningToolIds && event.longRunningToolIds.length > 0) {
-      this.log("   Long Running Tools: ".concat([...event.longRunningToolIds]));
-    }
-    return void 0;
-  }
-  async afterRunCallback({
-    invocationContext
-  }) {
-    var _a, _b;
-    this.log("\u2705 INVOCATION COMPLETED");
-    this.log("   Invocation ID: ".concat(invocationContext.invocationId));
-    this.log("   Final Agent: ".concat((_b = (_a = invocationContext.agent) == null ? void 0 : _a.name) != null ? _b : "Unknown"));
-    return void 0;
-  }
-  async beforeAgentCallback({
-    callbackContext
-  }) {
-    this.log("\u{1F916} AGENT STARTING");
-    this.log("   Agent Name: ".concat(callbackContext.agentName));
-    this.log("   Invocation ID: ".concat(callbackContext.invocationId));
-    if (callbackContext.invocationContext.branch) {
-      this.log("   Branch: ".concat(callbackContext.invocationContext.branch));
-    }
-    return void 0;
-  }
-  async afterAgentCallback({
-    callbackContext
-  }) {
-    this.log("\u{1F916} AGENT COMPLETED");
-    this.log("   Agent Name: ".concat(callbackContext.agentName));
-    this.log("   Invocation ID: ".concat(callbackContext.invocationId));
-    return void 0;
-  }
-  async beforeModelCallback({
-    callbackContext,
-    llmRequest
-  }) {
-    var _a;
-    this.log("\u{1F9E0} LLM REQUEST");
-    this.log("   Model: ".concat((_a = llmRequest.model) != null ? _a : "default"));
-    this.log("   Agent: ".concat(callbackContext.agentName));
-    if (llmRequest.config && llmRequest.config.systemInstruction) {
-      let sysInstruction = llmRequest.config.systemInstruction;
-      if (sysInstruction.length > 200) {
-        sysInstruction = sysInstruction.substring(0, 200) + "...";
-      }
-      this.log("   System Instruction: '".concat(sysInstruction, "'"));
-    }
-    if (llmRequest.toolsDict) {
-      const toolNames = Object.keys(llmRequest.toolsDict);
-      this.log("   Available Tools: ".concat(toolNames));
-    }
-    return void 0;
-  }
-  async afterModelCallback({
-    callbackContext,
-    llmResponse
-  }) {
-    this.log("\u{1F9E0} LLM RESPONSE");
-    this.log("   Agent: ".concat(callbackContext.agentName));
-    if (llmResponse.errorCode) {
-      this.log("   \u274C ERROR - Code: ".concat(llmResponse.errorCode));
-      this.log("   Error Message: ".concat(llmResponse.errorMessage));
-    } else {
-      this.log("   Content: ".concat(this.formatContent(llmResponse.content)));
-      if (llmResponse.partial) {
-        this.log("   Partial: ".concat(llmResponse.partial));
-      }
-      if (llmResponse.turnComplete !== void 0) {
-        this.log("   Turn Complete: ".concat(llmResponse.turnComplete));
-      }
-    }
-    if (llmResponse.usageMetadata) {
-      this.log(
-        "   Token Usage - Input: ".concat(llmResponse.usageMetadata.promptTokenCount, ", Output: ").concat(llmResponse.usageMetadata.candidatesTokenCount)
-      );
-    }
-    return void 0;
-  }
-  async beforeToolCallback({
-    tool,
-    toolArgs,
-    toolContext
-  }) {
-    this.log("\u{1F527} TOOL STARTING");
-    this.log("   Tool Name: ".concat(tool.name));
-    this.log("   Agent: ".concat(toolContext.agentName));
-    this.log("   Function Call ID: ".concat(toolContext.functionCallId));
-    this.log("   Arguments: ".concat(this.formatArgs(toolArgs)));
-    return void 0;
-  }
-  async afterToolCallback({
-    tool,
-    toolContext,
-    result
-  }) {
-    this.log("\u{1F527} TOOL COMPLETED");
-    this.log("   Tool Name: ".concat(tool.name));
-    this.log("   Agent: ".concat(toolContext.agentName));
-    this.log("   Function Call ID: ".concat(toolContext.functionCallId));
-    this.log("   Result: ".concat(this.formatArgs(result)));
-    return void 0;
-  }
-  async onModelErrorCallback({
-    callbackContext,
-    error
-  }) {
-    this.log("\u{1F9E0} LLM ERROR");
-    this.log("   Agent: ".concat(callbackContext.agentName));
-    this.log("   Error: ".concat(error));
-    return void 0;
-  }
-  async onToolErrorCallback({
-    tool,
-    toolArgs,
-    toolContext,
-    error
-  }) {
-    this.log("\u{1F527} TOOL ERROR");
-    this.log("   Tool Name: ".concat(tool.name));
-    this.log("   Agent: ".concat(toolContext.agentName));
-    this.log("   Function Call ID: ".concat(toolContext.functionCallId));
-    this.log("   Arguments: ".concat(this.formatArgs(toolArgs)));
-    this.log("   Error: ".concat(error));
-    return void 0;
-  }
-  log(message) {
-    const formattedMessage = "\x1B[90m[".concat(this.name, "] ").concat(message, "\x1B[0m");
-    logger.info(formattedMessage);
-  }
-  formatContent(content, maxLength = 200) {
-    if (!content || !content.parts) {
-      return "None";
-    }
-    const parts = [];
-    for (const part of content.parts) {
-      if (part.text) {
-        let text = part.text.trim();
-        if (text.length > maxLength) {
-          text = text.substring(0, maxLength) + "...";
-        }
-        parts.push("text: '".concat(text, "'"));
-      } else if (part.functionCall) {
-        parts.push("function_call: ".concat(part.functionCall.name));
-      } else if (part.functionResponse) {
-        parts.push("function_response: ".concat(part.functionResponse.name));
-      } else if (part.codeExecutionResult) {
-        parts.push("code_execution_result");
-      } else {
-        parts.push("other_part");
-      }
-    }
-    return parts.join(" | ");
-  }
-  formatArgs(args, maxLength = 300) {
-    if (!args) {
-      return "{}";
-    }
-    let formatted = JSON.stringify(args);
-    if (formatted.length > maxLength) {
-      formatted = formatted.substring(0, maxLength) + "...}";
-    }
-    return formatted;
-  }
-};
-__name(_LoggingPlugin, "LoggingPlugin");
-let LoggingPlugin = _LoggingPlugin;
-export {
-  LoggingPlugin
-};

package/dist/web/plugins/plugin_manager.js

@@ -1,318 +0,0 @@
-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";
-const _PluginManager = class _PluginManager {
-  /**
-   * Initializes the plugin service.
-   *
-   * @param plugins An optional list of plugins to register upon
-   *     initialization.
-   */
-  constructor(plugins) {
-    this.plugins = /* @__PURE__ */ new Set();
-    if (plugins) {
-      for (const plugin of plugins) {
-        this.registerPlugin(plugin);
-      }
-    }
-  }
-  /** Whether any plugin is registered. */
-  get hasPlugins() {
-    return this.plugins.size > 0;
-  }
-  /**
-   * Registers a new plugin.
-   *
-   * @param plugin The plugin instance to register.
-   * @throws If the same exact plugin or a plugin with the same name is already
-   *     registered.
-   */
-  registerPlugin(plugin) {
-    if (this.plugins.has(plugin)) {
-      throw new Error("Plugin '".concat(plugin.name, "' already registered."));
-    }
-    if (Array.from(this.plugins).some((p) => p.name === plugin.name)) {
-      throw new Error("Plugin with name '".concat(plugin.name, "' already registered."));
-    }
-    this.plugins.add(plugin);
-    logger.info("Plugin '".concat(plugin.name, "' registered."));
-  }
-  /**
-   * Retrieves a registered plugin by its name.
-   *
-   * @param pluginName The name of the plugin to retrieve.
-   * @returns The plugin instance if found, otherwise `undefined`.
-   */
-  getPlugin(pluginName) {
-    return Array.from(this.plugins).find((p) => p.name === pluginName);
-  }
-  /**
-   * Runs the same callback for all plugins. This is a utility method to reduce
-   * duplication below.
-   *
-   * @param plugins The set of plugins to run
-   * @param callback A closure containing the callback method to run on each
-   *     plugin
-   * @param callbackName The name of the function being called in the closure
-   *     above. Used for logging purposes.
-   * @returns A promise containing the plugin method result. Must be casted to
-   *     the proper type for the plugin method.
-   */
-  async runCallbacks(plugins, callback, callbackName) {
-    for (const plugin of plugins) {
-      try {
-        const result = await callback(plugin);
-        if (result !== void 0) {
-          logger.debug(
-            "Plugin '".concat(plugin.name, "' returned a value for callback '").concat(callbackName, "', exiting early.")
-          );
-          return result;
-        }
-      } catch (e) {
-        const errorMessage = "Error in plugin '".concat(plugin.name, "' during '").concat(callbackName, "' callback: ").concat(e);
-        logger.error(errorMessage);
-        throw new Error(errorMessage);
-      }
-    }
-    return void 0;
-  }
-  /**
-   * Runs the `onUserMessageCallback` for all plugins.
-   */
-  async runOnUserMessageCallback({
-    userMessage,
-    invocationContext
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.onUserMessageCallback({ userMessage, invocationContext }),
-      "onUserMessageCallback"
-    );
-  }
-  /**
-   * Runs the `beforeRunCallback` for all plugins.
-   */
-  async runBeforeRunCallback({
-    invocationContext
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.beforeRunCallback({ invocationContext }),
-      "beforeRunCallback"
-    );
-  }
-  /**
-   * Runs the `afterRunCallback` for all plugins.
-   */
-  async runAfterRunCallback({
-    invocationContext
-  }) {
-    await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.afterRunCallback({ invocationContext }),
-      "afterRunCallback"
-    );
-  }
-  /**
-   * Runs the `onEventCallback` for all plugins.
-   */
-  async runOnEventCallback({
-    invocationContext,
-    event
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.onEventCallback({ invocationContext, event }),
-      "onEventCallback"
-    );
-  }
-  /**
-   * Runs the `beforeAgentCallback` for all plugins.
-   */
-  async runBeforeAgentCallback({
-    agent,
-    callbackContext
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.beforeAgentCallback({ agent, callbackContext }),
-      "beforeAgentCallback"
-    );
-  }
-  /**
-   * Runs the `afterAgentCallback` for all plugins.
-   */
-  async runAfterAgentCallback({
-    agent,
-    callbackContext
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.afterAgentCallback({ agent, callbackContext }),
-      "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.
-   */
-  async runBeforeToolSelection({
-    callbackContext,
-    tools
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.beforeToolSelection({ callbackContext, tools }),
-      "beforeToolSelection"
-    );
-  }
-  /**
-   * Runs the `beforeContextCompaction` for all plugins.
-   */
-  async runBeforeContextCompaction({
-    invocationContext,
-    trigger
-  }) {
-    await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.beforeContextCompaction({ invocationContext, trigger }),
-      "beforeContextCompaction"
-    );
-  }
-  /**
-   * Runs the `afterContextCompaction` for all plugins.
-   */
-  async runAfterContextCompaction({
-    invocationContext,
-    trigger
-  }) {
-    await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.afterContextCompaction({ invocationContext, trigger }),
-      "afterContextCompaction"
-    );
-  }
-  /**
-   * Runs the `beforeToolCallback` for all plugins.
-   */
-  async runBeforeToolCallback({
-    tool,
-    toolArgs,
-    toolContext
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.beforeToolCallback({ tool, toolArgs, toolContext }),
-      "beforeToolCallback"
-    );
-  }
-  /**
-   * Runs the `afterToolCallback` for all plugins.
-   */
-  async runAfterToolCallback({
-    tool,
-    toolArgs,
-    toolContext,
-    result
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.afterToolCallback({ tool, toolArgs, toolContext, result }),
-      "afterToolCallback"
-    );
-  }
-  /**
-   * Runs the `onModelErrorCallback` for all plugins.
-   */
-  async runOnModelErrorCallback({
-    callbackContext,
-    llmRequest,
-    error
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.onModelErrorCallback({ callbackContext, llmRequest, error }),
-      "onModelErrorCallback"
-    );
-  }
-  /**
-   * Runs the `beforeModelCallback` for all plugins.
-   */
-  async runBeforeModelCallback({
-    callbackContext,
-    llmRequest
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.beforeModelCallback({ callbackContext, llmRequest }),
-      "beforeModelCallback"
-    );
-  }
-  /**
-   * Runs the `afterModelCallback` for all plugins.
-   */
-  async runAfterModelCallback({
-    callbackContext,
-    llmResponse
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.afterModelCallback({ callbackContext, llmResponse }),
-      "afterModelCallback"
-    );
-  }
-  /**
-   * Runs the `onToolErrorCallback` for all plugins.
-   */
-  async runOnToolErrorCallback({
-    tool,
-    toolArgs,
-    toolContext,
-    error
-  }) {
-    return await this.runCallbacks(
-      this.plugins,
-      (plugin) => plugin.onToolErrorCallback({ tool, toolArgs, toolContext, error }),
-      "onToolErrorCallback"
-    );
-  }
-};
-__name(_PluginManager, "PluginManager");
-let PluginManager = _PluginManager;
-export {
-  PluginManager
-};

package/dist/web/plugins/security_plugin.js

@@ -1,148 +0,0 @@
-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 { REQUEST_CONFIRMATION_FUNCTION_CALL_NAME } from "../agents/functions.js";
-import { BasePlugin } from "../plugins/base_plugin.js";
-const TOOL_CALL_SECURITY_CHECK_STATES = "orcas_tool_call_security_check_states";
-const INTERMEDIATE_REQUIRE_TOOL_CALL_CONFIRMATION_ERROR = "This tool call needs external confirmation before completion.";
-var PolicyOutcome = /* @__PURE__ */ ((PolicyOutcome2) => {
-  PolicyOutcome2["DENY"] = "DENY";
-  PolicyOutcome2["CONFIRM"] = "CONFIRM";
-  PolicyOutcome2["ALLOW"] = "ALLOW";
-  return PolicyOutcome2;
-})(PolicyOutcome || {});
-const _InMemoryPolicyEngine = class _InMemoryPolicyEngine {
-  /**
-   * Always returns {@link PolicyOutcome.ALLOW} for every tool call.
-   *
-   * @param _context - The tool and its arguments; ignored.
-   * @returns A promise resolving to an ALLOW result.
-   */
-  async evaluate(_context) {
-    return Promise.resolve({
-      outcome: "ALLOW" /* ALLOW */,
-      reason: "For prototyping purpose, all tool calls are allowed."
-    });
-  }
-};
-__name(_InMemoryPolicyEngine, "InMemoryPolicyEngine");
-let InMemoryPolicyEngine = _InMemoryPolicyEngine;
-const _SecurityPlugin = class _SecurityPlugin extends BasePlugin {
-  /**
-   * @param params - Optional configuration. Defaults to {@link InMemoryPolicyEngine}
-   *   when no policy engine is provided.
-   */
-  constructor(params) {
-    var _a;
-    super("security_plugin");
-    this.policyEngine = (_a = params == null ? void 0 : params.policyEngine) != null ? _a : new InMemoryPolicyEngine();
-  }
-  /**
-   * Intercepts tool calls, evaluating them against the policy engine before
-   * execution and handling confirmation flows for calls that require it.
-   *
-   * @returns A partial or error response object if the call is blocked or
-   *   awaiting confirmation, or `undefined` to allow execution to proceed.
-   */
-  async beforeToolCallback({
-    tool,
-    toolArgs,
-    toolContext
-  }) {
-    const toolCallCheckState = this.getToolCallCheckState(toolContext);
-    if (!toolCallCheckState) {
-      return this.checkToolCallPolicy({
-        tool,
-        toolArgs,
-        toolContext
-      });
-    }
-    if (toolCallCheckState !== "CONFIRM" /* CONFIRM */) {
-      return;
-    }
-    if (!toolContext.toolConfirmation) {
-      return { partial: INTERMEDIATE_REQUIRE_TOOL_CALL_CONFIRMATION_ERROR };
-    }
-    this.setToolCallCheckState(toolContext, toolContext.toolConfirmation);
-    if (!toolContext.toolConfirmation.confirmed) {
-      return {
-        error: "Tool call rejected from confirmation flow."
-      };
-    }
-    toolContext.toolConfirmation = void 0;
-    return;
-  }
-  getToolCallCheckState(toolContext) {
-    var _a;
-    const { functionCallId } = toolContext;
-    if (!functionCallId) {
-      return;
-    }
-    const toolCallStates = (_a = toolContext.state.get(TOOL_CALL_SECURITY_CHECK_STATES)) != null ? _a : {};
-    return toolCallStates[functionCallId];
-  }
-  setToolCallCheckState(toolContext, state) {
-    var _a;
-    const { functionCallId } = toolContext;
-    if (!functionCallId) {
-      return;
-    }
-    const toolCallStates = (_a = toolContext.state.get(TOOL_CALL_SECURITY_CHECK_STATES)) != null ? _a : {};
-    toolCallStates[functionCallId] = state;
-    toolContext.state.set(TOOL_CALL_SECURITY_CHECK_STATES, toolCallStates);
-  }
-  async checkToolCallPolicy({
-    tool,
-    toolArgs,
-    toolContext
-  }) {
-    const policyCheckResult = await this.policyEngine.evaluate({
-      tool,
-      toolArgs
-    });
-    this.setToolCallCheckState(toolContext, policyCheckResult.outcome);
-    switch (policyCheckResult.outcome) {
-      case "DENY" /* DENY */:
-        return {
-          error: "This tool call is rejected by policy engine. Reason: ".concat(policyCheckResult.reason)
-        };
-      case "CONFIRM" /* CONFIRM */:
-        toolContext.requestConfirmation({
-          hint: "Policy engine requires confirmation calling tool: ".concat(tool.name, ". Reason: ").concat(policyCheckResult.reason)
-        });
-        return { partial: INTERMEDIATE_REQUIRE_TOOL_CALL_CONFIRMATION_ERROR };
-      case "ALLOW" /* ALLOW */:
-        return;
-      default:
-        return;
-    }
-  }
-};
-__name(_SecurityPlugin, "SecurityPlugin");
-let SecurityPlugin = _SecurityPlugin;
-function getAskUserConfirmationFunctionCalls(event) {
-  if (!event.content || !event.content.parts) {
-    return [];
-  }
-  const results = [];
-  for (const part of event.content.parts) {
-    if (part && part.functionCall && part.functionCall.name === REQUEST_CONFIRMATION_FUNCTION_CALL_NAME) {
-      results.push(part.functionCall);
-    }
-  }
-  return results;
-}
-__name(getAskUserConfirmationFunctionCalls, "getAskUserConfirmationFunctionCalls");
-export {
-  InMemoryPolicyEngine,
-  PolicyOutcome,
-  REQUEST_CONFIRMATION_FUNCTION_CALL_NAME,
-  SecurityPlugin,
-  getAskUserConfirmationFunctionCalls
-};