@iqai/adk 0.0.9 → 0.0.10

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @iqai/adk
 
+## 0.0.10
+
+### Patch Changes
+
+- cb16b0c: Adds helper functions for creating sampling handlers
+
 ## 0.0.9
 
 ### Patch Changes

package/dist/index.d.mts

@@ -2699,6 +2699,11 @@ type McpConfig = {
         maxSize?: number;
     };
     debug?: boolean;
+    /**
+     * Sampling handler for processing MCP sampling requests.
+     * This allows MCP servers to request LLM completions through your ADK agent.
+     */
+    samplingHandler?: ADKSamplingHandler;
 };
 type McpTransportType = {
     mode: "stdio";
@@ -2732,7 +2737,7 @@ declare class McpError extends Error {
 }
 type SamplingRequest = z.infer<typeof CreateMessageRequestSchema>;
 type SamplingResponse = z.infer<typeof CreateMessageResultSchema>;
-
+type SamplingHandler = (request: SamplingRequest) => Promise<SamplingResponse>;
 /**
  * ADK sampling request format - what we pass to the user's handler
  */
@@ -2809,7 +2814,7 @@ declare class McpClientService {
     isConnected(): boolean;
     private setupSamplingHandler;
     /**
-     * Set an ADK sampling handler
+     * Set a new ADK sampling handler
      */
     setSamplingHandler(handler: ADKSamplingHandler): void;
     /**
@@ -2838,6 +2843,68 @@ declare function normalizeJsonSchema(schema: Record<string, any>): JSONSchema;
  */
 declare function mcpSchemaToParameters(mcpTool: Tool): JSONSchema;
 
+/**
+ * MCP Sampling Handler class that handles message format conversion
+ * between MCP format and ADK format
+ */
+declare class McpSamplingHandler {
+    private logger;
+    private adkHandler;
+    constructor(adkHandler: ADKSamplingHandler);
+    /**
+     * Handle MCP sampling request and convert between formats
+     */
+    handleSamplingRequest(request: SamplingRequest): Promise<SamplingResponse>;
+    /**
+     * Convert MCP messages to ADK message format
+     */
+    private convertMcpMessagesToADK;
+    /**
+     * Convert ADK response to MCP response format
+     */
+    private convertADKResponseToMcp;
+    /**
+     * Update the ADK handler
+     */
+    updateHandler(handler: ADKSamplingHandler): void;
+}
+/**
+ * Helper function to create a sampling handler with proper TypeScript types.
+ *
+ * @param handler - Function that handles sampling requests
+ * @returns Properly typed ADK sampling handler
+ *
+ * @example
+ * ```typescript
+ * import { createSamplingHandler, GoogleLLM, LLMRequest } from "@iqai/adk";
+ *
+ * const llm = new GoogleLLM("gemini-2.5-flash-preview-05-20");
+ *
+ * const samplingHandler = createSamplingHandler(async (request) => {
+ *   // request is properly typed with all the fields
+ *   const llmRequest = new LLMRequest({
+ *     messages: request.messages,
+ *     config: {
+ *       temperature: request.temperature || 0.7,
+ *       max_tokens: request.maxTokens,
+ *     }
+ *   });
+ *
+ *   const responses = [];
+ *   for await (const response of llm.generateContentAsync(llmRequest)) {
+ *     responses.push(response);
+ *   }
+ *
+ *   return {
+ *     model: llm.model,
+ *     content: responses[responses.length - 1].content,
+ *     stopReason: "endTurn"
+ *   };
+ * });
+ * ```
+ */
+declare function createSamplingHandler(handler: (request: ADKSamplingRequest) => Promise<ADKSamplingResponse>): ADKSamplingHandler;
+
 /**
  * A class for managing MCP tools similar to Python's MCPToolset.
  * Provides functionality to retrieve and use tools from an MCP server.
@@ -2858,6 +2925,17 @@ declare class McpToolset {
      * Initializes the client service and establishes a connection.
      */
     initialize(): Promise<McpClientService>;
+    /**
+     * Set a sampling handler for this MCP toolset.
+     * This allows MCP servers to request LLM completions through your ADK agent.
+     *
+     * @param handler - ADK sampling handler that receives ADK-formatted messages
+     */
+    setSamplingHandler(handler: ADKSamplingHandler): void;
+    /**
+     * Remove the sampling handler
+     */
+    removeSamplingHandler(): void;
     /**
      * Retrieves tools from the MCP server and converts them to BaseTool instances.
      * Similar to Python's get_tools method.
@@ -2897,6 +2975,9 @@ declare function getMcpTools(config: McpConfig, toolFilter?: string[] | ((tool:
  * Tools module exports
  */
 
+type index$2_ADKSamplingHandler = ADKSamplingHandler;
+type index$2_ADKSamplingRequest = ADKSamplingRequest;
+type index$2_ADKSamplingResponse = ADKSamplingResponse;
 type index$2_BaseTool = BaseTool;
 declare const index$2_BaseTool: typeof BaseTool;
 type index$2_BuildFunctionDeclarationOptions = BuildFunctionDeclarationOptions;
@@ -2920,9 +3001,12 @@ type index$2_McpError = McpError;
 declare const index$2_McpError: typeof McpError;
 type index$2_McpErrorType = McpErrorType;
 declare const index$2_McpErrorType: typeof McpErrorType;
+type index$2_McpSamplingHandler = McpSamplingHandler;
+declare const index$2_McpSamplingHandler: typeof McpSamplingHandler;
 type index$2_McpToolset = McpToolset;
 declare const index$2_McpToolset: typeof McpToolset;
 type index$2_McpTransportType = McpTransportType;
+type index$2_SamplingHandler = SamplingHandler;
 type index$2_SamplingRequest = SamplingRequest;
 type index$2_SamplingResponse = SamplingResponse;
 type index$2_ToolConfig = ToolConfig;
@@ -2935,12 +3019,13 @@ declare const index$2_UserInteractionTool: typeof UserInteractionTool;
 declare const index$2_adkToMcpToolType: typeof adkToMcpToolType;
 declare const index$2_buildFunctionDeclaration: typeof buildFunctionDeclaration;
 declare const index$2_createFunctionTool: typeof createFunctionTool;
+declare const index$2_createSamplingHandler: typeof createSamplingHandler;
 declare const index$2_getMcpTools: typeof getMcpTools;
 declare const index$2_jsonSchemaToDeclaration: typeof jsonSchemaToDeclaration;
 declare const index$2_mcpSchemaToParameters: typeof mcpSchemaToParameters;
 declare const index$2_normalizeJsonSchema: typeof normalizeJsonSchema;
 declare namespace index$2 {
-  export { index$2_BaseTool as BaseTool, type index$2_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, index$2_ExitLoopTool as ExitLoopTool, index$2_FileOperationsTool as FileOperationsTool, index$2_FunctionTool as FunctionTool, index$2_GetUserChoiceTool as GetUserChoiceTool, index$2_GoogleSearch as GoogleSearch, index$2_HttpRequestTool as HttpRequestTool, type index$2_IToolContext as IToolContext, index$2_LoadMemoryTool as LoadMemoryTool, type index$2_McpConfig as McpConfig, index$2_McpError as McpError, index$2_McpErrorType as McpErrorType, index$2_McpToolset as McpToolset, type index$2_McpTransportType as McpTransportType, type index$2_SamplingRequest as SamplingRequest, type index$2_SamplingResponse as SamplingResponse, type index$2_ToolConfig as ToolConfig, index$2_ToolContext as ToolContext, index$2_TransferToAgentTool as TransferToAgentTool, index$2_UserInteractionTool as UserInteractionTool, index$2_adkToMcpToolType as adkToMcpToolType, index$2_buildFunctionDeclaration as buildFunctionDeclaration, index$2_createFunctionTool as createFunctionTool, index$2_getMcpTools as getMcpTools, index$2_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$2_mcpSchemaToParameters as mcpSchemaToParameters, index$2_normalizeJsonSchema as normalizeJsonSchema };
+  export { type index$2_ADKSamplingHandler as ADKSamplingHandler, type index$2_ADKSamplingRequest as ADKSamplingRequest, type index$2_ADKSamplingResponse as ADKSamplingResponse, index$2_BaseTool as BaseTool, type index$2_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, index$2_ExitLoopTool as ExitLoopTool, index$2_FileOperationsTool as FileOperationsTool, index$2_FunctionTool as FunctionTool, index$2_GetUserChoiceTool as GetUserChoiceTool, index$2_GoogleSearch as GoogleSearch, index$2_HttpRequestTool as HttpRequestTool, type index$2_IToolContext as IToolContext, index$2_LoadMemoryTool as LoadMemoryTool, type index$2_McpConfig as McpConfig, index$2_McpError as McpError, index$2_McpErrorType as McpErrorType, index$2_McpSamplingHandler as McpSamplingHandler, index$2_McpToolset as McpToolset, type index$2_McpTransportType as McpTransportType, type index$2_SamplingHandler as SamplingHandler, type index$2_SamplingRequest as SamplingRequest, type index$2_SamplingResponse as SamplingResponse, type index$2_ToolConfig as ToolConfig, index$2_ToolContext as ToolContext, index$2_TransferToAgentTool as TransferToAgentTool, index$2_UserInteractionTool as UserInteractionTool, index$2_adkToMcpToolType as adkToMcpToolType, index$2_buildFunctionDeclaration as buildFunctionDeclaration, index$2_createFunctionTool as createFunctionTool, index$2_createSamplingHandler as createSamplingHandler, index$2_getMcpTools as getMcpTools, index$2_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$2_mcpSchemaToParameters as mcpSchemaToParameters, index$2_normalizeJsonSchema as normalizeJsonSchema };
 }
 
 /**
@@ -3499,4 +3584,4 @@ declare class InMemoryRunner extends Runner {
 
 declare const VERSION = "0.1.0";
 
-export { Agent, type AgentConfig, index$3 as Agents, AnthropicLLM, type AnthropicLLMConfig, AnthropicLLMConnection, ApiKeyCredential, ApiKeyScheme, type AudioTranscriptionConfig, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, BaseAgent, BaseLLM, BaseLLMConnection, type BaseMemoryService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BuildFunctionDeclarationOptions, ExitLoopTool, FileOperationsTool, type FunctionCall, type FunctionDeclaration, FunctionTool, GetUserChoiceTool, GoogleLLM, type GoogleLLMConfig, GoogleSearch, HttpRequestTool, HttpScheme, type IToolContext, type ImageContent, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, InvocationContext, type JSONSchema, LLMRegistry, LLMRequest, type LLMRequestConfig, LLMResponse, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionOptions, LoadMemoryTool, LoopAgent, type LoopAgentConfig, type McpConfig, McpError, McpErrorType, McpToolset, type McpTransportType, index$1 as Memory, type MemoryResult, type Message, type MessageContent, type MessageRole, index$4 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAILLM, type OpenAILLMConfig, OpenAILLMConnection, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PersistentMemoryService, PgLiteSessionService, PostgresSessionService, RunConfig, Runner, type SamplingRequest, type SamplingResponse, type SearchMemoryOptions, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionService, SessionState, index as Sessions, type SpeechConfig, SqliteSessionService, StreamingMode, type TextContent, type ToolCall, type ToolConfig, ToolContext, index$2 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, adkToMcpToolType, buildFunctionDeclaration, cloneSession, createFunctionTool, generateSessionId, getMcpTools, jsonSchemaToDeclaration, mcpSchemaToParameters, normalizeJsonSchema, registerProviders, validateSession };
+export { type ADKSamplingHandler, type ADKSamplingRequest, type ADKSamplingResponse, Agent, type AgentConfig, index$3 as Agents, AnthropicLLM, type AnthropicLLMConfig, AnthropicLLMConnection, ApiKeyCredential, ApiKeyScheme, type AudioTranscriptionConfig, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, BaseAgent, BaseLLM, BaseLLMConnection, type BaseMemoryService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BuildFunctionDeclarationOptions, ExitLoopTool, FileOperationsTool, type FunctionCall, type FunctionDeclaration, FunctionTool, GetUserChoiceTool, GoogleLLM, type GoogleLLMConfig, GoogleSearch, HttpRequestTool, HttpScheme, type IToolContext, type ImageContent, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, InvocationContext, type JSONSchema, LLMRegistry, LLMRequest, type LLMRequestConfig, LLMResponse, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionOptions, LoadMemoryTool, LoopAgent, type LoopAgentConfig, type McpConfig, McpError, McpErrorType, McpSamplingHandler, McpToolset, type McpTransportType, index$1 as Memory, type MemoryResult, type Message, type MessageContent, type MessageRole, index$4 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAILLM, type OpenAILLMConfig, OpenAILLMConnection, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PersistentMemoryService, PgLiteSessionService, PostgresSessionService, RunConfig, Runner, type SamplingHandler, type SamplingRequest, type SamplingResponse, type SearchMemoryOptions, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionService, SessionState, index as Sessions, type SpeechConfig, SqliteSessionService, StreamingMode, type TextContent, type ToolCall, type ToolConfig, ToolContext, index$2 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, adkToMcpToolType, buildFunctionDeclaration, cloneSession, createFunctionTool, createSamplingHandler, generateSessionId, getMcpTools, jsonSchemaToDeclaration, mcpSchemaToParameters, normalizeJsonSchema, registerProviders, validateSession };

package/dist/index.d.ts

@@ -2699,6 +2699,11 @@ type McpConfig = {
         maxSize?: number;
     };
     debug?: boolean;
+    /**
+     * Sampling handler for processing MCP sampling requests.
+     * This allows MCP servers to request LLM completions through your ADK agent.
+     */
+    samplingHandler?: ADKSamplingHandler;
 };
 type McpTransportType = {
     mode: "stdio";
@@ -2732,7 +2737,7 @@ declare class McpError extends Error {
 }
 type SamplingRequest = z.infer<typeof CreateMessageRequestSchema>;
 type SamplingResponse = z.infer<typeof CreateMessageResultSchema>;
-
+type SamplingHandler = (request: SamplingRequest) => Promise<SamplingResponse>;
 /**
  * ADK sampling request format - what we pass to the user's handler
  */
@@ -2809,7 +2814,7 @@ declare class McpClientService {
     isConnected(): boolean;
     private setupSamplingHandler;
     /**
-     * Set an ADK sampling handler
+     * Set a new ADK sampling handler
      */
     setSamplingHandler(handler: ADKSamplingHandler): void;
     /**
@@ -2838,6 +2843,68 @@ declare function normalizeJsonSchema(schema: Record<string, any>): JSONSchema;
  */
 declare function mcpSchemaToParameters(mcpTool: Tool): JSONSchema;
 
+/**
+ * MCP Sampling Handler class that handles message format conversion
+ * between MCP format and ADK format
+ */
+declare class McpSamplingHandler {
+    private logger;
+    private adkHandler;
+    constructor(adkHandler: ADKSamplingHandler);
+    /**
+     * Handle MCP sampling request and convert between formats
+     */
+    handleSamplingRequest(request: SamplingRequest): Promise<SamplingResponse>;
+    /**
+     * Convert MCP messages to ADK message format
+     */
+    private convertMcpMessagesToADK;
+    /**
+     * Convert ADK response to MCP response format
+     */
+    private convertADKResponseToMcp;
+    /**
+     * Update the ADK handler
+     */
+    updateHandler(handler: ADKSamplingHandler): void;
+}
+/**
+ * Helper function to create a sampling handler with proper TypeScript types.
+ *
+ * @param handler - Function that handles sampling requests
+ * @returns Properly typed ADK sampling handler
+ *
+ * @example
+ * ```typescript
+ * import { createSamplingHandler, GoogleLLM, LLMRequest } from "@iqai/adk";
+ *
+ * const llm = new GoogleLLM("gemini-2.5-flash-preview-05-20");
+ *
+ * const samplingHandler = createSamplingHandler(async (request) => {
+ *   // request is properly typed with all the fields
+ *   const llmRequest = new LLMRequest({
+ *     messages: request.messages,
+ *     config: {
+ *       temperature: request.temperature || 0.7,
+ *       max_tokens: request.maxTokens,
+ *     }
+ *   });
+ *
+ *   const responses = [];
+ *   for await (const response of llm.generateContentAsync(llmRequest)) {
+ *     responses.push(response);
+ *   }
+ *
+ *   return {
+ *     model: llm.model,
+ *     content: responses[responses.length - 1].content,
+ *     stopReason: "endTurn"
+ *   };
+ * });
+ * ```
+ */
+declare function createSamplingHandler(handler: (request: ADKSamplingRequest) => Promise<ADKSamplingResponse>): ADKSamplingHandler;
+
 /**
  * A class for managing MCP tools similar to Python's MCPToolset.
  * Provides functionality to retrieve and use tools from an MCP server.
@@ -2858,6 +2925,17 @@ declare class McpToolset {
      * Initializes the client service and establishes a connection.
      */
     initialize(): Promise<McpClientService>;
+    /**
+     * Set a sampling handler for this MCP toolset.
+     * This allows MCP servers to request LLM completions through your ADK agent.
+     *
+     * @param handler - ADK sampling handler that receives ADK-formatted messages
+     */
+    setSamplingHandler(handler: ADKSamplingHandler): void;
+    /**
+     * Remove the sampling handler
+     */
+    removeSamplingHandler(): void;
     /**
      * Retrieves tools from the MCP server and converts them to BaseTool instances.
      * Similar to Python's get_tools method.
@@ -2897,6 +2975,9 @@ declare function getMcpTools(config: McpConfig, toolFilter?: string[] | ((tool:
  * Tools module exports
  */
 
+type index$2_ADKSamplingHandler = ADKSamplingHandler;
+type index$2_ADKSamplingRequest = ADKSamplingRequest;
+type index$2_ADKSamplingResponse = ADKSamplingResponse;
 type index$2_BaseTool = BaseTool;
 declare const index$2_BaseTool: typeof BaseTool;
 type index$2_BuildFunctionDeclarationOptions = BuildFunctionDeclarationOptions;
@@ -2920,9 +3001,12 @@ type index$2_McpError = McpError;
 declare const index$2_McpError: typeof McpError;
 type index$2_McpErrorType = McpErrorType;
 declare const index$2_McpErrorType: typeof McpErrorType;
+type index$2_McpSamplingHandler = McpSamplingHandler;
+declare const index$2_McpSamplingHandler: typeof McpSamplingHandler;
 type index$2_McpToolset = McpToolset;
 declare const index$2_McpToolset: typeof McpToolset;
 type index$2_McpTransportType = McpTransportType;
+type index$2_SamplingHandler = SamplingHandler;
 type index$2_SamplingRequest = SamplingRequest;
 type index$2_SamplingResponse = SamplingResponse;
 type index$2_ToolConfig = ToolConfig;
@@ -2935,12 +3019,13 @@ declare const index$2_UserInteractionTool: typeof UserInteractionTool;
 declare const index$2_adkToMcpToolType: typeof adkToMcpToolType;
 declare const index$2_buildFunctionDeclaration: typeof buildFunctionDeclaration;
 declare const index$2_createFunctionTool: typeof createFunctionTool;
+declare const index$2_createSamplingHandler: typeof createSamplingHandler;
 declare const index$2_getMcpTools: typeof getMcpTools;
 declare const index$2_jsonSchemaToDeclaration: typeof jsonSchemaToDeclaration;
 declare const index$2_mcpSchemaToParameters: typeof mcpSchemaToParameters;
 declare const index$2_normalizeJsonSchema: typeof normalizeJsonSchema;
 declare namespace index$2 {
-  export { index$2_BaseTool as BaseTool, type index$2_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, index$2_ExitLoopTool as ExitLoopTool, index$2_FileOperationsTool as FileOperationsTool, index$2_FunctionTool as FunctionTool, index$2_GetUserChoiceTool as GetUserChoiceTool, index$2_GoogleSearch as GoogleSearch, index$2_HttpRequestTool as HttpRequestTool, type index$2_IToolContext as IToolContext, index$2_LoadMemoryTool as LoadMemoryTool, type index$2_McpConfig as McpConfig, index$2_McpError as McpError, index$2_McpErrorType as McpErrorType, index$2_McpToolset as McpToolset, type index$2_McpTransportType as McpTransportType, type index$2_SamplingRequest as SamplingRequest, type index$2_SamplingResponse as SamplingResponse, type index$2_ToolConfig as ToolConfig, index$2_ToolContext as ToolContext, index$2_TransferToAgentTool as TransferToAgentTool, index$2_UserInteractionTool as UserInteractionTool, index$2_adkToMcpToolType as adkToMcpToolType, index$2_buildFunctionDeclaration as buildFunctionDeclaration, index$2_createFunctionTool as createFunctionTool, index$2_getMcpTools as getMcpTools, index$2_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$2_mcpSchemaToParameters as mcpSchemaToParameters, index$2_normalizeJsonSchema as normalizeJsonSchema };
+  export { type index$2_ADKSamplingHandler as ADKSamplingHandler, type index$2_ADKSamplingRequest as ADKSamplingRequest, type index$2_ADKSamplingResponse as ADKSamplingResponse, index$2_BaseTool as BaseTool, type index$2_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, index$2_ExitLoopTool as ExitLoopTool, index$2_FileOperationsTool as FileOperationsTool, index$2_FunctionTool as FunctionTool, index$2_GetUserChoiceTool as GetUserChoiceTool, index$2_GoogleSearch as GoogleSearch, index$2_HttpRequestTool as HttpRequestTool, type index$2_IToolContext as IToolContext, index$2_LoadMemoryTool as LoadMemoryTool, type index$2_McpConfig as McpConfig, index$2_McpError as McpError, index$2_McpErrorType as McpErrorType, index$2_McpSamplingHandler as McpSamplingHandler, index$2_McpToolset as McpToolset, type index$2_McpTransportType as McpTransportType, type index$2_SamplingHandler as SamplingHandler, type index$2_SamplingRequest as SamplingRequest, type index$2_SamplingResponse as SamplingResponse, type index$2_ToolConfig as ToolConfig, index$2_ToolContext as ToolContext, index$2_TransferToAgentTool as TransferToAgentTool, index$2_UserInteractionTool as UserInteractionTool, index$2_adkToMcpToolType as adkToMcpToolType, index$2_buildFunctionDeclaration as buildFunctionDeclaration, index$2_createFunctionTool as createFunctionTool, index$2_createSamplingHandler as createSamplingHandler, index$2_getMcpTools as getMcpTools, index$2_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$2_mcpSchemaToParameters as mcpSchemaToParameters, index$2_normalizeJsonSchema as normalizeJsonSchema };
 }
 
 /**
@@ -3499,4 +3584,4 @@ declare class InMemoryRunner extends Runner {
 
 declare const VERSION = "0.1.0";
 
-export { Agent, type AgentConfig, index$3 as Agents, AnthropicLLM, type AnthropicLLMConfig, AnthropicLLMConnection, ApiKeyCredential, ApiKeyScheme, type AudioTranscriptionConfig, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, BaseAgent, BaseLLM, BaseLLMConnection, type BaseMemoryService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BuildFunctionDeclarationOptions, ExitLoopTool, FileOperationsTool, type FunctionCall, type FunctionDeclaration, FunctionTool, GetUserChoiceTool, GoogleLLM, type GoogleLLMConfig, GoogleSearch, HttpRequestTool, HttpScheme, type IToolContext, type ImageContent, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, InvocationContext, type JSONSchema, LLMRegistry, LLMRequest, type LLMRequestConfig, LLMResponse, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionOptions, LoadMemoryTool, LoopAgent, type LoopAgentConfig, type McpConfig, McpError, McpErrorType, McpToolset, type McpTransportType, index$1 as Memory, type MemoryResult, type Message, type MessageContent, type MessageRole, index$4 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAILLM, type OpenAILLMConfig, OpenAILLMConnection, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PersistentMemoryService, PgLiteSessionService, PostgresSessionService, RunConfig, Runner, type SamplingRequest, type SamplingResponse, type SearchMemoryOptions, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionService, SessionState, index as Sessions, type SpeechConfig, SqliteSessionService, StreamingMode, type TextContent, type ToolCall, type ToolConfig, ToolContext, index$2 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, adkToMcpToolType, buildFunctionDeclaration, cloneSession, createFunctionTool, generateSessionId, getMcpTools, jsonSchemaToDeclaration, mcpSchemaToParameters, normalizeJsonSchema, registerProviders, validateSession };
+export { type ADKSamplingHandler, type ADKSamplingRequest, type ADKSamplingResponse, Agent, type AgentConfig, index$3 as Agents, AnthropicLLM, type AnthropicLLMConfig, AnthropicLLMConnection, ApiKeyCredential, ApiKeyScheme, type AudioTranscriptionConfig, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, BaseAgent, BaseLLM, BaseLLMConnection, type BaseMemoryService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BuildFunctionDeclarationOptions, ExitLoopTool, FileOperationsTool, type FunctionCall, type FunctionDeclaration, FunctionTool, GetUserChoiceTool, GoogleLLM, type GoogleLLMConfig, GoogleSearch, HttpRequestTool, HttpScheme, type IToolContext, type ImageContent, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, InvocationContext, type JSONSchema, LLMRegistry, LLMRequest, type LLMRequestConfig, LLMResponse, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionOptions, LoadMemoryTool, LoopAgent, type LoopAgentConfig, type McpConfig, McpError, McpErrorType, McpSamplingHandler, McpToolset, type McpTransportType, index$1 as Memory, type MemoryResult, type Message, type MessageContent, type MessageRole, index$4 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAILLM, type OpenAILLMConfig, OpenAILLMConnection, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PersistentMemoryService, PgLiteSessionService, PostgresSessionService, RunConfig, Runner, type SamplingHandler, type SamplingRequest, type SamplingResponse, type SearchMemoryOptions, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionService, SessionState, index as Sessions, type SpeechConfig, SqliteSessionService, StreamingMode, type TextContent, type ToolCall, type ToolConfig, ToolContext, index$2 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, adkToMcpToolType, buildFunctionDeclaration, cloneSession, createFunctionTool, createSamplingHandler, generateSessionId, getMcpTools, jsonSchemaToDeclaration, mcpSchemaToParameters, normalizeJsonSchema, registerProviders, validateSession };

package/dist/index.js

@@ -2186,6 +2186,7 @@ __export(tools_exports, {
   LoadMemoryTool: () => LoadMemoryTool,
   McpError: () => McpError,
   McpErrorType: () => McpErrorType,
+  McpSamplingHandler: () => McpSamplingHandler,
   McpToolset: () => McpToolset,
   ToolContext: () => ToolContext,
   TransferToAgentTool: () => TransferToAgentTool,
@@ -2193,6 +2194,7 @@ __export(tools_exports, {
   adkToMcpToolType: () => adkToMcpToolType,
   buildFunctionDeclaration: () => buildFunctionDeclaration,
   createFunctionTool: () => createFunctionTool,
+  createSamplingHandler: () => createSamplingHandler,
   getMcpTools: () => getMcpTools,
   jsonSchemaToDeclaration: () => jsonSchemaToDeclaration,
   mcpSchemaToParameters: () => mcpSchemaToParameters,
@@ -2988,6 +2990,15 @@ var McpSamplingHandler = (_class17 = class {
    */
   async handleSamplingRequest(request) {
     try {
+      if (request.method !== "sampling/createMessage") {
+        this.logger.error(
+          `Invalid method for sampling handler: ${request.method}. Expected: sampling/createMessage`
+        );
+        throw new McpError(
+          `Invalid method: ${request.method}. This handler only processes sampling/createMessage requests.`,
+          "INVALID_REQUEST_ERROR" /* INVALID_REQUEST_ERROR */
+        );
+      }
       const validationResult = _typesjs.CreateMessageRequestSchema.safeParse(request);
       if (!validationResult.success) {
         this.logger.error(
@@ -3123,6 +3134,9 @@ var McpSamplingHandler = (_class17 = class {
     this.logger.debug("ADK sampling handler updated");
   }
 }, _class17);
+function createSamplingHandler(handler) {
+  return handler;
+}
 
 // src/tools/mcp/utils.ts
 function withRetry(fn, instance, reinitMethod, maxRetries = 1) {
@@ -3162,6 +3176,9 @@ var McpClientService = (_class18 = class {
   __init25() {this.logger = new Logger({ name: "McpClientService" })}
   constructor(config) {;_class18.prototype.__init21.call(this);_class18.prototype.__init22.call(this);_class18.prototype.__init23.call(this);_class18.prototype.__init24.call(this);_class18.prototype.__init25.call(this);
     this.config = config;
+    if (config.samplingHandler) {
+      this.mcpSamplingHandler = new McpSamplingHandler(config.samplingHandler);
+    }
   }
   /**
    * Initializes and returns an MCP client based on configuration.
@@ -3190,7 +3207,9 @@ var McpClientService = (_class18 = class {
           capabilities: {
             prompts: {},
             resources: {},
-            tools: {}
+            tools: {},
+            sampling: {}
+            // Enable sampling capability
           }
         }
       );
@@ -3396,13 +3415,13 @@ var McpClientService = (_class18 = class {
     }
   }
   /**
-   * Set an ADK sampling handler
+   * Set a new ADK sampling handler
    */
   setSamplingHandler(handler) {
     this.mcpSamplingHandler = new McpSamplingHandler(handler);
     if (this.client) {
       this.setupSamplingHandler(this.client).catch((error) => {
-        console.error("Failed to update sampling handler:", error);
+        console.error("Failed to update ADK sampling handler:", error);
       });
     }
   }
@@ -3753,6 +3772,32 @@ var McpToolset = (_class20 = class {
     await this.clientService.initialize();
     return this.clientService;
   }
+  /**
+   * Set a sampling handler for this MCP toolset.
+   * This allows MCP servers to request LLM completions through your ADK agent.
+   *
+   * @param handler - ADK sampling handler that receives ADK-formatted messages
+   */
+  setSamplingHandler(handler) {
+    if (!this.clientService) {
+      this.clientService = new McpClientService(this.config);
+    }
+    this.clientService.setSamplingHandler(handler);
+    if (this.config.debug) {
+      console.log("\u{1F3AF} Sampling handler set for MCP toolset");
+    }
+  }
+  /**
+   * Remove the sampling handler
+   */
+  removeSamplingHandler() {
+    if (this.clientService) {
+      this.clientService.removeSamplingHandler();
+      if (this.config.debug) {
+        console.log("\u{1F6AB} Sampling handler removed from MCP toolset");
+      }
+    }
+  }
   /**
    * Retrieves tools from the MCP server and converts them to BaseTool instances.
    * Similar to Python's get_tools method.
@@ -7202,4 +7247,6 @@ var VERSION = "0.1.0";
 
 
 
-exports.Agent = Agent; exports.Agents = agents_exports; exports.AnthropicLLM = AnthropicLLM; exports.AnthropicLLMConnection = AnthropicLLMConnection; exports.ApiKeyCredential = ApiKeyCredential; exports.ApiKeyScheme = ApiKeyScheme; exports.AuthConfig = AuthConfig; exports.AuthCredential = AuthCredential; exports.AuthCredentialType = AuthCredentialType; exports.AuthHandler = AuthHandler; exports.AuthScheme = AuthScheme; exports.AuthSchemeType = AuthSchemeType; exports.BaseAgent = BaseAgent; exports.BaseLLM = BaseLLM; exports.BaseLLMConnection = BaseLLMConnection; exports.BaseTool = BaseTool; exports.BasicAuthCredential = BasicAuthCredential; exports.BearerTokenCredential = BearerTokenCredential; exports.ExitLoopTool = ExitLoopTool; exports.FileOperationsTool = FileOperationsTool; exports.FunctionTool = FunctionTool; exports.GetUserChoiceTool = GetUserChoiceTool; exports.GoogleLLM = GoogleLLM; exports.GoogleSearch = GoogleSearch; exports.HttpRequestTool = HttpRequestTool; exports.HttpScheme = HttpScheme; exports.InMemoryMemoryService = InMemoryMemoryService; exports.InMemoryRunner = InMemoryRunner; exports.InMemorySessionService = InMemorySessionService; exports.InvocationContext = InvocationContext; exports.LLMRegistry = LLMRegistry; exports.LLMRequest = LLMRequest; exports.LLMResponse = LLMResponse; exports.LangGraphAgent = LangGraphAgent; exports.LoadMemoryTool = LoadMemoryTool; exports.LoopAgent = LoopAgent; exports.McpError = McpError; exports.McpErrorType = McpErrorType; exports.McpToolset = McpToolset; exports.Memory = memory_exports; exports.Models = models_exports; exports.OAuth2Credential = OAuth2Credential; exports.OAuth2Scheme = OAuth2Scheme; exports.OpenAILLM = OpenAILLM; exports.OpenAILLMConnection = OpenAILLMConnection; exports.OpenIdConnectScheme = OpenIdConnectScheme; exports.ParallelAgent = ParallelAgent; exports.PersistentMemoryService = PersistentMemoryService; exports.PgLiteSessionService = PgLiteSessionService; exports.PostgresSessionService = PostgresSessionService; exports.RunConfig = RunConfig; exports.Runner = Runner; exports.SequentialAgent = SequentialAgent; exports.SessionState = SessionState; exports.Sessions = sessions_exports; exports.SqliteSessionService = SqliteSessionService; exports.StreamingMode = StreamingMode; exports.ToolContext = ToolContext; exports.Tools = tools_exports; exports.TransferToAgentTool = TransferToAgentTool; exports.UserInteractionTool = UserInteractionTool; exports.VERSION = VERSION; exports.adkToMcpToolType = adkToMcpToolType; exports.buildFunctionDeclaration = buildFunctionDeclaration; exports.cloneSession = cloneSession; exports.createFunctionTool = createFunctionTool; exports.generateSessionId = generateSessionId; exports.getMcpTools = getMcpTools; exports.jsonSchemaToDeclaration = jsonSchemaToDeclaration; exports.mcpSchemaToParameters = mcpSchemaToParameters; exports.normalizeJsonSchema = normalizeJsonSchema; exports.registerProviders = registerProviders; exports.validateSession = validateSession;
+
+
+exports.Agent = Agent; exports.Agents = agents_exports; exports.AnthropicLLM = AnthropicLLM; exports.AnthropicLLMConnection = AnthropicLLMConnection; exports.ApiKeyCredential = ApiKeyCredential; exports.ApiKeyScheme = ApiKeyScheme; exports.AuthConfig = AuthConfig; exports.AuthCredential = AuthCredential; exports.AuthCredentialType = AuthCredentialType; exports.AuthHandler = AuthHandler; exports.AuthScheme = AuthScheme; exports.AuthSchemeType = AuthSchemeType; exports.BaseAgent = BaseAgent; exports.BaseLLM = BaseLLM; exports.BaseLLMConnection = BaseLLMConnection; exports.BaseTool = BaseTool; exports.BasicAuthCredential = BasicAuthCredential; exports.BearerTokenCredential = BearerTokenCredential; exports.ExitLoopTool = ExitLoopTool; exports.FileOperationsTool = FileOperationsTool; exports.FunctionTool = FunctionTool; exports.GetUserChoiceTool = GetUserChoiceTool; exports.GoogleLLM = GoogleLLM; exports.GoogleSearch = GoogleSearch; exports.HttpRequestTool = HttpRequestTool; exports.HttpScheme = HttpScheme; exports.InMemoryMemoryService = InMemoryMemoryService; exports.InMemoryRunner = InMemoryRunner; exports.InMemorySessionService = InMemorySessionService; exports.InvocationContext = InvocationContext; exports.LLMRegistry = LLMRegistry; exports.LLMRequest = LLMRequest; exports.LLMResponse = LLMResponse; exports.LangGraphAgent = LangGraphAgent; exports.LoadMemoryTool = LoadMemoryTool; exports.LoopAgent = LoopAgent; exports.McpError = McpError; exports.McpErrorType = McpErrorType; exports.McpSamplingHandler = McpSamplingHandler; exports.McpToolset = McpToolset; exports.Memory = memory_exports; exports.Models = models_exports; exports.OAuth2Credential = OAuth2Credential; exports.OAuth2Scheme = OAuth2Scheme; exports.OpenAILLM = OpenAILLM; exports.OpenAILLMConnection = OpenAILLMConnection; exports.OpenIdConnectScheme = OpenIdConnectScheme; exports.ParallelAgent = ParallelAgent; exports.PersistentMemoryService = PersistentMemoryService; exports.PgLiteSessionService = PgLiteSessionService; exports.PostgresSessionService = PostgresSessionService; exports.RunConfig = RunConfig; exports.Runner = Runner; exports.SequentialAgent = SequentialAgent; exports.SessionState = SessionState; exports.Sessions = sessions_exports; exports.SqliteSessionService = SqliteSessionService; exports.StreamingMode = StreamingMode; exports.ToolContext = ToolContext; exports.Tools = tools_exports; exports.TransferToAgentTool = TransferToAgentTool; exports.UserInteractionTool = UserInteractionTool; exports.VERSION = VERSION; exports.adkToMcpToolType = adkToMcpToolType; exports.buildFunctionDeclaration = buildFunctionDeclaration; exports.cloneSession = cloneSession; exports.createFunctionTool = createFunctionTool; exports.createSamplingHandler = createSamplingHandler; exports.generateSessionId = generateSessionId; exports.getMcpTools = getMcpTools; exports.jsonSchemaToDeclaration = jsonSchemaToDeclaration; exports.mcpSchemaToParameters = mcpSchemaToParameters; exports.normalizeJsonSchema = normalizeJsonSchema; exports.registerProviders = registerProviders; exports.validateSession = validateSession;

package/dist/index.mjs

@@ -2186,6 +2186,7 @@ __export(tools_exports, {
   LoadMemoryTool: () => LoadMemoryTool,
   McpError: () => McpError,
   McpErrorType: () => McpErrorType,
+  McpSamplingHandler: () => McpSamplingHandler,
   McpToolset: () => McpToolset,
   ToolContext: () => ToolContext,
   TransferToAgentTool: () => TransferToAgentTool,
@@ -2193,6 +2194,7 @@ __export(tools_exports, {
   adkToMcpToolType: () => adkToMcpToolType,
   buildFunctionDeclaration: () => buildFunctionDeclaration,
   createFunctionTool: () => createFunctionTool,
+  createSamplingHandler: () => createSamplingHandler,
   getMcpTools: () => getMcpTools,
   jsonSchemaToDeclaration: () => jsonSchemaToDeclaration,
   mcpSchemaToParameters: () => mcpSchemaToParameters,
@@ -2988,6 +2990,15 @@ var McpSamplingHandler = class {
    */
   async handleSamplingRequest(request) {
     try {
+      if (request.method !== "sampling/createMessage") {
+        this.logger.error(
+          `Invalid method for sampling handler: ${request.method}. Expected: sampling/createMessage`
+        );
+        throw new McpError(
+          `Invalid method: ${request.method}. This handler only processes sampling/createMessage requests.`,
+          "INVALID_REQUEST_ERROR" /* INVALID_REQUEST_ERROR */
+        );
+      }
       const validationResult = CreateMessageRequestSchema.safeParse(request);
       if (!validationResult.success) {
         this.logger.error(
@@ -3123,6 +3134,9 @@ var McpSamplingHandler = class {
     this.logger.debug("ADK sampling handler updated");
   }
 };
+function createSamplingHandler(handler) {
+  return handler;
+}
 
 // src/tools/mcp/utils.ts
 function withRetry(fn, instance, reinitMethod, maxRetries = 1) {
@@ -3162,6 +3176,9 @@ var McpClientService = class {
   logger = new Logger({ name: "McpClientService" });
   constructor(config) {
     this.config = config;
+    if (config.samplingHandler) {
+      this.mcpSamplingHandler = new McpSamplingHandler(config.samplingHandler);
+    }
   }
   /**
    * Initializes and returns an MCP client based on configuration.
@@ -3190,7 +3207,9 @@ var McpClientService = class {
           capabilities: {
             prompts: {},
             resources: {},
-            tools: {}
+            tools: {},
+            sampling: {}
+            // Enable sampling capability
           }
         }
       );
@@ -3396,13 +3415,13 @@ var McpClientService = class {
     }
   }
   /**
-   * Set an ADK sampling handler
+   * Set a new ADK sampling handler
    */
   setSamplingHandler(handler) {
     this.mcpSamplingHandler = new McpSamplingHandler(handler);
     if (this.client) {
       this.setupSamplingHandler(this.client).catch((error) => {
-        console.error("Failed to update sampling handler:", error);
+        console.error("Failed to update ADK sampling handler:", error);
       });
     }
   }
@@ -3753,6 +3772,32 @@ var McpToolset = class {
     await this.clientService.initialize();
     return this.clientService;
   }
+  /**
+   * Set a sampling handler for this MCP toolset.
+   * This allows MCP servers to request LLM completions through your ADK agent.
+   *
+   * @param handler - ADK sampling handler that receives ADK-formatted messages
+   */
+  setSamplingHandler(handler) {
+    if (!this.clientService) {
+      this.clientService = new McpClientService(this.config);
+    }
+    this.clientService.setSamplingHandler(handler);
+    if (this.config.debug) {
+      console.log("\u{1F3AF} Sampling handler set for MCP toolset");
+    }
+  }
+  /**
+   * Remove the sampling handler
+   */
+  removeSamplingHandler() {
+    if (this.clientService) {
+      this.clientService.removeSamplingHandler();
+      if (this.config.debug) {
+        console.log("\u{1F6AB} Sampling handler removed from MCP toolset");
+      }
+    }
+  }
   /**
    * Retrieves tools from the MCP server and converts them to BaseTool instances.
    * Similar to Python's get_tools method.
@@ -7167,6 +7212,7 @@ export {
   LoopAgent,
   McpError,
   McpErrorType,
+  McpSamplingHandler,
   McpToolset,
   memory_exports as Memory,
   models_exports as Models,
@@ -7195,6 +7241,7 @@ export {
   buildFunctionDeclaration,
   cloneSession,
   createFunctionTool,
+  createSamplingHandler,
   generateSessionId,
   getMcpTools,
   jsonSchemaToDeclaration,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@iqai/adk",
-  "version": "0.0.9",
+  "version": "0.0.10",
   "description": "Agent Development Kit for TypeScript with multi-provider LLM support",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",