@google/adk 0.4.0 → 0.5.0

package/dist/types/{tools/tool_context.d.ts → agents/context.d.ts}

@@ -3,34 +3,69 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { CallbackContext } from '../agents/callback_context.js';
-import { InvocationContext } from '../agents/invocation_context.js';
+import { Part } from '@google/genai';
 import { AuthCredential } from '../auth/auth_credential.js';
 import { AuthConfig } from '../auth/auth_tool.js';
 import { EventActions } from '../events/event_actions.js';
 import { SearchMemoryResponse } from '../memory/base_memory_service.js';
+import { State } from '../sessions/state.js';
 import { ToolConfirmation } from '../tools/tool_confirmation.js';
-export declare class ToolContext extends CallbackContext {
+import { InvocationContext } from './invocation_context.js';
+import { ReadonlyContext } from './readonly_context.js';
+/**
+ * The context of various callbacks within an agent run.
+ *
+ * This class provides the context for callbacks and tool invocations, including
+ * access to the invocation context, function call ID, event actions, and
+ * authentication response. It also provides methods for requesting credentials,
+ * retrieving authentication responses, loading and saving artifacts, and
+ * searching memory.
+ */
+export declare class Context extends ReadonlyContext {
+    private readonly _state;
+    readonly eventActions: EventActions;
     readonly functionCallId?: string;
     toolConfirmation?: ToolConfirmation;
     /**
-     * @param params.invocationContext The invocation context of the tool.
-     * @param params.eventActions The event actions of the current tool call.
-     * @param params.functionCallId The function call id of the current tool call.
+     * @param options The configuration options for the Context.
+     * @param options.invocationContext The invocation context.
+     * @param options.eventActions The event actions of the current call.
+     * @param options.functionCallId The function call id of the current tool call.
      *     This id was returned in the function call event from LLM to identify a
      *     function call. If LLM didn't return this id, ADK will assign one to it.
      *     This id is used to map function call response to the original function
      *     call.
-     * @param params.toolConfirmation The tool confirmation of the current tool
+     * @param options.toolConfirmation The tool confirmation of the current tool
      *     call.
      */
-    constructor(params: {
+    constructor(options: {
         invocationContext: InvocationContext;
         eventActions?: EventActions;
         functionCallId?: string;
         toolConfirmation?: ToolConfirmation;
     });
+    /**
+     * The delta-aware state of the current session.
+     */
+    get state(): State;
     get actions(): EventActions;
+    /**
+     * Loads an artifact attached to the current session.
+     *
+     * @param filename The filename of the artifact.
+     * @param version The version of the artifact. If not provided, the latest
+     *     version will be used.
+     * @return A promise that resolves to the loaded artifact.
+     */
+    loadArtifact(filename: string, version?: number): Promise<Part | undefined>;
+    /**
+     * Saves an artifact attached to the current session.
+     *
+     * @param filename The filename of the artifact.
+     * @param artifact The artifact to save.
+     * @return A promise that resolves to the version of the saved artifact.
+     */
+    saveArtifact(filename: string, artifact: Part): Promise<number>;
     requestCredential(authConfig: AuthConfig): void;
     /**
      * Gets the auth credential for the given auth config.

package/dist/types/agents/llm_agent.d.ts

@@ -15,10 +15,9 @@ import { LlmRequest } from '../models/llm_request.js';
 import { LlmResponse } from '../models/llm_response.js';
 import { BaseTool } from '../tools/base_tool.js';
 import { BaseToolset } from '../tools/base_toolset.js';
-import { ToolContext } from '../tools/tool_context.js';
+import { Context } from './context.js';
 import { BaseAgent, BaseAgentConfig } from './base_agent.js';
-import { BaseLlmRequestProcessor, BaseLlmResponseProcessor } from './base_llm_processor.js';
-import { CallbackContext } from './callback_context.js';
+import { BaseLlmRequestProcessor, BaseLlmResponseProcessor } from './processors/base_llm_processor.js';
 import { InvocationContext } from './invocation_context.js';
 import { ReadonlyContext } from './readonly_context.js';
 /**
@@ -36,7 +35,7 @@ export type InstructionProvider = (context: ReadonlyContext) => string | Promise
  *     will be skipped and the provided content will be returned to user.
  */
 export type SingleBeforeModelCallback = (params: {
-    context: CallbackContext;
+    context: Context;
     request: LlmRequest;
 }) => LlmResponse | undefined | Promise<LlmResponse | undefined>;
 /**
@@ -56,7 +55,7 @@ export type BeforeModelCallback = SingleBeforeModelCallback | SingleBeforeModelC
  *     user.
  */
 export type SingleAfterModelCallback = (params: {
-    context: CallbackContext;
+    context: Context;
     response: LlmResponse;
 }) => LlmResponse | undefined | Promise<LlmResponse | undefined>;
 /**
@@ -71,14 +70,14 @@ export type AfterModelCallback = SingleAfterModelCallback | SingleAfterModelCall
  *
  * @param params.tool The tool to be called.
  * @param params.args The arguments to the tool.
- * @param params.context ToolContext for the tool call.
+ * @param params.context Context for the tool call.
  * @returns The tool response. When present, the returned tool response will
  *     be used and the framework will skip calling the actual tool.
  */
 export type SingleBeforeToolCallback = (params: {
     tool: BaseTool;
     args: Record<string, unknown>;
-    context: ToolContext;
+    context: Context;
 }) => Record<string, unknown> | undefined | Promise<Record<string, unknown> | undefined>;
 /**
  * A single callback or a list of callbacks.
@@ -92,14 +91,14 @@ export type BeforeToolCallback = SingleBeforeToolCallback | SingleBeforeToolCall
  *
  * @param params.tool The tool to be called.
  * @param params.args The arguments to the tool.
- * @param params.context ToolContext for the tool call.
+ * @param params.context Context for the tool call.
  * @param params.response The response from the tool.
  * @returns When present, the returned record will be used as tool result.
  */
 export type SingleAfterToolCallback = (params: {
     tool: BaseTool;
     args: Record<string, unknown>;
-    context: ToolContext;
+    context: Context;
     response: Record<string, unknown>;
 }) => Record<string, unknown> | undefined | Promise<Record<string, unknown> | undefined>;
 /**
@@ -205,28 +204,6 @@ export interface LlmAgentConfig extends BaseAgentConfig {
      */
     codeExecutor?: BaseCodeExecutor;
 }
-declare class RequestConfirmationLlmRequestProcessor extends BaseLlmRequestProcessor {
-    /** Handles tool confirmation information to build the LLM request. */
-    runAsync(invocationContext: InvocationContext): AsyncGenerator<Event, void, void>;
-}
-export declare const REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR: RequestConfirmationLlmRequestProcessor;
-/**
- * Processor for code execution responses.
- */
-declare class CodeExecutionResponseProcessor implements BaseLlmResponseProcessor {
-    /**
-     * Processes the LLM response asynchronously.
-     *
-     * @param invocationContext The invocation context
-     * @param llmResponse The LLM response to process
-     * @returns An async generator yielding events
-     */
-    runAsync(invocationContext: InvocationContext, llmResponse: LlmResponse): AsyncGenerator<Event, void, unknown>;
-}
-/**
- * The exported response processor instance.
- */
-export declare const responseProcessor: CodeExecutionResponseProcessor;
 /**
  * A unique symbol to identify ADK agent classes.
  * Defined once and shared by all LlmAgent instances.

package/dist/types/agents/processors/agent_transfer_llm_request_processor.d.ts

@@ -0,0 +1,18 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Event } from '../../events/event.js';
+import { LlmRequest } from '../../models/llm_request.js';
+import { InvocationContext } from '../invocation_context.js';
+import { BaseLlmRequestProcessor } from './base_llm_processor.js';
+export declare class AgentTransferLlmRequestProcessor extends BaseLlmRequestProcessor {
+    private readonly toolName;
+    private readonly tool;
+    runAsync(invocationContext: InvocationContext, llmRequest: LlmRequest): AsyncGenerator<Event, void, void>;
+    private buildTargetAgentsInfo;
+    private buildTargetAgentsInstructions;
+    private getTransferTargets;
+}
+export declare const AGENT_TRANSFER_LLM_REQUEST_PROCESSOR: AgentTransferLlmRequestProcessor;

package/dist/types/agents/{base_llm_processor.d.ts → processors/base_llm_processor.d.ts}

@@ -3,10 +3,10 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { InvocationContext } from '../agents/invocation_context.js';
-import { Event } from '../events/event.js';
-import { LlmRequest } from '../models/llm_request.js';
-import { LlmResponse } from '../models/llm_response.js';
+import { Event } from '../../events/event.js';
+import { LlmRequest } from '../../models/llm_request.js';
+import { LlmResponse } from '../../models/llm_response.js';
+import { InvocationContext } from '../invocation_context.js';
 /**
  * Base class for LLM request processor.
  */

package/dist/types/agents/processors/basic_llm_request_processor.d.ts

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Event } from '../../events/event.js';
+import { LlmRequest } from '../../models/llm_request.js';
+import { InvocationContext } from '../invocation_context.js';
+import { BaseLlmRequestProcessor } from './base_llm_processor.js';
+export declare class BasicLlmRequestProcessor extends BaseLlmRequestProcessor {
+    runAsync(invocationContext: InvocationContext, llmRequest: LlmRequest): AsyncGenerator<Event, void, void>;
+}
+export declare const BASIC_LLM_REQUEST_PROCESSOR: BasicLlmRequestProcessor;

package/dist/types/agents/processors/code_execution_request_processor.d.ts

@@ -0,0 +1,34 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Event } from '../../events/event.js';
+import { LlmRequest } from '../../models/llm_request.js';
+import { LlmResponse } from '../../models/llm_response.js';
+import { InvocationContext } from '../invocation_context.js';
+import { BaseLlmRequestProcessor, BaseLlmResponseProcessor } from './base_llm_processor.js';
+/**
+ * Processes code execution requests.
+ */
+export declare class CodeExecutionRequestProcessor extends BaseLlmRequestProcessor {
+    runAsync(invocationContext: InvocationContext, llmRequest: LlmRequest): AsyncGenerator<Event, void, void>;
+}
+/**
+ * Processor for code execution responses.
+ */
+export declare class CodeExecutionResponseProcessor implements BaseLlmResponseProcessor {
+    /**
+     * Processes the LLM response asynchronously.
+     *
+     * @param invocationContext The invocation context
+     * @param llmResponse The LLM response to process
+     * @returns An async generator yielding events
+     */
+    runAsync(invocationContext: InvocationContext, llmResponse: LlmResponse): AsyncGenerator<Event, void, unknown>;
+}
+/**
+ * The exported response processor instance.
+ */
+export declare const responseProcessor: CodeExecutionResponseProcessor;
+export declare const CODE_EXECUTION_REQUEST_PROCESSOR: CodeExecutionRequestProcessor;

package/dist/types/agents/processors/content_request_processor.d.ts

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Event } from '../../events/event.js';
+import { LlmRequest } from '../../models/llm_request.js';
+import { InvocationContext } from '../invocation_context.js';
+import { BaseLlmRequestProcessor } from './base_llm_processor.js';
+export declare class ContentRequestProcessor implements BaseLlmRequestProcessor {
+    runAsync(invocationContext: InvocationContext, llmRequest: LlmRequest): AsyncGenerator<Event, void, void>;
+}
+export declare const CONTENT_REQUEST_PROCESSOR: ContentRequestProcessor;

package/dist/types/agents/processors/identity_llm_request_processor.d.ts

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Event } from '../../events/event.js';
+import { LlmRequest } from '../../models/llm_request.js';
+import { InvocationContext } from '../invocation_context.js';
+import { BaseLlmRequestProcessor } from './base_llm_processor.js';
+export declare class IdentityLlmRequestProcessor extends BaseLlmRequestProcessor {
+    runAsync(invocationContext: InvocationContext, llmRequest: LlmRequest): AsyncGenerator<Event, void, undefined>;
+}
+export declare const IDENTITY_LLM_REQUEST_PROCESSOR: IdentityLlmRequestProcessor;

package/dist/types/agents/processors/instructions_llm_request_processor.d.ts

@@ -0,0 +1,16 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Event } from '../../events/event.js';
+import { LlmRequest } from '../../models/llm_request.js';
+import { InvocationContext } from '../invocation_context.js';
+import { BaseLlmRequestProcessor } from './base_llm_processor.js';
+export declare class InstructionsLlmRequestProcessor extends BaseLlmRequestProcessor {
+    /**
+     * Handles instructions and global instructions for LLM flow.
+     */
+    runAsync(invocationContext: InvocationContext, llmRequest: LlmRequest): AsyncGenerator<Event, void, void>;
+}
+export declare const INSTRUCTIONS_LLM_REQUEST_PROCESSOR: InstructionsLlmRequestProcessor;

package/dist/types/agents/processors/request_confirmation_llm_request_processor.d.ts

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Event } from '../../events/event.js';
+import { InvocationContext } from '../invocation_context.js';
+import { BaseLlmRequestProcessor } from './base_llm_processor.js';
+export declare class RequestConfirmationLlmRequestProcessor extends BaseLlmRequestProcessor {
+    /** Handles tool confirmation information to build the LLM request. */
+    runAsync(invocationContext: InvocationContext): AsyncGenerator<Event, void, void>;
+}
+export declare const REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR: RequestConfirmationLlmRequestProcessor;

package/dist/types/auth/credential_service/base_credential_service.d.ts

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { ToolContext } from '../../tools/tool_context.js';
+import { Context } from '../../agents/context.js';
 import { AuthCredential } from '../auth_credential.js';
 import { AuthConfig } from '../auth_tool.js';
 /**
@@ -22,6 +22,6 @@ export interface BaseCredentialService {
      *     trying to load the credential.
      * @return A promise that resolves to the credential saved in the store.
      */
-    loadCredential(authConfig: AuthConfig, toolContext: ToolContext): Promise<AuthCredential | undefined>;
-    saveCredential(authConfig: AuthConfig, toolContext: ToolContext): Promise<void>;
+    loadCredential(authConfig: AuthConfig, toolContext: Context): Promise<AuthCredential | undefined>;
+    saveCredential(authConfig: AuthConfig, toolContext: Context): Promise<void>;
 }

package/dist/types/auth/credential_service/in_memory_credential_service.d.ts

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { ToolContext } from '../../tools/tool_context.js';
+import { Context } from '../../agents/context.js';
 import { AuthCredential } from '../auth_credential.js';
 import { AuthConfig } from '../auth_tool.js';
 import { BaseCredentialService } from './base_credential_service.js';
@@ -13,7 +13,7 @@ import { BaseCredentialService } from './base_credential_service.js';
  */
 export declare class InMemoryCredentialService implements BaseCredentialService {
     private readonly credentials;
-    loadCredential(authConfig: AuthConfig, toolContext: ToolContext): Promise<AuthCredential | undefined>;
-    saveCredential(authConfig: AuthConfig, toolContext: ToolContext): Promise<void>;
+    loadCredential(authConfig: AuthConfig, toolContext: Context): Promise<AuthCredential | undefined>;
+    saveCredential(authConfig: AuthConfig, toolContext: Context): Promise<void>;
     private getBucketForCurrentContext;
 }

package/dist/types/common.d.ts

@@ -7,8 +7,7 @@ export { ActiveStreamingTool } from './agents/active_streaming_tool.js';
 export type { ActiveStreamingToolParams } from './agents/active_streaming_tool.js';
 export { BaseAgent, isBaseAgent } from './agents/base_agent.js';
 export type { AfterAgentCallback, BaseAgentConfig, BeforeAgentCallback, SingleAgentCallback, } from './agents/base_agent.js';
-export { BaseLlmRequestProcessor, BaseLlmResponseProcessor, } from './agents/base_llm_processor.js';
-export { CallbackContext } from './agents/callback_context.js';
+export { Context } from './agents/context.js';
 export { functionsExportedForTestingOnly } from './agents/functions.js';
 export { InvocationContext } from './agents/invocation_context.js';
 export type { InvocationContextParams } from './agents/invocation_context.js';
@@ -19,6 +18,7 @@ export type { AfterModelCallback, AfterToolCallback, BeforeModelCallback, Before
 export { LoopAgent, isLoopAgent } from './agents/loop_agent.js';
 export type { LoopAgentConfig } from './agents/loop_agent.js';
 export { ParallelAgent, isParallelAgent } from './agents/parallel_agent.js';
+export { BaseLlmRequestProcessor, BaseLlmResponseProcessor, } from './agents/processors/base_llm_processor.js';
 export { ReadonlyContext } from './agents/readonly_context.js';
 export { StreamingMode } from './agents/run_config.js';
 export type { RunConfig } from './agents/run_config.js';
@@ -74,14 +74,14 @@ export { AgentTool, isAgentTool } from './tools/agent_tool.js';
 export type { AgentToolConfig } from './tools/agent_tool.js';
 export { BaseTool, isBaseTool } from './tools/base_tool.js';
 export type { BaseToolParams, RunAsyncToolRequest, ToolProcessLlmRequest, } from './tools/base_tool.js';
-export { BaseToolset } from './tools/base_toolset.js';
+export { BaseToolset, isBaseToolset } from './tools/base_toolset.js';
 export type { ToolPredicate } from './tools/base_toolset.js';
+export { EXIT_LOOP, ExitLoopTool } from './tools/exit_loop_tool.js';
 export { FunctionTool, isFunctionTool } from './tools/function_tool.js';
 export type { ToolExecuteArgument, ToolExecuteFunction, ToolInputParameters, ToolOptions, } from './tools/function_tool.js';
 export { GOOGLE_SEARCH, GoogleSearchTool } from './tools/google_search_tool.js';
 export { LongRunningFunctionTool } from './tools/long_running_tool.js';
 export { ToolConfirmation } from './tools/tool_confirmation.js';
-export { ToolContext } from './tools/tool_context.js';
 export { LogLevel, getLogger, setLogLevel, setLogger } from './utils/logger.js';
 export type { Logger } from './utils/logger.js';
 export { isGemini2OrAbove } from './utils/model_name.js';

package/dist/types/models/llm_response.d.ts

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { Content, FinishReason, GenerateContentResponse, GenerateContentResponseUsageMetadata, GroundingMetadata, LiveServerSessionResumptionUpdate, Transcription } from '@google/genai';
+import { CitationMetadata, Content, FinishReason, GenerateContentResponse, GenerateContentResponseUsageMetadata, GroundingMetadata, LiveServerSessionResumptionUpdate, Transcription } from '@google/genai';
 /**
  * LLM response class that provides the first candidate response from the
  * model if available. Otherwise, returns error code and message.
@@ -17,6 +17,10 @@ export interface LlmResponse {
      * The grounding metadata of the response.
      */
     groundingMetadata?: GroundingMetadata;
+    /**
+     * The citation metadata of the response.
+     */
+    citationMetadata?: CitationMetadata;
     /**
      * Indicates whether the text content is part of a unfinished text stream.
      * Only used for streaming mode and when the content is plain text.

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

@@ -5,13 +5,12 @@
  */
 import { Content } from '@google/genai';
 import { BaseAgent } from '../agents/base_agent.js';
-import { CallbackContext } from '../agents/callback_context.js';
+import { Context } from '../agents/context.js';
 import { InvocationContext } from '../agents/invocation_context.js';
 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 { ToolContext } from '../tools/tool_context.js';
 /**
  * Base class for creating plugins.
  *
@@ -60,7 +59,7 @@ import { ToolContext } from '../tools/tool_context.js';
  *     {tool, toolArgs, toolContext}: {
  *       tool: BaseTool,
  *       toolArgs: Record<string, unknown>,
- *       toolContext: ToolContext,
+ *       toolContext: Context,
  *     },
  *   ): Promise<Record<string, unknown> | undefined> {
  *     this.logger.info(
@@ -75,7 +74,7 @@ import { ToolContext } from '../tools/tool_context.js';
  *     {tool, toolArgs, toolContext, result}: {
  *       tool: BaseTool,
  *       toolArgs: Record<string, unknown>,
- *       toolContext: ToolContext,
+ *       toolContext: Context,
  *       result: Record<string, unknown>,
  *     },
  *   ): Promise<Record<string, unknown> | undefined> {
@@ -177,7 +176,7 @@ export declare abstract class BasePlugin {
      */
     beforeAgentCallback(params: {
         agent: BaseAgent;
-        callbackContext: CallbackContext;
+        callbackContext: Context;
     }): Promise<Content | undefined>;
     /**
      * Callback executed after an agent's primary logic has completed.
@@ -193,7 +192,7 @@ export declare abstract class BasePlugin {
      */
     afterAgentCallback(params: {
         agent: BaseAgent;
-        callbackContext: CallbackContext;
+        callbackContext: Context;
     }): Promise<Content | undefined>;
     /**
      * Callback executed before a request is sent to the model.
@@ -209,7 +208,7 @@ export declare abstract class BasePlugin {
      *     `undefined` allows the LLM request to proceed normally.
      */
     beforeModelCallback(params: {
-        callbackContext: CallbackContext;
+        callbackContext: Context;
         llmRequest: LlmRequest;
     }): Promise<LlmResponse | undefined>;
     /**
@@ -225,7 +224,7 @@ export declare abstract class BasePlugin {
      *     allows the original response to be used.
      */
     afterModelCallback(params: {
-        callbackContext: CallbackContext;
+        callbackContext: Context;
         llmResponse: LlmResponse;
     }): Promise<LlmResponse | undefined>;
     /**
@@ -243,7 +242,7 @@ export declare abstract class BasePlugin {
      *     the original error to be raised.
      */
     onModelErrorCallback(params: {
-        callbackContext: CallbackContext;
+        callbackContext: Context;
         llmRequest: LlmRequest;
         error: Error;
     }): Promise<LlmResponse | undefined>;
@@ -264,7 +263,7 @@ export declare abstract class BasePlugin {
     beforeToolCallback(params: {
         tool: BaseTool;
         toolArgs: Record<string, unknown>;
-        toolContext: ToolContext;
+        toolContext: Context;
     }): Promise<Record<string, unknown> | undefined>;
     /**
      * Callback executed after a tool has been called.
@@ -284,14 +283,14 @@ export declare abstract class BasePlugin {
     afterToolCallback(params: {
         tool: BaseTool;
         toolArgs: Record<string, unknown>;
-        toolContext: ToolContext;
+        toolContext: Context;
         result: Record<string, unknown>;
     }): Promise<Record<string, unknown> | undefined>;
     /**
      * Callback executed when a tool call encounters an error.
       tool: BaseTool;
       toolArgs: Record<string, unknown>;
-      toolContext: ToolContext;
+      toolContext: Context;
       result: Record<string, unknown>;
     }): Promise<Record<string, unknown> | undefined> {
       return;
@@ -314,7 +313,7 @@ export declare abstract class BasePlugin {
     onToolErrorCallback(params: {
         tool: BaseTool;
         toolArgs: Record<string, unknown>;
-        toolContext: ToolContext;
+        toolContext: Context;
         error: Error;
     }): Promise<Record<string, unknown> | undefined>;
 }

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

@@ -5,13 +5,12 @@
  */
 import { Content } from '@google/genai';
 import { BaseAgent } from '../agents/base_agent.js';
-import { CallbackContext } from '../agents/callback_context.js';
+import { Context } from '../agents/context.js';
 import { InvocationContext } from '../agents/invocation_context.js';
 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 { ToolContext } from '../tools/tool_context.js';
 import { BasePlugin } from './base_plugin.js';
 /**
  * A plugin that logs important information at each callback point.
@@ -62,40 +61,40 @@ export declare class LoggingPlugin extends BasePlugin {
     }): Promise<void>;
     beforeAgentCallback({ callbackContext, }: {
         agent: BaseAgent;
-        callbackContext: CallbackContext;
+        callbackContext: Context;
     }): Promise<Content | undefined>;
     afterAgentCallback({ callbackContext, }: {
         agent: BaseAgent;
-        callbackContext: CallbackContext;
+        callbackContext: Context;
     }): Promise<Content | undefined>;
     beforeModelCallback({ callbackContext, llmRequest, }: {
-        callbackContext: CallbackContext;
+        callbackContext: Context;
         llmRequest: LlmRequest;
     }): Promise<LlmResponse | undefined>;
     afterModelCallback({ callbackContext, llmResponse, }: {
-        callbackContext: CallbackContext;
+        callbackContext: Context;
         llmResponse: LlmResponse;
     }): Promise<LlmResponse | undefined>;
     beforeToolCallback({ tool, toolArgs, toolContext, }: {
         tool: BaseTool;
         toolArgs: Record<string, unknown>;
-        toolContext: ToolContext;
+        toolContext: Context;
     }): Promise<Record<string, unknown> | undefined>;
     afterToolCallback({ tool, toolContext, result, }: {
         tool: BaseTool;
         toolArgs: Record<string, unknown>;
-        toolContext: ToolContext;
+        toolContext: Context;
         result: Record<string, unknown>;
     }): Promise<Record<string, unknown> | undefined>;
     onModelErrorCallback({ callbackContext, error, }: {
-        callbackContext: CallbackContext;
+        callbackContext: Context;
         llmRequest: LlmRequest;
         error: Error;
     }): Promise<LlmResponse | undefined>;
     onToolErrorCallback({ tool, toolArgs, toolContext, error, }: {
         tool: BaseTool;
         toolArgs: Record<string, unknown>;
-        toolContext: ToolContext;
+        toolContext: Context;
         error: Error;
     }): Promise<Record<string, unknown> | undefined>;
     private log;