@machina.ai/cell-cli-core 1.0.13-rc9 → 1.0.21-rc1

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

@@ -5,11 +5,18 @@
  */
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
+import { Prompt, GetPromptResult } from '@modelcontextprotocol/sdk/types.js';
 import { MCPServerConfig } from '../config/config.js';
 import { DiscoveredMCPTool } from './mcp-tool.js';
 import { FunctionDeclaration } from '@google/genai';
 import { ToolRegistry } from './tool-registry.js';
+import { PromptRegistry } from '../prompts/prompt-registry.js';
+import { WorkspaceContext } from '../utils/workspaceContext.js';
 export declare const MCP_DEFAULT_TIMEOUT_MSEC: number;
+export type DiscoveredMCPPrompt = Prompt & {
+    serverName: string;
+    invoke: (params: Record<string, unknown>) => Promise<GetPromptResult>;
+};
 /**
  * Enum representing the connection status of an MCP server
  */
@@ -32,6 +39,49 @@ export declare enum MCPDiscoveryState {
     /** Discovery has completed (with or without errors) */
     COMPLETED = "completed"
 }
+/**
+ * A client for a single MCP server.
+ *
+ * This class is responsible for connecting to, discovering tools from, and
+ * managing the state of a single MCP server.
+ */
+export declare class McpClient {
+    private readonly serverName;
+    private readonly serverConfig;
+    private readonly toolRegistry;
+    private readonly promptRegistry;
+    private readonly workspaceContext;
+    private readonly debugMode;
+    private client;
+    private transport;
+    private status;
+    private isDisconnecting;
+    constructor(serverName: string, serverConfig: MCPServerConfig, toolRegistry: ToolRegistry, promptRegistry: PromptRegistry, workspaceContext: WorkspaceContext, debugMode: boolean);
+    /**
+     * Connects to the MCP server.
+     */
+    connect(): Promise<void>;
+    /**
+     * Discovers tools and prompts from the MCP server.
+     */
+    discover(): Promise<void>;
+    /**
+     * Disconnects from the MCP server.
+     */
+    disconnect(): Promise<void>;
+    /**
+     * Returns the current status of the client.
+     */
+    getStatus(): MCPServerStatus;
+    private updateStatus;
+    private createTransport;
+    private discoverTools;
+    private discoverPrompts;
+}
+/**
+ * Map to track which MCP servers have been discovered to require OAuth
+ */
+export declare const mcpServerRequiresOAuth: Map<string, boolean>;
 /**
  * Event listeners for MCP server status changes
  */
@@ -44,6 +94,10 @@ export declare function addMCPStatusChangeListener(listener: StatusChangeListene
  * Remove a listener for MCP server status changes
  */
 export declare function removeMCPStatusChangeListener(listener: StatusChangeListener): void;
+/**
+ * Update the status of an MCP server
+ */
+export declare function updateMCPServerStatus(serverName: string, status: MCPServerStatus): void;
 /**
  * Get the current status of an MCP server
  */
@@ -66,7 +120,7 @@ export declare function getMCPDiscoveryState(): MCPDiscoveryState;
  * @param toolRegistry The central registry where discovered tools will be registered.
  * @returns A promise that resolves when the discovery process has been attempted for all servers.
  */
-export declare function discoverMcpTools(mcpServers: Record<string, MCPServerConfig>, mcpServerCommand: string | undefined, toolRegistry: ToolRegistry, debugMode: boolean): Promise<void>;
+export declare function discoverMcpTools(mcpServers: Record<string, MCPServerConfig>, mcpServerCommand: string | undefined, toolRegistry: ToolRegistry, promptRegistry: PromptRegistry, debugMode: boolean, workspaceContext: WorkspaceContext): Promise<void>;
 /** Visible for Testing */
 export declare function populateMcpServerCommand(mcpServers: Record<string, MCPServerConfig>, mcpServerCommand: string | undefined): Record<string, MCPServerConfig>;
 /**
@@ -79,7 +133,17 @@ export declare function populateMcpServerCommand(mcpServers: Record<string, MCPS
  * @param toolRegistry The registry to register discovered tools with
  * @returns Promise that resolves when discovery is complete
  */
-export declare function connectAndDiscover(mcpServerName: string, mcpServerConfig: MCPServerConfig, toolRegistry: ToolRegistry, debugMode: boolean): Promise<void>;
+export declare function connectAndDiscover(mcpServerName: string, mcpServerConfig: MCPServerConfig, toolRegistry: ToolRegistry, promptRegistry: PromptRegistry, debugMode: boolean, workspaceContext: WorkspaceContext): Promise<void>;
+/**
+ * Recursively validates that a JSON schema and all its nested properties and
+ * items have a `type` defined.
+ *
+ * @param schema The JSON schema to validate.
+ * @returns `true` if the schema is valid, `false` otherwise.
+ *
+ * @visiblefortesting
+ */
+export declare function hasValidTypes(schema: unknown): boolean;
 /**
  * Discovers and sanitizes tools from a connected MCP client.
  * It retrieves function declarations from the client, filters out disabled tools,
@@ -92,6 +156,24 @@ export declare function connectAndDiscover(mcpServerName: string, mcpServerConfi
  * @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): Promise<DiscoveredMCPTool[]>;
+/**
+ * Discovers and logs prompts from a connected MCP client.
+ * It retrieves prompt declarations from the client and logs their names.
+ *
+ * @param mcpServerName The name of the MCP server.
+ * @param mcpClient The active MCP client instance.
+ */
+export declare function discoverPrompts(mcpServerName: string, mcpClient: Client, promptRegistry: PromptRegistry): Promise<Prompt[]>;
+/**
+ * Invokes a prompt on a connected MCP client.
+ *
+ * @param mcpServerName The name of the MCP server.
+ * @param mcpClient The active MCP client instance.
+ * @param promptName The name of the prompt to invoke.
+ * @param promptParams The parameters to pass to the prompt.
+ * @returns A promise that resolves to the result of the prompt invocation.
+ */
+export declare function invokeMcpPrompt(mcpServerName: string, mcpClient: Client, promptName: string, promptParams: Record<string, unknown>): Promise<GetPromptResult>;
 /**
  * Creates and connects an MCP client to a server based on the provided configuration.
  * It determines the appropriate transport (Stdio, SSE, or Streamable HTTP) and
@@ -102,9 +184,9 @@ export declare function discoverTools(mcpServerName: string, mcpServerConfig: MC
  * @returns A promise that resolves to a connected MCP `Client` instance.
  * @throws An error if the connection fails or the configuration is invalid.
  */
-export declare function connectToMcpServer(mcpServerName: string, mcpServerConfig: MCPServerConfig, debugMode: boolean): Promise<Client>;
+export declare function connectToMcpServer(mcpServerName: string, mcpServerConfig: MCPServerConfig, debugMode: boolean, workspaceContext: WorkspaceContext): Promise<Client>;
 /** Visible for Testing */
-export declare function createTransport(mcpServerName: string, mcpServerConfig: MCPServerConfig, debugMode: boolean): Transport;
+export declare function createTransport(mcpServerName: string, mcpServerConfig: MCPServerConfig, debugMode: boolean): Promise<Transport>;
 /** Visible for testing */
 export declare function isEnabled(funcDecl: FunctionDeclaration, mcpServerName: string, mcpServerConfig: MCPServerConfig): boolean;
 export {};