@iqai/adk 0.1.19 → 0.1.21

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # @iqai/adk
 
+## 0.1.21
+
+### Patch Changes
+
+- 22c1cc6: Adds support for input and output schemas for agents, now output schema would update the instruction with the given schema to ristrict model into giving the desired output and validates it before producing output. Agent builder is wired to provide better type inference of the schema given by withOutputSchema
+- f141bc0: Improves error handling for missing models in workflows
+
+## 0.1.20
+
+### Patch Changes
+
+- 85473c7: Fix OpenAI and AI SDK LLMs not taking in the schema from MCP tools
+
 ## 0.1.19
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1,8 +1,8 @@
 import { Part, Content, Blob, SpeechConfig, AudioTranscriptionConfig, RealtimeInputConfig, ProactivityConfig, FunctionDeclaration, GroundingMetadata, GenerateContentResponseUsageMetadata, GenerateContentConfig, Schema, LiveConnectConfig, GoogleGenAI, FunctionCall } from '@google/genai';
 export { Blob, Content, FunctionDeclaration, Schema as JSONSchema } from '@google/genai';
 import { LanguageModel } from 'ai';
-import * as z from 'zod';
-import { z as z$1 } from 'zod';
+import * as zod from 'zod';
+import { z } from 'zod';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { CreateMessageRequestSchema, CreateMessageResultSchema, Tool } from '@modelcontextprotocol/sdk/types.js';
 import { Kysely, Generated } from 'kysely';
@@ -868,7 +868,7 @@ interface CreateToolConfig<T extends Record<string, any> = Record<string, never>
     /** A description of what the tool does */
     description: string;
     /** Zod schema for validating tool arguments (optional) */
-    schema?: z.ZodSchema<T>;
+    schema?: zod.ZodSchema<T>;
     /** The function to execute (can be sync or async) */
     fn: (args: T, context: ToolContext) => any;
     /** Whether the tool is a long running operation */
@@ -887,7 +887,7 @@ interface CreateToolConfigWithSchema<T extends Record<string, any>> {
     /** A description of what the tool does */
     description: string;
     /** Zod schema for validating tool arguments */
-    schema: z.ZodSchema<T>;
+    schema: zod.ZodSchema<T>;
     /** The function to execute (can be sync or async) */
     fn: (args: T, context: ToolContext) => any;
     /** Whether the tool is a long running operation */
@@ -1257,7 +1257,7 @@ declare abstract class BaseLlm {
      * @param llmRequest LlmRequest, the request to send to the LLM.
      * @returns BaseLLMConnection, the connection to the LLM.
      */
-    connect(llmRequest: LlmRequest): BaseLLMConnection;
+    connect(_llmRequest: LlmRequest): BaseLLMConnection;
 }
 
 /**
@@ -1292,6 +1292,38 @@ type InstructionProvider = (ctx: ReadonlyContext) => string | Promise<string>;
  * Union type for tools (supporting functions, tools, and toolsets)
  */
 type ToolUnion = BaseTool | ((...args: any[]) => any);
+/**
+ * Single before model callback type
+ */
+type SingleBeforeModelCallback = (callbackContext: CallbackContext, llmRequest: LlmRequest) => LlmResponse | null | Promise<LlmResponse | null>;
+/**
+ * Before model callback type (single or array)
+ */
+type BeforeModelCallback = SingleBeforeModelCallback | SingleBeforeModelCallback[];
+/**
+ * Single after model callback type
+ */
+type SingleAfterModelCallback = (callbackContext: CallbackContext, llmResponse: LlmResponse) => LlmResponse | null | Promise<LlmResponse | null>;
+/**
+ * After model callback type (single or array)
+ */
+type AfterModelCallback = SingleAfterModelCallback | SingleAfterModelCallback[];
+/**
+ * Single before tool callback type
+ */
+type SingleBeforeToolCallback = (tool: BaseTool, args: Record<string, any>, toolContext: ToolContext) => Record<string, any> | null | Promise<Record<string, any> | null>;
+/**
+ * Before tool callback type (single or array)
+ */
+type BeforeToolCallback = SingleBeforeToolCallback | SingleBeforeToolCallback[];
+/**
+ * Single after tool callback type
+ */
+type SingleAfterToolCallback = (tool: BaseTool, args: Record<string, any>, toolContext: ToolContext, toolResponse: Record<string, any>) => Record<string, any> | null | Promise<Record<string, any> | null>;
+/**
+ * After tool callback type (single or array)
+ */
+type AfterToolCallback = SingleAfterToolCallback | SingleAfterToolCallback[];
 /**
  * Configuration for LlmAgent
  */
@@ -1387,12 +1419,28 @@ interface LlmAgentConfig<T extends BaseLlm = BaseLlm> {
     /**
      * The input schema when agent is used as a tool
      */
-    inputSchema?: any;
+    inputSchema?: z.ZodSchema;
     /**
      * The output schema when agent replies
      * NOTE: when this is set, agent can ONLY reply and CANNOT use any tools
      */
-    outputSchema?: any;
+    outputSchema?: z.ZodSchema;
+    /**
+     * Callback or list of callbacks to be called before calling the LLM
+     */
+    beforeModelCallback?: BeforeModelCallback;
+    /**
+     * Callback or list of callbacks to be called after calling the LLM
+     */
+    afterModelCallback?: AfterModelCallback;
+    /**
+     * Callback or list of callbacks to be called before calling a tool
+     */
+    beforeToolCallback?: BeforeToolCallback;
+    /**
+     * Callback or list of callbacks to be called after calling a tool
+     */
+    afterToolCallback?: AfterToolCallback;
 }
 /**
  * LLM-based Agent
@@ -1467,11 +1515,27 @@ declare class LlmAgent<T extends BaseLlm = BaseLlm> extends BaseAgent {
     /**
      * The input schema when agent is used as a tool
      */
-    inputSchema?: any;
+    inputSchema?: z.ZodSchema;
     /**
      * The output schema when agent replies
      */
-    outputSchema?: any;
+    outputSchema?: z.ZodSchema;
+    /**
+     * Callback or list of callbacks to be called before calling the LLM
+     */
+    beforeModelCallback?: BeforeModelCallback;
+    /**
+     * Callback or list of callbacks to be called after calling the LLM
+     */
+    afterModelCallback?: AfterModelCallback;
+    /**
+     * Callback or list of callbacks to be called before calling a tool
+     */
+    beforeToolCallback?: BeforeToolCallback;
+    /**
+     * Callback or list of callbacks to be called after calling a tool
+     */
+    afterToolCallback?: AfterToolCallback;
     protected logger: Logger;
     /**
      * Constructor for LlmAgent
@@ -1497,6 +1561,27 @@ declare class LlmAgent<T extends BaseLlm = BaseLlm> extends BaseAgent {
      * This method is only for use by Agent Development Kit
      */
     canonicalTools(ctx?: ReadonlyContext): Promise<BaseTool[]>;
+    /**
+     * Gets the canonical before model callbacks as an array
+     */
+    get canonicalBeforeModelCallbacks(): SingleBeforeModelCallback[];
+    /**
+     * Gets the canonical after model callbacks as an array
+     */
+    get canonicalAfterModelCallbacks(): SingleAfterModelCallback[];
+    /**
+     * Gets the canonical before tool callbacks as an array
+     */
+    get canonicalBeforeToolCallbacks(): SingleBeforeToolCallback[];
+    /**
+     * Gets the canonical after tool callbacks as an array
+     */
+    get canonicalAfterToolCallbacks(): SingleAfterToolCallback[];
+    /**
+     * Validates output schema configuration
+     * This matches the Python implementation's __check_output_schema
+     */
+    private validateOutputSchemaConfig;
     /**
      * Gets the appropriate LLM flow for this agent
      * This matches the Python implementation's _llm_flow property
@@ -2006,8 +2091,8 @@ declare class McpError extends Error {
     originalError?: Error;
     constructor(message: string, type: McpErrorType, originalError?: Error);
 }
-type McpSamplingRequest = z$1.infer<typeof CreateMessageRequestSchema>;
-type McpSamplingResponse = z$1.infer<typeof CreateMessageResultSchema>;
+type McpSamplingRequest = z.infer<typeof CreateMessageRequestSchema>;
+type McpSamplingResponse = z.infer<typeof CreateMessageResultSchema>;
 type SamplingHandler = (request: LlmRequest) => Promise<string | LlmResponse>;
 
 declare class McpClientService {
@@ -2577,76 +2662,46 @@ declare class LlmRequest {
 }
 
 /**
- * Google LLM Variant enum
- */
-declare enum GoogleLLMVariant {
-    VERTEX_AI = "VERTEX_AI",
-    GEMINI_API = "GEMINI_API"
-}
-/**
- * Integration for Gemini models.
+ * AI SDK integration that accepts a pre-configured LanguageModel.
+ * Enables ADK to work with any provider supported by Vercel's AI SDK.
  */
-declare class GoogleLlm extends BaseLlm {
-    private _apiClient?;
-    private _liveApiClient?;
-    private _apiBackend?;
-    private _trackingHeaders?;
+declare class AiSdkLlm extends BaseLlm {
+    private modelInstance;
+    protected logger: Logger;
     /**
-     * Constructor for Gemini
+     * Constructor accepts a pre-configured LanguageModel instance
+     * @param model - Pre-configured LanguageModel from provider(modelName)
      */
-    constructor(model?: string);
+    constructor(modelInstance: LanguageModel);
     /**
-     * Provides the list of supported models.
+     * Returns empty array - following Python ADK pattern
      */
     static supportedModels(): string[];
+    protected generateContentAsyncImpl(request: LlmRequest, stream?: boolean): AsyncGenerator<LlmResponse, void, unknown>;
     /**
-     * Main content generation method - handles both streaming and non-streaming
-     */
-    protected generateContentAsyncImpl(llmRequest: LlmRequest, stream?: boolean): AsyncGenerator<LlmResponse, void, unknown>;
-    /**
-     * Connects to the Gemini model and returns an llm connection.
-     */
-    connect(_llmRequest: LlmRequest): BaseLLMConnection;
-    /**
-     * Check if response has inline data
-     */
-    private hasInlineData;
-    /**
-     * Convert LlmRequest contents to GoogleGenAI format
-     */
-    private convertContents;
-    /**
-     * Preprocesses the request based on the API backend.
-     */
-    private preprocessRequest;
-    /**
-     * Sets display_name to null for the Gemini API (non-Vertex) backend.
-     */
-    private removeDisplayNameIfPresent;
-    /**
-     * Builds function declaration log string.
+     * Convert ADK LlmRequest to AI SDK CoreMessage format
      */
-    private buildFunctionDeclarationLog;
+    private convertToAiSdkMessages;
     /**
-     * Provides the api client.
+     * Transform JSON schema to use lowercase types for AI SDK compatibility
      */
-    get apiClient(): GoogleGenAI;
+    private transformSchemaForAiSdk;
     /**
-     * Gets the API backend type.
+     * Convert ADK tools to AI SDK tools format
      */
-    get apiBackend(): GoogleLLMVariant;
+    private convertToAiSdkTools;
     /**
-     * Gets the tracking headers.
+     * Convert ADK Content to AI SDK CoreMessage
      */
-    get trackingHeaders(): Record<string, string>;
+    private contentToAiSdkMessage;
     /**
-     * Gets the live API version.
+     * Map ADK role to AI SDK role
      */
-    get liveApiVersion(): string;
+    private mapRole;
     /**
-     * Gets the live API client.
+     * Map AI SDK finish reason to ADK finish reason
      */
-    get liveApiClient(): GoogleGenAI;
+    private mapFinishReason;
 }
 
 /**
@@ -2709,6 +2764,79 @@ declare class AnthropicLlm extends BaseLlm {
     private get client();
 }
 
+/**
+ * Google LLM Variant enum
+ */
+declare enum GoogleLLMVariant {
+    VERTEX_AI = "VERTEX_AI",
+    GEMINI_API = "GEMINI_API"
+}
+/**
+ * Integration for Gemini models.
+ */
+declare class GoogleLlm extends BaseLlm {
+    private _apiClient?;
+    private _liveApiClient?;
+    private _apiBackend?;
+    private _trackingHeaders?;
+    /**
+     * Constructor for Gemini
+     */
+    constructor(model?: string);
+    /**
+     * Provides the list of supported models.
+     */
+    static supportedModels(): string[];
+    /**
+     * Main content generation method - handles both streaming and non-streaming
+     */
+    protected generateContentAsyncImpl(llmRequest: LlmRequest, stream?: boolean): AsyncGenerator<LlmResponse, void, unknown>;
+    /**
+     * Connects to the Gemini model and returns an llm connection.
+     */
+    connect(_llmRequest: LlmRequest): BaseLLMConnection;
+    /**
+     * Check if response has inline data
+     */
+    private hasInlineData;
+    /**
+     * Convert LlmRequest contents to GoogleGenAI format
+     */
+    private convertContents;
+    /**
+     * Preprocesses the request based on the API backend.
+     */
+    private preprocessRequest;
+    /**
+     * Sets display_name to null for the Gemini API (non-Vertex) backend.
+     */
+    private removeDisplayNameIfPresent;
+    /**
+     * Builds function declaration log string.
+     */
+    private buildFunctionDeclarationLog;
+    /**
+     * Provides the api client.
+     */
+    get apiClient(): GoogleGenAI;
+    /**
+     * Gets the API backend type.
+     */
+    get apiBackend(): GoogleLLMVariant;
+    /**
+     * Gets the tracking headers.
+     */
+    get trackingHeaders(): Record<string, string>;
+    /**
+     * Gets the live API version.
+     */
+    get liveApiVersion(): string;
+    /**
+     * Gets the live API client.
+     */
+    get liveApiClient(): GoogleGenAI;
+}
+
 /**
  * OpenAI LLM implementation using GPT models
  * Enhanced with comprehensive debug logging similar to Google LLM
@@ -2747,6 +2875,10 @@ declare class OpenAiLlm extends BaseLlm {
      * Convert ADK Part to OpenAI message content
      */
     private partToOpenAiContent;
+    /**
+     * Transform JSON schema to use lowercase types for OpenAI compatibility
+     */
+    private transformSchemaForOpenAi;
     /**
      * Convert ADK function declaration to OpenAI tool
      */
@@ -2782,45 +2914,6 @@ declare class OpenAiLlm extends BaseLlm {
     private get client();
 }
 
-/**
- * AI SDK integration that accepts a pre-configured LanguageModel.
- * Enables ADK to work with any provider supported by Vercel's AI SDK.
- */
-declare class AiSdkLlm extends BaseLlm {
-    private modelInstance;
-    protected logger: Logger;
-    /**
-     * Constructor accepts a pre-configured LanguageModel instance
-     * @param model - Pre-configured LanguageModel from provider(modelName)
-     */
-    constructor(modelInstance: LanguageModel);
-    /**
-     * Returns empty array - following Python ADK pattern
-     */
-    static supportedModels(): string[];
-    protected generateContentAsyncImpl(request: LlmRequest, stream?: boolean): AsyncGenerator<LlmResponse, void, unknown>;
-    /**
-     * Convert ADK LlmRequest to AI SDK CoreMessage format
-     */
-    private convertToAiSdkMessages;
-    /**
-     * Convert ADK tools to AI SDK tools format
-     */
-    private convertToAiSdkTools;
-    /**
-     * Convert ADK Content to AI SDK CoreMessage
-     */
-    private contentToAiSdkMessage;
-    /**
-     * Map ADK role to AI SDK role
-     */
-    private mapRole;
-    /**
-     * Map AI SDK finish reason to ADK finish reason
-     */
-    private mapFinishReason;
-}
-
 /**
  * Type for LLM constructor with static methods
  */
@@ -3943,6 +4036,8 @@ interface AgentBuilderConfig {
     nodes?: LangGraphNode[];
     rootNode?: string;
     outputKey?: string;
+    inputSchema?: zod.ZodSchema;
+    outputSchema?: zod.ZodSchema;
 }
 /**
  * Session configuration options
@@ -3968,26 +4063,35 @@ interface FullMessage extends Content {
 /**
  * Enhanced runner interface with simplified API
  */
-interface EnhancedRunner {
-    ask(message: string | FullMessage | LlmRequest): Promise<string>;
+interface EnhancedRunner<T = string> {
+    ask(message: string | FullMessage | LlmRequest): Promise<T>;
     runAsync(params: {
         userId: string;
         sessionId: string;
         newMessage: FullMessage;
     }): AsyncIterable<Event>;
+    __outputSchema?: zod.ZodSchema;
 }
 /**
  * Built agent result containing the agent and runner/session
  */
-interface BuiltAgent {
+interface BuiltAgent<T = string> {
     agent: BaseAgent;
-    runner: EnhancedRunner;
+    runner: EnhancedRunner<T>;
     session: Session;
 }
 /**
  * Agent types that can be built
  */
 type AgentType = "llm" | "sequential" | "parallel" | "loop" | "langgraph";
+/**
+ * AgentBuilder with typed output schema
+ */
+interface AgentBuilderWithSchema<T> extends Omit<AgentBuilder, 'build' | 'ask'> {
+    build(): Promise<BuiltAgent<T>>;
+    buildWithSchema<U = T>(): Promise<BuiltAgent<U>>;
+    ask(message: string | FullMessage): Promise<T>;
+}
 /**
  * AgentBuilder - A fluent interface for creating AI agents with automatic session management
  *
@@ -4074,6 +4178,8 @@ declare class AgentBuilder {
      * @returns This builder instance for chaining
      */
     withInstruction(instruction: string): this;
+    withInputSchema(schema: zod.ZodSchema): this;
+    withOutputSchema<T>(schema: zod.ZodType<T>): AgentBuilderWithSchema<T>;
     /**
      * Add tools to the agent
      * @param tools Tools to add to the agent
@@ -4179,7 +4285,12 @@ declare class AgentBuilder {
      * Build the agent and optionally create runner and session
      * @returns Built agent with optional runner and session
      */
-    build(): Promise<BuiltAgent>;
+    build<T = string>(): Promise<BuiltAgent<T>>;
+    /**
+     * Type-safe build method for agents with output schemas
+     * Provides better type inference for the ask method return type
+     */
+    buildWithSchema<T>(): Promise<BuiltAgent<T>>;
     /**
      * Quick execution helper - build and run a message
      * @param message Message to send to the agent (string or full message object)
@@ -4202,7 +4313,7 @@ declare class AgentBuilder {
      */
     private generateDefaultAppName;
     /**
-     * Create enhanced runner with simplified API
+     * Create enhanced runner with simplified API and proper typing
      * @param baseRunner The base runner instance
      * @param session The session instance
      * @returns Enhanced runner with simplified API
@@ -4211,17 +4322,22 @@ declare class AgentBuilder {
 }
 
 type index$4_AfterAgentCallback = AfterAgentCallback;
+type index$4_AfterModelCallback = AfterModelCallback;
+type index$4_AfterToolCallback = AfterToolCallback;
 type index$4_AgentBuilder = AgentBuilder;
 declare const index$4_AgentBuilder: typeof AgentBuilder;
 type index$4_AgentBuilderConfig = AgentBuilderConfig;
+type index$4_AgentBuilderWithSchema<T> = AgentBuilderWithSchema<T>;
 type index$4_AgentType = AgentType;
 type index$4_BaseAgent = BaseAgent;
 declare const index$4_BaseAgent: typeof BaseAgent;
 type index$4_BeforeAgentCallback = BeforeAgentCallback;
-type index$4_BuiltAgent = BuiltAgent;
+type index$4_BeforeModelCallback = BeforeModelCallback;
+type index$4_BeforeToolCallback = BeforeToolCallback;
+type index$4_BuiltAgent<T = string> = BuiltAgent<T>;
 type index$4_CallbackContext = CallbackContext;
 declare const index$4_CallbackContext: typeof CallbackContext;
-type index$4_EnhancedRunner = EnhancedRunner;
+type index$4_EnhancedRunner<T = string> = EnhancedRunner<T>;
 type index$4_FullMessage = FullMessage;
 type index$4_InstructionProvider = InstructionProvider;
 type index$4_InvocationContext = InvocationContext;
@@ -4250,7 +4366,11 @@ type index$4_SequentialAgent = SequentialAgent;
 declare const index$4_SequentialAgent: typeof SequentialAgent;
 type index$4_SequentialAgentConfig = SequentialAgentConfig;
 type index$4_SessionOptions = SessionOptions;
+type index$4_SingleAfterModelCallback = SingleAfterModelCallback;
+type index$4_SingleAfterToolCallback = SingleAfterToolCallback;
 type index$4_SingleAgentCallback = SingleAgentCallback;
+type index$4_SingleBeforeModelCallback = SingleBeforeModelCallback;
+type index$4_SingleBeforeToolCallback = SingleBeforeToolCallback;
 type index$4_StreamingMode = StreamingMode;
 declare const index$4_StreamingMode: typeof StreamingMode;
 type index$4_ToolUnion = ToolUnion;
@@ -4258,7 +4378,7 @@ declare const index$4_createBranchContextForSubAgent: typeof createBranchContext
 declare const index$4_mergeAgentRun: typeof mergeAgentRun;
 declare const index$4_newInvocationContextId: typeof newInvocationContextId;
 declare namespace index$4 {
-  export { type index$4_AfterAgentCallback as AfterAgentCallback, LlmAgent as Agent, index$4_AgentBuilder as AgentBuilder, type index$4_AgentBuilderConfig as AgentBuilderConfig, type index$4_AgentType as AgentType, index$4_BaseAgent as BaseAgent, type index$4_BeforeAgentCallback as BeforeAgentCallback, type index$4_BuiltAgent as BuiltAgent, index$4_CallbackContext as CallbackContext, type index$4_EnhancedRunner as EnhancedRunner, type index$4_FullMessage as FullMessage, type index$4_InstructionProvider as InstructionProvider, index$4_InvocationContext as InvocationContext, index$4_LangGraphAgent as LangGraphAgent, type index$4_LangGraphAgentConfig as LangGraphAgentConfig, type index$4_LangGraphNode as LangGraphNode, index$4_LlmAgent as LlmAgent, type index$4_LlmAgentConfig as LlmAgentConfig, index$4_LlmCallsLimitExceededError as LlmCallsLimitExceededError, index$4_LoopAgent as LoopAgent, type index$4_LoopAgentConfig as LoopAgentConfig, type index$4_MessagePart as MessagePart, index$4_ParallelAgent as ParallelAgent, type index$4_ParallelAgentConfig as ParallelAgentConfig, index$4_ReadonlyContext as ReadonlyContext, index$4_RunConfig as RunConfig, index$4_SequentialAgent as SequentialAgent, type index$4_SequentialAgentConfig as SequentialAgentConfig, type index$4_SessionOptions as SessionOptions, type index$4_SingleAgentCallback as SingleAgentCallback, index$4_StreamingMode as StreamingMode, type index$4_ToolUnion as ToolUnion, index$4_createBranchContextForSubAgent as createBranchContextForSubAgent, index$4_mergeAgentRun as mergeAgentRun, index$4_newInvocationContextId as newInvocationContextId };
+  export { type index$4_AfterAgentCallback as AfterAgentCallback, type index$4_AfterModelCallback as AfterModelCallback, type index$4_AfterToolCallback as AfterToolCallback, LlmAgent as Agent, index$4_AgentBuilder as AgentBuilder, type index$4_AgentBuilderConfig as AgentBuilderConfig, type index$4_AgentBuilderWithSchema as AgentBuilderWithSchema, type index$4_AgentType as AgentType, index$4_BaseAgent as BaseAgent, type index$4_BeforeAgentCallback as BeforeAgentCallback, type index$4_BeforeModelCallback as BeforeModelCallback, type index$4_BeforeToolCallback as BeforeToolCallback, type index$4_BuiltAgent as BuiltAgent, index$4_CallbackContext as CallbackContext, type index$4_EnhancedRunner as EnhancedRunner, type index$4_FullMessage as FullMessage, type index$4_InstructionProvider as InstructionProvider, index$4_InvocationContext as InvocationContext, index$4_LangGraphAgent as LangGraphAgent, type index$4_LangGraphAgentConfig as LangGraphAgentConfig, type index$4_LangGraphNode as LangGraphNode, index$4_LlmAgent as LlmAgent, type index$4_LlmAgentConfig as LlmAgentConfig, index$4_LlmCallsLimitExceededError as LlmCallsLimitExceededError, index$4_LoopAgent as LoopAgent, type index$4_LoopAgentConfig as LoopAgentConfig, type index$4_MessagePart as MessagePart, index$4_ParallelAgent as ParallelAgent, type index$4_ParallelAgentConfig as ParallelAgentConfig, index$4_ReadonlyContext as ReadonlyContext, index$4_RunConfig as RunConfig, index$4_SequentialAgent as SequentialAgent, type index$4_SequentialAgentConfig as SequentialAgentConfig, type index$4_SessionOptions as SessionOptions, type index$4_SingleAfterModelCallback as SingleAfterModelCallback, type index$4_SingleAfterToolCallback as SingleAfterToolCallback, type index$4_SingleAgentCallback as SingleAgentCallback, type index$4_SingleBeforeModelCallback as SingleBeforeModelCallback, type index$4_SingleBeforeToolCallback as SingleBeforeToolCallback, index$4_StreamingMode as StreamingMode, type index$4_ToolUnion as ToolUnion, index$4_createBranchContextForSubAgent as createBranchContextForSubAgent, index$4_mergeAgentRun as mergeAgentRun, index$4_newInvocationContextId as newInvocationContextId };
 }
 
 /**
@@ -5200,4 +5320,4 @@ declare const traceLlmCall: (invocationContext: InvocationContext, eventId: stri
 
 declare const VERSION = "0.1.0";
 
-export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, AgentTool, type AgentToolConfig, type AgentType, index$4 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, type BaseAgentType, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, Event, EventActions, index$1 as Events, ExitLoopTool, type File, FileOperationsTool, index as Flows, type FullMessage, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, InvocationContext, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntents, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, index$3 as Memory, type MessagePart, index$5 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PlanReActPlanner, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RunConfig, Runner, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionOptions, index$2 as Sessions, type SingleAgentCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$6 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, createAuthToolArguments, createBranchContextForSubAgent, createDatabaseSessionService, createFunctionTool, createMysqlSessionService, createPostgresSessionService, createSamplingHandler, createSqliteSessionService, createTool, generateAuthEvent, generateClientFunctionCallId, getLongRunningFunctionCalls, getMcpTools, handleFunctionCallsAsync, handleFunctionCallsLive, requestProcessor$5 as identityRequestProcessor, initializeTelemetry, injectSessionState, requestProcessor$4 as instructionsRequestProcessor, isEnhancedAuthConfig, jsonSchemaToDeclaration, mcpSchemaToParameters, mergeAgentRun, mergeParallelFunctionResponseEvents, newInvocationContextId, requestProcessor$1 as nlPlanningRequestProcessor, responseProcessor$1 as nlPlanningResponseProcessor, normalizeJsonSchema, populateClientFunctionCallId, registerProviders, removeClientFunctionCallId, requestProcessor$7 as requestProcessor, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };
+export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, type AfterModelCallback, type AfterToolCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentBuilderWithSchema, AgentTool, type AgentToolConfig, type AgentType, index$4 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, type BaseAgentType, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BeforeModelCallback, type BeforeToolCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, Event, EventActions, index$1 as Events, ExitLoopTool, type File, FileOperationsTool, index as Flows, type FullMessage, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, InvocationContext, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntents, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, index$3 as Memory, type MessagePart, index$5 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PlanReActPlanner, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RunConfig, Runner, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionOptions, index$2 as Sessions, type SingleAfterModelCallback, type SingleAfterToolCallback, type SingleAgentCallback, type SingleBeforeModelCallback, type SingleBeforeToolCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$6 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, createAuthToolArguments, createBranchContextForSubAgent, createDatabaseSessionService, createFunctionTool, createMysqlSessionService, createPostgresSessionService, createSamplingHandler, createSqliteSessionService, createTool, generateAuthEvent, generateClientFunctionCallId, getLongRunningFunctionCalls, getMcpTools, handleFunctionCallsAsync, handleFunctionCallsLive, requestProcessor$5 as identityRequestProcessor, initializeTelemetry, injectSessionState, requestProcessor$4 as instructionsRequestProcessor, isEnhancedAuthConfig, jsonSchemaToDeclaration, mcpSchemaToParameters, mergeAgentRun, mergeParallelFunctionResponseEvents, newInvocationContextId, requestProcessor$1 as nlPlanningRequestProcessor, responseProcessor$1 as nlPlanningResponseProcessor, normalizeJsonSchema, populateClientFunctionCallId, registerProviders, removeClientFunctionCallId, requestProcessor$7 as requestProcessor, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };