@iqai/adk 0.1.8 → 0.1.10

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # @iqai/adk
 
+## 0.1.10
+
+### Patch Changes
+
+- 83e8a58: Add create tool function to easily create tools with zod schema
+
+## 0.1.9
+
+### Patch Changes
+
+- 2711998: - adds @iqai/mcp-discord mcp definition
+  - update EnhancedRunner.ask() to take in message type LlmRequest
+  - update SamplingHandler response type to be LlmResponse | string for more flexibility when used with enhanced runner
+
 ## 0.1.8
 
 ### Patch Changes

package/README.md

@@ -58,7 +58,7 @@ npm install @iqai/adk
 import { AgentBuilder } from '@iqai/adk';
 
 const response = await AgentBuilder
-  .withModel("gpt-4")
+  .withModel("gpt-4.1")
   .ask("What is the primary function of an AI agent?");
 
 console.log(response);
@@ -129,7 +129,6 @@ async function quickQuery() {
     .create("query_assistant")
     .withModel("gemini-2.5-flash")
     .withInstruction("You are a helpful AI. Respond clearly and concisely.")
-    .withQuickSession("my-app", "user-123")
     .ask("What is the capital of France?");
 
   console.log(response);

package/dist/index.d.mts

@@ -1,9 +1,10 @@
 import { Part, Content as Content$1, Blob, SpeechConfig, AudioTranscriptionConfig, RealtimeInputConfig, ProactivityConfig, GenerateContentConfig, LiveConnectConfig, GroundingMetadata, GenerateContentResponseUsageMetadata, GoogleGenAI, FunctionCall } from '@google/genai';
 export { Blob, Content } from '@google/genai';
 import { LanguageModel } from 'ai';
+import * as z from 'zod/v4';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { CreateMessageRequestSchema, CreateMessageResultSchema, Tool } from '@modelcontextprotocol/sdk/types.js';
-import { z } from 'zod';
+import { z as z$1 } from 'zod';
 import { Kysely, Generated } from 'kysely';
 import { StorageOptions } from '@google-cloud/storage';
 import { Tracer } from '@opentelemetry/api';
@@ -904,6 +905,65 @@ declare abstract class BaseTool {
     private findToolWithFunctionDeclarations;
 }
 
+/**
+ * Configuration for creating a tool
+ */
+interface CreateToolConfig<T extends Record<string, any>> {
+    /** The name of the tool */
+    name: string;
+    /** A description of what the tool does */
+    description: string;
+    /** Zod schema for validating tool arguments */
+    schema: z.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 */
+    isLongRunning?: boolean;
+    /** Whether the tool execution should be retried on failure */
+    shouldRetryOnFailure?: boolean;
+    /** Maximum retry attempts */
+    maxRetryAttempts?: number;
+}
+/**
+ * Creates a tool from a configuration object.
+ *
+ * This is a more user-friendly alternative to FunctionTool that provides:
+ * - Automatic argument validation using Zod schemas
+ * - Clear error messages for invalid inputs
+ * - Automatic JSON Schema generation for LLM function declarations
+ * - Support for both sync and async functions
+ * - Optional ToolContext parameter support
+ *
+ * @param config The tool configuration object
+ * @returns A BaseTool instance ready for use with agents
+ *
+ * @example
+ * ```typescript
+ * import { createTool } from '@iqai/adk';
+ * import { z } from 'zod';
+ *
+ * const calculatorTool = createTool({
+ *   name: 'calculator',
+ *   description: 'Performs basic arithmetic operations',
+ *   schema: z.object({
+ *     operation: z.enum(['add', 'subtract', 'multiply', 'divide']),
+ *     a: z.number().describe('First number'),
+ *     b: z.number().describe('Second number')
+ *   }),
+ *   fn: ({ operation, a, b }) => {
+ *     switch (operation) {
+ *       case 'add': return { result: a + b };
+ *       case 'subtract': return { result: a - b };
+ *       case 'multiply': return { result: a * b };
+ *       case 'divide': return { result: b !== 0 ? a / b : 'Cannot divide by zero' };
+ *       default: return { error: 'Unknown operation' };
+ *     }
+ *   }
+ * });
+ * ```
+ */
+declare function createTool<T extends Record<string, any>>(config: CreateToolConfig<T>): BaseTool;
+
 /**
  * A tool that wraps a user-defined TypeScript function.
  *
@@ -1322,9 +1382,9 @@ declare class McpError extends Error {
     originalError?: Error;
     constructor(message: string, type: McpErrorType, originalError?: Error);
 }
-type McpSamplingRequest = z.infer<typeof CreateMessageRequestSchema>;
-type McpSamplingResponse = z.infer<typeof CreateMessageResultSchema>;
-type SamplingHandler = (messages: LlmRequest) => Promise<LlmResponse>;
+type McpSamplingRequest = z$1.infer<typeof CreateMessageRequestSchema>;
+type McpSamplingResponse = z$1.infer<typeof CreateMessageResultSchema>;
+type SamplingHandler = (request: LlmRequest) => Promise<string | LlmResponse>;
 
 declare class McpClientService {
     private config;
@@ -1443,15 +1503,23 @@ declare class McpSamplingHandler {
  *
  * const llm = new Gemini("gemini-2.0-flash-exp");
  *
- * const samplingHandler = createSamplingHandler(async (request) => {
- *   // request is properly typed with all the fields
+ * // Example 1: Return full LlmResponse
+ * const samplingHandler1 = createSamplingHandler(async (request) => {
  *   const responses = [];
  *   for await (const response of llm.generateContentAsync(request)) {
  *     responses.push(response);
  *   }
- *
  *   return responses[responses.length - 1];
  * });
+ *
+ * // Example 2: Return simple string
+ * const samplingHandler2 = createSamplingHandler(async (request) => {
+ *   const lastMessage = request.contents[request.contents.length - 1].parts[0].text;
+ *   return await runner.ask(lastMessage);
+ * });
+ *
+ * // Example 3: Direct function reference
+ * const samplingHandler3 = createSamplingHandler(runner.ask);
  * ```
  */
 declare function createSamplingHandler(handler: SamplingHandler): SamplingHandler;
@@ -1633,6 +1701,13 @@ declare function McpOdos(config?: McpServerConfig): McpToolset;
  * Required env vars: TELEGRAM_BOT_TOKEN
  */
 declare function McpTelegram(config?: McpServerConfig): McpToolset;
+/**
+ * MCP Discord - Interact with Discord via MCP protocol
+ *
+ * Required env vars: DISCORD_TOKEN
+ * Optional env vars: PATH
+ */
+declare function McpDiscord(config?: McpServerConfig): McpToolset;
 /**
  * MCP CoinGecko - Access cryptocurrency market data and analytics
  *
@@ -1737,6 +1812,7 @@ declare function getMcpTools(config: McpConfig, toolFilter?: string[] | ((tool:
 type index$6_BaseTool = BaseTool;
 declare const index$6_BaseTool: typeof BaseTool;
 type index$6_BuildFunctionDeclarationOptions = BuildFunctionDeclarationOptions;
+type index$6_CreateToolConfig<T extends Record<string, any>> = CreateToolConfig<T>;
 type index$6_ExitLoopTool = ExitLoopTool;
 declare const index$6_ExitLoopTool: typeof ExitLoopTool;
 type index$6_FileOperationsTool = FileOperationsTool;
@@ -1758,6 +1834,7 @@ declare const index$6_McpAtp: typeof McpAtp;
 declare const index$6_McpBamm: typeof McpBamm;
 declare const index$6_McpCoinGecko: typeof McpCoinGecko;
 type index$6_McpConfig = McpConfig;
+declare const index$6_McpDiscord: typeof McpDiscord;
 type index$6_McpError = McpError;
 declare const index$6_McpError: typeof McpError;
 type index$6_McpErrorType = McpErrorType;
@@ -1791,12 +1868,13 @@ declare const index$6_adkToMcpToolType: typeof adkToMcpToolType;
 declare const index$6_buildFunctionDeclaration: typeof buildFunctionDeclaration;
 declare const index$6_createFunctionTool: typeof createFunctionTool;
 declare const index$6_createSamplingHandler: typeof createSamplingHandler;
+declare const index$6_createTool: typeof createTool;
 declare const index$6_getMcpTools: typeof getMcpTools;
 declare const index$6_jsonSchemaToDeclaration: typeof jsonSchemaToDeclaration;
 declare const index$6_mcpSchemaToParameters: typeof mcpSchemaToParameters;
 declare const index$6_normalizeJsonSchema: typeof normalizeJsonSchema;
 declare namespace index$6 {
-  export { index$6_BaseTool as BaseTool, type index$6_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, index$6_ExitLoopTool as ExitLoopTool, index$6_FileOperationsTool as FileOperationsTool, index$6_FunctionTool as FunctionTool, index$6_GetUserChoiceTool as GetUserChoiceTool, index$6_GoogleSearch as GoogleSearch, index$6_HttpRequestTool as HttpRequestTool, index$6_LoadArtifactsTool as LoadArtifactsTool, index$6_LoadMemoryTool as LoadMemoryTool, index$6_McpAbi as McpAbi, index$6_McpAtp as McpAtp, index$6_McpBamm as McpBamm, index$6_McpCoinGecko as McpCoinGecko, type index$6_McpConfig as McpConfig, index$6_McpError as McpError, index$6_McpErrorType as McpErrorType, index$6_McpFilesystem as McpFilesystem, index$6_McpFraxlend as McpFraxlend, index$6_McpGeneric as McpGeneric, index$6_McpIqWiki as McpIqWiki, index$6_McpMemory as McpMemory, index$6_McpNearAgent as McpNearAgent, index$6_McpNearIntentSwaps as McpNearIntentSwaps, index$6_McpOdos as McpOdos, index$6_McpSamplingHandler as McpSamplingHandler, type index$6_McpSamplingRequest as McpSamplingRequest, type index$6_McpSamplingResponse as McpSamplingResponse, type index$6_McpServerConfig as McpServerConfig, index$6_McpTelegram as McpTelegram, index$6_McpToolset as McpToolset, type index$6_McpTransportType as McpTransportType, type index$6_SamplingHandler as SamplingHandler, type index$6_ToolConfig as ToolConfig, index$6_ToolContext as ToolContext, index$6_TransferToAgentTool as TransferToAgentTool, index$6_UserInteractionTool as UserInteractionTool, index$6_adkToMcpToolType as adkToMcpToolType, index$6_buildFunctionDeclaration as buildFunctionDeclaration, index$6_createFunctionTool as createFunctionTool, index$6_createSamplingHandler as createSamplingHandler, index$6_getMcpTools as getMcpTools, index$6_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$6_mcpSchemaToParameters as mcpSchemaToParameters, index$6_normalizeJsonSchema as normalizeJsonSchema };
+  export { index$6_BaseTool as BaseTool, type index$6_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, type index$6_CreateToolConfig as CreateToolConfig, index$6_ExitLoopTool as ExitLoopTool, index$6_FileOperationsTool as FileOperationsTool, index$6_FunctionTool as FunctionTool, index$6_GetUserChoiceTool as GetUserChoiceTool, index$6_GoogleSearch as GoogleSearch, index$6_HttpRequestTool as HttpRequestTool, index$6_LoadArtifactsTool as LoadArtifactsTool, index$6_LoadMemoryTool as LoadMemoryTool, index$6_McpAbi as McpAbi, index$6_McpAtp as McpAtp, index$6_McpBamm as McpBamm, index$6_McpCoinGecko as McpCoinGecko, type index$6_McpConfig as McpConfig, index$6_McpDiscord as McpDiscord, index$6_McpError as McpError, index$6_McpErrorType as McpErrorType, index$6_McpFilesystem as McpFilesystem, index$6_McpFraxlend as McpFraxlend, index$6_McpGeneric as McpGeneric, index$6_McpIqWiki as McpIqWiki, index$6_McpMemory as McpMemory, index$6_McpNearAgent as McpNearAgent, index$6_McpNearIntentSwaps as McpNearIntentSwaps, index$6_McpOdos as McpOdos, index$6_McpSamplingHandler as McpSamplingHandler, type index$6_McpSamplingRequest as McpSamplingRequest, type index$6_McpSamplingResponse as McpSamplingResponse, type index$6_McpServerConfig as McpServerConfig, index$6_McpTelegram as McpTelegram, index$6_McpToolset as McpToolset, type index$6_McpTransportType as McpTransportType, type index$6_SamplingHandler as SamplingHandler, type index$6_ToolConfig as ToolConfig, index$6_ToolContext as ToolContext, index$6_TransferToAgentTool as TransferToAgentTool, index$6_UserInteractionTool as UserInteractionTool, index$6_adkToMcpToolType as adkToMcpToolType, index$6_buildFunctionDeclaration as buildFunctionDeclaration, index$6_createFunctionTool as createFunctionTool, index$6_createSamplingHandler as createSamplingHandler, index$6_createTool as createTool, index$6_getMcpTools as getMcpTools, index$6_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$6_mcpSchemaToParameters as mcpSchemaToParameters, index$6_normalizeJsonSchema as normalizeJsonSchema };
 }
 
 /**
@@ -3801,7 +3879,7 @@ interface FullMessage extends Content$1 {
  * Enhanced runner interface with simplified API
  */
 interface EnhancedRunner {
-    ask(message: string | FullMessage): Promise<string>;
+    ask(message: string | FullMessage | LlmRequest): Promise<string>;
     runAsync(params: {
         userId: string;
         sessionId: string;
@@ -4411,15 +4489,6 @@ declare abstract class BaseLlmFlow {
     _handleBeforeModelCallback(invocationContext: InvocationContext, llmRequest: LlmRequest, modelResponseEvent: Event): Promise<LlmResponse | undefined>;
     _handleAfterModelCallback(invocationContext: InvocationContext, llmResponse: LlmResponse, modelResponseEvent: Event): Promise<LlmResponse | undefined>;
     _finalizeModelResponseEvent(llmRequest: LlmRequest, llmResponse: LlmResponse, modelResponseEvent: Event): Event;
-    /**
-     * Logs data in a visually appealing format that works well in any terminal size.
-     * Uses vertical layout for better readability and respects debug settings.
-     */
-    _formatContentPreview(content: any): string;
-    /**
-     * Formats response content preview for debug logging
-     */
-    _formatResponsePreview(llmResponse: LlmResponse): string;
     __getLlm(invocationContext: InvocationContext): BaseLlm;
 }
 
@@ -4975,4 +5044,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, type AgentType, index$4 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, 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, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, Event, EventActions, index$1 as Events, ExitLoopTool, type File, FileOperationsTool, index as Flows, type FullMessage, type FunctionDeclaration, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, InvocationContext, type JSONSchema, 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, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntentSwaps, 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 SessionConfig, 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, 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, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentType, index$4 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, 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, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, Event, EventActions, index$1 as Events, ExitLoopTool, type File, FileOperationsTool, index as Flows, type FullMessage, type FunctionDeclaration, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, InvocationContext, type JSONSchema, 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, McpNearIntentSwaps, 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 SessionConfig, 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 };

package/dist/index.d.ts

@@ -1,9 +1,10 @@
 import { Part, Content as Content$1, Blob, SpeechConfig, AudioTranscriptionConfig, RealtimeInputConfig, ProactivityConfig, GenerateContentConfig, LiveConnectConfig, GroundingMetadata, GenerateContentResponseUsageMetadata, GoogleGenAI, FunctionCall } from '@google/genai';
 export { Blob, Content } from '@google/genai';
 import { LanguageModel } from 'ai';
+import * as z from 'zod/v4';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { CreateMessageRequestSchema, CreateMessageResultSchema, Tool } from '@modelcontextprotocol/sdk/types.js';
-import { z } from 'zod';
+import { z as z$1 } from 'zod';
 import { Kysely, Generated } from 'kysely';
 import { StorageOptions } from '@google-cloud/storage';
 import { Tracer } from '@opentelemetry/api';
@@ -904,6 +905,65 @@ declare abstract class BaseTool {
     private findToolWithFunctionDeclarations;
 }
 
+/**
+ * Configuration for creating a tool
+ */
+interface CreateToolConfig<T extends Record<string, any>> {
+    /** The name of the tool */
+    name: string;
+    /** A description of what the tool does */
+    description: string;
+    /** Zod schema for validating tool arguments */
+    schema: z.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 */
+    isLongRunning?: boolean;
+    /** Whether the tool execution should be retried on failure */
+    shouldRetryOnFailure?: boolean;
+    /** Maximum retry attempts */
+    maxRetryAttempts?: number;
+}
+/**
+ * Creates a tool from a configuration object.
+ *
+ * This is a more user-friendly alternative to FunctionTool that provides:
+ * - Automatic argument validation using Zod schemas
+ * - Clear error messages for invalid inputs
+ * - Automatic JSON Schema generation for LLM function declarations
+ * - Support for both sync and async functions
+ * - Optional ToolContext parameter support
+ *
+ * @param config The tool configuration object
+ * @returns A BaseTool instance ready for use with agents
+ *
+ * @example
+ * ```typescript
+ * import { createTool } from '@iqai/adk';
+ * import { z } from 'zod';
+ *
+ * const calculatorTool = createTool({
+ *   name: 'calculator',
+ *   description: 'Performs basic arithmetic operations',
+ *   schema: z.object({
+ *     operation: z.enum(['add', 'subtract', 'multiply', 'divide']),
+ *     a: z.number().describe('First number'),
+ *     b: z.number().describe('Second number')
+ *   }),
+ *   fn: ({ operation, a, b }) => {
+ *     switch (operation) {
+ *       case 'add': return { result: a + b };
+ *       case 'subtract': return { result: a - b };
+ *       case 'multiply': return { result: a * b };
+ *       case 'divide': return { result: b !== 0 ? a / b : 'Cannot divide by zero' };
+ *       default: return { error: 'Unknown operation' };
+ *     }
+ *   }
+ * });
+ * ```
+ */
+declare function createTool<T extends Record<string, any>>(config: CreateToolConfig<T>): BaseTool;
+
 /**
  * A tool that wraps a user-defined TypeScript function.
  *
@@ -1322,9 +1382,9 @@ declare class McpError extends Error {
     originalError?: Error;
     constructor(message: string, type: McpErrorType, originalError?: Error);
 }
-type McpSamplingRequest = z.infer<typeof CreateMessageRequestSchema>;
-type McpSamplingResponse = z.infer<typeof CreateMessageResultSchema>;
-type SamplingHandler = (messages: LlmRequest) => Promise<LlmResponse>;
+type McpSamplingRequest = z$1.infer<typeof CreateMessageRequestSchema>;
+type McpSamplingResponse = z$1.infer<typeof CreateMessageResultSchema>;
+type SamplingHandler = (request: LlmRequest) => Promise<string | LlmResponse>;
 
 declare class McpClientService {
     private config;
@@ -1443,15 +1503,23 @@ declare class McpSamplingHandler {
  *
  * const llm = new Gemini("gemini-2.0-flash-exp");
  *
- * const samplingHandler = createSamplingHandler(async (request) => {
- *   // request is properly typed with all the fields
+ * // Example 1: Return full LlmResponse
+ * const samplingHandler1 = createSamplingHandler(async (request) => {
  *   const responses = [];
  *   for await (const response of llm.generateContentAsync(request)) {
  *     responses.push(response);
  *   }
- *
  *   return responses[responses.length - 1];
  * });
+ *
+ * // Example 2: Return simple string
+ * const samplingHandler2 = createSamplingHandler(async (request) => {
+ *   const lastMessage = request.contents[request.contents.length - 1].parts[0].text;
+ *   return await runner.ask(lastMessage);
+ * });
+ *
+ * // Example 3: Direct function reference
+ * const samplingHandler3 = createSamplingHandler(runner.ask);
  * ```
  */
 declare function createSamplingHandler(handler: SamplingHandler): SamplingHandler;
@@ -1633,6 +1701,13 @@ declare function McpOdos(config?: McpServerConfig): McpToolset;
  * Required env vars: TELEGRAM_BOT_TOKEN
  */
 declare function McpTelegram(config?: McpServerConfig): McpToolset;
+/**
+ * MCP Discord - Interact with Discord via MCP protocol
+ *
+ * Required env vars: DISCORD_TOKEN
+ * Optional env vars: PATH
+ */
+declare function McpDiscord(config?: McpServerConfig): McpToolset;
 /**
  * MCP CoinGecko - Access cryptocurrency market data and analytics
  *
@@ -1737,6 +1812,7 @@ declare function getMcpTools(config: McpConfig, toolFilter?: string[] | ((tool:
 type index$6_BaseTool = BaseTool;
 declare const index$6_BaseTool: typeof BaseTool;
 type index$6_BuildFunctionDeclarationOptions = BuildFunctionDeclarationOptions;
+type index$6_CreateToolConfig<T extends Record<string, any>> = CreateToolConfig<T>;
 type index$6_ExitLoopTool = ExitLoopTool;
 declare const index$6_ExitLoopTool: typeof ExitLoopTool;
 type index$6_FileOperationsTool = FileOperationsTool;
@@ -1758,6 +1834,7 @@ declare const index$6_McpAtp: typeof McpAtp;
 declare const index$6_McpBamm: typeof McpBamm;
 declare const index$6_McpCoinGecko: typeof McpCoinGecko;
 type index$6_McpConfig = McpConfig;
+declare const index$6_McpDiscord: typeof McpDiscord;
 type index$6_McpError = McpError;
 declare const index$6_McpError: typeof McpError;
 type index$6_McpErrorType = McpErrorType;
@@ -1791,12 +1868,13 @@ declare const index$6_adkToMcpToolType: typeof adkToMcpToolType;
 declare const index$6_buildFunctionDeclaration: typeof buildFunctionDeclaration;
 declare const index$6_createFunctionTool: typeof createFunctionTool;
 declare const index$6_createSamplingHandler: typeof createSamplingHandler;
+declare const index$6_createTool: typeof createTool;
 declare const index$6_getMcpTools: typeof getMcpTools;
 declare const index$6_jsonSchemaToDeclaration: typeof jsonSchemaToDeclaration;
 declare const index$6_mcpSchemaToParameters: typeof mcpSchemaToParameters;
 declare const index$6_normalizeJsonSchema: typeof normalizeJsonSchema;
 declare namespace index$6 {
-  export { index$6_BaseTool as BaseTool, type index$6_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, index$6_ExitLoopTool as ExitLoopTool, index$6_FileOperationsTool as FileOperationsTool, index$6_FunctionTool as FunctionTool, index$6_GetUserChoiceTool as GetUserChoiceTool, index$6_GoogleSearch as GoogleSearch, index$6_HttpRequestTool as HttpRequestTool, index$6_LoadArtifactsTool as LoadArtifactsTool, index$6_LoadMemoryTool as LoadMemoryTool, index$6_McpAbi as McpAbi, index$6_McpAtp as McpAtp, index$6_McpBamm as McpBamm, index$6_McpCoinGecko as McpCoinGecko, type index$6_McpConfig as McpConfig, index$6_McpError as McpError, index$6_McpErrorType as McpErrorType, index$6_McpFilesystem as McpFilesystem, index$6_McpFraxlend as McpFraxlend, index$6_McpGeneric as McpGeneric, index$6_McpIqWiki as McpIqWiki, index$6_McpMemory as McpMemory, index$6_McpNearAgent as McpNearAgent, index$6_McpNearIntentSwaps as McpNearIntentSwaps, index$6_McpOdos as McpOdos, index$6_McpSamplingHandler as McpSamplingHandler, type index$6_McpSamplingRequest as McpSamplingRequest, type index$6_McpSamplingResponse as McpSamplingResponse, type index$6_McpServerConfig as McpServerConfig, index$6_McpTelegram as McpTelegram, index$6_McpToolset as McpToolset, type index$6_McpTransportType as McpTransportType, type index$6_SamplingHandler as SamplingHandler, type index$6_ToolConfig as ToolConfig, index$6_ToolContext as ToolContext, index$6_TransferToAgentTool as TransferToAgentTool, index$6_UserInteractionTool as UserInteractionTool, index$6_adkToMcpToolType as adkToMcpToolType, index$6_buildFunctionDeclaration as buildFunctionDeclaration, index$6_createFunctionTool as createFunctionTool, index$6_createSamplingHandler as createSamplingHandler, index$6_getMcpTools as getMcpTools, index$6_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$6_mcpSchemaToParameters as mcpSchemaToParameters, index$6_normalizeJsonSchema as normalizeJsonSchema };
+  export { index$6_BaseTool as BaseTool, type index$6_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, type index$6_CreateToolConfig as CreateToolConfig, index$6_ExitLoopTool as ExitLoopTool, index$6_FileOperationsTool as FileOperationsTool, index$6_FunctionTool as FunctionTool, index$6_GetUserChoiceTool as GetUserChoiceTool, index$6_GoogleSearch as GoogleSearch, index$6_HttpRequestTool as HttpRequestTool, index$6_LoadArtifactsTool as LoadArtifactsTool, index$6_LoadMemoryTool as LoadMemoryTool, index$6_McpAbi as McpAbi, index$6_McpAtp as McpAtp, index$6_McpBamm as McpBamm, index$6_McpCoinGecko as McpCoinGecko, type index$6_McpConfig as McpConfig, index$6_McpDiscord as McpDiscord, index$6_McpError as McpError, index$6_McpErrorType as McpErrorType, index$6_McpFilesystem as McpFilesystem, index$6_McpFraxlend as McpFraxlend, index$6_McpGeneric as McpGeneric, index$6_McpIqWiki as McpIqWiki, index$6_McpMemory as McpMemory, index$6_McpNearAgent as McpNearAgent, index$6_McpNearIntentSwaps as McpNearIntentSwaps, index$6_McpOdos as McpOdos, index$6_McpSamplingHandler as McpSamplingHandler, type index$6_McpSamplingRequest as McpSamplingRequest, type index$6_McpSamplingResponse as McpSamplingResponse, type index$6_McpServerConfig as McpServerConfig, index$6_McpTelegram as McpTelegram, index$6_McpToolset as McpToolset, type index$6_McpTransportType as McpTransportType, type index$6_SamplingHandler as SamplingHandler, type index$6_ToolConfig as ToolConfig, index$6_ToolContext as ToolContext, index$6_TransferToAgentTool as TransferToAgentTool, index$6_UserInteractionTool as UserInteractionTool, index$6_adkToMcpToolType as adkToMcpToolType, index$6_buildFunctionDeclaration as buildFunctionDeclaration, index$6_createFunctionTool as createFunctionTool, index$6_createSamplingHandler as createSamplingHandler, index$6_createTool as createTool, index$6_getMcpTools as getMcpTools, index$6_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$6_mcpSchemaToParameters as mcpSchemaToParameters, index$6_normalizeJsonSchema as normalizeJsonSchema };
 }
 
 /**
@@ -3801,7 +3879,7 @@ interface FullMessage extends Content$1 {
  * Enhanced runner interface with simplified API
  */
 interface EnhancedRunner {
-    ask(message: string | FullMessage): Promise<string>;
+    ask(message: string | FullMessage | LlmRequest): Promise<string>;
     runAsync(params: {
         userId: string;
         sessionId: string;
@@ -4411,15 +4489,6 @@ declare abstract class BaseLlmFlow {
     _handleBeforeModelCallback(invocationContext: InvocationContext, llmRequest: LlmRequest, modelResponseEvent: Event): Promise<LlmResponse | undefined>;
     _handleAfterModelCallback(invocationContext: InvocationContext, llmResponse: LlmResponse, modelResponseEvent: Event): Promise<LlmResponse | undefined>;
     _finalizeModelResponseEvent(llmRequest: LlmRequest, llmResponse: LlmResponse, modelResponseEvent: Event): Event;
-    /**
-     * Logs data in a visually appealing format that works well in any terminal size.
-     * Uses vertical layout for better readability and respects debug settings.
-     */
-    _formatContentPreview(content: any): string;
-    /**
-     * Formats response content preview for debug logging
-     */
-    _formatResponsePreview(llmResponse: LlmResponse): string;
     __getLlm(invocationContext: InvocationContext): BaseLlm;
 }
 
@@ -4975,4 +5044,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, type AgentType, index$4 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, 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, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, Event, EventActions, index$1 as Events, ExitLoopTool, type File, FileOperationsTool, index as Flows, type FullMessage, type FunctionDeclaration, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, InvocationContext, type JSONSchema, 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, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntentSwaps, 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 SessionConfig, 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, 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, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentType, index$4 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, 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, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, Event, EventActions, index$1 as Events, ExitLoopTool, type File, FileOperationsTool, index as Flows, type FullMessage, type FunctionDeclaration, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, InvocationContext, type JSONSchema, 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, McpNearIntentSwaps, 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 SessionConfig, 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 };