@google/adk 0.3.0 → 0.5.0

package/dist/types/artifacts/registry.d.ts

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseArtifactService } from './base_artifact_service.js';
+export declare function getArtifactServiceFromUri(uri: string): BaseArtifactService;

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,23 +7,24 @@ 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';
 export { LiveRequestQueue } from './agents/live_request_queue.js';
 export type { LiveRequest } from './agents/live_request_queue.js';
 export { LlmAgent, isLlmAgent } from './agents/llm_agent.js';
-export type { AfterModelCallback, AfterToolCallback, BeforeModelCallback, BeforeToolCallback, InstructionProvider, LlmAgentConfig, SingleAfterModelCallback, SingleAfterToolCallback, SingleBeforeModelCallback, SingleBeforeToolCallback, ToolUnion, } from './agents/llm_agent.js';
+export type { AfterModelCallback, AfterToolCallback, BeforeModelCallback, BeforeToolCallback, InstructionProvider, LlmAgentConfig, LlmAgentSchema, SingleAfterModelCallback, SingleAfterToolCallback, SingleBeforeModelCallback, SingleBeforeToolCallback, ToolUnion, } from './agents/llm_agent.js';
 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';
 export { SequentialAgent, isSequentialAgent } from './agents/sequential_agent.js';
 export type { TranscriptionEntry } from './agents/transcription_entry.js';
+export type { BaseArtifactService, DeleteArtifactRequest, ListArtifactKeysRequest, ListVersionsRequest, LoadArtifactRequest, SaveArtifactRequest, } from './artifacts/base_artifact_service.js';
 export { InMemoryArtifactService } from './artifacts/in_memory_artifact_service.js';
 export { AuthCredentialTypes } from './auth/auth_credential.js';
 export type { AuthCredential, HttpAuth, HttpCredentials, OAuth2Auth, ServiceAccount, ServiceAccountCredential, } from './auth/auth_credential.js';
@@ -38,13 +39,18 @@ export { createEvent, getFunctionCalls, getFunctionResponses, hasTrailingCodeExe
 export type { Event } from './events/event.js';
 export { createEventActions } from './events/event_actions.js';
 export type { EventActions } from './events/event_actions.js';
+export { EventType, toStructuredEvents } from './events/structured_events.js';
+export type { ActivityEvent, CallCodeEvent, CodeResultEvent, ContentEvent, ErrorEvent, FinishedEvent, StructuredEvent, ThoughtEvent, ToolCallEvent, ToolConfirmationEvent, ToolResultEvent, } from './events/structured_events.js';
 export { BaseExampleProvider, isBaseExampleProvider, } from './examples/base_example_provider.js';
 export type { Example } from './examples/example.js';
+export type { BaseMemoryService, SearchMemoryRequest, SearchMemoryResponse, } from './memory/base_memory_service.js';
 export { InMemoryMemoryService } from './memory/in_memory_memory_service.js';
 export type { MemoryEntry } from './memory/memory_entry.js';
+export { ApigeeLlm } from './models/apigee_llm.js';
+export type { ApigeeLlmParams } from './models/apigee_llm.js';
 export { BaseLlm, isBaseLlm } from './models/base_llm.js';
 export type { BaseLlmConnection } from './models/base_llm_connection.js';
-export { Gemini } from './models/google_llm.js';
+export { Gemini, geminiInitParams } from './models/google_llm.js';
 export type { GeminiParams } from './models/google_llm.js';
 export type { LlmRequest } from './models/llm_request.js';
 export type { LlmResponse } from './models/llm_response.js';
@@ -58,6 +64,8 @@ export type { BasePolicyEngine, PolicyCheckResult, ToolCallPolicyContext, } from
 export { InMemoryRunner } from './runner/in_memory_runner.js';
 export { Runner } from './runner/runner.js';
 export type { RunnerConfig } from './runner/runner.js';
+export { BaseSessionService } from './sessions/base_session_service.js';
+export type { AppendEventRequest, CreateSessionRequest, DeleteSessionRequest, GetSessionConfig, GetSessionRequest, ListSessionsRequest, ListSessionsResponse, } from './sessions/base_session_service.js';
 export { InMemorySessionService } from './sessions/in_memory_session_service.js';
 export { createSession } from './sessions/session.js';
 export type { Session } from './sessions/session.js';
@@ -65,14 +73,15 @@ export { State } from './sessions/state.js';
 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 { BaseToolset } from './tools/base_toolset.js';
+export type { BaseToolParams, RunAsyncToolRequest, ToolProcessLlmRequest, } from './tools/base_tool.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/events/event.d.ts

@@ -1,6 +1,6 @@
 /**
  * @license
- * Copyright 2025 Google LLC
+ * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 import { FunctionCall, FunctionResponse } from '@google/genai';
@@ -86,3 +86,17 @@ export declare function stringifyContent(event: Event): string;
  * Generates a new unique ID for the event.
  */
 export declare function createNewEventId(): string;
+/**
+ * Transforms a snake_cased event object to a camelCased Event object.
+ *
+ * @param event The snake_cased event object.
+ * @returns The camelCased Event object.
+ */
+export declare function transformToCamelCaseEvent(event: Record<string, unknown>): Event;
+/**
+ * Transforms a camelCased event object to a snake_cased Event object.
+ *
+ * @param event The camelCased event object.
+ * @returns The snake_cased Event object.
+ */
+export declare function transformToSnakeCaseEvent(event: Event): Record<string, unknown>;

package/dist/types/events/event_actions.d.ts

@@ -1,6 +1,6 @@
 /**
  * @license
- * Copyright 2025 Google LLC
+ * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
 import { ToolConfirmation } from '../tools/tool_confirmation.js';

package/dist/types/events/structured_events.d.ts

@@ -0,0 +1,106 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { CodeExecutionResult, ExecutableCode, FunctionCall, FunctionResponse } from '@google/genai';
+import { Event } from './event.js';
+/**
+ * The types of events that can be parsed from a raw Event.
+ */
+export declare enum EventType {
+    THOUGHT = "thought",
+    CONTENT = "content",
+    TOOL_CALL = "tool_call",
+    TOOL_RESULT = "tool_result",
+    CALL_CODE = "call_code",
+    CODE_RESULT = "code_result",
+    ERROR = "error",
+    ACTIVITY = "activity",
+    TOOL_CONFIRMATION = "tool_confirmation",
+    FINISHED = "finished"
+}
+/**
+ * Represents a reasoning trace (thought) from the agent.
+ */
+export interface ThoughtEvent {
+    type: EventType.THOUGHT;
+    content: string;
+}
+/**
+ * Represents partial content (text delta) intended for the user.
+ */
+export interface ContentEvent {
+    type: EventType.CONTENT;
+    content: string;
+}
+/**
+ * Represents a request to execute a tool.
+ */
+export interface ToolCallEvent {
+    type: EventType.TOOL_CALL;
+    call: FunctionCall;
+}
+/**
+ * Represents the result of a tool execution.
+ */
+export interface ToolResultEvent {
+    type: EventType.TOOL_RESULT;
+    result: FunctionResponse;
+}
+/**
+ * Represents a request to execute code.
+ */
+export interface CallCodeEvent {
+    type: EventType.CALL_CODE;
+    code: ExecutableCode;
+}
+/**
+ * Represents the result of code execution.
+ */
+export interface CodeResultEvent {
+    type: EventType.CODE_RESULT;
+    result: CodeExecutionResult;
+}
+/**
+ * Represents a runtime error.
+ */
+export interface ErrorEvent {
+    type: EventType.ERROR;
+    error: Error;
+}
+/**
+ * Represents a generic activity or status update.
+ */
+export interface ActivityEvent {
+    type: EventType.ACTIVITY;
+    kind: string;
+    detail: Record<string, unknown>;
+}
+/**
+ * Represents a request for tool confirmation.
+ */
+export interface ToolConfirmationEvent {
+    type: EventType.TOOL_CONFIRMATION;
+    confirmations: Record<string, unknown>;
+}
+/**
+ * Represents the final completion of the agent's task.
+ */
+export interface FinishedEvent {
+    type: EventType.FINISHED;
+    output?: unknown;
+}
+/**
+ * A standard structured event parsed from the raw Event stream.
+ */
+export type StructuredEvent = ThoughtEvent | ContentEvent | ToolCallEvent | ToolResultEvent | CallCodeEvent | CodeResultEvent | ErrorEvent | ActivityEvent | ToolConfirmationEvent | FinishedEvent;
+/**
+ * Converts an internal Event to a list of structured events.
+ * This is an optional utility for callers who want to easily identify
+ * the type of event they are handling.
+ *
+ * @param event - The raw event to convert.
+ * @returns The structured events.
+ */
+export declare function toStructuredEvents(event: Event): StructuredEvent[];

package/dist/types/index.d.ts

@@ -3,8 +3,12 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-export * from './artifacts/gcs_artifact_service.js';
+export { FileArtifactService } from './artifacts/file_artifact_service.js';
+export { GcsArtifactService } from './artifacts/gcs_artifact_service.js';
+export { getArtifactServiceFromUri } from './artifacts/registry.js';
 export * from './common.js';
+export { DatabaseSessionService } from './sessions/database_session_service.js';
+export { getSessionServiceFromUri } from './sessions/registry.js';
 export * from './telemetry/google_cloud.js';
 export * from './telemetry/setup.js';
 export * from './tools/mcp/mcp_session_manager.js';

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

@@ -0,0 +1,59 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { HttpOptions } from '@google/genai';
+import { BaseLlmConnection } from './base_llm_connection.js';
+import { Gemini, GeminiParams } from './google_llm.js';
+import { LlmRequest } from './llm_request.js';
+import { LlmResponse } from './llm_response.js';
+export interface ApigeeLlmParams extends GeminiParams {
+    /**
+     * The name of the model to use. The model string specifies the LLM provider
+     * (e.g., Vertex AI, Gemini), API version, and the model ID. Supported format:
+     *     `apigee/[<provider>/][<version>/]<model_id>`
+     *     Components:
+     *       `provider` (optional): `vertex_ai` or `gemini`.
+     *       `version` (optional): The API version (e.g., `v1`, `v1beta`). If not
+     *         provided, a default version will selected based on the provider.
+     *       `model_id` (required): The model identifier (e.g.,
+     *         `gemini-2.5-flash`).
+     *     Examples:
+     *       - `apigee/gemini-2.5-flash`
+     *       - `apigee/v1/gemini-2.5-flash`
+     *       - `apigee/vertex_ai/gemini-2.5-flash`
+     *       - `apigee/gemini/v1/gemini-2.5-flash`
+     *       - `apigee/vertex_ai/v1beta/gemini-2.5-flash`
+     */
+    model: string;
+    /**
+     * The proxy URL for the provider API. If not provided, it will look for
+     * the APIGEE_PROXY_URL environment variable.
+     */
+    proxyUrl?: string;
+    /**
+     * API key to use. If not provided, it will look for
+     * the GOOGLE_GENAI_API_KEY or GEMINI_API_KEY environment variable. If gemini
+     * provider is selected and no key is provided, the fake key "-" will be
+     * used for the "x-goog-api-key" header.
+     */
+    apiKey?: string;
+}
+export declare class ApigeeLlm extends Gemini {
+    private readonly proxyUrl;
+    /**
+     * A list of model name patterns that are supported by this LLM.
+     *
+     * @returns A list of supported models.
+     */
+    static readonly supportedModels: Array<string | RegExp>;
+    constructor({ model, proxyUrl, apiKey, vertexai, location, project, headers, }: ApigeeLlmParams);
+    protected getHttpOptions(): HttpOptions;
+    protected getLiveHttpOptions(): HttpOptions;
+    private identifyApiVersion;
+    private _apigeeLiveApiVersion?;
+    get liveApiVersion(): string;
+    generateContentAsync(llmRequest: LlmRequest, stream?: boolean): AsyncGenerator<LlmResponse, void>;
+    connect(llmRequest: LlmRequest): Promise<BaseLlmConnection>;
+}

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

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { GoogleGenAI } from '@google/genai';
+import { GoogleGenAI, HttpOptions } from '@google/genai';
 import { GoogleLLMVariant } from '../utils/variant_utils.js';
 import { BaseLlm } from './base_llm.js';
 import { BaseLlmConnection } from './base_llm_connection.js';
@@ -45,7 +45,7 @@ export interface GeminiParams {
  */
 export declare class Gemini extends BaseLlm {
     private readonly apiKey?;
-    private readonly vertexai;
+    protected readonly vertexai: boolean;
     private readonly project?;
     private readonly location?;
     private readonly headers?;
@@ -72,9 +72,11 @@ export declare class Gemini extends BaseLlm {
      * @yields LlmResponse: The model response.
      */
     generateContentAsync(llmRequest: LlmRequest, stream?: boolean): AsyncGenerator<LlmResponse, void>;
+    protected getHttpOptions(): HttpOptions;
     get apiClient(): GoogleGenAI;
     get apiBackend(): GoogleLLMVariant;
     get liveApiVersion(): string;
+    protected getLiveHttpOptions(): HttpOptions;
     get liveApiClient(): GoogleGenAI;
     /**
      * Connects to the Gemini model and returns an llm connection.
@@ -85,3 +87,4 @@ export declare class Gemini extends BaseLlm {
     connect(llmRequest: LlmRequest): Promise<BaseLlmConnection>;
     private preprocessRequest;
 }
+export declare function geminiInitParams({ model, vertexai, project, location, apiKey, }: GeminiParams): GeminiParams;

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;

package/dist/types/plugins/plugin_manager.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';
 /**
  * Manages the registration and execution of plugins.
@@ -95,14 +94,14 @@ export declare class PluginManager {
      */
     runBeforeAgentCallback({ agent, callbackContext, }: {
         agent: BaseAgent;
-        callbackContext: CallbackContext;
+        callbackContext: Context;
     }): Promise<Content | undefined>;
     /**
      * Runs the `afterAgentCallback` for all plugins.
      */
     runAfterAgentCallback({ agent, callbackContext, }: {
         agent: BaseAgent;
-        callbackContext: CallbackContext;
+        callbackContext: Context;
     }): Promise<Content | undefined>;
     /**
      * Runs the `beforeToolCallback` for all plugins.
@@ -110,7 +109,7 @@ export declare class PluginManager {
     runBeforeToolCallback({ tool, toolArgs, toolContext, }: {
         tool: BaseTool;
         toolArgs: Record<string, unknown>;
-        toolContext: ToolContext;
+        toolContext: Context;
     }): Promise<Record<string, unknown> | undefined>;
     /**
      * Runs the `afterToolCallback` for all plugins.
@@ -118,14 +117,14 @@ export declare class PluginManager {
     runAfterToolCallback({ tool, toolArgs, toolContext, result, }: {
         tool: BaseTool;
         toolArgs: Record<string, unknown>;
-        toolContext: ToolContext;
+        toolContext: Context;
         result: Record<string, unknown>;
     }): Promise<Record<string, unknown> | undefined>;
     /**
      * Runs the `onModelErrorCallback` for all plugins.
      */
     runOnModelErrorCallback({ callbackContext, llmRequest, error, }: {
-        callbackContext: CallbackContext;
+        callbackContext: Context;
         llmRequest: LlmRequest;
         error: Error;
     }): Promise<LlmResponse | undefined>;
@@ -133,14 +132,14 @@ export declare class PluginManager {
      * Runs the `beforeModelCallback` for all plugins.
      */
     runBeforeModelCallback({ callbackContext, llmRequest, }: {
-        callbackContext: CallbackContext;
+        callbackContext: Context;
         llmRequest: LlmRequest;
     }): Promise<LlmResponse | undefined>;
     /**
      * Runs the `afterModelCallback` for all plugins.
      */
     runAfterModelCallback({ callbackContext, llmResponse, }: {
-        callbackContext: CallbackContext;
+        callbackContext: Context;
         llmResponse: LlmResponse;
     }): Promise<LlmResponse | undefined>;
     /**
@@ -149,7 +148,7 @@ export declare class PluginManager {
     runOnToolErrorCallback({ tool, toolArgs, toolContext, error, }: {
         tool: BaseTool;
         toolArgs: Record<string, unknown>;
-        toolContext: ToolContext;
+        toolContext: Context;
         error: Error;
     }): Promise<Record<string, unknown> | undefined>;
 }