@iqai/adk 0.0.5 → 0.0.7

package/dist/index.d.ts

@@ -1,10 +1,12 @@
 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';
 import { NodePgDatabase } from 'drizzle-orm/node-postgres';
 import { PGlite } from '@electric-sql/pglite';
+import Database from 'better-sqlite3';
 
 /**
  * Function call result from LLM
@@ -439,6 +441,7 @@ declare class AnthropicLLM extends BaseLLM {
      * Default parameters for requests
      */
     private defaultParams;
+    private logger;
     /**
      * Constructor for AnthropicLLM
      */
@@ -515,6 +518,7 @@ declare class AnthropicLLMConnection extends BaseLLMConnection {
     private responseCallback?;
     private errorCallback?;
     private endCallback?;
+    private logger;
     /**
      * Constructor
      */
@@ -711,6 +715,7 @@ declare class OpenAILLM extends BaseLLM {
      * Default parameters for requests
      */
     private defaultParams;
+    private logger;
     /**
      * Constructor for OpenAILLM
      */
@@ -1994,6 +1999,7 @@ declare class Agent extends BaseAgent {
      * The minimum relevance score for memory augmentation (0-1)
      */
     private memoryRelevanceThreshold;
+    private logger;
     /**
      * Constructor for Agent
      */
@@ -2068,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
      */
@@ -2112,6 +2119,7 @@ interface ParallelAgentConfig {
  * All sub-agents execute independently with the same input
  */
 declare class ParallelAgent extends BaseAgent {
+    private logger;
     /**
      * Constructor for ParallelAgent
      */
@@ -2180,6 +2188,7 @@ declare class LoopAgent extends BaseAgent {
      * Custom condition check function
      */
     private conditionCheck?;
+    private logger;
     /**
      * Constructor for LoopAgent
      */
@@ -2277,6 +2286,7 @@ declare class LangGraphAgent extends BaseAgent {
      * Results from node executions
      */
     private results;
+    private logger;
     /**
      * Constructor for LangGraphAgent
      */
@@ -2438,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
      */
@@ -2590,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
      */
@@ -2608,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
      */
@@ -2631,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
      */
@@ -2651,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
      */
@@ -2684,6 +2699,7 @@ type McpConfig = {
         maxSize?: number;
     };
     debug?: boolean;
+    samplingHandler?: SamplingHandler;
 };
 type McpTransportType = {
     mode: "stdio";
@@ -2703,7 +2719,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
@@ -2713,12 +2731,17 @@ 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>;
+type SamplingHandler = (request: SamplingRequest) => Promise<SamplingResponse>;
 
 declare class McpClientService {
     private config;
     private client;
     private transport;
     private isClosing;
+    private samplingHandler;
+    private logger;
     constructor(config: McpConfig);
     /**
      * Initializes and returns an MCP client based on configuration.
@@ -2753,6 +2776,9 @@ declare class McpClientService {
      * Checks if the client is currently connected
      */
     isConnected(): boolean;
+    private setupSamplingHandler;
+    setSamplingHandler(handler: SamplingHandler): void;
+    removeSamplingHandler(): void;
 }
 
 /**
@@ -2860,6 +2886,9 @@ 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_SamplingHandler = SamplingHandler;
+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;
@@ -2875,7 +2904,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_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_getMcpTools as getMcpTools, index$2_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$2_mcpSchemaToParameters as mcpSchemaToParameters, index$2_normalizeJsonSchema as normalizeJsonSchema };
 }
 
 /**
@@ -2954,6 +2983,7 @@ declare class PersistentMemoryService implements BaseMemoryService {
      * File prefix for memory files
      */
     private filePrefix;
+    private logger;
     /**
      * Constructor for PersistentMemoryService
      */
@@ -3079,7 +3109,7 @@ declare class InMemorySessionService implements SessionService {
     appendEvent(session: Session, event: Event): Promise<Event>;
 }
 
-declare const sessionsSchema$1: drizzle_orm_pg_core.PgTableWithColumns<{
+declare const sessionsSchema: drizzle_orm_pg_core.PgTableWithColumns<{
     name: "sessions";
     schema: undefined;
     columns: {
@@ -3215,7 +3245,7 @@ declare const sessionsSchema$1: drizzle_orm_pg_core.PgTableWithColumns<{
     };
     dialect: "pg";
 }>;
-type SessionsTable = typeof sessionsSchema$1;
+type SessionsTable = typeof sessionsSchema;
 /**
  * Configuration for DatabaseSessionService with Drizzle
  */
@@ -3251,142 +3281,6 @@ declare class PostgresSessionService implements SessionService {
     appendEvent(session: Session, event: Event): Promise<Event>;
 }
 
-declare const sessionsSchema: drizzle_orm_pg_core.PgTableWithColumns<{
-    name: "sessions";
-    schema: undefined;
-    columns: {
-        id: drizzle_orm_pg_core.PgColumn<{
-            name: "id";
-            tableName: "sessions";
-            dataType: "string";
-            columnType: "PgVarchar";
-            data: string;
-            driverParam: string;
-            notNull: true;
-            hasDefault: false;
-            isPrimaryKey: true;
-            isAutoincrement: false;
-            hasRuntimeDefault: false;
-            enumValues: [string, ...string[]];
-            baseColumn: never;
-            identity: undefined;
-            generated: undefined;
-        }, {}, {
-            length: 255;
-        }>;
-        userId: drizzle_orm_pg_core.PgColumn<{
-            name: "user_id";
-            tableName: "sessions";
-            dataType: "string";
-            columnType: "PgVarchar";
-            data: string;
-            driverParam: string;
-            notNull: true;
-            hasDefault: false;
-            isPrimaryKey: false;
-            isAutoincrement: false;
-            hasRuntimeDefault: false;
-            enumValues: [string, ...string[]];
-            baseColumn: never;
-            identity: undefined;
-            generated: undefined;
-        }, {}, {
-            length: 255;
-        }>;
-        messages: drizzle_orm_pg_core.PgColumn<{
-            name: "messages";
-            tableName: "sessions";
-            dataType: "json";
-            columnType: "PgJsonb";
-            data: Message[];
-            driverParam: unknown;
-            notNull: false;
-            hasDefault: true;
-            isPrimaryKey: false;
-            isAutoincrement: false;
-            hasRuntimeDefault: false;
-            enumValues: undefined;
-            baseColumn: never;
-            identity: undefined;
-            generated: undefined;
-        }, {}, {
-            $type: Message[];
-        }>;
-        metadata: drizzle_orm_pg_core.PgColumn<{
-            name: "metadata";
-            tableName: "sessions";
-            dataType: "json";
-            columnType: "PgJsonb";
-            data: Record<string, any>;
-            driverParam: unknown;
-            notNull: false;
-            hasDefault: true;
-            isPrimaryKey: false;
-            isAutoincrement: false;
-            hasRuntimeDefault: false;
-            enumValues: undefined;
-            baseColumn: never;
-            identity: undefined;
-            generated: undefined;
-        }, {}, {
-            $type: Record<string, any>;
-        }>;
-        createdAt: drizzle_orm_pg_core.PgColumn<{
-            name: "created_at";
-            tableName: "sessions";
-            dataType: "date";
-            columnType: "PgTimestamp";
-            data: Date;
-            driverParam: string;
-            notNull: true;
-            hasDefault: true;
-            isPrimaryKey: false;
-            isAutoincrement: false;
-            hasRuntimeDefault: false;
-            enumValues: undefined;
-            baseColumn: never;
-            identity: undefined;
-            generated: undefined;
-        }, {}, {}>;
-        updatedAt: drizzle_orm_pg_core.PgColumn<{
-            name: "updated_at";
-            tableName: "sessions";
-            dataType: "date";
-            columnType: "PgTimestamp";
-            data: Date;
-            driverParam: string;
-            notNull: true;
-            hasDefault: true;
-            isPrimaryKey: false;
-            isAutoincrement: false;
-            hasRuntimeDefault: false;
-            enumValues: undefined;
-            baseColumn: never;
-            identity: undefined;
-            generated: undefined;
-        }, {}, {}>;
-        state: drizzle_orm_pg_core.PgColumn<{
-            name: "state";
-            tableName: "sessions";
-            dataType: "json";
-            columnType: "PgJsonb";
-            data: Record<string, any>;
-            driverParam: unknown;
-            notNull: false;
-            hasDefault: true;
-            isPrimaryKey: false;
-            isAutoincrement: false;
-            hasRuntimeDefault: false;
-            enumValues: undefined;
-            baseColumn: never;
-            identity: undefined;
-            generated: undefined;
-        }, {}, {
-            $type: Record<string, any>;
-        }>;
-    };
-    dialect: "pg";
-}>;
 /**
  * Configuration for PgLiteSessionService
  */
@@ -3423,6 +3317,43 @@ declare class PgLiteSessionService implements SessionService {
     appendEvent(session: Session, event: Event): Promise<Event>;
 }
 
+/**
+ * Configuration for SqliteSessionService
+ */
+interface SqliteSessionServiceConfig {
+    /**
+     * An initialized better-sqlite3 Database instance.
+     * The service will handle all Drizzle ORM setup internally.
+     */
+    sqlite: Database.Database;
+    /**
+     * Optional: Skip automatic table creation if you handle migrations externally
+     */
+    skipTableCreation?: boolean;
+}
+declare class SqliteSessionService implements SessionService {
+    private db;
+    private sessionsTable;
+    private initialized;
+    private sqliteInstance;
+    constructor(config: SqliteSessionServiceConfig);
+    /**
+     * Initialize the database by creating required tables if they don't exist
+     */
+    private initializeDatabase;
+    /**
+     * Ensure database is initialized before any operation
+     */
+    private ensureInitialized;
+    private generateSessionId;
+    createSession(userId: string, metadata?: Record<string, any>): Promise<Session>;
+    getSession(sessionId: string): Promise<Session | undefined>;
+    updateSession(session: Session): Promise<void>;
+    listSessions(userId: string, options?: ListSessionOptions): Promise<Session[]>;
+    deleteSession(sessionId: string): Promise<void>;
+    appendEvent(session: Session, event: Event): Promise<Event>;
+}
+
 /**
  * Utility functions for working with sessions
  */
@@ -3460,12 +3391,13 @@ type index_Session = Session;
 type index_SessionService = SessionService;
 type index_SessionState = SessionState;
 declare const index_SessionState: typeof SessionState;
+type index_SqliteSessionService = SqliteSessionService;
+declare const index_SqliteSessionService: typeof SqliteSessionService;
 declare const index_cloneSession: typeof cloneSession;
 declare const index_generateSessionId: typeof generateSessionId;
-declare const index_sessionsSchema: typeof sessionsSchema;
 declare const index_validateSession: typeof validateSession;
 declare namespace index {
-  export { index_InMemorySessionService as InMemorySessionService, type index_ListSessionOptions as ListSessionOptions, index_PgLiteSessionService as PgLiteSessionService, index_PostgresSessionService as PostgresSessionService, type index_Session as Session, type index_SessionService as SessionService, index_SessionState as SessionState, index_cloneSession as cloneSession, index_generateSessionId as generateSessionId, index_sessionsSchema as sessionsSchema, index_validateSession as validateSession };
+  export { index_InMemorySessionService as InMemorySessionService, type index_ListSessionOptions as ListSessionOptions, index_PgLiteSessionService as PgLiteSessionService, index_PostgresSessionService as PostgresSessionService, type index_Session as Session, type index_SessionService as SessionService, index_SessionState as SessionState, index_SqliteSessionService as SqliteSessionService, index_cloneSession as cloneSession, index_generateSessionId as generateSessionId, index_validateSession as validateSession };
 }
 
 /**
@@ -3531,4 +3463,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, 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, sessionsSchema, 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 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, generateSessionId, getMcpTools, jsonSchemaToDeclaration, mcpSchemaToParameters, normalizeJsonSchema, registerProviders, validateSession };