@office-ai/aioncli-core 0.2.3 → 0.18.4

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

@@ -4,14 +4,14 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 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 type { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
+import type { GetPromptResult, Prompt } from '@modelcontextprotocol/sdk/types.js';
+import type { Config, 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';
+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';
 export declare const MCP_DEFAULT_TIMEOUT_MSEC: number;
 export type DiscoveredMCPPrompt = Prompt & {
     serverName: string;
@@ -23,6 +23,8 @@ export type DiscoveredMCPPrompt = Prompt & {
 export declare enum MCPServerStatus {
     /** Server is disconnected or experiencing errors */
     DISCONNECTED = "disconnected",
+    /** Server is actively disconnecting */
+    DISCONNECTING = "disconnecting",
     /** Server is in the process of connecting */
     CONNECTING = "connecting",
     /** Server is connected and ready to use */
@@ -55,7 +57,6 @@ export declare class McpClient {
     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.
@@ -64,7 +65,7 @@ export declare class McpClient {
     /**
      * Discovers tools and prompts from the MCP server.
      */
-    discover(): Promise<void>;
+    discover(cliConfig: Config): Promise<void>;
     /**
      * Disconnects from the MCP server.
      */
@@ -74,9 +75,10 @@ export declare class McpClient {
      */
     getStatus(): MCPServerStatus;
     private updateStatus;
-    private createTransport;
+    private assertConnected;
     private discoverTools;
     private discoverPrompts;
+    getServerConfig(): MCPServerConfig;
 }
 /**
  * Map to track which MCP servers have been discovered to require OAuth
@@ -120,7 +122,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, promptRegistry: PromptRegistry, debugMode: boolean, workspaceContext: WorkspaceContext): Promise<void>;
+export declare function discoverMcpTools(mcpServers: Record<string, MCPServerConfig>, mcpServerCommand: string | undefined, toolRegistry: ToolRegistry, promptRegistry: PromptRegistry, debugMode: boolean, workspaceContext: WorkspaceContext, cliConfig: Config): Promise<void>;
 /** Visible for Testing */
 export declare function populateMcpServerCommand(mcpServers: Record<string, MCPServerConfig>, mcpServerCommand: string | undefined): Record<string, MCPServerConfig>;
 /**
@@ -133,17 +135,7 @@ 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, 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;
+export declare function connectAndDiscover(mcpServerName: string, mcpServerConfig: MCPServerConfig, toolRegistry: ToolRegistry, promptRegistry: PromptRegistry, debugMode: boolean, workspaceContext: WorkspaceContext, cliConfig: Config): Promise<void>;
 /**
  * Discovers and sanitizes tools from a connected MCP client.
  * It retrieves function declarations from the client, filters out disabled tools,
@@ -152,10 +144,12 @@ export declare function hasValidTypes(schema: unknown): boolean;
  * @param mcpServerName The name of the MCP server.
  * @param mcpServerConfig The configuration for the MCP server.
  * @param mcpClient The active MCP client instance.
+ * @param cliConfig The CLI configuration object.
+ * @param messageBus Optional message bus for policy engine integration.
  * @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): Promise<DiscoveredMCPTool[]>;
+export declare function discoverTools(mcpServerName: string, mcpServerConfig: MCPServerConfig, mcpClient: Client, cliConfig: Config, messageBus?: MessageBus): Promise<DiscoveredMCPTool[]>;
 /**
  * Discovers and logs prompts from a connected MCP client.
  * It retrieves prompt declarations from the client and logs their names.
@@ -174,6 +168,13 @@ export declare function discoverPrompts(mcpServerName: string, mcpClient: Client
  * @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>;
+/**
+ * @visiblefortesting
+ * Checks if the MCP server configuration has a network transport URL (SSE or HTTP).
+ * @param config The MCP server configuration.
+ * @returns True if a `url` or `httpUrl` is present, false otherwise.
+ */
+export declare function hasNetworkTransport(config: MCPServerConfig): boolean;
 /**
  * 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
@@ -187,6 +188,9 @@ export declare function invokeMcpPrompt(mcpServerName: string, mcpClient: 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): Promise<Transport>;
+interface NamedTool {
+    name?: string;
+}
 /** Visible for testing */
-export declare function isEnabled(funcDecl: FunctionDeclaration, mcpServerName: string, mcpServerConfig: MCPServerConfig): boolean;
+export declare function isEnabled(funcDecl: NamedTool, mcpServerName: string, mcpServerConfig: MCPServerConfig): boolean;
 export {};