@iqai/adk 0.4.0 → 0.5.0

package/CHANGELOG.md

@@ -1,5 +1,25 @@
 # @iqai/adk
 
+## 0.5.0
+
+### Minor Changes
+
+- 9c8441c: Enhanced CoinGecko MCP server support with remote endpoints
+
+  - **Enhanced createMcpConfig function**: Now automatically detects URLs and uses `mcp-remote` for remote MCP endpoints while maintaining backward compatibility with npm packages
+  - **Updated McpCoinGecko**: Now uses the remote CoinGecko MCP API endpoint (`https://mcp.api.coingecko.com/mcp`) instead of the npm package
+  - **Added McpCoinGeckoPro**: New function for accessing the professional CoinGecko MCP API endpoint (`https://mcp.pro-api.coingecko.com/mcp`) with enhanced features and higher rate limits
+  - **Improved code maintainability**: Refactored both CoinGecko functions to use the enhanced `createMcpConfig`, eliminating code duplication
+  - **Added documentation**: Updated module documentation with examples showing how to use both CoinGecko functions
+
+  This change enables seamless integration with CoinGecko's remote MCP endpoints while providing a cleaner, more maintainable codebase for future remote endpoint integrations.
+
+## 0.4.1
+
+### Patch Changes
+
+- 1b00e47: Add export to VertexAiRagMemoryService
+
 ## 0.4.0
 
 ### Minor Changes

package/dist/index.d.mts

@@ -674,6 +674,18 @@ declare class ReadonlyContext {
      * The name of the agent that is currently running.
      */
     get agentName(): string;
+    /**
+     * The application name for this invocation. READONLY field.
+     */
+    get appName(): string;
+    /**
+     * The user ID for this invocation. READONLY field.
+     */
+    get userId(): string;
+    /**
+     * The session ID for this invocation. READONLY field.
+     */
+    get sessionId(): string;
     /**
      * The state of the current session. READONLY field.
      */
@@ -1255,7 +1267,7 @@ type ToolUnion = BaseTool | ((...args: any[]) => any);
 type SingleBeforeModelCallback = (args: {
     callbackContext: CallbackContext;
     llmRequest: LlmRequest;
-}) => LlmResponse | null | Promise<LlmResponse | null>;
+}) => LlmResponse | null | undefined | Promise<LlmResponse | null | undefined>;
 /**
  * Before model callback type (single or array)
  */
@@ -1266,7 +1278,7 @@ type BeforeModelCallback = SingleBeforeModelCallback | SingleBeforeModelCallback
 type SingleAfterModelCallback = (args: {
     callbackContext: CallbackContext;
     llmResponse: LlmResponse;
-}) => LlmResponse | null | Promise<LlmResponse | null>;
+}) => LlmResponse | null | undefined | Promise<LlmResponse | null | undefined>;
 /**
  * After model callback type (single or array)
  */
@@ -1274,7 +1286,7 @@ type AfterModelCallback = SingleAfterModelCallback | SingleAfterModelCallback[];
 /**
  * Single before tool callback type
  */
-type SingleBeforeToolCallback = (tool: BaseTool, args: Record<string, any>, toolContext: ToolContext) => Record<string, any> | null | Promise<Record<string, any> | null>;
+type SingleBeforeToolCallback = (tool: BaseTool, args: Record<string, any>, toolContext: ToolContext) => Record<string, any> | null | undefined | Promise<Record<string, any> | null | undefined>;
 /**
  * Before tool callback type (single or array)
  */
@@ -1282,7 +1294,7 @@ type BeforeToolCallback = SingleBeforeToolCallback | SingleBeforeToolCallback[];
 /**
  * Single after tool callback type
  */
-type SingleAfterToolCallback = (tool: BaseTool, args: Record<string, any>, toolContext: ToolContext, toolResponse: Record<string, any>) => Record<string, any> | null | Promise<Record<string, any> | null>;
+type SingleAfterToolCallback = (tool: BaseTool, args: Record<string, any>, toolContext: ToolContext, toolResponse: Record<string, any>) => Record<string, any> | null | undefined | Promise<Record<string, any> | null | undefined>;
 /**
  * After tool callback type (single or array)
  */
@@ -2275,6 +2287,24 @@ declare function createSamplingHandler(handler: SamplingHandler): SamplingHandle
  *
  * @example
  * ```typescript
+ * // Using remote MCP endpoints (CoinGecko):
+ * const coinGeckoTools = await McpCoinGecko().getTools();
+ *
+ * const coinGeckoProTools = await McpCoinGeckoPro({
+ *   env: {
+ *     COINGECKO_PRO_API_KEY: env.COINGECKO_PRO_API_KEY
+ *   }
+ * }).getTools();
+ *
+ * const cryptoAgent = new LlmAgent({
+ *   name: "crypto_assistant",
+ *   model: "gemini-2.5-flash",
+ *   tools: [...coinGeckoTools, ...coinGeckoProTools],
+ * });
+ * ```
+ *
+ * @example
+ * ```typescript
  * // Using MCP servers with sampling handlers:
  * import { createSamplingHandler, LlmResponse } from "@iqai/adk";
  *
@@ -2388,11 +2418,18 @@ declare function McpTelegram(config?: McpServerConfig): McpToolset;
  */
 declare function McpDiscord(config?: McpServerConfig): McpToolset;
 /**
- * MCP CoinGecko - Access cryptocurrency market data and analytics
+ * MCP CoinGecko - Access cryptocurrency market data and analytics via remote endpoint
  *
- * Optional env vars: COINGECKO_PRO_API_KEY, COINGECKO_DEMO_API_KEY, COINGECKO_ENVIRONMENT
+ * Uses the public CoinGecko MCP API endpoint. No API key required for basic functionality.
  */
 declare function McpCoinGecko(config?: McpServerConfig): McpToolset;
+/**
+ * MCP CoinGecko Pro - Access premium cryptocurrency market data and analytics via remote endpoint
+ *
+ * Uses the professional CoinGecko MCP API endpoint with enhanced features and higher rate limits.
+ * Requires a CoinGecko Pro API subscription.
+ */
+declare function McpCoinGeckoPro(config?: McpServerConfig): McpToolset;
 /**
  * MCP Upbit - Interact with the Upbit cryptocurrency exchange
  *
@@ -2528,6 +2565,7 @@ declare const index$7_McpAbi: typeof McpAbi;
 declare const index$7_McpAtp: typeof McpAtp;
 declare const index$7_McpBamm: typeof McpBamm;
 declare const index$7_McpCoinGecko: typeof McpCoinGecko;
+declare const index$7_McpCoinGeckoPro: typeof McpCoinGeckoPro;
 type index$7_McpConfig = McpConfig;
 declare const index$7_McpDiscord: typeof McpDiscord;
 type index$7_McpError = McpError;
@@ -2571,7 +2609,7 @@ declare const index$7_jsonSchemaToDeclaration: typeof jsonSchemaToDeclaration;
 declare const index$7_mcpSchemaToParameters: typeof mcpSchemaToParameters;
 declare const index$7_normalizeJsonSchema: typeof normalizeJsonSchema;
 declare namespace index$7 {
-  export { index$7_AgentTool as AgentTool, type index$7_AgentToolConfig as AgentToolConfig, type index$7_BaseAgentType as BaseAgentType, index$7_BaseTool as BaseTool, type index$7_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, type index$7_CreateToolConfig as CreateToolConfig, type index$7_CreateToolConfigWithSchema as CreateToolConfigWithSchema, type index$7_CreateToolConfigWithoutSchema as CreateToolConfigWithoutSchema, index$7_ExitLoopTool as ExitLoopTool, index$7_FileOperationsTool as FileOperationsTool, index$7_FunctionTool as FunctionTool, index$7_GetUserChoiceTool as GetUserChoiceTool, index$7_GoogleSearch as GoogleSearch, index$7_HttpRequestTool as HttpRequestTool, index$7_LoadArtifactsTool as LoadArtifactsTool, index$7_LoadMemoryTool as LoadMemoryTool, index$7_McpAbi as McpAbi, index$7_McpAtp as McpAtp, index$7_McpBamm as McpBamm, index$7_McpCoinGecko as McpCoinGecko, type index$7_McpConfig as McpConfig, index$7_McpDiscord as McpDiscord, index$7_McpError as McpError, index$7_McpErrorType as McpErrorType, index$7_McpFilesystem as McpFilesystem, index$7_McpFraxlend as McpFraxlend, index$7_McpGeneric as McpGeneric, index$7_McpIqWiki as McpIqWiki, index$7_McpMemory as McpMemory, index$7_McpNearAgent as McpNearAgent, index$7_McpNearIntents as McpNearIntents, index$7_McpOdos as McpOdos, index$7_McpSamplingHandler as McpSamplingHandler, type index$7_McpSamplingRequest as McpSamplingRequest, type index$7_McpSamplingResponse as McpSamplingResponse, type index$7_McpServerConfig as McpServerConfig, index$7_McpTelegram as McpTelegram, index$7_McpToolset as McpToolset, type index$7_McpTransportType as McpTransportType, index$7_McpUpbit as McpUpbit, type index$7_SamplingHandler as SamplingHandler, type index$7_ToolConfig as ToolConfig, index$7_ToolContext as ToolContext, index$7_TransferToAgentTool as TransferToAgentTool, index$7_UserInteractionTool as UserInteractionTool, index$7_adkToMcpToolType as adkToMcpToolType, index$7_buildFunctionDeclaration as buildFunctionDeclaration, index$7_convertMcpToolToBaseTool as convertMcpToolToBaseTool, index$7_createFunctionTool as createFunctionTool, index$7_createSamplingHandler as createSamplingHandler, index$7_createTool as createTool, index$7_getMcpTools as getMcpTools, index$7_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$7_mcpSchemaToParameters as mcpSchemaToParameters, index$7_normalizeJsonSchema as normalizeJsonSchema };
+  export { index$7_AgentTool as AgentTool, type index$7_AgentToolConfig as AgentToolConfig, type index$7_BaseAgentType as BaseAgentType, index$7_BaseTool as BaseTool, type index$7_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, type index$7_CreateToolConfig as CreateToolConfig, type index$7_CreateToolConfigWithSchema as CreateToolConfigWithSchema, type index$7_CreateToolConfigWithoutSchema as CreateToolConfigWithoutSchema, index$7_ExitLoopTool as ExitLoopTool, index$7_FileOperationsTool as FileOperationsTool, index$7_FunctionTool as FunctionTool, index$7_GetUserChoiceTool as GetUserChoiceTool, index$7_GoogleSearch as GoogleSearch, index$7_HttpRequestTool as HttpRequestTool, index$7_LoadArtifactsTool as LoadArtifactsTool, index$7_LoadMemoryTool as LoadMemoryTool, index$7_McpAbi as McpAbi, index$7_McpAtp as McpAtp, index$7_McpBamm as McpBamm, index$7_McpCoinGecko as McpCoinGecko, index$7_McpCoinGeckoPro as McpCoinGeckoPro, type index$7_McpConfig as McpConfig, index$7_McpDiscord as McpDiscord, index$7_McpError as McpError, index$7_McpErrorType as McpErrorType, index$7_McpFilesystem as McpFilesystem, index$7_McpFraxlend as McpFraxlend, index$7_McpGeneric as McpGeneric, index$7_McpIqWiki as McpIqWiki, index$7_McpMemory as McpMemory, index$7_McpNearAgent as McpNearAgent, index$7_McpNearIntents as McpNearIntents, index$7_McpOdos as McpOdos, index$7_McpSamplingHandler as McpSamplingHandler, type index$7_McpSamplingRequest as McpSamplingRequest, type index$7_McpSamplingResponse as McpSamplingResponse, type index$7_McpServerConfig as McpServerConfig, index$7_McpTelegram as McpTelegram, index$7_McpToolset as McpToolset, type index$7_McpTransportType as McpTransportType, index$7_McpUpbit as McpUpbit, type index$7_SamplingHandler as SamplingHandler, type index$7_ToolConfig as ToolConfig, index$7_ToolContext as ToolContext, index$7_TransferToAgentTool as TransferToAgentTool, index$7_UserInteractionTool as UserInteractionTool, index$7_adkToMcpToolType as adkToMcpToolType, index$7_buildFunctionDeclaration as buildFunctionDeclaration, index$7_convertMcpToolToBaseTool as convertMcpToolToBaseTool, index$7_createFunctionTool as createFunctionTool, index$7_createSamplingHandler as createSamplingHandler, index$7_createTool as createTool, index$7_getMcpTools as getMcpTools, index$7_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$7_mcpSchemaToParameters as mcpSchemaToParameters, index$7_normalizeJsonSchema as normalizeJsonSchema };
 }
 
 /**
@@ -3997,6 +4035,10 @@ interface AgentBuilderConfig {
     subAgents?: BaseAgent[];
     beforeAgentCallback?: BeforeAgentCallback;
     afterAgentCallback?: AfterAgentCallback;
+    beforeModelCallback?: BeforeModelCallback;
+    afterModelCallback?: AfterModelCallback;
+    beforeToolCallback?: BeforeToolCallback;
+    afterToolCallback?: AfterToolCallback;
     maxIterations?: number;
     nodes?: LangGraphNode[];
     rootNode?: string;
@@ -4207,6 +4249,30 @@ declare class AgentBuilder<TOut = string, TMulti extends boolean = false> {
      * @returns This builder instance for chaining
      */
     withAfterAgentCallback(callback: AfterAgentCallback): this;
+    /**
+     * Set the before model callback for LLM interaction
+     * @param callback Callback to invoke before calling the LLM
+     * @returns This builder instance for chaining
+     */
+    withBeforeModelCallback(callback: BeforeModelCallback): this;
+    /**
+     * Set the after model callback for LLM interaction
+     * @param callback Callback to invoke after receiving LLM response
+     * @returns This builder instance for chaining
+     */
+    withAfterModelCallback(callback: AfterModelCallback): this;
+    /**
+     * Set the before tool callback for tool execution
+     * @param callback Callback to invoke before running a tool
+     * @returns This builder instance for chaining
+     */
+    withBeforeToolCallback(callback: BeforeToolCallback): this;
+    /**
+     * Set the after tool callback for tool execution
+     * @param callback Callback to invoke after running a tool
+     * @returns This builder instance for chaining
+     */
+    withAfterToolCallback(callback: AfterToolCallback): this;
     /**
      * Provide an already constructed agent instance. Further definition-mutating calls
      * (model/tools/instruction/etc.) will be ignored with a dev warning.
@@ -4422,14 +4488,46 @@ declare class InMemoryMemoryService implements BaseMemoryService {
     clear(): void;
 }
 
+/**
+ * A memory service that uses Vertex AI RAG for storage and retrieval.
+ */
+declare class VertexAiRagMemoryService implements BaseMemoryService {
+    private _vertexRagStore;
+    /**
+     * Initializes a VertexAiRagMemoryService.
+     *
+     * @param ragCorpus The name of the Vertex AI RAG corpus to use. Format:
+     *   `projects/{project}/locations/{location}/ragCorpora/{rag_corpus_id}`
+     *   or `{rag_corpus_id}`
+     * @param similarityTopK The number of contexts to retrieve.
+     * @param vectorDistanceThreshold Only returns contexts with vector distance
+     *   smaller than the threshold.
+     */
+    constructor(ragCorpus?: string, similarityTopK?: number, vectorDistanceThreshold?: number);
+    /**
+     * Adds a session to the memory service
+     */
+    addSessionToMemory(session: Session): Promise<void>;
+    /**
+     * Searches for sessions that match the query using rag.retrieval_query
+     */
+    searchMemory(options: {
+        appName: string;
+        userId: string;
+        query: string;
+    }): Promise<SearchMemoryResponse>;
+}
+
 /**
  * Memory Services for the Agent Development Kit
  */
 
 type index$4_InMemoryMemoryService = InMemoryMemoryService;
 declare const index$4_InMemoryMemoryService: typeof InMemoryMemoryService;
+type index$4_VertexAiRagMemoryService = VertexAiRagMemoryService;
+declare const index$4_VertexAiRagMemoryService: typeof VertexAiRagMemoryService;
 declare namespace index$4 {
-  export { index$4_InMemoryMemoryService as InMemoryMemoryService };
+  export { index$4_InMemoryMemoryService as InMemoryMemoryService, index$4_VertexAiRagMemoryService as VertexAiRagMemoryService };
 }
 
 /**
@@ -5575,4 +5673,4 @@ declare const traceLlmCall: (invocationContext: InvocationContext, eventId: stri
 
 declare const VERSION = "0.1.0";
 
-export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, type AfterModelCallback, type AfterToolCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentBuilderWithSchema, AgentEvaluator, AgentTool, type AgentToolConfig, type AgentType, index$5 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, type BaseAgentType, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BeforeModelCallback, type BeforeToolCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, type EvalCase, type EvalCaseResult, type EvalMetric, type EvalMetricResult, type EvalMetricResultPerInvocation, EvalResult, type EvalSet, type EvalSetResult, EvalStatus, type EvaluateConfig, index as Evaluation, type EvaluationResult, Evaluator, Event, EventActions, index$2 as Events, ExitLoopTool, type File, FileOperationsTool, FinalResponseMatchV2Evaluator, index$1 as Flows, type FullMessage, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, type IntermediateData, type Interval, type Invocation, InvocationContext, type JudgeModelOptions, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, type LlmModel, type LlmModelConfig, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LocalEvalService, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntents, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, McpUpbit, index$4 as Memory, type MessagePart, type MetricInfo, type MetricValueInfo, index$6 as Models, type MultiAgentResponse, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, type PerInvocationResult, PlanReActPlanner, PrebuiltMetrics, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RougeEvaluator, RunConfig, Runner, type RunnerAskReturn, SafetyEvaluatorV1, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionInput, type SessionOptions, index$3 as Sessions, type SingleAfterModelCallback, type SingleAfterToolCallback, type SingleAgentCallback, type SingleBeforeModelCallback, type SingleBeforeToolCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$7 as Tools, TrajectoryEvaluator, 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, convertMcpToolToBaseTool, createAuthToolArguments, createBranchContextForSubAgent, createDatabaseSessionService, createFunctionTool, createMysqlSessionService, createPostgresSessionService, createSamplingHandler, createSqliteSessionService, createTool, generateAuthEvent, generateClientFunctionCallId, getLongRunningFunctionCalls, getMcpTools, handleFunctionCallsAsync, handleFunctionCallsLive, requestProcessor$5 as identityRequestProcessor, initializeTelemetry, injectSessionState, requestProcessor$4 as instructionsRequestProcessor, isEnhancedAuthConfig, jsonSchemaToDeclaration, mcpSchemaToParameters, mergeAgentRun, mergeParallelFunctionResponseEvents, newInvocationContextId, requestProcessor$1 as nlPlanningRequestProcessor, responseProcessor$1 as nlPlanningResponseProcessor, normalizeJsonSchema, populateClientFunctionCallId, registerProviders, removeClientFunctionCallId, requestProcessor$7 as requestProcessor, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };
+export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, type AfterModelCallback, type AfterToolCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentBuilderWithSchema, AgentEvaluator, AgentTool, type AgentToolConfig, type AgentType, index$5 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, type BaseAgentType, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BeforeModelCallback, type BeforeToolCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, type EvalCase, type EvalCaseResult, type EvalMetric, type EvalMetricResult, type EvalMetricResultPerInvocation, EvalResult, type EvalSet, type EvalSetResult, EvalStatus, type EvaluateConfig, index as Evaluation, type EvaluationResult, Evaluator, Event, EventActions, index$2 as Events, ExitLoopTool, type File, FileOperationsTool, FinalResponseMatchV2Evaluator, index$1 as Flows, type FullMessage, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, type IntermediateData, type Interval, type Invocation, InvocationContext, type JudgeModelOptions, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, type LlmModel, type LlmModelConfig, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LocalEvalService, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, McpCoinGeckoPro, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntents, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, McpUpbit, index$4 as Memory, type MessagePart, type MetricInfo, type MetricValueInfo, index$6 as Models, type MultiAgentResponse, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, type PerInvocationResult, PlanReActPlanner, PrebuiltMetrics, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RougeEvaluator, RunConfig, Runner, type RunnerAskReturn, SafetyEvaluatorV1, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionInput, type SessionOptions, index$3 as Sessions, type SingleAfterModelCallback, type SingleAfterToolCallback, type SingleAgentCallback, type SingleBeforeModelCallback, type SingleBeforeToolCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$7 as Tools, TrajectoryEvaluator, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiRagMemoryService, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, convertMcpToolToBaseTool, 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

@@ -674,6 +674,18 @@ declare class ReadonlyContext {
      * The name of the agent that is currently running.
      */
     get agentName(): string;
+    /**
+     * The application name for this invocation. READONLY field.
+     */
+    get appName(): string;
+    /**
+     * The user ID for this invocation. READONLY field.
+     */
+    get userId(): string;
+    /**
+     * The session ID for this invocation. READONLY field.
+     */
+    get sessionId(): string;
     /**
      * The state of the current session. READONLY field.
      */
@@ -1255,7 +1267,7 @@ type ToolUnion = BaseTool | ((...args: any[]) => any);
 type SingleBeforeModelCallback = (args: {
     callbackContext: CallbackContext;
     llmRequest: LlmRequest;
-}) => LlmResponse | null | Promise<LlmResponse | null>;
+}) => LlmResponse | null | undefined | Promise<LlmResponse | null | undefined>;
 /**
  * Before model callback type (single or array)
  */
@@ -1266,7 +1278,7 @@ type BeforeModelCallback = SingleBeforeModelCallback | SingleBeforeModelCallback
 type SingleAfterModelCallback = (args: {
     callbackContext: CallbackContext;
     llmResponse: LlmResponse;
-}) => LlmResponse | null | Promise<LlmResponse | null>;
+}) => LlmResponse | null | undefined | Promise<LlmResponse | null | undefined>;
 /**
  * After model callback type (single or array)
  */
@@ -1274,7 +1286,7 @@ type AfterModelCallback = SingleAfterModelCallback | SingleAfterModelCallback[];
 /**
  * Single before tool callback type
  */
-type SingleBeforeToolCallback = (tool: BaseTool, args: Record<string, any>, toolContext: ToolContext) => Record<string, any> | null | Promise<Record<string, any> | null>;
+type SingleBeforeToolCallback = (tool: BaseTool, args: Record<string, any>, toolContext: ToolContext) => Record<string, any> | null | undefined | Promise<Record<string, any> | null | undefined>;
 /**
  * Before tool callback type (single or array)
  */
@@ -1282,7 +1294,7 @@ type BeforeToolCallback = SingleBeforeToolCallback | SingleBeforeToolCallback[];
 /**
  * Single after tool callback type
  */
-type SingleAfterToolCallback = (tool: BaseTool, args: Record<string, any>, toolContext: ToolContext, toolResponse: Record<string, any>) => Record<string, any> | null | Promise<Record<string, any> | null>;
+type SingleAfterToolCallback = (tool: BaseTool, args: Record<string, any>, toolContext: ToolContext, toolResponse: Record<string, any>) => Record<string, any> | null | undefined | Promise<Record<string, any> | null | undefined>;
 /**
  * After tool callback type (single or array)
  */
@@ -2275,6 +2287,24 @@ declare function createSamplingHandler(handler: SamplingHandler): SamplingHandle
  *
  * @example
  * ```typescript
+ * // Using remote MCP endpoints (CoinGecko):
+ * const coinGeckoTools = await McpCoinGecko().getTools();
+ *
+ * const coinGeckoProTools = await McpCoinGeckoPro({
+ *   env: {
+ *     COINGECKO_PRO_API_KEY: env.COINGECKO_PRO_API_KEY
+ *   }
+ * }).getTools();
+ *
+ * const cryptoAgent = new LlmAgent({
+ *   name: "crypto_assistant",
+ *   model: "gemini-2.5-flash",
+ *   tools: [...coinGeckoTools, ...coinGeckoProTools],
+ * });
+ * ```
+ *
+ * @example
+ * ```typescript
  * // Using MCP servers with sampling handlers:
  * import { createSamplingHandler, LlmResponse } from "@iqai/adk";
  *
@@ -2388,11 +2418,18 @@ declare function McpTelegram(config?: McpServerConfig): McpToolset;
  */
 declare function McpDiscord(config?: McpServerConfig): McpToolset;
 /**
- * MCP CoinGecko - Access cryptocurrency market data and analytics
+ * MCP CoinGecko - Access cryptocurrency market data and analytics via remote endpoint
  *
- * Optional env vars: COINGECKO_PRO_API_KEY, COINGECKO_DEMO_API_KEY, COINGECKO_ENVIRONMENT
+ * Uses the public CoinGecko MCP API endpoint. No API key required for basic functionality.
  */
 declare function McpCoinGecko(config?: McpServerConfig): McpToolset;
+/**
+ * MCP CoinGecko Pro - Access premium cryptocurrency market data and analytics via remote endpoint
+ *
+ * Uses the professional CoinGecko MCP API endpoint with enhanced features and higher rate limits.
+ * Requires a CoinGecko Pro API subscription.
+ */
+declare function McpCoinGeckoPro(config?: McpServerConfig): McpToolset;
 /**
  * MCP Upbit - Interact with the Upbit cryptocurrency exchange
  *
@@ -2528,6 +2565,7 @@ declare const index$7_McpAbi: typeof McpAbi;
 declare const index$7_McpAtp: typeof McpAtp;
 declare const index$7_McpBamm: typeof McpBamm;
 declare const index$7_McpCoinGecko: typeof McpCoinGecko;
+declare const index$7_McpCoinGeckoPro: typeof McpCoinGeckoPro;
 type index$7_McpConfig = McpConfig;
 declare const index$7_McpDiscord: typeof McpDiscord;
 type index$7_McpError = McpError;
@@ -2571,7 +2609,7 @@ declare const index$7_jsonSchemaToDeclaration: typeof jsonSchemaToDeclaration;
 declare const index$7_mcpSchemaToParameters: typeof mcpSchemaToParameters;
 declare const index$7_normalizeJsonSchema: typeof normalizeJsonSchema;
 declare namespace index$7 {
-  export { index$7_AgentTool as AgentTool, type index$7_AgentToolConfig as AgentToolConfig, type index$7_BaseAgentType as BaseAgentType, index$7_BaseTool as BaseTool, type index$7_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, type index$7_CreateToolConfig as CreateToolConfig, type index$7_CreateToolConfigWithSchema as CreateToolConfigWithSchema, type index$7_CreateToolConfigWithoutSchema as CreateToolConfigWithoutSchema, index$7_ExitLoopTool as ExitLoopTool, index$7_FileOperationsTool as FileOperationsTool, index$7_FunctionTool as FunctionTool, index$7_GetUserChoiceTool as GetUserChoiceTool, index$7_GoogleSearch as GoogleSearch, index$7_HttpRequestTool as HttpRequestTool, index$7_LoadArtifactsTool as LoadArtifactsTool, index$7_LoadMemoryTool as LoadMemoryTool, index$7_McpAbi as McpAbi, index$7_McpAtp as McpAtp, index$7_McpBamm as McpBamm, index$7_McpCoinGecko as McpCoinGecko, type index$7_McpConfig as McpConfig, index$7_McpDiscord as McpDiscord, index$7_McpError as McpError, index$7_McpErrorType as McpErrorType, index$7_McpFilesystem as McpFilesystem, index$7_McpFraxlend as McpFraxlend, index$7_McpGeneric as McpGeneric, index$7_McpIqWiki as McpIqWiki, index$7_McpMemory as McpMemory, index$7_McpNearAgent as McpNearAgent, index$7_McpNearIntents as McpNearIntents, index$7_McpOdos as McpOdos, index$7_McpSamplingHandler as McpSamplingHandler, type index$7_McpSamplingRequest as McpSamplingRequest, type index$7_McpSamplingResponse as McpSamplingResponse, type index$7_McpServerConfig as McpServerConfig, index$7_McpTelegram as McpTelegram, index$7_McpToolset as McpToolset, type index$7_McpTransportType as McpTransportType, index$7_McpUpbit as McpUpbit, type index$7_SamplingHandler as SamplingHandler, type index$7_ToolConfig as ToolConfig, index$7_ToolContext as ToolContext, index$7_TransferToAgentTool as TransferToAgentTool, index$7_UserInteractionTool as UserInteractionTool, index$7_adkToMcpToolType as adkToMcpToolType, index$7_buildFunctionDeclaration as buildFunctionDeclaration, index$7_convertMcpToolToBaseTool as convertMcpToolToBaseTool, index$7_createFunctionTool as createFunctionTool, index$7_createSamplingHandler as createSamplingHandler, index$7_createTool as createTool, index$7_getMcpTools as getMcpTools, index$7_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$7_mcpSchemaToParameters as mcpSchemaToParameters, index$7_normalizeJsonSchema as normalizeJsonSchema };
+  export { index$7_AgentTool as AgentTool, type index$7_AgentToolConfig as AgentToolConfig, type index$7_BaseAgentType as BaseAgentType, index$7_BaseTool as BaseTool, type index$7_BuildFunctionDeclarationOptions as BuildFunctionDeclarationOptions, type index$7_CreateToolConfig as CreateToolConfig, type index$7_CreateToolConfigWithSchema as CreateToolConfigWithSchema, type index$7_CreateToolConfigWithoutSchema as CreateToolConfigWithoutSchema, index$7_ExitLoopTool as ExitLoopTool, index$7_FileOperationsTool as FileOperationsTool, index$7_FunctionTool as FunctionTool, index$7_GetUserChoiceTool as GetUserChoiceTool, index$7_GoogleSearch as GoogleSearch, index$7_HttpRequestTool as HttpRequestTool, index$7_LoadArtifactsTool as LoadArtifactsTool, index$7_LoadMemoryTool as LoadMemoryTool, index$7_McpAbi as McpAbi, index$7_McpAtp as McpAtp, index$7_McpBamm as McpBamm, index$7_McpCoinGecko as McpCoinGecko, index$7_McpCoinGeckoPro as McpCoinGeckoPro, type index$7_McpConfig as McpConfig, index$7_McpDiscord as McpDiscord, index$7_McpError as McpError, index$7_McpErrorType as McpErrorType, index$7_McpFilesystem as McpFilesystem, index$7_McpFraxlend as McpFraxlend, index$7_McpGeneric as McpGeneric, index$7_McpIqWiki as McpIqWiki, index$7_McpMemory as McpMemory, index$7_McpNearAgent as McpNearAgent, index$7_McpNearIntents as McpNearIntents, index$7_McpOdos as McpOdos, index$7_McpSamplingHandler as McpSamplingHandler, type index$7_McpSamplingRequest as McpSamplingRequest, type index$7_McpSamplingResponse as McpSamplingResponse, type index$7_McpServerConfig as McpServerConfig, index$7_McpTelegram as McpTelegram, index$7_McpToolset as McpToolset, type index$7_McpTransportType as McpTransportType, index$7_McpUpbit as McpUpbit, type index$7_SamplingHandler as SamplingHandler, type index$7_ToolConfig as ToolConfig, index$7_ToolContext as ToolContext, index$7_TransferToAgentTool as TransferToAgentTool, index$7_UserInteractionTool as UserInteractionTool, index$7_adkToMcpToolType as adkToMcpToolType, index$7_buildFunctionDeclaration as buildFunctionDeclaration, index$7_convertMcpToolToBaseTool as convertMcpToolToBaseTool, index$7_createFunctionTool as createFunctionTool, index$7_createSamplingHandler as createSamplingHandler, index$7_createTool as createTool, index$7_getMcpTools as getMcpTools, index$7_jsonSchemaToDeclaration as jsonSchemaToDeclaration, index$7_mcpSchemaToParameters as mcpSchemaToParameters, index$7_normalizeJsonSchema as normalizeJsonSchema };
 }
 
 /**
@@ -3997,6 +4035,10 @@ interface AgentBuilderConfig {
     subAgents?: BaseAgent[];
     beforeAgentCallback?: BeforeAgentCallback;
     afterAgentCallback?: AfterAgentCallback;
+    beforeModelCallback?: BeforeModelCallback;
+    afterModelCallback?: AfterModelCallback;
+    beforeToolCallback?: BeforeToolCallback;
+    afterToolCallback?: AfterToolCallback;
     maxIterations?: number;
     nodes?: LangGraphNode[];
     rootNode?: string;
@@ -4207,6 +4249,30 @@ declare class AgentBuilder<TOut = string, TMulti extends boolean = false> {
      * @returns This builder instance for chaining
      */
     withAfterAgentCallback(callback: AfterAgentCallback): this;
+    /**
+     * Set the before model callback for LLM interaction
+     * @param callback Callback to invoke before calling the LLM
+     * @returns This builder instance for chaining
+     */
+    withBeforeModelCallback(callback: BeforeModelCallback): this;
+    /**
+     * Set the after model callback for LLM interaction
+     * @param callback Callback to invoke after receiving LLM response
+     * @returns This builder instance for chaining
+     */
+    withAfterModelCallback(callback: AfterModelCallback): this;
+    /**
+     * Set the before tool callback for tool execution
+     * @param callback Callback to invoke before running a tool
+     * @returns This builder instance for chaining
+     */
+    withBeforeToolCallback(callback: BeforeToolCallback): this;
+    /**
+     * Set the after tool callback for tool execution
+     * @param callback Callback to invoke after running a tool
+     * @returns This builder instance for chaining
+     */
+    withAfterToolCallback(callback: AfterToolCallback): this;
     /**
      * Provide an already constructed agent instance. Further definition-mutating calls
      * (model/tools/instruction/etc.) will be ignored with a dev warning.
@@ -4422,14 +4488,46 @@ declare class InMemoryMemoryService implements BaseMemoryService {
     clear(): void;
 }
 
+/**
+ * A memory service that uses Vertex AI RAG for storage and retrieval.
+ */
+declare class VertexAiRagMemoryService implements BaseMemoryService {
+    private _vertexRagStore;
+    /**
+     * Initializes a VertexAiRagMemoryService.
+     *
+     * @param ragCorpus The name of the Vertex AI RAG corpus to use. Format:
+     *   `projects/{project}/locations/{location}/ragCorpora/{rag_corpus_id}`
+     *   or `{rag_corpus_id}`
+     * @param similarityTopK The number of contexts to retrieve.
+     * @param vectorDistanceThreshold Only returns contexts with vector distance
+     *   smaller than the threshold.
+     */
+    constructor(ragCorpus?: string, similarityTopK?: number, vectorDistanceThreshold?: number);
+    /**
+     * Adds a session to the memory service
+     */
+    addSessionToMemory(session: Session): Promise<void>;
+    /**
+     * Searches for sessions that match the query using rag.retrieval_query
+     */
+    searchMemory(options: {
+        appName: string;
+        userId: string;
+        query: string;
+    }): Promise<SearchMemoryResponse>;
+}
+
 /**
  * Memory Services for the Agent Development Kit
  */
 
 type index$4_InMemoryMemoryService = InMemoryMemoryService;
 declare const index$4_InMemoryMemoryService: typeof InMemoryMemoryService;
+type index$4_VertexAiRagMemoryService = VertexAiRagMemoryService;
+declare const index$4_VertexAiRagMemoryService: typeof VertexAiRagMemoryService;
 declare namespace index$4 {
-  export { index$4_InMemoryMemoryService as InMemoryMemoryService };
+  export { index$4_InMemoryMemoryService as InMemoryMemoryService, index$4_VertexAiRagMemoryService as VertexAiRagMemoryService };
 }
 
 /**
@@ -5575,4 +5673,4 @@ declare const traceLlmCall: (invocationContext: InvocationContext, eventId: stri
 
 declare const VERSION = "0.1.0";
 
-export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, type AfterModelCallback, type AfterToolCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentBuilderWithSchema, AgentEvaluator, AgentTool, type AgentToolConfig, type AgentType, index$5 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, type BaseAgentType, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BeforeModelCallback, type BeforeToolCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, type EvalCase, type EvalCaseResult, type EvalMetric, type EvalMetricResult, type EvalMetricResultPerInvocation, EvalResult, type EvalSet, type EvalSetResult, EvalStatus, type EvaluateConfig, index as Evaluation, type EvaluationResult, Evaluator, Event, EventActions, index$2 as Events, ExitLoopTool, type File, FileOperationsTool, FinalResponseMatchV2Evaluator, index$1 as Flows, type FullMessage, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, type IntermediateData, type Interval, type Invocation, InvocationContext, type JudgeModelOptions, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, type LlmModel, type LlmModelConfig, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LocalEvalService, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntents, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, McpUpbit, index$4 as Memory, type MessagePart, type MetricInfo, type MetricValueInfo, index$6 as Models, type MultiAgentResponse, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, type PerInvocationResult, PlanReActPlanner, PrebuiltMetrics, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RougeEvaluator, RunConfig, Runner, type RunnerAskReturn, SafetyEvaluatorV1, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionInput, type SessionOptions, index$3 as Sessions, type SingleAfterModelCallback, type SingleAfterToolCallback, type SingleAgentCallback, type SingleBeforeModelCallback, type SingleBeforeToolCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$7 as Tools, TrajectoryEvaluator, 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, convertMcpToolToBaseTool, createAuthToolArguments, createBranchContextForSubAgent, createDatabaseSessionService, createFunctionTool, createMysqlSessionService, createPostgresSessionService, createSamplingHandler, createSqliteSessionService, createTool, generateAuthEvent, generateClientFunctionCallId, getLongRunningFunctionCalls, getMcpTools, handleFunctionCallsAsync, handleFunctionCallsLive, requestProcessor$5 as identityRequestProcessor, initializeTelemetry, injectSessionState, requestProcessor$4 as instructionsRequestProcessor, isEnhancedAuthConfig, jsonSchemaToDeclaration, mcpSchemaToParameters, mergeAgentRun, mergeParallelFunctionResponseEvents, newInvocationContextId, requestProcessor$1 as nlPlanningRequestProcessor, responseProcessor$1 as nlPlanningResponseProcessor, normalizeJsonSchema, populateClientFunctionCallId, registerProviders, removeClientFunctionCallId, requestProcessor$7 as requestProcessor, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };
+export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, type AfterModelCallback, type AfterToolCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentBuilderWithSchema, AgentEvaluator, AgentTool, type AgentToolConfig, type AgentType, index$5 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, type BaseAgentType, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BeforeModelCallback, type BeforeToolCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, type EvalCase, type EvalCaseResult, type EvalMetric, type EvalMetricResult, type EvalMetricResultPerInvocation, EvalResult, type EvalSet, type EvalSetResult, EvalStatus, type EvaluateConfig, index as Evaluation, type EvaluationResult, Evaluator, Event, EventActions, index$2 as Events, ExitLoopTool, type File, FileOperationsTool, FinalResponseMatchV2Evaluator, index$1 as Flows, type FullMessage, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, type IntermediateData, type Interval, type Invocation, InvocationContext, type JudgeModelOptions, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, type LlmModel, type LlmModelConfig, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LocalEvalService, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, McpCoinGeckoPro, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntents, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, McpUpbit, index$4 as Memory, type MessagePart, type MetricInfo, type MetricValueInfo, index$6 as Models, type MultiAgentResponse, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, type PerInvocationResult, PlanReActPlanner, PrebuiltMetrics, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RougeEvaluator, RunConfig, Runner, type RunnerAskReturn, SafetyEvaluatorV1, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionInput, type SessionOptions, index$3 as Sessions, type SingleAfterModelCallback, type SingleAfterToolCallback, type SingleAgentCallback, type SingleBeforeModelCallback, type SingleBeforeToolCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$7 as Tools, TrajectoryEvaluator, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiRagMemoryService, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, convertMcpToolToBaseTool, 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.js

@@ -3521,6 +3521,24 @@ var ReadonlyContext = class {
   get agentName() {
     return this._invocationContext.agent.name;
   }
+  /**
+   * The application name for this invocation. READONLY field.
+   */
+  get appName() {
+    return this._invocationContext.appName;
+  }
+  /**
+   * The user ID for this invocation. READONLY field.
+   */
+  get userId() {
+    return this._invocationContext.userId;
+  }
+  /**
+   * The session ID for this invocation. READONLY field.
+   */
+  get sessionId() {
+    return this._invocationContext.session.id;
+  }
   /**
    * The state of the current session. READONLY field.
    */
@@ -4079,6 +4097,7 @@ __export(tools_exports, {
   McpAtp: () => McpAtp,
   McpBamm: () => McpBamm,
   McpCoinGecko: () => McpCoinGecko,
+  McpCoinGeckoPro: () => McpCoinGeckoPro,
   McpDiscord: () => McpDiscord,
   McpError: () => McpError,
   McpErrorType: () => McpErrorType,
@@ -6123,7 +6142,7 @@ var McpToolAdapter = (_class23 = class extends BaseTool {
 }, _class23);
 
 // src/tools/mcp/servers.ts
-function createMcpConfig(name, packageName, config = {}) {
+function createMcpConfig(name, packageNameOrUrl, config = {}) {
   const {
     debug,
     description,
@@ -6140,17 +6159,25 @@ function createMcpConfig(name, packageName, config = {}) {
   if (!env.PATH) {
     env.PATH = process.env.PATH || "";
   }
+  let isUrl;
+  try {
+    const url = new URL(packageNameOrUrl);
+    isUrl = url.protocol === "http:" || url.protocol === "https:";
+  } catch (e5) {
+    isUrl = false;
+  }
+  const transport = {
+    mode: "stdio",
+    command: "npx",
+    args: isUrl ? ["-y", "mcp-remote@latest", packageNameOrUrl] : ["-y", packageNameOrUrl],
+    env
+  };
   return {
     name,
     description: description || `Client for ${name}`,
     debug: debug || false,
     retryOptions: retryOptions || { maxRetries: 2, initialDelay: 200 },
-    transport: {
-      mode: "stdio",
-      command: "npx",
-      args: ["-y", packageName],
-      env
-    },
+    transport,
     samplingHandler
   };
 }
@@ -6229,7 +6256,15 @@ function McpDiscord(config = {}) {
 function McpCoinGecko(config = {}) {
   const mcpConfig = createMcpConfig(
     "CoinGecko MCP Client",
-    "@coingecko/coingecko-mcp",
+    "https://mcp.api.coingecko.com/mcp",
+    config
+  );
+  return new McpToolset(mcpConfig);
+}
+function McpCoinGeckoPro(config = {}) {
+  const mcpConfig = createMcpConfig(
+    "CoinGecko Pro MCP Client",
+    "https://mcp.pro-api.coingecko.com/mcp",
     config
   );
   return new McpToolset(mcpConfig);
@@ -7480,7 +7515,7 @@ var CodeExecutionUtils = class _CodeExecutionUtils {
   static isBase64Encoded(str) {
     try {
       return btoa(atob(str)) === str;
-    } catch (e5) {
+    } catch (e6) {
       return false;
     }
   }
@@ -8520,7 +8555,7 @@ var InstructionsLlmRequestProcessor = class extends BaseLlmRequestProcessor {
         llmRequest.appendInstructions([
           'IMPORTANT: After any tool calls, function calls, or agent transfers have completed, produce ONE final assistant message whose entire content is ONLY the JSON object that conforms to the schema provided above. Do NOT include any explanatory text, markdown, or additional messages. Do NOT wrap the JSON in code fences (for example, do NOT use ```json or ```). If you cannot produce valid JSON that matches the schema, return a JSON object with an "error" field describing the problem.'
         ]);
-      } catch (e6) {
+      } catch (e7) {
       }
     }
     for await (const _ of []) {
@@ -10860,6 +10895,46 @@ var AgentBuilder = (_class33 = class _AgentBuilder {
     this.config.afterAgentCallback = callback;
     return this;
   }
+  /**
+   * Set the before model callback for LLM interaction
+   * @param callback Callback to invoke before calling the LLM
+   * @returns This builder instance for chaining
+   */
+  withBeforeModelCallback(callback) {
+    this.warnIfLocked("withBeforeModelCallback");
+    this.config.beforeModelCallback = callback;
+    return this;
+  }
+  /**
+   * Set the after model callback for LLM interaction
+   * @param callback Callback to invoke after receiving LLM response
+   * @returns This builder instance for chaining
+   */
+  withAfterModelCallback(callback) {
+    this.warnIfLocked("withAfterModelCallback");
+    this.config.afterModelCallback = callback;
+    return this;
+  }
+  /**
+   * Set the before tool callback for tool execution
+   * @param callback Callback to invoke before running a tool
+   * @returns This builder instance for chaining
+   */
+  withBeforeToolCallback(callback) {
+    this.warnIfLocked("withBeforeToolCallback");
+    this.config.beforeToolCallback = callback;
+    return this;
+  }
+  /**
+   * Set the after tool callback for tool execution
+   * @param callback Callback to invoke after running a tool
+   * @returns This builder instance for chaining
+   */
+  withAfterToolCallback(callback) {
+    this.warnIfLocked("withAfterToolCallback");
+    this.config.afterToolCallback = callback;
+    return this;
+  }
   /**
    * Provide an already constructed agent instance. Further definition-mutating calls
    * (model/tools/instruction/etc.) will be ignored with a dev warning.
@@ -11133,6 +11208,10 @@ var AgentBuilder = (_class33 = class _AgentBuilder {
           subAgents: this.config.subAgents,
           beforeAgentCallback: this.config.beforeAgentCallback,
           afterAgentCallback: this.config.afterAgentCallback,
+          beforeModelCallback: this.config.beforeModelCallback,
+          afterModelCallback: this.config.afterModelCallback,
+          beforeToolCallback: this.config.beforeToolCallback,
+          afterToolCallback: this.config.afterToolCallback,
           memoryService: this.memoryService,
           artifactService: this.artifactService,
           outputKey: this.config.outputKey,
@@ -11295,9 +11374,183 @@ var AgentBuilder = (_class33 = class _AgentBuilder {
 // src/memory/index.ts
 var memory_exports = {};
 __export(memory_exports, {
-  InMemoryMemoryService: () => InMemoryMemoryService
+  InMemoryMemoryService: () => InMemoryMemoryService,
+  VertexAiRagMemoryService: () => VertexAiRagMemoryService
 });
 
+// src/memory/vertex-ai-rag-memory-service.ts
+
+var _fs = require('fs');
+var _os = require('os');
+
+var rag = {
+  async upload_file(options) {
+    console.log("Mock upload_file:", options);
+  },
+  async retrieval_query(options) {
+    console.log("Mock retrieval_query:", options);
+    return { contexts: { contexts: [] } };
+  }
+};
+function _mergeEventLists(eventLists) {
+  const merged = [];
+  while (eventLists.length > 0) {
+    const current = eventLists.shift();
+    const currentTs = new Set(current.map((event) => event.timestamp));
+    let mergeFound = true;
+    while (mergeFound) {
+      mergeFound = false;
+      const remaining = [];
+      for (const other of eventLists) {
+        const otherTs = new Set(other.map((event) => event.timestamp));
+        const hasOverlap = Array.from(currentTs).some((ts) => otherTs.has(ts));
+        if (hasOverlap) {
+          const newEvents = other.filter((e) => !currentTs.has(e.timestamp));
+          current.push(...newEvents);
+          newEvents.forEach((e) => currentTs.add(e.timestamp));
+          mergeFound = true;
+        } else {
+          remaining.push(other);
+        }
+      }
+      eventLists.splice(0, eventLists.length, ...remaining);
+    }
+    merged.push(current);
+  }
+  return merged;
+}
+var VertexAiRagMemoryService = class {
+  
+  /**
+   * Initializes a VertexAiRagMemoryService.
+   *
+   * @param ragCorpus The name of the Vertex AI RAG corpus to use. Format:
+   *   `projects/{project}/locations/{location}/ragCorpora/{rag_corpus_id}`
+   *   or `{rag_corpus_id}`
+   * @param similarityTopK The number of contexts to retrieve.
+   * @param vectorDistanceThreshold Only returns contexts with vector distance
+   *   smaller than the threshold.
+   */
+  constructor(ragCorpus, similarityTopK, vectorDistanceThreshold = 10) {
+    this._vertexRagStore = {
+      rag_resources: ragCorpus ? [{ rag_corpus: ragCorpus }] : [],
+      similarity_top_k: similarityTopK,
+      vector_distance_threshold: vectorDistanceThreshold
+    };
+  }
+  /**
+   * Adds a session to the memory service
+   */
+  async addSessionToMemory(session) {
+    const tempFileName = `temp_${_crypto.randomUUID.call(void 0, )}.txt`;
+    const tempFilePath = _path.join.call(void 0, _os.tmpdir.call(void 0, ), tempFileName);
+    try {
+      const outputLines = [];
+      for (const event of session.events) {
+        if (!event.content || !event.content.parts) {
+          continue;
+        }
+        const textParts = event.content.parts.filter((part) => part.text).map((part) => part.text.replace(/\n/g, " "));
+        if (textParts.length > 0) {
+          outputLines.push(
+            JSON.stringify({
+              author: event.author,
+              timestamp: event.timestamp,
+              text: textParts.join(".")
+            })
+          );
+        }
+      }
+      const outputString = outputLines.join("\n");
+      _fs.writeFileSync.call(void 0, tempFilePath, outputString, "utf8");
+      if (!this._vertexRagStore.rag_resources || this._vertexRagStore.rag_resources.length === 0) {
+        throw new Error("Rag resources must be set.");
+      }
+      for (const ragResource of this._vertexRagStore.rag_resources) {
+        await rag.upload_file({
+          corpus_name: ragResource.rag_corpus,
+          path: tempFilePath,
+          // This is the temp workaround as upload file does not support
+          // adding metadata, thus use display_name to store the session info.
+          display_name: `${session.appName}.${session.userId}.${session.id}`
+        });
+      }
+    } finally {
+      try {
+        _fs.unlinkSync.call(void 0, tempFilePath);
+      } catch (error) {
+        console.warn("Failed to delete temporary file:", tempFilePath, error);
+      }
+    }
+  }
+  /**
+   * Searches for sessions that match the query using rag.retrieval_query
+   */
+  async searchMemory(options) {
+    const { appName, userId, query } = options;
+    const response = await rag.retrieval_query({
+      text: query,
+      rag_resources: this._vertexRagStore.rag_resources,
+      rag_corpora: this._vertexRagStore.rag_corpora,
+      similarity_top_k: this._vertexRagStore.similarity_top_k,
+      vector_distance_threshold: this._vertexRagStore.vector_distance_threshold
+    });
+    const memoryResults = [];
+    const sessionEventsMap = /* @__PURE__ */ new Map();
+    for (const context4 of response.contexts.contexts) {
+      if (!context4.source_display_name.startsWith(`${appName}.${userId}.`)) {
+        continue;
+      }
+      const sessionId = context4.source_display_name.split(".").pop();
+      const events = [];
+      if (context4.text) {
+        const lines = context4.text.split("\n");
+        for (const line of lines) {
+          const trimmedLine = line.trim();
+          if (!trimmedLine) {
+            continue;
+          }
+          try {
+            const eventData = JSON.parse(trimmedLine);
+            const author = eventData.author || "";
+            const timestamp = Number.parseFloat(eventData.timestamp || "0");
+            const text = eventData.text || "";
+            const content = {
+              parts: [{ text }]
+            };
+            const event = {
+              author,
+              timestamp,
+              content
+            };
+            events.push(event);
+          } catch (e8) {
+          }
+        }
+      }
+      if (sessionEventsMap.has(sessionId)) {
+        sessionEventsMap.get(sessionId).push(events);
+      } else {
+        sessionEventsMap.set(sessionId, [events]);
+      }
+    }
+    for (const [sessionId, eventLists] of sessionEventsMap.entries()) {
+      const mergedEventLists = _mergeEventLists(eventLists);
+      for (const events of mergedEventLists) {
+        const sortedEvents = events.sort((a, b) => a.timestamp - b.timestamp).filter((event) => event.content);
+        memoryResults.push(
+          ...sortedEvents.map((event) => ({
+            author: event.author,
+            content: event.content,
+            timestamp: formatTimestamp(event.timestamp)
+          }))
+        );
+      }
+    }
+    return { memories: memoryResults };
+  }
+};
+
 // src/sessions/index.ts
 var sessions_exports = {};
 __export(sessions_exports, {
@@ -11703,7 +11956,7 @@ var DatabaseSessionService = (_class34 = class extends BaseSessionService {
     if (!jsonString) return defaultValue;
     try {
       return JSON.parse(jsonString);
-    } catch (e7) {
+    } catch (e9) {
       return defaultValue;
     }
   }
@@ -13729,4 +13982,6 @@ var VERSION = "0.1.0";
 
 
 
-exports.AF_FUNCTION_CALL_ID_PREFIX = AF_FUNCTION_CALL_ID_PREFIX; exports.Agent = LlmAgent; exports.AgentBuilder = AgentBuilder; exports.AgentEvaluator = AgentEvaluator; exports.AgentTool = AgentTool; exports.Agents = agents_exports; exports.AiSdkLlm = AiSdkLlm; exports.AnthropicLlm = AnthropicLlm; exports.ApiKeyCredential = ApiKeyCredential; exports.ApiKeyScheme = ApiKeyScheme; exports.AuthConfig = AuthConfig; exports.AuthCredential = AuthCredential; exports.AuthCredentialType = AuthCredentialType; exports.AuthHandler = AuthHandler; exports.AuthScheme = AuthScheme; exports.AuthSchemeType = AuthSchemeType; exports.AuthTool = AuthTool; exports.AutoFlow = AutoFlow; exports.BaseAgent = BaseAgent; exports.BaseCodeExecutor = BaseCodeExecutor; exports.BaseLLMConnection = BaseLLMConnection; exports.BaseLlm = BaseLlm; exports.BaseLlmFlow = BaseLlmFlow; exports.BaseLlmRequestProcessor = BaseLlmRequestProcessor; exports.BaseLlmResponseProcessor = BaseLlmResponseProcessor; exports.BasePlanner = BasePlanner; exports.BaseSessionService = BaseSessionService; exports.BaseTool = BaseTool; exports.BasicAuthCredential = BasicAuthCredential; exports.BearerTokenCredential = BearerTokenCredential; exports.BuiltInCodeExecutor = BuiltInCodeExecutor; exports.BuiltInPlanner = BuiltInPlanner; exports.CallbackContext = CallbackContext; exports.CodeExecutionUtils = CodeExecutionUtils; exports.CodeExecutorContext = CodeExecutorContext; exports.DatabaseSessionService = DatabaseSessionService; exports.EnhancedAuthConfig = EnhancedAuthConfig; exports.EvalResult = EvalResult; exports.EvalStatus = EvalStatus; exports.Evaluation = evaluation_exports; exports.Evaluator = Evaluator; exports.Event = Event; exports.EventActions = EventActions; exports.Events = events_exports; exports.ExitLoopTool = ExitLoopTool; exports.FileOperationsTool = FileOperationsTool; exports.FinalResponseMatchV2Evaluator = FinalResponseMatchV2Evaluator; exports.Flows = flows_exports; exports.FunctionTool = FunctionTool; exports.GcsArtifactService = GcsArtifactService; exports.GetUserChoiceTool = GetUserChoiceTool; exports.GoogleLlm = GoogleLlm; exports.GoogleSearch = GoogleSearch; exports.HttpRequestTool = HttpRequestTool; exports.HttpScheme = HttpScheme; exports.InMemoryArtifactService = InMemoryArtifactService; exports.InMemoryMemoryService = InMemoryMemoryService; exports.InMemoryRunner = InMemoryRunner; exports.InMemorySessionService = InMemorySessionService; exports.InvocationContext = InvocationContext; exports.LLMRegistry = LLMRegistry; exports.LangGraphAgent = LangGraphAgent; exports.LlmAgent = LlmAgent; exports.LlmCallsLimitExceededError = LlmCallsLimitExceededError; exports.LlmRequest = LlmRequest; exports.LlmResponse = LlmResponse; exports.LoadArtifactsTool = LoadArtifactsTool; exports.LoadMemoryTool = LoadMemoryTool; exports.LocalEvalService = LocalEvalService; exports.LoopAgent = LoopAgent; exports.McpAbi = McpAbi; exports.McpAtp = McpAtp; exports.McpBamm = McpBamm; exports.McpCoinGecko = McpCoinGecko; exports.McpDiscord = McpDiscord; exports.McpError = McpError; exports.McpErrorType = McpErrorType; exports.McpFilesystem = McpFilesystem; exports.McpFraxlend = McpFraxlend; exports.McpGeneric = McpGeneric; exports.McpIqWiki = McpIqWiki; exports.McpMemory = McpMemory; exports.McpNearAgent = McpNearAgent; exports.McpNearIntents = McpNearIntents; exports.McpOdos = McpOdos; exports.McpSamplingHandler = McpSamplingHandler; exports.McpTelegram = McpTelegram; exports.McpToolset = McpToolset; exports.McpUpbit = McpUpbit; exports.Memory = memory_exports; exports.Models = models_exports; exports.OAuth2Credential = OAuth2Credential; exports.OAuth2Scheme = OAuth2Scheme; exports.OpenAiLlm = OpenAiLlm; exports.OpenIdConnectScheme = OpenIdConnectScheme; exports.ParallelAgent = ParallelAgent; exports.PlanReActPlanner = PlanReActPlanner; exports.PrebuiltMetrics = PrebuiltMetrics; exports.REQUEST_EUC_FUNCTION_CALL_NAME = REQUEST_EUC_FUNCTION_CALL_NAME; exports.ReadonlyContext = ReadonlyContext; exports.RougeEvaluator = RougeEvaluator; exports.RunConfig = RunConfig; exports.Runner = Runner; exports.SafetyEvaluatorV1 = SafetyEvaluatorV1; exports.SequentialAgent = SequentialAgent; exports.Sessions = sessions_exports; exports.SingleFlow = SingleFlow; exports.State = State; exports.StreamingMode = StreamingMode; exports.TelemetryService = TelemetryService; exports.ToolContext = ToolContext; exports.Tools = tools_exports; exports.TrajectoryEvaluator = TrajectoryEvaluator; exports.TransferToAgentTool = TransferToAgentTool; exports.UserInteractionTool = UserInteractionTool; exports.VERSION = VERSION; exports.VertexAiSessionService = VertexAiSessionService; exports._findFunctionCallEventIfLastEventIsFunctionResponse = _findFunctionCallEventIfLastEventIsFunctionResponse; exports.adkToMcpToolType = adkToMcpToolType; exports.agentTransferRequestProcessor = requestProcessor8; exports.basicRequestProcessor = requestProcessor2; exports.buildFunctionDeclaration = buildFunctionDeclaration; exports.codeExecutionRequestProcessor = requestProcessor3; exports.codeExecutionResponseProcessor = responseProcessor; exports.contentRequestProcessor = requestProcessor4; exports.convertMcpToolToBaseTool = convertMcpToolToBaseTool; exports.createAuthToolArguments = createAuthToolArguments; exports.createBranchContextForSubAgent = createBranchContextForSubAgent; exports.createDatabaseSessionService = createDatabaseSessionService; exports.createFunctionTool = createFunctionTool; exports.createMysqlSessionService = createMysqlSessionService; exports.createPostgresSessionService = createPostgresSessionService; exports.createSamplingHandler = createSamplingHandler; exports.createSqliteSessionService = createSqliteSessionService; exports.createTool = createTool; exports.generateAuthEvent = generateAuthEvent; exports.generateClientFunctionCallId = generateClientFunctionCallId; exports.getLongRunningFunctionCalls = getLongRunningFunctionCalls; exports.getMcpTools = getMcpTools; exports.handleFunctionCallsAsync = handleFunctionCallsAsync; exports.handleFunctionCallsLive = handleFunctionCallsLive; exports.identityRequestProcessor = requestProcessor5; exports.initializeTelemetry = initializeTelemetry; exports.injectSessionState = injectSessionState; exports.instructionsRequestProcessor = requestProcessor6; exports.isEnhancedAuthConfig = isEnhancedAuthConfig; exports.jsonSchemaToDeclaration = jsonSchemaToDeclaration; exports.mcpSchemaToParameters = mcpSchemaToParameters; exports.mergeAgentRun = mergeAgentRun; exports.mergeParallelFunctionResponseEvents = mergeParallelFunctionResponseEvents; exports.newInvocationContextId = newInvocationContextId; exports.nlPlanningRequestProcessor = requestProcessor7; exports.nlPlanningResponseProcessor = responseProcessor2; exports.normalizeJsonSchema = normalizeJsonSchema; exports.populateClientFunctionCallId = populateClientFunctionCallId; exports.registerProviders = registerProviders; exports.removeClientFunctionCallId = removeClientFunctionCallId; exports.requestProcessor = requestProcessor; exports.shutdownTelemetry = shutdownTelemetry; exports.telemetryService = telemetryService; exports.traceLlmCall = traceLlmCall; exports.traceToolCall = traceToolCall; exports.tracer = tracer;
+
+
+exports.AF_FUNCTION_CALL_ID_PREFIX = AF_FUNCTION_CALL_ID_PREFIX; exports.Agent = LlmAgent; exports.AgentBuilder = AgentBuilder; exports.AgentEvaluator = AgentEvaluator; exports.AgentTool = AgentTool; exports.Agents = agents_exports; exports.AiSdkLlm = AiSdkLlm; exports.AnthropicLlm = AnthropicLlm; exports.ApiKeyCredential = ApiKeyCredential; exports.ApiKeyScheme = ApiKeyScheme; exports.AuthConfig = AuthConfig; exports.AuthCredential = AuthCredential; exports.AuthCredentialType = AuthCredentialType; exports.AuthHandler = AuthHandler; exports.AuthScheme = AuthScheme; exports.AuthSchemeType = AuthSchemeType; exports.AuthTool = AuthTool; exports.AutoFlow = AutoFlow; exports.BaseAgent = BaseAgent; exports.BaseCodeExecutor = BaseCodeExecutor; exports.BaseLLMConnection = BaseLLMConnection; exports.BaseLlm = BaseLlm; exports.BaseLlmFlow = BaseLlmFlow; exports.BaseLlmRequestProcessor = BaseLlmRequestProcessor; exports.BaseLlmResponseProcessor = BaseLlmResponseProcessor; exports.BasePlanner = BasePlanner; exports.BaseSessionService = BaseSessionService; exports.BaseTool = BaseTool; exports.BasicAuthCredential = BasicAuthCredential; exports.BearerTokenCredential = BearerTokenCredential; exports.BuiltInCodeExecutor = BuiltInCodeExecutor; exports.BuiltInPlanner = BuiltInPlanner; exports.CallbackContext = CallbackContext; exports.CodeExecutionUtils = CodeExecutionUtils; exports.CodeExecutorContext = CodeExecutorContext; exports.DatabaseSessionService = DatabaseSessionService; exports.EnhancedAuthConfig = EnhancedAuthConfig; exports.EvalResult = EvalResult; exports.EvalStatus = EvalStatus; exports.Evaluation = evaluation_exports; exports.Evaluator = Evaluator; exports.Event = Event; exports.EventActions = EventActions; exports.Events = events_exports; exports.ExitLoopTool = ExitLoopTool; exports.FileOperationsTool = FileOperationsTool; exports.FinalResponseMatchV2Evaluator = FinalResponseMatchV2Evaluator; exports.Flows = flows_exports; exports.FunctionTool = FunctionTool; exports.GcsArtifactService = GcsArtifactService; exports.GetUserChoiceTool = GetUserChoiceTool; exports.GoogleLlm = GoogleLlm; exports.GoogleSearch = GoogleSearch; exports.HttpRequestTool = HttpRequestTool; exports.HttpScheme = HttpScheme; exports.InMemoryArtifactService = InMemoryArtifactService; exports.InMemoryMemoryService = InMemoryMemoryService; exports.InMemoryRunner = InMemoryRunner; exports.InMemorySessionService = InMemorySessionService; exports.InvocationContext = InvocationContext; exports.LLMRegistry = LLMRegistry; exports.LangGraphAgent = LangGraphAgent; exports.LlmAgent = LlmAgent; exports.LlmCallsLimitExceededError = LlmCallsLimitExceededError; exports.LlmRequest = LlmRequest; exports.LlmResponse = LlmResponse; exports.LoadArtifactsTool = LoadArtifactsTool; exports.LoadMemoryTool = LoadMemoryTool; exports.LocalEvalService = LocalEvalService; exports.LoopAgent = LoopAgent; exports.McpAbi = McpAbi; exports.McpAtp = McpAtp; exports.McpBamm = McpBamm; exports.McpCoinGecko = McpCoinGecko; exports.McpCoinGeckoPro = McpCoinGeckoPro; exports.McpDiscord = McpDiscord; exports.McpError = McpError; exports.McpErrorType = McpErrorType; exports.McpFilesystem = McpFilesystem; exports.McpFraxlend = McpFraxlend; exports.McpGeneric = McpGeneric; exports.McpIqWiki = McpIqWiki; exports.McpMemory = McpMemory; exports.McpNearAgent = McpNearAgent; exports.McpNearIntents = McpNearIntents; exports.McpOdos = McpOdos; exports.McpSamplingHandler = McpSamplingHandler; exports.McpTelegram = McpTelegram; exports.McpToolset = McpToolset; exports.McpUpbit = McpUpbit; exports.Memory = memory_exports; exports.Models = models_exports; exports.OAuth2Credential = OAuth2Credential; exports.OAuth2Scheme = OAuth2Scheme; exports.OpenAiLlm = OpenAiLlm; exports.OpenIdConnectScheme = OpenIdConnectScheme; exports.ParallelAgent = ParallelAgent; exports.PlanReActPlanner = PlanReActPlanner; exports.PrebuiltMetrics = PrebuiltMetrics; exports.REQUEST_EUC_FUNCTION_CALL_NAME = REQUEST_EUC_FUNCTION_CALL_NAME; exports.ReadonlyContext = ReadonlyContext; exports.RougeEvaluator = RougeEvaluator; exports.RunConfig = RunConfig; exports.Runner = Runner; exports.SafetyEvaluatorV1 = SafetyEvaluatorV1; exports.SequentialAgent = SequentialAgent; exports.Sessions = sessions_exports; exports.SingleFlow = SingleFlow; exports.State = State; exports.StreamingMode = StreamingMode; exports.TelemetryService = TelemetryService; exports.ToolContext = ToolContext; exports.Tools = tools_exports; exports.TrajectoryEvaluator = TrajectoryEvaluator; exports.TransferToAgentTool = TransferToAgentTool; exports.UserInteractionTool = UserInteractionTool; exports.VERSION = VERSION; exports.VertexAiRagMemoryService = VertexAiRagMemoryService; exports.VertexAiSessionService = VertexAiSessionService; exports._findFunctionCallEventIfLastEventIsFunctionResponse = _findFunctionCallEventIfLastEventIsFunctionResponse; exports.adkToMcpToolType = adkToMcpToolType; exports.agentTransferRequestProcessor = requestProcessor8; exports.basicRequestProcessor = requestProcessor2; exports.buildFunctionDeclaration = buildFunctionDeclaration; exports.codeExecutionRequestProcessor = requestProcessor3; exports.codeExecutionResponseProcessor = responseProcessor; exports.contentRequestProcessor = requestProcessor4; exports.convertMcpToolToBaseTool = convertMcpToolToBaseTool; exports.createAuthToolArguments = createAuthToolArguments; exports.createBranchContextForSubAgent = createBranchContextForSubAgent; exports.createDatabaseSessionService = createDatabaseSessionService; exports.createFunctionTool = createFunctionTool; exports.createMysqlSessionService = createMysqlSessionService; exports.createPostgresSessionService = createPostgresSessionService; exports.createSamplingHandler = createSamplingHandler; exports.createSqliteSessionService = createSqliteSessionService; exports.createTool = createTool; exports.generateAuthEvent = generateAuthEvent; exports.generateClientFunctionCallId = generateClientFunctionCallId; exports.getLongRunningFunctionCalls = getLongRunningFunctionCalls; exports.getMcpTools = getMcpTools; exports.handleFunctionCallsAsync = handleFunctionCallsAsync; exports.handleFunctionCallsLive = handleFunctionCallsLive; exports.identityRequestProcessor = requestProcessor5; exports.initializeTelemetry = initializeTelemetry; exports.injectSessionState = injectSessionState; exports.instructionsRequestProcessor = requestProcessor6; exports.isEnhancedAuthConfig = isEnhancedAuthConfig; exports.jsonSchemaToDeclaration = jsonSchemaToDeclaration; exports.mcpSchemaToParameters = mcpSchemaToParameters; exports.mergeAgentRun = mergeAgentRun; exports.mergeParallelFunctionResponseEvents = mergeParallelFunctionResponseEvents; exports.newInvocationContextId = newInvocationContextId; exports.nlPlanningRequestProcessor = requestProcessor7; exports.nlPlanningResponseProcessor = responseProcessor2; exports.normalizeJsonSchema = normalizeJsonSchema; exports.populateClientFunctionCallId = populateClientFunctionCallId; exports.registerProviders = registerProviders; exports.removeClientFunctionCallId = removeClientFunctionCallId; exports.requestProcessor = requestProcessor; exports.shutdownTelemetry = shutdownTelemetry; exports.telemetryService = telemetryService; exports.traceLlmCall = traceLlmCall; exports.traceToolCall = traceToolCall; exports.tracer = tracer;

package/dist/index.mjs

@@ -3521,6 +3521,24 @@ var ReadonlyContext = class {
   get agentName() {
     return this._invocationContext.agent.name;
   }
+  /**
+   * The application name for this invocation. READONLY field.
+   */
+  get appName() {
+    return this._invocationContext.appName;
+  }
+  /**
+   * The user ID for this invocation. READONLY field.
+   */
+  get userId() {
+    return this._invocationContext.userId;
+  }
+  /**
+   * The session ID for this invocation. READONLY field.
+   */
+  get sessionId() {
+    return this._invocationContext.session.id;
+  }
   /**
    * The state of the current session. READONLY field.
    */
@@ -4079,6 +4097,7 @@ __export(tools_exports, {
   McpAtp: () => McpAtp,
   McpBamm: () => McpBamm,
   McpCoinGecko: () => McpCoinGecko,
+  McpCoinGeckoPro: () => McpCoinGeckoPro,
   McpDiscord: () => McpDiscord,
   McpError: () => McpError,
   McpErrorType: () => McpErrorType,
@@ -6123,7 +6142,7 @@ var McpToolAdapter = class extends BaseTool {
 };
 
 // src/tools/mcp/servers.ts
-function createMcpConfig(name, packageName, config = {}) {
+function createMcpConfig(name, packageNameOrUrl, config = {}) {
   const {
     debug,
     description,
@@ -6140,17 +6159,25 @@ function createMcpConfig(name, packageName, config = {}) {
   if (!env.PATH) {
     env.PATH = process.env.PATH || "";
   }
+  let isUrl;
+  try {
+    const url = new URL(packageNameOrUrl);
+    isUrl = url.protocol === "http:" || url.protocol === "https:";
+  } catch {
+    isUrl = false;
+  }
+  const transport = {
+    mode: "stdio",
+    command: "npx",
+    args: isUrl ? ["-y", "mcp-remote@latest", packageNameOrUrl] : ["-y", packageNameOrUrl],
+    env
+  };
   return {
     name,
     description: description || `Client for ${name}`,
     debug: debug || false,
     retryOptions: retryOptions || { maxRetries: 2, initialDelay: 200 },
-    transport: {
-      mode: "stdio",
-      command: "npx",
-      args: ["-y", packageName],
-      env
-    },
+    transport,
     samplingHandler
   };
 }
@@ -6229,7 +6256,15 @@ function McpDiscord(config = {}) {
 function McpCoinGecko(config = {}) {
   const mcpConfig = createMcpConfig(
     "CoinGecko MCP Client",
-    "@coingecko/coingecko-mcp",
+    "https://mcp.api.coingecko.com/mcp",
+    config
+  );
+  return new McpToolset(mcpConfig);
+}
+function McpCoinGeckoPro(config = {}) {
+  const mcpConfig = createMcpConfig(
+    "CoinGecko Pro MCP Client",
+    "https://mcp.pro-api.coingecko.com/mcp",
     config
   );
   return new McpToolset(mcpConfig);
@@ -10860,6 +10895,46 @@ var AgentBuilder = class _AgentBuilder {
     this.config.afterAgentCallback = callback;
     return this;
   }
+  /**
+   * Set the before model callback for LLM interaction
+   * @param callback Callback to invoke before calling the LLM
+   * @returns This builder instance for chaining
+   */
+  withBeforeModelCallback(callback) {
+    this.warnIfLocked("withBeforeModelCallback");
+    this.config.beforeModelCallback = callback;
+    return this;
+  }
+  /**
+   * Set the after model callback for LLM interaction
+   * @param callback Callback to invoke after receiving LLM response
+   * @returns This builder instance for chaining
+   */
+  withAfterModelCallback(callback) {
+    this.warnIfLocked("withAfterModelCallback");
+    this.config.afterModelCallback = callback;
+    return this;
+  }
+  /**
+   * Set the before tool callback for tool execution
+   * @param callback Callback to invoke before running a tool
+   * @returns This builder instance for chaining
+   */
+  withBeforeToolCallback(callback) {
+    this.warnIfLocked("withBeforeToolCallback");
+    this.config.beforeToolCallback = callback;
+    return this;
+  }
+  /**
+   * Set the after tool callback for tool execution
+   * @param callback Callback to invoke after running a tool
+   * @returns This builder instance for chaining
+   */
+  withAfterToolCallback(callback) {
+    this.warnIfLocked("withAfterToolCallback");
+    this.config.afterToolCallback = callback;
+    return this;
+  }
   /**
    * Provide an already constructed agent instance. Further definition-mutating calls
    * (model/tools/instruction/etc.) will be ignored with a dev warning.
@@ -11133,6 +11208,10 @@ var AgentBuilder = class _AgentBuilder {
           subAgents: this.config.subAgents,
           beforeAgentCallback: this.config.beforeAgentCallback,
           afterAgentCallback: this.config.afterAgentCallback,
+          beforeModelCallback: this.config.beforeModelCallback,
+          afterModelCallback: this.config.afterModelCallback,
+          beforeToolCallback: this.config.beforeToolCallback,
+          afterToolCallback: this.config.afterToolCallback,
           memoryService: this.memoryService,
           artifactService: this.artifactService,
           outputKey: this.config.outputKey,
@@ -11295,9 +11374,183 @@ var AgentBuilder = class _AgentBuilder {
 // src/memory/index.ts
 var memory_exports = {};
 __export(memory_exports, {
-  InMemoryMemoryService: () => InMemoryMemoryService
+  InMemoryMemoryService: () => InMemoryMemoryService,
+  VertexAiRagMemoryService: () => VertexAiRagMemoryService
 });
 
+// src/memory/vertex-ai-rag-memory-service.ts
+import { randomUUID as randomUUID2 } from "crypto";
+import { unlinkSync, writeFileSync } from "fs";
+import { tmpdir } from "os";
+import { join } from "path";
+var rag = {
+  async upload_file(options) {
+    console.log("Mock upload_file:", options);
+  },
+  async retrieval_query(options) {
+    console.log("Mock retrieval_query:", options);
+    return { contexts: { contexts: [] } };
+  }
+};
+function _mergeEventLists(eventLists) {
+  const merged = [];
+  while (eventLists.length > 0) {
+    const current = eventLists.shift();
+    const currentTs = new Set(current.map((event) => event.timestamp));
+    let mergeFound = true;
+    while (mergeFound) {
+      mergeFound = false;
+      const remaining = [];
+      for (const other of eventLists) {
+        const otherTs = new Set(other.map((event) => event.timestamp));
+        const hasOverlap = Array.from(currentTs).some((ts) => otherTs.has(ts));
+        if (hasOverlap) {
+          const newEvents = other.filter((e) => !currentTs.has(e.timestamp));
+          current.push(...newEvents);
+          newEvents.forEach((e) => currentTs.add(e.timestamp));
+          mergeFound = true;
+        } else {
+          remaining.push(other);
+        }
+      }
+      eventLists.splice(0, eventLists.length, ...remaining);
+    }
+    merged.push(current);
+  }
+  return merged;
+}
+var VertexAiRagMemoryService = class {
+  _vertexRagStore;
+  /**
+   * Initializes a VertexAiRagMemoryService.
+   *
+   * @param ragCorpus The name of the Vertex AI RAG corpus to use. Format:
+   *   `projects/{project}/locations/{location}/ragCorpora/{rag_corpus_id}`
+   *   or `{rag_corpus_id}`
+   * @param similarityTopK The number of contexts to retrieve.
+   * @param vectorDistanceThreshold Only returns contexts with vector distance
+   *   smaller than the threshold.
+   */
+  constructor(ragCorpus, similarityTopK, vectorDistanceThreshold = 10) {
+    this._vertexRagStore = {
+      rag_resources: ragCorpus ? [{ rag_corpus: ragCorpus }] : [],
+      similarity_top_k: similarityTopK,
+      vector_distance_threshold: vectorDistanceThreshold
+    };
+  }
+  /**
+   * Adds a session to the memory service
+   */
+  async addSessionToMemory(session) {
+    const tempFileName = `temp_${randomUUID2()}.txt`;
+    const tempFilePath = join(tmpdir(), tempFileName);
+    try {
+      const outputLines = [];
+      for (const event of session.events) {
+        if (!event.content || !event.content.parts) {
+          continue;
+        }
+        const textParts = event.content.parts.filter((part) => part.text).map((part) => part.text.replace(/\n/g, " "));
+        if (textParts.length > 0) {
+          outputLines.push(
+            JSON.stringify({
+              author: event.author,
+              timestamp: event.timestamp,
+              text: textParts.join(".")
+            })
+          );
+        }
+      }
+      const outputString = outputLines.join("\n");
+      writeFileSync(tempFilePath, outputString, "utf8");
+      if (!this._vertexRagStore.rag_resources || this._vertexRagStore.rag_resources.length === 0) {
+        throw new Error("Rag resources must be set.");
+      }
+      for (const ragResource of this._vertexRagStore.rag_resources) {
+        await rag.upload_file({
+          corpus_name: ragResource.rag_corpus,
+          path: tempFilePath,
+          // This is the temp workaround as upload file does not support
+          // adding metadata, thus use display_name to store the session info.
+          display_name: `${session.appName}.${session.userId}.${session.id}`
+        });
+      }
+    } finally {
+      try {
+        unlinkSync(tempFilePath);
+      } catch (error) {
+        console.warn("Failed to delete temporary file:", tempFilePath, error);
+      }
+    }
+  }
+  /**
+   * Searches for sessions that match the query using rag.retrieval_query
+   */
+  async searchMemory(options) {
+    const { appName, userId, query } = options;
+    const response = await rag.retrieval_query({
+      text: query,
+      rag_resources: this._vertexRagStore.rag_resources,
+      rag_corpora: this._vertexRagStore.rag_corpora,
+      similarity_top_k: this._vertexRagStore.similarity_top_k,
+      vector_distance_threshold: this._vertexRagStore.vector_distance_threshold
+    });
+    const memoryResults = [];
+    const sessionEventsMap = /* @__PURE__ */ new Map();
+    for (const context4 of response.contexts.contexts) {
+      if (!context4.source_display_name.startsWith(`${appName}.${userId}.`)) {
+        continue;
+      }
+      const sessionId = context4.source_display_name.split(".").pop();
+      const events = [];
+      if (context4.text) {
+        const lines = context4.text.split("\n");
+        for (const line of lines) {
+          const trimmedLine = line.trim();
+          if (!trimmedLine) {
+            continue;
+          }
+          try {
+            const eventData = JSON.parse(trimmedLine);
+            const author = eventData.author || "";
+            const timestamp = Number.parseFloat(eventData.timestamp || "0");
+            const text = eventData.text || "";
+            const content = {
+              parts: [{ text }]
+            };
+            const event = {
+              author,
+              timestamp,
+              content
+            };
+            events.push(event);
+          } catch {
+          }
+        }
+      }
+      if (sessionEventsMap.has(sessionId)) {
+        sessionEventsMap.get(sessionId).push(events);
+      } else {
+        sessionEventsMap.set(sessionId, [events]);
+      }
+    }
+    for (const [sessionId, eventLists] of sessionEventsMap.entries()) {
+      const mergedEventLists = _mergeEventLists(eventLists);
+      for (const events of mergedEventLists) {
+        const sortedEvents = events.sort((a, b) => a.timestamp - b.timestamp).filter((event) => event.content);
+        memoryResults.push(
+          ...sortedEvents.map((event) => ({
+            author: event.author,
+            content: event.content,
+            timestamp: formatTimestamp(event.timestamp)
+          }))
+        );
+      }
+    }
+    return { memories: memoryResults };
+  }
+};
+
 // src/sessions/index.ts
 var sessions_exports = {};
 __export(sessions_exports, {
@@ -13640,6 +13893,7 @@ export {
   McpAtp,
   McpBamm,
   McpCoinGecko,
+  McpCoinGeckoPro,
   McpDiscord,
   McpError,
   McpErrorType,
@@ -13682,6 +13936,7 @@ export {
   TransferToAgentTool,
   UserInteractionTool,
   VERSION,
+  VertexAiRagMemoryService,
   VertexAiSessionService,
   _findFunctionCallEventIfLastEventIsFunctionResponse,
   adkToMcpToolType,

package/package.json

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