@google/adk 0.4.0 → 0.6.0

package/dist/types/a2a/executor_context.d.ts

@@ -0,0 +1,33 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { RequestContext } from '@a2a-js/sdk/server';
+import { Content } from '@google/genai';
+import { Event } from '../events/event.js';
+import { Session } from '../sessions/session.js';
+/**
+ * The A2A Agent Executor context.
+ */
+export interface ExecutorContext {
+    userId: string;
+    sessionId: string;
+    appName: string;
+    readonlyState: Record<string, unknown>;
+    events: Event[];
+    userContent: Content;
+    requestContext: RequestContext;
+}
+/**
+ * Creates an A2A Agent Executor context from the given parameters.
+ * @param session The session.
+ * @param userContent The content of the user.
+ * @param requestContext The request context.
+ * @returns The A2A Agent Executor context.
+ */
+export declare function createExecutorContext({ session, userContent, requestContext, }: {
+    session: Session;
+    userContent: Content;
+    requestContext: RequestContext;
+}): ExecutorContext;

package/dist/types/a2a/metadata_converter_utils.d.ts

@@ -0,0 +1,63 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Event as AdkEvent } from '../events/event.js';
+import { EventActions as AdkEventActions } from '../events/event_actions.js';
+import { A2AEvent } from './a2a_event.js';
+/**
+ * Keys for metadata that will be stored in A2A message metadata and related to ADK events.
+ */
+export declare enum A2AMetadataKeys {
+    APP_NAME = "adk_app_name",
+    USER_ID = "adk_user_id",
+    SESSION_ID = "adk_session_id",
+    INVOCATION_ID = "adk_invocation_id",
+    AUTHOR = "adk_author",
+    BRANCH = "adk_branch",
+    DATA_PART_TYPE = "adk_type",
+    PARTIAL = "adk_partial",
+    ESCALATE = "adk_escalate",
+    TRANSFER_TO_AGENT = "adk_transfer_to_agent",
+    IS_LONG_RUNNING = "adk_is_long_running",
+    THOUGHT = "adk_thought",
+    ERROR_CODE = "adk_error_code",
+    ERROR_MESSAGE = "adk_error_message",
+    CITATION_METADATA = "adk_citation_metadata",
+    GROUNDING_METADATA = "adk_grounding_metadata",
+    USAGE_METADATA = "adk_usage_metadata",
+    CUSTOM_METADATA = "adk_custom_metadata",
+    VIDEO_METADATA = "adk_video_metadata"
+}
+/**
+ * Keys for metadata that will be stored in ADK event metadata and related to A2A messages.
+ */
+export declare enum AdkMetadataKeys {
+    TASK_ID = "a2a:task_id",
+    CONTEXT_ID = "a2a:context_id"
+}
+/**
+ * Creates ADK Event metadata from an A2A Event.
+ */
+export declare function getAdkEventMetadata(a2aEvent: A2AEvent): Record<string, unknown>;
+/**
+ * Creates A2A Event metadata from an ADK Event.
+ */
+export declare function getA2AEventMetadata(adkEvent: AdkEvent, { appName, userId, sessionId, }: {
+    appName: string;
+    userId: string;
+    sessionId: string;
+}): Record<string, unknown>;
+/**
+ * Creates A2A Session metadata from ADK Event invocation metadata.
+ */
+export declare function getA2ASessionMetadata({ appName, userId, sessionId, }: {
+    appName: string;
+    userId: string;
+    sessionId: string;
+}): Record<string, unknown>;
+/**
+ * Creates A2A Event metadata from ADK Event actions.
+ */
+export declare function getA2AEventMetadataFromActions(actions: AdkEventActions): Record<string, unknown>;

package/dist/types/a2a/part_converter_utils.d.ts

@@ -3,12 +3,12 @@
  * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { DataPart as A2ADataPart, FilePart as A2AFilePart, Part as A2APart, TextPart as A2ATextPart } from '@a2a-js/sdk';
-import { Part as GenAIPart } from '@google/genai';
+import { DataPart as A2ADataPart, FilePart as A2AFilePart, Part as A2APart, TextPart as A2ATextPart, Message } from '@a2a-js/sdk';
+import { Content as GenAIContent, Part as GenAIPart } from '@google/genai';
 /**
  * Converts GenAI Parts to A2A Parts.
  */
-export declare function toA2AParts(parts: GenAIPart[], longRunningToolIDs?: string[]): A2APart[];
+export declare function toA2AParts(parts?: GenAIPart[], longRunningToolIDs?: string[]): A2APart[];
 /**
  * Converts a GenAI Part to an A2A Part.
  */
@@ -25,6 +25,7 @@ export declare function toA2AFilePart(part: GenAIPart): A2APart;
  * Converts a GenAI Data Part to an A2A Data Part.
  */
 export declare function toA2ADataPart(part: GenAIPart, longRunningToolIDs?: string[]): A2APart;
+export declare function toGenAIContent(a2aMessage: Message): GenAIContent;
 /**
  * Converts an A2A Part to a GenAI Part.
  */

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

@@ -5,12 +5,12 @@
  */
 import { Content } from '@google/genai';
 import { Event } from '../events/event.js';
-import { CallbackContext } from './callback_context.js';
+import { Context } from './context.js';
 import { InvocationContext } from './invocation_context.js';
 /**
  * A single callback function for an agent.
  */
-export type SingleAgentCallback = (context: CallbackContext) => Promise<Content | undefined> | (Content | undefined);
+export type SingleAgentCallback = (context: Context) => Promise<Content | undefined> | (Content | undefined);
 /**
  * Type for before agent callbacks, which can be a single callback or
  * an array of callbacks.

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,10 @@ 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 { BaseContextCompactor } from '../context/base_context_compactor.js';
 import { InvocationContext } from './invocation_context.js';
 import { ReadonlyContext } from './readonly_context.js';
 /**
@@ -36,7 +36,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 +56,7 @@ export type BeforeModelCallback = SingleBeforeModelCallback | SingleBeforeModelC
  *     user.
  */
 export type SingleAfterModelCallback = (params: {
-    context: CallbackContext;
+    context: Context;
     response: LlmResponse;
 }) => LlmResponse | undefined | Promise<LlmResponse | undefined>;
 /**
@@ -71,14 +71,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 +92,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>;
 /**
@@ -201,32 +201,15 @@ export interface LlmAgentConfig extends BaseAgentConfig {
      */
     responseProcessors?: BaseLlmResponseProcessor[];
     /**
-     * Instructs the agent to make a plan and execute it step by step.
+     * A list of context compactors to evaluate in priority order.
+     * Modifies the session history to keep context overhead within limits.
      */
-    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 {
+    contextCompactors?: BaseContextCompactor[];
     /**
-     * Processes the LLM response asynchronously.
-     *
-     * @param invocationContext The invocation context
-     * @param llmResponse The LLM response to process
-     * @returns An async generator yielding events
+     * Instructs the agent to make a plan and execute it step by step.
      */
-    runAsync(invocationContext: InvocationContext, llmResponse: LlmResponse): AsyncGenerator<Event, void, unknown>;
+    codeExecutor?: BaseCodeExecutor;
 }
-/**
- * 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/loop_agent.d.ts

@@ -39,7 +39,7 @@ export declare class LoopAgent extends BaseAgent {
      * A unique symbol to identify ADK loop agent class.
      */
     readonly [LOOP_AGENT_SIGNATURE_SYMBOL] = true;
-    private readonly maxIterations;
+    readonly maxIterations: number;
     constructor(config: LoopAgentConfig);
     protected runAsyncImpl(context: InvocationContext): AsyncGenerator<Event, void, void>;
     protected runLiveImpl(_context: InvocationContext): AsyncGenerator<Event, void, void>;

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/{content_processor_utils.d.ts → processors/content_processor_utils.d.ts}

@@ -4,7 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { Content } from '@google/genai';
-import { Event } from '../events/event.js';
+import { Event } from '../../events/event.js';
 /**
  * Get the contents for the LLM request.
  *

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/context_compactor_request_processor.d.ts

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseContextCompactor } from '../../context/base_context_compactor.js';
+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';
+/**
+ * A processor that evaluates a set of compactors to optionally compact
+ * the conversation history (events) prior to generating an LLM request.
+ *
+ * It evaluates each compactor in priority order. The first one that indicates
+ * it should compact will perform the compaction and iteration stops.
+ */
+export declare class ContextCompactorRequestProcessor implements BaseLlmRequestProcessor {
+    private compactors;
+    constructor(compactors: BaseContextCompactor[]);
+    runAsync(invocationContext: InvocationContext, _llmRequest: LlmRequest): AsyncGenerator<Event, void, void>;
+}

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/artifacts/file_artifact_service.d.ts

@@ -41,3 +41,7 @@ export declare class FileArtifactService implements BaseArtifactService {
     listArtifactVersions({ userId, sessionId, filename, }: ListVersionsRequest): Promise<ArtifactVersion[]>;
     getArtifactVersion({ userId, sessionId, filename, version, }: LoadArtifactRequest): Promise<ArtifactVersion | undefined>;
 }
+export declare function assertSafeSegment(value: string, label: string): void;
+export declare function assertInsideRoot(resolvedPath: string, rootDir: string, label: string): void;
+export declare function getUserRoot(rootDir: string, userId: string): string;
+export declare function getSessionArtifactsDir(baseRoot: string, sessionId: string): string;

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,9 @@ 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 { CONTENT_REQUEST_PROCESSOR, ContentRequestProcessor, } from './agents/processors/content_request_processor.js';
+export { ContextCompactorRequestProcessor } from './agents/processors/context_compactor_request_processor.js';
 export { ReadonlyContext } from './agents/readonly_context.js';
 export { StreamingMode } from './agents/run_config.js';
 export type { RunConfig } from './agents/run_config.js';
@@ -35,6 +37,16 @@ export { BaseCodeExecutor } from './code_executors/base_code_executor.js';
 export type { ExecuteCodeParams } from './code_executors/base_code_executor.js';
 export { BuiltInCodeExecutor } from './code_executors/built_in_code_executor.js';
 export type { CodeExecutionInput, CodeExecutionResult, File, } from './code_executors/code_execution_utils.js';
+export type { BaseContextCompactor } from './context/base_context_compactor.js';
+export type { BaseSummarizer } from './context/summarizers/base_summarizer.js';
+export { LlmSummarizer } from './context/summarizers/llm_summarizer.js';
+export type { LlmSummarizerOptions } from './context/summarizers/llm_summarizer.js';
+export { TokenBasedContextCompactor } from './context/token_based_context_compactor.js';
+export type { TokenBasedContextCompactorOptions } from './context/token_based_context_compactor.js';
+export { TruncatingContextCompactor } from './context/truncating_context_compactor.js';
+export type { TruncatingContextCompactorOptions } from './context/truncating_context_compactor.js';
+export { isCompactedEvent } from './events/compacted_event.js';
+export type { CompactedEvent } from './events/compacted_event.js';
 export { createEvent, getFunctionCalls, getFunctionResponses, hasTrailingCodeExecutionResult, isFinalResponse, stringifyContent, } from './events/event.js';
 export type { Event } from './events/event.js';
 export { createEventActions } from './events/event_actions.js';
@@ -62,7 +74,7 @@ export { PluginManager } from './plugins/plugin_manager.js';
 export { InMemoryPolicyEngine, PolicyOutcome, REQUEST_CONFIRMATION_FUNCTION_CALL_NAME, SecurityPlugin, getAskUserConfirmationFunctionCalls, } from './plugins/security_plugin.js';
 export type { BasePolicyEngine, PolicyCheckResult, ToolCallPolicyContext, } from './plugins/security_plugin.js';
 export { InMemoryRunner } from './runner/in_memory_runner.js';
-export { Runner } from './runner/runner.js';
+export { Runner, isRunner } 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';
@@ -74,14 +86,17 @@ 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 { LOAD_ARTIFACTS, LoadArtifactsTool, } from './tools/load_artifacts_tool.js';
+export { LOAD_MEMORY, LoadMemoryTool } from './tools/load_memory_tool.js';
 export { LongRunningFunctionTool } from './tools/long_running_tool.js';
+export { PRELOAD_MEMORY, PreloadMemoryTool, } from './tools/preload_memory_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';