@iqai/adk 0.0.6 → 0.0.8

package/CHANGELOG.md

@@ -1,13 +1,13 @@
 # @iqai/adk
 
-## 0.0.6
+## 0.0.8
 
 ### Patch Changes
 
-- 54902c8: Adds debug logs
+- 74ee653: Simplified sampling handler with adk message and response typings
 
-## 0.0.5
+## 0.0.7
 
 ### Patch Changes
 
-- 5448661: Updates pglite session to take pglite instance as input instead of drizzle
+- 35cb95d: Adds MCP Sampling request handling

package/README.md

@@ -0,0 +1,177 @@
+# @iqai/adk - Agent Development Kit Core
+
+`@iqai/adk` is the core TypeScript library for the Agent Development Kit, providing the foundational tools and abstractions to build sophisticated AI agents. It enables seamless integration with multiple Large Language Models (LLMs), advanced tool usage, and persistent memory capabilities.
+
+<p align="center">
+  <a href="https://www.npmjs.com/package/@iqai/adk">
+    <img src="https://img.shields.io/npm/v/@iqai/adk" alt="NPM Version" />
+  </a>
+  <a href="https://www.npmjs.com/package/@iqai/adk">
+    <img src="https://img.shields.io/npm/dm/@iqai/adk" alt="NPM Downloads" />
+  </a>
+  <a href="https://github.com/IQAIcom/adk-ts/blob/main/LICENSE.md">
+    <img src="https://img.shields.io/npm/l/@iqai/adk" alt="License" />
+  </a>
+  <a href="https://github.com/IQAIcom/adk-ts">
+    <img src="https://img.shields.io/github/stars/IQAIcom/adk-ts?style=social" alt="GitHub Stars" />
+  </a>
+</p>
+
+## 🚀 Core Features
+
+The `@iqai/adk` package empowers your AI agent development with:
+
+*   **Multi-Provider LLM Support:** Flexibly integrate and switch between leading LLM providers like OpenAI, Anthropic, and Google.
+*   **Extensible Tool System:** Define and utilize custom tools with declarative schemas, allowing LLMs to intelligently leverage external functionalities.
+*   **Advanced Agent Reasoning Loop:** A complete reasoning loop implementation for complex task execution and iterative problem-solving.
+*   **Real-Time Streaming:** Support for streaming responses from LLMs for dynamic user interactions.
+*   **Flexible Authentication:** Mechanisms for securing agent API access.
+*   **Persistent Memory Systems:** Capabilities for agents to retain context and learn from past interactions.
+
+## 🚦 Installation
+
+Install the `@iqai/adk` package using your preferred package manager:
+
+```bash
+# Using npm
+npm install @iqai/adk
+
+# Using yarn
+yarn add @iqai/adk
+
+# Using pnpm
+pnpm add @iqai/adk
+```
+
+## ⚙️ Environment Configuration
+
+For the library to connect to LLM providers, you need to set up API keys. Create a `.env` file in the root of your project and add your keys:
+
+```env
+OPENAI_API_KEY=your_openai_api_key
+ANTHROPIC_API_KEY=your_anthropic_api_key
+GOOGLE_API_KEY=your_google_api_key
+```
+The library uses `dotenv` to load these variables automatically if `dotenv.config()` is called in your application.
+
+## 📖 Basic Usage
+
+Here's a fundamental example of creating and running an agent:
+
+```typescript
+import { Agent } from '@iqai/adk';
+import dotenv from 'dotenv';
+
+// Load environment variables
+dotenv.config();
+
+// Instantiate the agent
+const myAgent = new Agent({
+  name: "simple_query_assistant",
+  model: "gemini-2.5-flash-preview-05-20", // Or "gpt-4-turbo", "claude-3-opus"
+  description: "A basic assistant to answer questions.",
+  instructions: "You are a helpful AI. Respond clearly and concisely."
+});
+
+// Asynchronously run the agent
+async function runQuery() {
+  try {
+    const query = "What is the capital of France?";
+    console.log(`User: ${query}`);
+
+    const response = await myAgent.run({
+      messages: [{ role: 'user', content: query }]
+    });
+
+    console.log(`Agent: ${response.content}`);
+  } catch (error) {
+    console.error("Error during agent execution:", error);
+  }
+}
+
+runQuery();
+```
+
+## 🛠️ Using Tools with an Agent
+
+Extend your agent's capabilities by defining and integrating custom tools.
+
+```typescript
+import { Agent, BaseTool } from '@iqai/adk';
+import dotenv from 'dotenv';
+
+dotenv.config();
+
+// Define a simple calculator tool
+class CalculatorTool extends BaseTool {
+  constructor() {
+    super({
+      name: 'calculator',
+      description: 'Performs basic arithmetic operations: add, subtract, multiply, divide.'
+    });
+  }
+
+  getDeclaration() {
+    return {
+      name: this.name,
+      description: this.description,
+      parameters: {
+        type: 'object',
+        properties: {
+          operation: {
+            type: 'string',
+            enum: ['add', 'subtract', 'multiply', 'divide']
+          },
+          operand1: { type: 'number' },
+          operand2: { type: 'number' }
+        },
+        required: ['operation', 'operand1', 'operand2']
+      }
+    };
+  }
+
+  async runAsync(args: { operation: string; operand1: number; operand2: number }) {
+    const { operation, operand1, operand2 } = args;
+    switch (operation) {
+      case 'add': return { result: operand1 + operand2 };
+      case 'subtract': return { result: operand1 - operand2 };
+      case 'multiply': return { result: operand1 * operand2 };
+      case 'divide':
+        if (operand2 === 0) return { error: 'Cannot divide by zero.' };
+        return { result: operand1 / operand2 };
+      default: return { error: `Unknown operation: ${operation}` };
+    }
+  }
+}
+
+// Create an agent equipped with the calculator tool
+const mathAgent = new Agent({
+  name: "math_assistant_agent",
+  model: "gpt-4-turbo", // Choose a model proficient with tool usage
+  instructions:
+    "You are a helpful assistant. Use the calculator tool for any mathematical calculations requested.",
+  tools: [new CalculatorTool()]
+});
+
+async function performCalculation() {
+  const response = await mathAgent.run({
+    messages: [
+      { role: 'user', content: 'What is 15 multiplied by 4?' }
+    ]
+  });
+  // The response.content will likely include the thought process and the tool's output.
+  console.log(JSON.stringify(response, null, 2));
+}
+
+performCalculation().catch(console.error);
+```
+
+More detailed examples and advanced usage patterns can be found in the `apps/examples` directory of the main [ADK TypeScript repository](https://github.com/IQAIcom/adk-ts).
+
+## 🤝 Contributing
+
+While this README focuses on the `@iqai/adk` package, contributions to the overall ADK TypeScript project are welcome. Please see the [Contributing Guide](https://github.com/IQAIcom/adk-ts/blob/main/CONTRIBUTING.md) in the main repository for details on how to contribute.
+
+## 📜 License
+
+This package is licensed under the [MIT License](https://github.com/IQAIcom/adk-ts/blob/main/LICENSE.md).

package/dist/index.d.mts

@@ -1,5 +1,6 @@
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
-import { Tool } from '@modelcontextprotocol/sdk/types.js';
+import { CreateMessageRequestSchema, CreateMessageResultSchema, Tool } from '@modelcontextprotocol/sdk/types.js';
+import { z } from 'zod';
 import { AxiosInstance } from 'axios';
 import OpenAI from 'openai';
 import * as drizzle_orm_pg_core from 'drizzle-orm/pg-core';
@@ -440,6 +441,7 @@ declare class AnthropicLLM extends BaseLLM {
      * Default parameters for requests
      */
     private defaultParams;
+    private logger;
     /**
      * Constructor for AnthropicLLM
      */
@@ -516,6 +518,7 @@ declare class AnthropicLLMConnection extends BaseLLMConnection {
     private responseCallback?;
     private errorCallback?;
     private endCallback?;
+    private logger;
     /**
      * Constructor
      */
@@ -712,6 +715,7 @@ declare class OpenAILLM extends BaseLLM {
      * Default parameters for requests
      */
     private defaultParams;
+    private logger;
     /**
      * Constructor for OpenAILLM
      */
@@ -1995,6 +1999,7 @@ declare class Agent extends BaseAgent {
      * The minimum relevance score for memory augmentation (0-1)
      */
     private memoryRelevanceThreshold;
+    private logger;
     /**
      * Constructor for Agent
      */
@@ -2069,6 +2074,7 @@ interface EnhancedLLMResponse extends LLMResponse {
  * Each sub-agent's output becomes input to the next agent
  */
 declare class SequentialAgent extends BaseAgent {
+    private logger;
     /**
      * Constructor for SequentialAgent
      */
@@ -2113,6 +2119,7 @@ interface ParallelAgentConfig {
  * All sub-agents execute independently with the same input
  */
 declare class ParallelAgent extends BaseAgent {
+    private logger;
     /**
      * Constructor for ParallelAgent
      */
@@ -2181,6 +2188,7 @@ declare class LoopAgent extends BaseAgent {
      * Custom condition check function
      */
     private conditionCheck?;
+    private logger;
     /**
      * Constructor for LoopAgent
      */
@@ -2278,6 +2286,7 @@ declare class LangGraphAgent extends BaseAgent {
      * Results from node executions
      */
     private results;
+    private logger;
     /**
      * Constructor for LangGraphAgent
      */
@@ -2439,6 +2448,7 @@ declare function createFunctionTool(func: (...args: any[]) => any, options?: {
  * Simple GoogleSearch tool implementation
  */
 declare class GoogleSearch extends BaseTool {
+    private logger;
     /**
      * Constructor for GoogleSearch
      */
@@ -2591,6 +2601,7 @@ declare module "./tool-context" {
  * Tool that allows an agent to exit the current execution loop
  */
 declare class ExitLoopTool extends BaseTool {
+    private logger;
     /**
      * Constructor for ExitLoopTool
      */
@@ -2609,6 +2620,7 @@ declare class ExitLoopTool extends BaseTool {
  * Tool that allows an agent to get a choice from the user
  */
 declare class GetUserChoiceTool extends BaseTool {
+    private logger;
     /**
      * Constructor for GetUserChoiceTool
      */
@@ -2632,6 +2644,7 @@ declare class GetUserChoiceTool extends BaseTool {
  * Tool that allows an agent to transfer control to another agent
  */
 declare class TransferToAgentTool extends BaseTool {
+    private logger;
     /**
      * Constructor for TransferToAgentTool
      */
@@ -2652,6 +2665,7 @@ declare class TransferToAgentTool extends BaseTool {
  * Tool that allows an agent to load memories relevant to a query
  */
 declare class LoadMemoryTool extends BaseTool {
+    private logger;
     /**
      * Constructor for LoadMemoryTool
      */
@@ -2670,7 +2684,6 @@ declare class LoadMemoryTool extends BaseTool {
 
 type McpConfig = {
     name: string;
-    description: string;
     transport: McpTransportType;
     timeout?: number;
     retryOptions?: {
@@ -2704,7 +2717,9 @@ declare enum McpErrorType {
     TOOL_EXECUTION_ERROR = "tool_execution_error",
     RESOURCE_CLOSED_ERROR = "resource_closed_error",
     TIMEOUT_ERROR = "timeout_error",
-    INVALID_SCHEMA_ERROR = "invalid_schema_error"
+    INVALID_SCHEMA_ERROR = "invalid_schema_error",
+    SAMPLING_ERROR = "SAMPLING_ERROR",
+    INVALID_REQUEST_ERROR = "INVALID_REQUEST_ERROR"
 }
 /**
  * Custom error class for MCP-related errors
@@ -2714,12 +2729,49 @@ declare class McpError extends Error {
     originalError?: Error;
     constructor(message: string, type: McpErrorType, originalError?: Error);
 }
+type SamplingRequest = z.infer<typeof CreateMessageRequestSchema>;
+type SamplingResponse = z.infer<typeof CreateMessageResultSchema>;
+
+/**
+ * ADK sampling request format - what we pass to the user's handler
+ */
+interface ADKSamplingRequest {
+    messages: Message[];
+    systemPrompt?: string;
+    modelPreferences?: {
+        hints?: Array<{
+            name?: string;
+        }>;
+        costPriority?: number;
+        speedPriority?: number;
+        intelligencePriority?: number;
+    };
+    includeContext?: "none" | "thisServer" | "allServers";
+    temperature?: number;
+    maxTokens: number;
+    stopSequences?: string[];
+    metadata?: Record<string, unknown>;
+}
+/**
+ * ADK sampling response format - what we expect from the user's handler
+ */
+interface ADKSamplingResponse {
+    model: string;
+    stopReason?: "endTurn" | "stopSequence" | "maxTokens" | string;
+    content: string | null;
+}
+/**
+ * ADK sampling handler function type - receives ADK formatted messages
+ */
+type ADKSamplingHandler = (request: ADKSamplingRequest) => Promise<ADKSamplingResponse>;
 
 declare class McpClientService {
     private config;
     private client;
     private transport;
     private isClosing;
+    private mcpSamplingHandler;
+    private logger;
     constructor(config: McpConfig);
     /**
      * Initializes and returns an MCP client based on configuration.
@@ -2754,6 +2806,15 @@ declare class McpClientService {
      * Checks if the client is currently connected
      */
     isConnected(): boolean;
+    private setupSamplingHandler;
+    /**
+     * Set an ADK sampling handler
+     */
+    setSamplingHandler(handler: ADKSamplingHandler): void;
+    /**
+     * Remove the sampling handler
+     */
+    removeSamplingHandler(): void;
 }
 
 /**
@@ -2861,6 +2922,8 @@ declare const index$2_McpErrorType: typeof McpErrorType;
 type index$2_McpToolset = McpToolset;
 declare const index$2_McpToolset: typeof McpToolset;
 type index$2_McpTransportType = McpTransportType;
+type index$2_SamplingRequest = SamplingRequest;
+type index$2_SamplingResponse = SamplingResponse;
 type index$2_ToolConfig = ToolConfig;
 type index$2_ToolContext = ToolContext;
 declare const index$2_ToolContext: typeof ToolContext;
@@ -2876,7 +2939,7 @@ 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_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 { 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 };
 }
 
 /**
@@ -2955,6 +3018,7 @@ declare class PersistentMemoryService implements BaseMemoryService {
      * File prefix for memory files
      */
     private filePrefix;
+    private logger;
     /**
      * Constructor for PersistentMemoryService
      */
@@ -3434,4 +3498,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 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 { 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 };

package/dist/index.d.ts

@@ -1,5 +1,6 @@
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
-import { Tool } from '@modelcontextprotocol/sdk/types.js';
+import { CreateMessageRequestSchema, CreateMessageResultSchema, Tool } from '@modelcontextprotocol/sdk/types.js';
+import { z } from 'zod';
 import { AxiosInstance } from 'axios';
 import OpenAI from 'openai';
 import * as drizzle_orm_pg_core from 'drizzle-orm/pg-core';
@@ -440,6 +441,7 @@ declare class AnthropicLLM extends BaseLLM {
      * Default parameters for requests
      */
     private defaultParams;
+    private logger;
     /**
      * Constructor for AnthropicLLM
      */
@@ -516,6 +518,7 @@ declare class AnthropicLLMConnection extends BaseLLMConnection {
     private responseCallback?;
     private errorCallback?;
     private endCallback?;
+    private logger;
     /**
      * Constructor
      */
@@ -712,6 +715,7 @@ declare class OpenAILLM extends BaseLLM {
      * Default parameters for requests
      */
     private defaultParams;
+    private logger;
     /**
      * Constructor for OpenAILLM
      */
@@ -1995,6 +1999,7 @@ declare class Agent extends BaseAgent {
      * The minimum relevance score for memory augmentation (0-1)
      */
     private memoryRelevanceThreshold;
+    private logger;
     /**
      * Constructor for Agent
      */
@@ -2069,6 +2074,7 @@ interface EnhancedLLMResponse extends LLMResponse {
  * Each sub-agent's output becomes input to the next agent
  */
 declare class SequentialAgent extends BaseAgent {
+    private logger;
     /**
      * Constructor for SequentialAgent
      */
@@ -2113,6 +2119,7 @@ interface ParallelAgentConfig {
  * All sub-agents execute independently with the same input
  */
 declare class ParallelAgent extends BaseAgent {
+    private logger;
     /**
      * Constructor for ParallelAgent
      */
@@ -2181,6 +2188,7 @@ declare class LoopAgent extends BaseAgent {
      * Custom condition check function
      */
     private conditionCheck?;
+    private logger;
     /**
      * Constructor for LoopAgent
      */
@@ -2278,6 +2286,7 @@ declare class LangGraphAgent extends BaseAgent {
      * Results from node executions
      */
     private results;
+    private logger;
     /**
      * Constructor for LangGraphAgent
      */
@@ -2439,6 +2448,7 @@ declare function createFunctionTool(func: (...args: any[]) => any, options?: {
  * Simple GoogleSearch tool implementation
  */
 declare class GoogleSearch extends BaseTool {
+    private logger;
     /**
      * Constructor for GoogleSearch
      */
@@ -2591,6 +2601,7 @@ declare module "./tool-context" {
  * Tool that allows an agent to exit the current execution loop
  */
 declare class ExitLoopTool extends BaseTool {
+    private logger;
     /**
      * Constructor for ExitLoopTool
      */
@@ -2609,6 +2620,7 @@ declare class ExitLoopTool extends BaseTool {
  * Tool that allows an agent to get a choice from the user
  */
 declare class GetUserChoiceTool extends BaseTool {
+    private logger;
     /**
      * Constructor for GetUserChoiceTool
      */
@@ -2632,6 +2644,7 @@ declare class GetUserChoiceTool extends BaseTool {
  * Tool that allows an agent to transfer control to another agent
  */
 declare class TransferToAgentTool extends BaseTool {
+    private logger;
     /**
      * Constructor for TransferToAgentTool
      */
@@ -2652,6 +2665,7 @@ declare class TransferToAgentTool extends BaseTool {
  * Tool that allows an agent to load memories relevant to a query
  */
 declare class LoadMemoryTool extends BaseTool {
+    private logger;
     /**
      * Constructor for LoadMemoryTool
      */
@@ -2670,7 +2684,6 @@ declare class LoadMemoryTool extends BaseTool {
 
 type McpConfig = {
     name: string;
-    description: string;
     transport: McpTransportType;
     timeout?: number;
     retryOptions?: {
@@ -2704,7 +2717,9 @@ declare enum McpErrorType {
     TOOL_EXECUTION_ERROR = "tool_execution_error",
     RESOURCE_CLOSED_ERROR = "resource_closed_error",
     TIMEOUT_ERROR = "timeout_error",
-    INVALID_SCHEMA_ERROR = "invalid_schema_error"
+    INVALID_SCHEMA_ERROR = "invalid_schema_error",
+    SAMPLING_ERROR = "SAMPLING_ERROR",
+    INVALID_REQUEST_ERROR = "INVALID_REQUEST_ERROR"
 }
 /**
  * Custom error class for MCP-related errors
@@ -2714,12 +2729,49 @@ declare class McpError extends Error {
     originalError?: Error;
     constructor(message: string, type: McpErrorType, originalError?: Error);
 }
+type SamplingRequest = z.infer<typeof CreateMessageRequestSchema>;
+type SamplingResponse = z.infer<typeof CreateMessageResultSchema>;
+
+/**
+ * ADK sampling request format - what we pass to the user's handler
+ */
+interface ADKSamplingRequest {
+    messages: Message[];
+    systemPrompt?: string;
+    modelPreferences?: {
+        hints?: Array<{
+            name?: string;
+        }>;
+        costPriority?: number;
+        speedPriority?: number;
+        intelligencePriority?: number;
+    };
+    includeContext?: "none" | "thisServer" | "allServers";
+    temperature?: number;
+    maxTokens: number;
+    stopSequences?: string[];
+    metadata?: Record<string, unknown>;
+}
+/**
+ * ADK sampling response format - what we expect from the user's handler
+ */
+interface ADKSamplingResponse {
+    model: string;
+    stopReason?: "endTurn" | "stopSequence" | "maxTokens" | string;
+    content: string | null;
+}
+/**
+ * ADK sampling handler function type - receives ADK formatted messages
+ */
+type ADKSamplingHandler = (request: ADKSamplingRequest) => Promise<ADKSamplingResponse>;
 
 declare class McpClientService {
     private config;
     private client;
     private transport;
     private isClosing;
+    private mcpSamplingHandler;
+    private logger;
     constructor(config: McpConfig);
     /**
      * Initializes and returns an MCP client based on configuration.
@@ -2754,6 +2806,15 @@ declare class McpClientService {
      * Checks if the client is currently connected
      */
     isConnected(): boolean;
+    private setupSamplingHandler;
+    /**
+     * Set an ADK sampling handler
+     */
+    setSamplingHandler(handler: ADKSamplingHandler): void;
+    /**
+     * Remove the sampling handler
+     */
+    removeSamplingHandler(): void;
 }
 
 /**
@@ -2861,6 +2922,8 @@ declare const index$2_McpErrorType: typeof McpErrorType;
 type index$2_McpToolset = McpToolset;
 declare const index$2_McpToolset: typeof McpToolset;
 type index$2_McpTransportType = McpTransportType;
+type index$2_SamplingRequest = SamplingRequest;
+type index$2_SamplingResponse = SamplingResponse;
 type index$2_ToolConfig = ToolConfig;
 type index$2_ToolContext = ToolContext;
 declare const index$2_ToolContext: typeof ToolContext;
@@ -2876,7 +2939,7 @@ 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_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 { 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 };
 }
 
 /**
@@ -2955,6 +3018,7 @@ declare class PersistentMemoryService implements BaseMemoryService {
      * File prefix for memory files
      */
     private filePrefix;
+    private logger;
     /**
      * Constructor for PersistentMemoryService
      */
@@ -3434,4 +3498,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 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 { 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 };