@google/adk 0.6.1 → 1.0.0

package/dist/types/auth/oauth2/oauth2_discovery.d.ts

@@ -0,0 +1,38 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { z } from 'zod';
+/**
+ * Represents the OAuth2 authorization server metadata per RFC8414.
+ */
+export declare const AuthorizationServerMetadataSchema: z.ZodObject<{
+    issuer: z.ZodString;
+    authorization_endpoint: z.ZodString;
+    token_endpoint: z.ZodString;
+    scopes_supported: z.ZodOptional<z.ZodArray<z.ZodString>>;
+    registration_endpoint: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export type AuthorizationServerMetadata = z.infer<typeof AuthorizationServerMetadataSchema>;
+/**
+ * Represents the OAuth2 protected resource metadata per RFC9728.
+ */
+export declare const ProtectedResourceMetadataSchema: z.ZodObject<{
+    resource: z.ZodString;
+    authorization_servers: z.ZodDefault<z.ZodArray<z.ZodString>>;
+}, z.core.$strip>;
+export type ProtectedResourceMetadata = z.infer<typeof ProtectedResourceMetadataSchema>;
+/**
+ * Implements Metadata discovery for OAuth2 following RFC8414 and RFC9728.
+ */
+export declare class OAuth2DiscoveryManager {
+    /**
+     * Discovers the OAuth2 authorization server metadata.
+     */
+    discoverAuthServerMetadata(issuerUrl: string): Promise<AuthorizationServerMetadata | undefined>;
+    /**
+     * Discovers the OAuth2 protected resource metadata.
+     */
+    discoverResourceMetadata(resourceUrl: string): Promise<ProtectedResourceMetadata | undefined>;
+}

package/dist/types/auth/oauth2/oauth2_utils.d.ts

@@ -0,0 +1,56 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { OAuth2Auth } from '../auth_credential.js';
+import { AuthScheme } from '../auth_schemes.js';
+/**
+ * Returns the token endpoint for the given auth scheme.
+ */
+export declare function getTokenEndpoint(authScheme: AuthScheme): string | undefined;
+/**
+ * Fetches OAuth2 tokens from the endpoint using the given body.
+ */
+export declare function fetchOAuth2Tokens(endpoint: string, body: URLSearchParams): Promise<OAuth2Auth>;
+/**
+ * Parses the authorization code from an authorization response URI.
+ */
+export declare function parseAuthorizationCode(uri: string): string | undefined;
+/**
+ * Parameters for a Client Credentials token request.
+ */
+export interface ClientCredentialsParams {
+    grantType: 'client_credentials';
+    clientId: string;
+    clientSecret: string;
+}
+/**
+ * Parameters for an Authorization Code token request.
+ */
+export interface AuthorizationCodeParams {
+    grantType: 'authorization_code';
+    clientId: string;
+    clientSecret: string;
+    code: string;
+    redirectUri?: string;
+    codeVerifier?: string;
+}
+/**
+ * Parameters for a Refresh Token request.
+ */
+export interface RefreshTokenParams {
+    grantType: 'refresh_token';
+    clientId: string;
+    clientSecret: string;
+    refreshToken: string;
+}
+/**
+ * Parameters for creating an OAuth2 token request body.
+ */
+export type OAuth2TokenRequestParams = ClientCredentialsParams | AuthorizationCodeParams | RefreshTokenParams;
+/**
+ * Creates URLSearchParams for an OAuth2 token request.
+ */
+export declare function createOAuth2TokenRequestBody(params: OAuth2TokenRequestParams): URLSearchParams;
+export declare function isTokenExpired(token: OAuth2Auth, leeway?: number): boolean;

package/dist/types/auth/refresher/base_credential_refresher.d.ts

@@ -0,0 +1,38 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthCredential } from '../auth_credential.js';
+import { AuthScheme } from '../auth_schemes.js';
+/**
+ * Base exception for credential refresh errors.
+ */
+export declare class CredentialRefresherError extends Error {
+    constructor(message: string);
+}
+/**
+ * Base interface for credential refreshers.
+ *
+ * Credential refreshers are responsible for checking if a credential is expired
+ * or needs to be refreshed, and for refreshing it if necessary.
+ */
+export interface BaseCredentialRefresher {
+    /**
+     * Checks if a credential needs to be refreshed.
+     *
+     * @param authCredential The credential to check.
+     * @param authScheme The authentication scheme (optional).
+     * @returns True if the credential needs to be refreshed, False otherwise.
+     */
+    isRefreshNeeded(authCredential: AuthCredential, authScheme?: AuthScheme): Promise<boolean>;
+    /**
+     * Refreshes a credential if needed.
+     *
+     * @param authCredential The credential to refresh.
+     * @param authScheme The authentication scheme (optional).
+     * @returns The refreshed credential.
+     * @throws {CredentialRefresherError} If credential refresh fails.
+     */
+    refresh(authCredential: AuthCredential, authScheme?: AuthScheme): Promise<AuthCredential>;
+}

package/dist/types/auth/refresher/credential_refresher_registry.d.ts

@@ -0,0 +1,27 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthCredentialTypes } from '../auth_credential.js';
+import { BaseCredentialRefresher } from './base_credential_refresher.js';
+/**
+ * Registry for credential refresher instances.
+ */
+export declare class CredentialRefresherRegistry {
+    private readonly refreshers;
+    /**
+     * Register a refresher instance for a credential type.
+     *
+     * @param credentialType The credential type to register for.
+     * @param refresherInstance The refresher instance to register.
+     */
+    register(credentialType: AuthCredentialTypes, refresherInstance: BaseCredentialRefresher): void;
+    /**
+     * Get the refresher instance for a credential type.
+     *
+     * @param credentialType The credential type to get refresher for.
+     * @returns The refresher instance if registered, undefined otherwise.
+     */
+    getRefresher(credentialType: AuthCredentialTypes): BaseCredentialRefresher | undefined;
+}

package/dist/types/code_executors/base_code_executor.d.ts

@@ -51,11 +51,11 @@ export declare abstract class BaseCodeExecutor {
     errorRetryAttempts: number;
     /**
      * The list of the enclosing delimiters to identify the code blocks.
-     * For example, the delimiter('```python\\n', '\\n```') can be  used to
-     * identify code blocks with the following format::
+     * For example, the delimiter('```javascript\\n', '\\n```') can be used to
+     * identify code blocks with the following format:
      *
-     * ```python
-     *  print("hello")
+     * ```javascript
+     *  console.log("hello")
      * ```
      */
     codeBlockDelimiters: Array<[string, string]>;

package/dist/types/code_executors/code_execution_utils.d.ts

@@ -4,6 +4,10 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { Content, Part } from '@google/genai';
+export declare enum FileContentEncoding {
+    UTF8 = "utf-8",
+    BASE64 = "base64"
+}
 /**
  * A structure that contains a file name and its content
  */
@@ -13,14 +17,30 @@ export interface File {
      * */
     name: string;
     /**
-     * The base64 - encoded bytes of the file content.
+     * The encoded bytes of the file content.
      * */
     content: string;
+    /**
+     * The encoding of the file content.
+     */
+    contentEncoding?: FileContentEncoding;
     /**
      * The mime type of the file (e.g., ' image / png')
      * */
     mimeType: string;
 }
+/**
+ * The language of the code to execute.
+ */
+export declare enum CodeExecutionLanguage {
+    UNSPECIFIED = "unspecified",
+    PYTHON = "python",
+    JAVASCRIPT = "javascript",
+    TYPESCRIPT = "typescript",
+    SHELL = "shell",
+    POWERSHELL = "powershell",
+    WINDOWS_CMD = "cmd"
+}
 /**
  * A structure that contains the input of code execution.
  * */
@@ -29,6 +49,10 @@ export interface CodeExecutionInput {
      * The code to execute.
      * */
     code: string;
+    /**
+     * The language of the code to execute.
+     */
+    language: CodeExecutionLanguage;
     /**
      * The input files available to the code.
      * */
@@ -37,6 +61,10 @@ export interface CodeExecutionInput {
      * The execution ID for the stateful code execution.
      * */
     executionId?: string;
+    /**
+     * Optional arguments to pass to the executed code/script.
+     */
+    args?: string[] | Record<string, string | number | boolean>;
 }
 /**
  * A structure that contains the result of code execution.

package/dist/types/code_executors/unsafe_local_code_executor.d.ts

@@ -0,0 +1,48 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseCodeExecutor, ExecuteCodeParams } from './base_code_executor.js';
+import { CodeExecutionResult } from './code_execution_utils.js';
+/**
+ * Options for UnsafeLocalCodeExecutor.
+ */
+export interface UnsafeLocalCodeExecutorOptions {
+    /**
+     * Timeout for code execution in seconds. Default is 30.
+     */
+    timeoutSeconds?: number;
+    /**
+     * The command to run JavaScript code. Default is `process.execPath` (Node.js).
+     */
+    commandPath?: string;
+    /**
+     * The command to run Python code. Default is `python3`.
+     */
+    pythonCommandPath?: string;
+    /**
+     * The command to run Shell code. Default is `bash`.
+     */
+    shellCommandPath?: string;
+}
+/**
+ * A code executor that unsafely executes code in the local context.
+ * Supports JavaScript, Python, and Shell capabilities cross-platform.
+ *
+ * **Execution Details**:
+ * - **JavaScript**: Executed via `node` (defaults to `process.execPath`).
+ * - **Python**: Executed via `python3` on Unix, and `python` on Windows.
+ * - **Shell**: Executed via `bash` on Unix, and defaults to `powershell` (injecting ExecutionPolicy Bypass) or `cmd.exe` on Windows.
+ *
+ * WARNING: This executor runs code in the local environment without sandboxing or security restrictions.
+ * Use with caution and only for trusted code.
+ */
+export declare class UnsafeLocalCodeExecutor extends BaseCodeExecutor {
+    private readonly timeoutSeconds;
+    private readonly nodeCommandPath;
+    private readonly pythonCommandPath;
+    private readonly shellCommandPath;
+    constructor(options?: UnsafeLocalCodeExecutorOptions);
+    executeCode(params: ExecuteCodeParams): Promise<CodeExecutionResult>;
+}

package/dist/types/common.d.ts

@@ -13,15 +13,18 @@ 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 { LlmAgent as Agent, LlmAgent, isLlmAgent } 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 { AgentTransferLlmRequestProcessor } from './agents/processors/agent_transfer_llm_request_processor.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 { RoutedAgent, isRoutedAgent } from './agents/routed_agent.js';
+export type { AgentRouter, RoutedAgentConfig } from './agents/routed_agent.js';
 export { StreamingMode } from './agents/run_config.js';
 export type { RunConfig } from './agents/run_config.js';
 export { SequentialAgent, isSequentialAgent } from './agents/sequential_agent.js';
@@ -30,13 +33,25 @@ export type { BaseArtifactService, DeleteArtifactRequest, ListArtifactKeysReques
 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';
+export { AuthHandler } from './auth/auth_handler.js';
+export { AuthProviderRegistry } from './auth/auth_provider_registry.js';
+export { OAuthGrantType } from './auth/auth_schemes.js';
 export type { AuthScheme, OpenIdConnectWithConfig } from './auth/auth_schemes.js';
 export type { AuthConfig } from './auth/auth_tool.js';
+export type { BaseAuthProvider } from './auth/base_auth_provider.js';
 export type { BaseCredentialService } from './auth/credential_service/base_credential_service.js';
+export { InMemoryCredentialService } from './auth/credential_service/in_memory_credential_service.js';
+export { SessionStateCredentialService } from './auth/credential_service/session_state_credential_service.js';
+export { CredentialExchangeError } from './auth/exchanger/base_credential_exchanger.js';
+export type { BaseCredentialExchanger, ExchangeResult, } from './auth/exchanger/base_credential_exchanger.js';
+export { OAuth2CredentialExchanger } from './auth/oauth2/oauth2_credential_exchanger.js';
+export { OAuth2DiscoveryManager } from './auth/oauth2/oauth2_discovery.js';
+export type { BaseCredentialRefresher } from './auth/refresher/base_credential_refresher.js';
+export { CredentialRefresherRegistry } from './auth/refresher/credential_refresher_registry.js';
 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 { CodeExecutionLanguage, FileContentEncoding, type CodeExecutionInput, type CodeExecutionResult, type 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';
@@ -68,7 +83,9 @@ export type { LlmRequest } from './models/llm_request.js';
 export type { LlmResponse } from './models/llm_response.js';
 export { LLMRegistry } from './models/registry.js';
 export type { BaseLlmType } from './models/registry.js';
-export { BasePlugin } from './plugins/base_plugin.js';
+export { RoutedLlm } from './models/routed_llm.js';
+export type { LlmRouter } from './models/routed_llm.js';
+export { BasePlugin, ContextCompactionTrigger } from './plugins/base_plugin.js';
 export { LoggingPlugin } from './plugins/logging_plugin.js';
 export { PluginManager } from './plugins/plugin_manager.js';
 export { InMemoryPolicyEngine, PolicyOutcome, REQUEST_CONFIRMATION_FUNCTION_CALL_NAME, SecurityPlugin, getAskUserConfirmationFunctionCalls, } from './plugins/security_plugin.js';
@@ -103,7 +120,13 @@ export { isGemini2OrAbove } from './utils/model_name.js';
 export { zodObjectToSchema } from './utils/simple_zod_to_json.js';
 export { GoogleLLMVariant } from './utils/variant_utils.js';
 export { version } from './version.js';
+export type { Frontmatter, Resources, Script, Skill } from './skills/skill.js';
+export { ListSkillsTool } from './tools/skill/list_skills_tool.js';
+export { LoadSkillResourceTool } from './tools/skill/load_skill_resource_tool.js';
+export { LoadSkillTool } from './tools/skill/load_skill_tool.js';
+export { SkillToolset } from './tools/skill/skill_toolset.js';
 export * from './artifacts/base_artifact_service.js';
+export * from './features/feature_registry.js';
 export * from './memory/base_memory_service.js';
 export * from './sessions/base_session_service.js';
 export * from './tools/base_tool.js';

package/dist/types/features/feature_registry.d.ts

@@ -0,0 +1,66 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Feature names.
+ */
+export declare enum FeatureName {
+    PROGRESSIVE_SSE_STREAMING = "PROGRESSIVE_SSE_STREAMING"
+}
+/**
+ * Feature lifecycle stages.
+ */
+export declare enum FeatureStage {
+    WIP = "wip",
+    EXPERIMENTAL = "experimental",
+    STABLE = "stable"
+}
+/**
+ * Feature configuration.
+ */
+export interface FeatureConfig {
+    stage: FeatureStage;
+    defaultOn: boolean;
+}
+/**
+ * Get the configuration of a feature from the registry.
+ *
+ * @param featureName The feature name.
+ * @returns The feature config from the registry, or undefined if not found.
+ */
+export declare function getFeatureConfig(featureName: FeatureName): FeatureConfig | undefined;
+/**
+ * Register a feature with a specific config.
+ *
+ * @param featureName The feature name.
+ * @param config The feature config to register.
+ */
+export declare function registerFeature(featureName: FeatureName, config: FeatureConfig): void;
+/**
+ * Programmatically override a feature's enabled state.
+ *
+ * This override takes highest priority, superseding environment variables
+ * and registry defaults.
+ *
+ * @param featureName The feature name to override.
+ * @param enabled Whether the feature should be enabled.
+ */
+export declare function overrideFeatureEnabled(featureName: FeatureName, enabled: boolean | undefined): void;
+/**
+ * Check if a feature is enabled at runtime.
+ *
+ * Priority order (highest to lowest):
+ * 1. Programmatic overrides
+ * 2. Environment variables (ADK_ENABLE_* / ADK_DISABLE_*)
+ * 3. Registry defaults
+ *
+ * @param featureName The feature name.
+ * @returns True if the feature is enabled, false otherwise.
+ */
+export declare function isFeatureEnabled(featureName: FeatureName): boolean;
+/**
+ * Temporarily overrides a feature for the duration of a callback.
+ */
+export declare function withTemporaryFeatureOverride<T>(featureName: FeatureName, enabled: boolean, callback: () => Promise<T> | T): Promise<T>;

package/dist/types/index.d.ts

@@ -14,9 +14,13 @@ export type { ExecutorContext } from './a2a/executor_context.js';
 export { FileArtifactService } from './artifacts/file_artifact_service.js';
 export { GcsArtifactService } from './artifacts/gcs_artifact_service.js';
 export { getArtifactServiceFromUri } from './artifacts/registry.js';
+export { UnsafeLocalCodeExecutor, type UnsafeLocalCodeExecutorOptions, } from './code_executors/unsafe_local_code_executor.js';
 export * from './common.js';
 export { DatabaseSessionService } from './sessions/database_session_service.js';
 export { getSessionServiceFromUri } from './sessions/registry.js';
+export { loadAllSkillsInDir, loadSkillFromDir, validateSkillDir, } from './skills/loader.js';
+export { RunSkillInlineScriptTool } from './tools/skill/run_skill_inline_script_tool.js';
+export { RunSkillScriptTool } from './tools/skill/run_skill_script_tool.js';
 export * from './telemetry/google_cloud.js';
 export * from './telemetry/setup.js';
 export * from './tools/mcp/mcp_session_manager.js';

package/dist/types/memory/base_memory_service.d.ts

@@ -18,8 +18,14 @@ export interface SearchMemoryResponse {
  * The parameters for `searchMemory`.
  */
 export interface SearchMemoryRequest {
+    /** The app name associated with the memory to search. */
     appName: string;
+    /** The user ID whose memory is being searched. */
     userId: string;
+    /**
+     * The natural language query used to retrieve relevant memories.
+     * Implementations may use keyword matching or semantic search.
+     */
     query: string;
 }
 /**

package/dist/types/memory/memory_entry.d.ts

@@ -5,20 +5,26 @@
  */
 import { Content } from '@google/genai';
 /**
- * Represents one memory entry.
+ * Represents one memory entry retrieved from a memory service.
+ *
+ * Memory entries are created from session events and surfaced to the agent
+ * to provide relevant context from past interactions.
  */
 export interface MemoryEntry {
     /**
-     * The content of the memory entry.
+     * The content of the memory entry, as originally produced during a session.
      */
     content: Content;
     /**
-     * The author of the memory.
+     * The author of the memory. Common values are `'user'` and `'model'`, but
+     * this can also be the name of an agent when the content was produced by a
+     * named sub-agent.
      */
     author?: string;
     /**
-     * The timestamp when the original content of this memory happened.
-     * This string will be forwarded to LLM. Preferred format is ISO 8601 format.
+     * The time when the original content was produced.
+     * Forwarded to the LLM as part of the memory context.
+     * Preferred format is ISO 8601 (e.g. `'2024-01-15T10:30:00.000Z'`).
      */
     timestamp?: string;
 }

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

@@ -54,6 +54,6 @@ export declare class ApigeeLlm extends Gemini {
     private identifyApiVersion;
     private _apigeeLiveApiVersion?;
     get liveApiVersion(): string;
-    generateContentAsync(llmRequest: LlmRequest, stream?: boolean): AsyncGenerator<LlmResponse, void>;
+    generateContentAsync(llmRequest: LlmRequest, stream?: boolean, abortSignal?: AbortSignal): AsyncGenerator<LlmResponse, void>;
     connect(llmRequest: LlmRequest): Promise<BaseLlmConnection>;
 }

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

@@ -47,7 +47,7 @@ export declare abstract class BaseLlm {
      * For non-streaming call, it will only yield one Content.
      * @return A generator of LlmResponse.
      */
-    abstract generateContentAsync(llmRequest: LlmRequest, stream?: boolean): AsyncGenerator<LlmResponse, void>;
+    abstract generateContentAsync(llmRequest: LlmRequest, stream?: boolean, abortSignal?: AbortSignal): AsyncGenerator<LlmResponse, void>;
     /**
      * Creates a live connection to the LLM.
      *

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

@@ -71,7 +71,7 @@ export declare class Gemini extends BaseLlm {
      * @param stream bool = false, whether to do streaming call.
      * @yields LlmResponse: The model response.
      */
-    generateContentAsync(llmRequest: LlmRequest, stream?: boolean): AsyncGenerator<LlmResponse, void>;
+    generateContentAsync(llmRequest: LlmRequest, stream?: boolean, abortSignal?: AbortSignal): AsyncGenerator<LlmResponse, void>;
     protected getHttpOptions(): HttpOptions;
     get apiClient(): GoogleGenAI;
     get apiBackend(): GoogleLLMVariant;

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

@@ -30,6 +30,10 @@ export interface LlmRequest {
     toolsDict: {
         [key: string]: BaseTool;
     };
+    /**
+     * The set of allowed tools, populated by request processors.
+     */
+    allowedTools?: string[];
 }
 /**
  * Appends instructions to the system instruction.

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

@@ -0,0 +1,38 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseLlm } from './base_llm.js';
+import { BaseLlmConnection } from './base_llm_connection.js';
+import { LlmRequest } from './llm_request.js';
+import { LlmResponse } from './llm_response.js';
+/**
+ * Type definition for a function that selects a model based on the request.
+ */
+export type LlmRouter = (models: Readonly<Record<string, BaseLlm>>, request: LlmRequest, errorContext?: {
+    failedKeys: ReadonlySet<string>;
+    lastError: unknown;
+}) => Promise<string | undefined> | string | undefined;
+/**
+ * A BaseLlm implementation that delegates to one of multiple models based on a router function.
+ */
+export declare class RoutedLlm extends BaseLlm {
+    private readonly models;
+    private readonly router;
+    constructor({ models, router, modelName, }: {
+        models: Readonly<Record<string, BaseLlm>> | BaseLlm[];
+        router: LlmRouter;
+        modelName?: string;
+    });
+    /**
+     * Generates content by delegating to the selected model.
+     */
+    generateContentAsync(llmRequest: LlmRequest, stream?: boolean): AsyncGenerator<LlmResponse, void>;
+    /**
+     * Creates a live connection to the LLM by delegating to the selected model.
+     * This live connection cannot be switched mid-stream, it is tied to the model
+     * selected at the time of connection.
+     */
+    connect(llmRequest: LlmRequest): Promise<BaseLlmConnection>;
+}

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

@@ -11,6 +11,13 @@ 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';
+/**
+ * Trigger for context compaction.
+ */
+export declare enum ContextCompactionTrigger {
+    Manual = "Manual",
+    Auto = "Auto"
+}
 /**
  * Base class for creating plugins.
  *
@@ -246,6 +253,48 @@ export declare abstract class BasePlugin {
         llmRequest: LlmRequest;
         error: Error;
     }): Promise<LlmResponse | undefined>;
+    /**
+     * Callback executed before a tool is selected.
+     *
+     * This callback provides an opportunity to inspect, log, or modify the
+     * available tools before they are selected.
+     *
+     * @param params.callbackContext The context for the current agent call.
+     * @param params.tools The available tools.
+     * @returns An optional value. A non-`undefined` return may be used by the
+     *     framework to modify or replace the available tools. Returning
+     *     `undefined` allows the original tools to be used.
+     */
+    beforeToolSelection(params: {
+        callbackContext: Context;
+        tools: Readonly<Record<string, BaseTool>>;
+    }): Promise<Readonly<Record<string, BaseTool>> | undefined>;
+    /**
+     * Callback executed before context compaction.
+     *
+     * This callback provides an opportunity to inspect or modify the context
+     * before it is compacted.
+     *
+     * @param params.invocationContext The context for the entire invocation.
+     * @param params.trigger The trigger for the context compaction.
+     */
+    beforeContextCompaction(params: {
+        invocationContext: InvocationContext;
+        trigger: ContextCompactionTrigger;
+    }): Promise<void>;
+    /**
+     * Callback executed after context compaction.
+     *
+     * This callback provides an opportunity to inspect the context
+     * after it has been compacted.
+     *
+     * @param params.invocationContext The context for the entire invocation.
+     * @param params.trigger The trigger for the context compaction.
+     */
+    afterContextCompaction(params: {
+        invocationContext: InvocationContext;
+        trigger: ContextCompactionTrigger;
+    }): Promise<void>;
     /**
      * Callback executed before a tool is called.
      *