@google/adk 0.6.1 → 1.1.0

package/dist/types/plugins/plugin_manager.d.ts

@@ -11,7 +11,7 @@ import { Event } from '../events/event.js';
 import { LlmRequest } from '../models/llm_request.js';
 import { LlmResponse } from '../models/llm_response.js';
 import { BaseTool } from '../tools/base_tool.js';
-import { BasePlugin } from './base_plugin.js';
+import { BasePlugin, ContextCompactionTrigger } from './base_plugin.js';
 /**
  * Manages the registration and execution of plugins.
  *
@@ -103,6 +103,27 @@ export declare class PluginManager {
         agent: BaseAgent;
         callbackContext: Context;
     }): Promise<Content | undefined>;
+    /**
+     * Runs the `beforeToolSelection` for all plugins.
+     */
+    runBeforeToolSelection({ callbackContext, tools, }: {
+        callbackContext: Context;
+        tools: Readonly<Record<string, BaseTool>>;
+    }): Promise<Readonly<Record<string, BaseTool>> | undefined>;
+    /**
+     * Runs the `beforeContextCompaction` for all plugins.
+     */
+    runBeforeContextCompaction({ invocationContext, trigger, }: {
+        invocationContext: InvocationContext;
+        trigger: ContextCompactionTrigger;
+    }): Promise<void>;
+    /**
+     * Runs the `afterContextCompaction` for all plugins.
+     */
+    runAfterContextCompaction({ invocationContext, trigger, }: {
+        invocationContext: InvocationContext;
+        trigger: ContextCompactionTrigger;
+    }): Promise<void>;
     /**
      * Runs the `beforeToolCallback` for all plugins.
      */

package/dist/types/runner/runner.d.ts

@@ -84,6 +84,7 @@ export declare class Runner {
         newMessage: Content;
         stateDelta?: Record<string, unknown>;
         runConfig?: RunConfig;
+        abortSignal?: AbortSignal;
     }): AsyncGenerator<Event, void, undefined>;
     /**
      * Saves artifacts from the message parts and replaces the inline data with

package/dist/types/skills/loader.d.ts

@@ -0,0 +1,38 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Frontmatter, Skill } from './skill.js';
+/**
+ * Parses SKILL.md from a raw content string, extracting the YAML frontmatter and the body.
+ *
+ * @param content - The raw content of the SKILL.md file.
+ * @returns An object containing the parsed frontmatter and the remaining markdown body.
+ * @throws {Error} If the content is not properly formatted with YAML frontmatter.
+ */
+export declare function parseSkillMdContent(content: string): {
+    frontmatter: Frontmatter;
+    body: string;
+};
+/**
+ * Load a complete skill, including its instructions and resources, from a directory.
+ *
+ * @param skillDir - The path to the directory containing the skill definition.
+ * @returns A promise that resolves to the fully loaded Skill object.
+ */
+export declare function loadSkillFromDir(skillDir: string): Promise<Skill>;
+/**
+ * Validates a skill directory structure and frontmatter without fully loading all resources.
+ *
+ * @param skillDir - The path to the skill directory to validate.
+ * @returns A promise that resolves to an array of validation error messages, or an empty array if valid.
+ */
+export declare function validateSkillDir(skillDir: string): Promise<string[]>;
+/**
+ * Loads all skills located within subdirectories of a given base path.
+ *
+ * @param skillsBasePath - The base directory containing individual skill subdirectories.
+ * @returns A promise that resolves to a map of skill names to their corresponding Skill objects.
+ */
+export declare function loadAllSkillsInDir(skillsBasePath: string): Promise<Record<string, Skill>>;

package/dist/types/skills/prompt.d.ts

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Frontmatter, Skill } from './skill.js';
+/**
+ * Formats available skills into a standard XML string.
+ *
+ * @param skills A list of skill frontmatter or full skill objects.
+ * @returns XML string with <available_skills> block.
+ */
+export declare function formatSkillsAsXml(skills: Array<Skill | Frontmatter>): string;

package/dist/types/skills/skill.d.ts

@@ -0,0 +1,48 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { z } from 'zod';
+export declare const SNAKE_OR_KEBAB_NAME_PATTERN: RegExp;
+/**
+ * Schema and Type for Skill Frontmatter metadata.
+ */
+export declare const FrontmatterSchema: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodObject<{
+    name: z.ZodString;
+    description: z.ZodString;
+    license: z.ZodOptional<z.ZodString>;
+    compatibility: z.ZodOptional<z.ZodString>;
+    'allowed-tools': z.ZodOptional<z.ZodString>;
+    metadata: z.ZodDefault<z.ZodRecord<z.ZodString, z.ZodAny>>;
+}, z.core.$loose>>;
+export interface Frontmatter {
+    name: string;
+    description: string;
+    license?: string;
+    compatibility?: string;
+    allowedTools?: string;
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Wrapper for script content.
+ */
+export interface Script {
+    src: string;
+}
+/**
+ * L3 skill content: additional instructions, assets, and scripts.
+ */
+export interface Resources {
+    references?: Record<string, string | Buffer>;
+    assets?: Record<string, string | Buffer>;
+    scripts?: Record<string, Script>;
+}
+/**
+ * Complete skill representation including frontmatter, instructions, and resources.
+ */
+export interface Skill {
+    frontmatter: Frontmatter;
+    instructions: string;
+    resources?: Resources;
+}

package/dist/types/tools/mcp/mcp_tool.d.ts

@@ -20,11 +20,17 @@ import { MCPSessionManager } from './mcp_session_manager.js';
  * invoked, which in turn establishes an MCP session, sends a `callTool`
  * request with the provided arguments, and returns the result from the
  * remote tool.
+ *
+ * The originalName parameter allows the tool to track the native tool name
+ * exposed by the MCP server. This is critical when the toolset applies a
+ * prefix to tool names (e.g., for LLM namespace disambiguation), ensuring
+ * the correct original name is used when executing on the server.
  */
 export declare class MCPTool extends BaseTool {
     private readonly mcpTool;
     private readonly mcpSessionManager;
-    constructor(mcpTool: Tool, mcpSessionManager: MCPSessionManager);
+    private readonly originalName;
+    constructor(mcpTool: Tool, mcpSessionManager: MCPSessionManager, originalName?: string);
     _getDeclaration(): FunctionDeclaration;
     runAsync(request: RunAsyncToolRequest): Promise<unknown>;
 }

package/dist/types/tools/mcp/mcp_toolset.d.ts

@@ -17,6 +17,11 @@ import { MCPConnectionParams } from './mcp_session_manager.js';
  * The toolset can be configured with a filter to selectively expose a subset
  * of the tools provided by the MCP server.
  *
+ * It can also be configured with a prefix. If provided, all tools discovered
+ * from the MCP server will have their names prefixed with `${prefix}_`. When the
+ * LLM invokes the prefixed tool, this toolset transparently strips the prefix
+ * before sending the request to the underlying MCP server.
+ *
  * Usage:
  *   import { MCPToolset } from '@google/adk';
  *   import { StreamableHTTPConnectionParamsSchema } from '@google/adk';

package/dist/types/tools/skill/list_skills_tool.d.ts

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { FunctionDeclaration } from '@google/genai';
+import { BaseTool, RunAsyncToolRequest } from '../base_tool.js';
+import { SkillToolset } from './skill_toolset.js';
+export declare class ListSkillsTool extends BaseTool {
+    private toolset;
+    constructor(toolset: SkillToolset);
+    _getDeclaration(): FunctionDeclaration;
+    runAsync(_request: RunAsyncToolRequest): Promise<unknown>;
+}

package/dist/types/tools/skill/load_skill_resource_tool.d.ts

@@ -0,0 +1,15 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { FunctionDeclaration } from '@google/genai';
+import { BaseTool, RunAsyncToolRequest, ToolProcessLlmRequest } from '../base_tool.js';
+import { SkillToolset } from './skill_toolset.js';
+export declare class LoadSkillResourceTool extends BaseTool {
+    private toolset;
+    constructor(toolset: SkillToolset);
+    _getDeclaration(): FunctionDeclaration;
+    runAsync({ args }: RunAsyncToolRequest): Promise<unknown>;
+    processLlmRequest(request: ToolProcessLlmRequest): Promise<void>;
+}

package/dist/types/tools/skill/load_skill_tool.d.ts

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { FunctionDeclaration } from '@google/genai';
+import { BaseTool, RunAsyncToolRequest } from '../base_tool.js';
+import { SkillToolset } from './skill_toolset.js';
+export declare class LoadSkillTool extends BaseTool {
+    private toolset;
+    constructor(toolset: SkillToolset);
+    _getDeclaration(): FunctionDeclaration;
+    runAsync({ args, toolContext, }: RunAsyncToolRequest): Promise<unknown>;
+}

package/dist/types/tools/skill/run_skill_inline_script_tool.d.ts

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { FunctionDeclaration } from '@google/genai';
+import { BaseTool, RunAsyncToolRequest } from '../base_tool.js';
+import { SkillToolset } from './skill_toolset.js';
+export declare class RunSkillInlineScriptTool extends BaseTool {
+    private toolset;
+    constructor(toolset: SkillToolset);
+    _getDeclaration(): FunctionDeclaration;
+    runAsync({ args, toolContext, }: RunAsyncToolRequest): Promise<unknown>;
+}

package/dist/types/tools/skill/run_skill_script_tool.d.ts

@@ -0,0 +1,17 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { FunctionDeclaration } from '@google/genai';
+import { File } from '../../code_executors/code_execution_utils.js';
+import { Skill } from '../../skills/skill.js';
+import { BaseTool, RunAsyncToolRequest } from '../base_tool.js';
+import { SkillToolset } from './skill_toolset.js';
+export declare class RunSkillScriptTool extends BaseTool {
+    private toolset;
+    constructor(toolset: SkillToolset);
+    _getDeclaration(): FunctionDeclaration;
+    runAsync({ args, toolContext, }: RunAsyncToolRequest): Promise<unknown>;
+}
+export declare function getSkillResourceFiles(skill: Skill): File[];

package/dist/types/tools/skill/skill_toolset.d.ts

@@ -0,0 +1,28 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Context } from '../../agents/context.js';
+import { ReadonlyContext } from '../../agents/readonly_context.js';
+import { BaseCodeExecutor } from '../../code_executors/base_code_executor.js';
+import { LlmRequest } from '../../models/llm_request.js';
+import { Skill } from '../../skills/skill.js';
+import { BaseTool } from '../base_tool.js';
+import { BaseToolset } from '../base_toolset.js';
+export declare class SkillToolset extends BaseToolset {
+    skills: Record<string, Skill>;
+    private tools;
+    additionalTools: Array<BaseTool | BaseToolset>;
+    codeExecutor?: BaseCodeExecutor;
+    private toolCache;
+    constructor(skills: Record<string, Skill> | Skill[], options?: {
+        codeExecutor?: BaseCodeExecutor;
+        additionalTools?: Array<BaseTool | BaseToolset>;
+    });
+    getTools(context?: ReadonlyContext): Promise<BaseTool[]>;
+    close(): Promise<void>;
+    getSkill(name: string): Skill | undefined;
+    processLlmRequest(toolContext: Context, llmRequest: LlmRequest): Promise<void>;
+    private resolveAdditionalTools;
+}

package/dist/types/tools/url_context_tool.d.ts

@@ -0,0 +1,17 @@
+import { BaseTool, ToolProcessLlmRequest } from './base_tool.js';
+/**
+ * A built-in tool that allows Gemini 2+ models to retrieve content from URLs
+ * provided in the conversation.
+ *
+ * This tool operates internally within the model and does not require or
+ * perform local code execution.
+ */
+export declare class UrlContextTool extends BaseTool {
+    constructor();
+    runAsync(): Promise<unknown>;
+    processLlmRequest({ llmRequest, }: ToolProcessLlmRequest): Promise<void>;
+}
+/**
+ * A global instance of {@link UrlContextTool}.
+ */
+export declare const URL_CONTEXT: UrlContextTool;

package/dist/types/tools/vertex_ai_search_tool.d.ts

@@ -0,0 +1,55 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ReadonlyContext } from '../agents/readonly_context.js';
+import { BaseTool, ToolProcessLlmRequest } from './base_tool.js';
+export interface VertexAISearchDataStoreSpec {
+    dataStore?: string;
+}
+export interface VertexAISearchConfig {
+    datastore?: string;
+    dataStoreSpecs?: VertexAISearchDataStoreSpec[];
+    engine?: string;
+    filter?: string;
+    maxResults?: number;
+}
+export interface BaseVertexAiSearchToolParams {
+    filter?: string;
+    maxResults?: number;
+    bypassMultiToolsLimit?: boolean;
+}
+export interface DataStoreParams extends BaseVertexAiSearchToolParams {
+    dataStoreId: string;
+    searchEngineId?: never;
+    dataStoreSpecs?: never;
+}
+export interface SearchEngineParams extends BaseVertexAiSearchToolParams {
+    searchEngineId: string;
+    dataStoreId?: never;
+    dataStoreSpecs?: VertexAISearchDataStoreSpec[];
+}
+export type VertexAiSearchToolParams = DataStoreParams | SearchEngineParams;
+/**
+ * A built-in tool using Vertex AI Search.
+ */
+export declare class VertexAiSearchTool extends BaseTool {
+    readonly dataStoreId?: string;
+    readonly dataStoreSpecs?: VertexAISearchDataStoreSpec[];
+    readonly searchEngineId?: string;
+    readonly filter?: string;
+    readonly maxResults?: number;
+    readonly bypassMultiToolsLimit: boolean;
+    constructor(params: VertexAiSearchToolParams);
+    runAsync(): Promise<unknown>;
+    /**
+     * Builds the VertexAISearch configuration.
+     *
+     * Override this method in a subclass to dynamically customize the search
+     * configuration based on the context (e.g., set filter based on session
+     * state).
+     */
+    protected buildVertexAiSearchConfig(_readonlyContext: ReadonlyContext): VertexAISearchConfig;
+    processLlmRequest({ toolContext, llmRequest, }: ToolProcessLlmRequest): Promise<void>;
+}

package/dist/types/utils/experimental.d.ts

@@ -0,0 +1,11 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Decorator to mark classes and methods as experimental.
+ * Logs a warning once per item when the class is instantiated or the method is called.
+ */
+export type Constructor = new (...args: unknown[]) => any;
+export declare function experimental<T extends Constructor | object, P extends string | symbol | undefined = undefined>(target: T, propertyKey?: P, descriptor?: PropertyDescriptor): P extends undefined ? T : PropertyDescriptor;

package/dist/types/utils/failover_utils.d.ts

@@ -0,0 +1,17 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Type definition for a function that selects an item based on the context.
+ */
+export type Router<T, C> = (items: Readonly<Record<string, T>>, context: C, errorContext?: {
+    failedKeys: ReadonlySet<string>;
+    lastError: unknown;
+}) => Promise<string | undefined> | string | undefined;
+/**
+ * Runs a core operation with selection and failover support.
+ * Internal helper to unify Promise and Generator logic.
+ */
+export declare function runWithRouting<T, C, TYield>(items: Readonly<Record<string, T>>, context: C, router: Router<T, C>, runFn: (item: T) => AsyncGenerator<TYield, void, void> | Promise<TYield>): AsyncGenerator<TYield, void, void>;

package/dist/types/utils/file_extension_utils.d.ts

@@ -0,0 +1,21 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { CodeExecutionLanguage, FileContentEncoding } from '../code_executors/code_execution_utils.js';
+/**
+ * Gets the MIME type and file content encoding for a given file extension.
+ * @param ext The file extension (e.g., '.js', '.png').
+ * @returns An object containing the mimeType and encoding.
+ */
+export declare function getMimeTypeAndEncoding(ext: string): {
+    mimeType: string;
+    encoding: FileContentEncoding;
+};
+/**
+ * Gets the code execution language for a given file extension.
+ * @param ext The file extension (e.g., '.js', '.py').
+ * @returns The code execution language.
+ */
+export declare function getScriptLanguageByExtension(ext: string): CodeExecutionLanguage;

package/dist/types/utils/file_utils.d.ts

@@ -0,0 +1,11 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { File } from '../code_executors/code_execution_utils.js';
+/**
+ * Creates files with the given paths in the current working directory.
+ * @param files The files to materialize.
+ */
+export declare function materializeFiles(files: File[], dir?: string): Promise<File[]>;

package/dist/types/utils/model_name.d.ts

@@ -32,3 +32,7 @@ export declare function isGemini1Model(modelString: string): boolean;
  * @return true if it's a Gemini 2.x model, false otherwise.
  */
 export declare function isGemini2OrAbove(modelString: string): boolean;
+/**
+ * Returns True when Gemini model-id validation should be bypassed.
+ */
+export declare function isGeminiModelIdCheckDisabled(): boolean;

package/dist/types/utils/streaming_utils.d.ts

@@ -0,0 +1,40 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { GenerateContentResponse } from '@google/genai';
+import { LlmResponse } from '../models/llm_response.js';
+/**
+ * Aggregates partial streaming responses.
+ *
+ * It aggregates content from partial responses, and generates LlmResponses for
+ * individual (partial) model responses, as well as for aggregated content.
+ */
+export declare class StreamingResponseAggregator {
+    private readonly isProgressiveMode;
+    private usageMetadata?;
+    private groundingMetadata?;
+    private citationMetadata?;
+    private response?;
+    private text;
+    private thoughtText;
+    private partsSequence;
+    private currentTextBuffer;
+    private currentTextIsThought?;
+    private finishReason?;
+    private currentFcName?;
+    private currentFcArgs;
+    private currentFcId?;
+    private currentThoughtSignature?;
+    private lastThoughtSignature?;
+    constructor(isProgressiveMode?: boolean);
+    private flushTextBufferToSequence;
+    private getValueFromPartialArg;
+    private setValueByJsonPath;
+    private flushFunctionCallToSequence;
+    private processStreamingFunctionCall;
+    private processFunctionCallPart;
+    processResponse(response: GenerateContentResponse): AsyncGenerator<LlmResponse, void, void>;
+    close(): LlmResponse | undefined;
+}

package/dist/types/version.d.ts

@@ -3,4 +3,4 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-export declare const version = "0.6.1";
+export declare const version = "1.1.0";

package/dist/web/a2a/a2a_event.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/web/a2a/a2a_remote_agent.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 var __knownSymbol = (name, symbol) => (symbol = Symbol[name]) ? symbol : Symbol.for("Symbol." + name);
 var __await = function(promise, isYieldStar) {
   this[0] = promise;

package/dist/web/a2a/a2a_remote_agent_run_processor.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/web/a2a/a2a_remote_agent_utils.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/web/a2a/agent_card.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/web/a2a/agent_executor.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 var __knownSymbol = (name, symbol) => (symbol = Symbol[name]) ? symbol : Symbol.for("Symbol." + name);
 var __forAwait = (obj, it, method) => (it = obj[__knownSymbol("asyncIterator")]) ? it.call(obj) : (obj = obj[__knownSymbol("iterator")](), it = {}, method = (key, fn) => (fn = obj[key]) && (it[key] = (arg) => new Promise((yes, no, done) => (arg = fn.call(obj, arg), done = arg.done, Promise.resolve(arg.value).then((value) => yes({ value, done }), no)))), method("next"), method("return"), it);
 /**

package/dist/web/a2a/agent_to_a2a.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/web/a2a/event_converter_utils.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 var __defProp = Object.defineProperty;
 var __defProps = Object.defineProperties;
 var __getOwnPropDescs = Object.getOwnPropertyDescriptors;

package/dist/web/a2a/event_processor_utils.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 var __defProp = Object.defineProperty;
 var __defProps = Object.defineProperties;
 var __getOwnPropDescs = Object.getOwnPropertyDescriptors;

package/dist/web/a2a/executor_context.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/web/a2a/metadata_converter_utils.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 var __defProp = Object.defineProperty;
 var __defProps = Object.defineProperties;
 var __getOwnPropDescs = Object.getOwnPropertyDescriptors;

package/dist/web/a2a/part_converter_utils.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2026 Google LLC

package/dist/web/agents/active_streaming_tool.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 /**
  * @license
  * Copyright 2025 Google LLC

package/dist/web/agents/base_agent.js

@@ -1,3 +1,5 @@
+import {createRequire as topLevelCreateRequire} from 'module';
+const require = topLevelCreateRequire(import.meta.url);
 var __defProp = Object.defineProperty;
 var __defProps = Object.defineProperties;
 var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
@@ -125,12 +127,13 @@ class BaseAgent {
           this,
           function() {
             return __asyncGenerator(this, null, function* () {
+              var _a2, _b;
               const context2 = this.createInvocationContext(parentContext);
               const beforeAgentCallbackEvent = yield new __await(this.handleBeforeAgentCallback(context2));
               if (beforeAgentCallbackEvent) {
                 yield beforeAgentCallbackEvent;
               }
-              if (context2.endInvocation) {
+              if (context2.endInvocation || ((_a2 = parentContext.abortSignal) == null ? void 0 : _a2.aborted)) {
                 return;
               }
               traceAgentInvocation({ agent: this, invocationContext: context2 });
@@ -149,7 +152,7 @@ class BaseAgent {
                     throw error[0];
                 }
               }
-              if (context2.endInvocation) {
+              if (context2.endInvocation || ((_b = parentContext.abortSignal) == null ? void 0 : _b.aborted)) {
                 return;
               }
               const afterAgentCallbackEvent = yield new __await(this.handleAfterAgentCallback(context2));
@@ -236,12 +239,16 @@ class BaseAgent {
    *     generated.
    */
   async handleBeforeAgentCallback(invocationContext) {
+    var _a2;
     if (this.beforeAgentCallback.length === 0) {
       return void 0;
     }
     const callbackContext = new Context({ invocationContext });
     for (const callback of this.beforeAgentCallback) {
       const content = await callback(callbackContext);
+      if ((_a2 = invocationContext.abortSignal) == null ? void 0 : _a2.aborted) {
+        return;
+      }
       if (content) {
         invocationContext.endInvocation = true;
         return createEvent({
@@ -271,12 +278,16 @@ class BaseAgent {
    *     generated.
    */
   async handleAfterAgentCallback(invocationContext) {
+    var _a2;
     if (this.afterAgentCallback.length === 0) {
       return void 0;
     }
     const callbackContext = new Context({ invocationContext });
     for (const callback of this.afterAgentCallback) {
       const content = await callback(callbackContext);
+      if ((_a2 = invocationContext.abortSignal) == null ? void 0 : _a2.aborted) {
+        return;
+      }
       if (content) {
         return createEvent({
           invocationId: invocationContext.invocationId,