@machina.ai/cell-cli-core 1.20.2-rc1 → 1.22.5-rc1

package/dist/src/tools/mcp-client.d.ts

@@ -5,13 +5,14 @@
  */
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import type { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
-import type { GetPromptResult, Prompt } from '@modelcontextprotocol/sdk/types.js';
+import type { GetPromptResult, Prompt, ReadResourceResult, Resource } from '@modelcontextprotocol/sdk/types.js';
 import type { Config, MCPServerConfig } from '../config/config.js';
 import { DiscoveredMCPTool } from './mcp-tool.js';
 import type { PromptRegistry } from '../prompts/prompt-registry.js';
 import type { WorkspaceContext } from '../utils/workspaceContext.js';
 import type { ToolRegistry } from './tool-registry.js';
 import type { MessageBus } from '../confirmation-bus/message-bus.js';
+import type { ResourceRegistry } from '../resources/resource-registry.js';
 export declare const MCP_DEFAULT_TIMEOUT_MSEC: number;
 export type DiscoveredMCPPrompt = Prompt & {
     serverName: string;
@@ -52,12 +53,19 @@ export declare class McpClient {
     private readonly serverConfig;
     private readonly toolRegistry;
     private readonly promptRegistry;
+    private readonly resourceRegistry;
     private readonly workspaceContext;
+    private readonly cliConfig;
     private readonly debugMode;
+    private readonly onToolsUpdated?;
     private client;
     private transport;
     private status;
-    constructor(serverName: string, serverConfig: MCPServerConfig, toolRegistry: ToolRegistry, promptRegistry: PromptRegistry, workspaceContext: WorkspaceContext, debugMode: boolean);
+    private isRefreshingTools;
+    private pendingToolRefresh;
+    private isRefreshingResources;
+    private pendingResourceRefresh;
+    constructor(serverName: string, serverConfig: MCPServerConfig, toolRegistry: ToolRegistry, promptRegistry: PromptRegistry, resourceRegistry: ResourceRegistry, workspaceContext: WorkspaceContext, cliConfig: Config, debugMode: boolean, onToolsUpdated?: ((signal?: AbortSignal) => Promise<void>) | undefined);
     /**
      * Connects to the MCP server.
      */
@@ -78,8 +86,32 @@ export declare class McpClient {
     private assertConnected;
     private discoverTools;
     private discoverPrompts;
+    private discoverResources;
+    private updateResourceRegistry;
+    readResource(uri: string): Promise<ReadResourceResult>;
+    /**
+     * Registers notification handlers for dynamic updates from the MCP server.
+     * This includes handlers for tool list changes and resource list changes.
+     */
+    private registerNotificationHandlers;
+    /**
+     * Refreshes the resources for this server by re-querying the MCP `resources/list` endpoint.
+     *
+     * This method implements a **Coalescing Pattern** to handle rapid bursts of notifications
+     * (e.g., during server startup or bulk updates) without overwhelming the server or
+     * creating race conditions in the ResourceRegistry.
+     */
+    private refreshResources;
     getServerConfig(): MCPServerConfig;
     getInstructions(): string | undefined;
+    /**
+     * Refreshes the tools for this server by re-querying the MCP `tools/list` endpoint.
+     *
+     * This method implements a **Coalescing Pattern** to handle rapid bursts of notifications
+     * (e.g., during server startup or bulk updates) without overwhelming the server or
+     * creating race conditions in the global ToolRegistry.
+     */
+    private refreshTools;
 }
 /**
  * Map to track which MCP servers have been discovered to require OAuth
@@ -150,7 +182,10 @@ export declare function connectAndDiscover(mcpServerName: string, mcpServerConfi
  * @returns A promise that resolves to an array of discovered and enabled tools.
  * @throws An error if no enabled tools are found or if the server provides invalid function declarations.
  */
-export declare function discoverTools(mcpServerName: string, mcpServerConfig: MCPServerConfig, mcpClient: Client, cliConfig: Config, messageBus?: MessageBus): Promise<DiscoveredMCPTool[]>;
+export declare function discoverTools(mcpServerName: string, mcpServerConfig: MCPServerConfig, mcpClient: Client, cliConfig: Config, messageBus?: MessageBus, options?: {
+    timeout?: number;
+    signal?: AbortSignal;
+}): Promise<DiscoveredMCPTool[]>;
 /**
  * Discovers and logs prompts from a connected MCP client.
  * It retrieves prompt declarations from the client and logs their names.
@@ -159,6 +194,7 @@ export declare function discoverTools(mcpServerName: string, mcpServerConfig: MC
  * @param mcpClient The active MCP client instance.
  */
 export declare function discoverPrompts(mcpServerName: string, mcpClient: Client, promptRegistry: PromptRegistry): Promise<Prompt[]>;
+export declare function discoverResources(mcpServerName: string, mcpClient: Client): Promise<Resource[]>;
 /**
  * Invokes a prompt on a connected MCP client.
  *