@genkit-ai/mcp 1.14.1-rc.0

package/lib/client/client.d.mts

@@ -0,0 +1,177 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { StdioServerParameters } from '@modelcontextprotocol/sdk/client/stdio.js';
+import { StreamableHTTPClientTransportOptions } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
+import { Root } from '@modelcontextprotocol/sdk/types.js';
+import { Genkit, ToolAction, DynamicResourceAction, PromptGenerateOptions, ExecutablePrompt } from 'genkit';
+
+/**
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+interface McpServerRef {
+    client: Client;
+    transport: Transport;
+    error?: string;
+}
+interface McpServerControls {
+    /** when true, the server will be stopped and its registered components will
+     * not appear in lists/plugins/etc */
+    disabled?: boolean;
+    roots?: Root[];
+}
+type McpStdioServerConfig = StdioServerParameters & {
+    url?: never;
+    transport?: never;
+};
+type McpStreamableHttpConfig = {
+    url: string;
+    command?: never;
+    transport?: never;
+} & Omit<StreamableHTTPClientTransportOptions, 'sessionId'>;
+type McpTransportServerConfig = {
+    transport: Transport;
+    command?: never;
+    url?: never;
+};
+/**
+ * Configuration for an individual MCP server. The interface should be familiar
+ * and compatible with existing tool configurations e.g. Cursor or Claude
+ * Desktop.
+ *
+ * In addition to stdio servers, remote servers are supported via URL and
+ * custom/arbitary transports are supported as well.
+ */
+type McpServerConfig = (McpStdioServerConfig | McpStreamableHttpConfig | McpTransportServerConfig) & McpServerControls;
+/**
+ * Configuration options for an individual `GenkitMcpClient` instance.
+ * This defines how the client connects to a single MCP server and how it behaves.
+ */
+type McpClientOptions = {
+    /** Client name to advertise to the server. */
+    name: string;
+    /** Name for the server, defaults to the server's advertised name. */
+    serverName?: string;
+    /**
+     * An optional version number for this client. This is primarily for logging
+     * and identification purposes. Defaults to '1.0.0'.
+     */
+    version?: string;
+    /**
+     * If true, tool responses from the MCP server will be returned in their raw
+     * MCP format. Otherwise (default), they are processed and potentially
+     * simplified for better compatibility with Genkit's typical data structures.
+     */
+    rawToolResponses?: boolean;
+    /** The server configuration to connect. */
+    mcpServer: McpServerConfig;
+    /** Manually supply a session id for HTTP streaming clients if desired. */
+    sessionId?: string;
+};
+/**
+ * Represents a client connection to a single MCP (Model Context Protocol) server.
+ * It handles the lifecycle of the connection (connect, disconnect, disable, re-enable, reconnect)
+ * and provides methods to fetch tools from the connected server.
+ *
+ * An instance of `GenkitMcpClient` is typically managed by a `GenkitMcpHost`
+ * when dealing with multiple MCP server connections.
+ */
+declare class GenkitMcpClient {
+    _server?: McpServerRef;
+    sessionId?: string;
+    readonly name: string;
+    readonly suppliedServerName?: string;
+    private version;
+    private serverConfig;
+    private rawToolResponses?;
+    private disabled;
+    private roots?;
+    private _readyListeners;
+    private _ready;
+    constructor(options: McpClientOptions);
+    get serverName(): string;
+    updateRoots(roots: Root[]): Promise<void>;
+    /**
+     * Sets up a connection based on a provided map of server configurations.
+     * - Reconnects existing servers if their configuration appears to have
+     *   changed (implicitly handled by `connectServer`).
+     * - Sets the client's ready state once all connection attempts are complete.
+     * @param mcpServers A record mapping server names to their configurations.
+     */
+    private _initializeConnection;
+    /**
+     * Returns a Promise that resolves when the client has attempted to connect
+     * to all configured servers, or rejects if a critical error occurs during
+     * the initial connection phase.
+     */
+    ready(): Promise<void>;
+    /**
+     * Connects to a single MCP server defined by the provided configuration.
+     * @param config The configuration object for the server.
+     */
+    private _connect;
+    /**
+     * Disconnects the MCP server and removes its registration
+     * from this client instance.
+     */
+    _disconnect(): Promise<void>;
+    /**
+     * Disables a server. Closes the underlying transport and server's configuration. Does nothing if the server is
+     * already disabled.
+     */
+    disable(): Promise<void>;
+    /**
+     * Whether this client-server connection is enabled or not.
+     */
+    isEnabled(): boolean;
+    /**
+     * Enables a server connection, including previously disabled ones. Does nothing if the
+     * server is not disabled.
+     */
+    enable(): Promise<void>;
+    /**
+     * Closes and then restarts the transport connection for the specified server.
+     * Useful for attempting to recover from connection issues without full
+     * reconfiguration.
+     */
+    restart(): Promise<void>;
+    /**
+     * Fetches all tools available through this client, if the server
+     * configuration is not disabled.
+     */
+    getActiveTools(ai: Genkit): Promise<ToolAction[]>;
+    /**
+     * Fetches all resources available through this client, if the server
+     * configuration is not disabled.
+     */
+    getActiveResources(ai: Genkit): Promise<DynamicResourceAction[]>;
+    /**
+     * Fetches all active prompts available through this client, if the server
+     * configuration supports prompts.
+     * @param ai The Genkit instance.
+     * @param options Optional prompt generation options.
+     * @returns A promise that resolves to an array of ExecutablePrompt.
+     */
+    getActivePrompts(ai: Genkit, options?: PromptGenerateOptions): Promise<ExecutablePrompt[]>;
+    /**
+     * Get the specified prompt as an `ExecutablePrompt` available through this
+     * client. If no such prompt is found, return undefined.
+     */
+    getPrompt(ai: Genkit, promptName: string, opts?: PromptGenerateOptions): Promise<ExecutablePrompt | undefined>;
+    /** Returns the underlying MCP SDK client if one has been initialized. */
+    get mcpClient(): Client | undefined;
+}
+
+export { GenkitMcpClient, type McpClientOptions, type McpServerConfig, type McpServerControls, type McpStdioServerConfig, type McpStreamableHttpConfig, type McpTransportServerConfig };

package/lib/client/client.d.ts

@@ -0,0 +1,177 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { StdioServerParameters } from '@modelcontextprotocol/sdk/client/stdio.js';
+import { StreamableHTTPClientTransportOptions } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
+import { Root } from '@modelcontextprotocol/sdk/types.js';
+import { Genkit, ToolAction, DynamicResourceAction, PromptGenerateOptions, ExecutablePrompt } from 'genkit';
+
+/**
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+interface McpServerRef {
+    client: Client;
+    transport: Transport;
+    error?: string;
+}
+interface McpServerControls {
+    /** when true, the server will be stopped and its registered components will
+     * not appear in lists/plugins/etc */
+    disabled?: boolean;
+    roots?: Root[];
+}
+type McpStdioServerConfig = StdioServerParameters & {
+    url?: never;
+    transport?: never;
+};
+type McpStreamableHttpConfig = {
+    url: string;
+    command?: never;
+    transport?: never;
+} & Omit<StreamableHTTPClientTransportOptions, 'sessionId'>;
+type McpTransportServerConfig = {
+    transport: Transport;
+    command?: never;
+    url?: never;
+};
+/**
+ * Configuration for an individual MCP server. The interface should be familiar
+ * and compatible with existing tool configurations e.g. Cursor or Claude
+ * Desktop.
+ *
+ * In addition to stdio servers, remote servers are supported via URL and
+ * custom/arbitary transports are supported as well.
+ */
+type McpServerConfig = (McpStdioServerConfig | McpStreamableHttpConfig | McpTransportServerConfig) & McpServerControls;
+/**
+ * Configuration options for an individual `GenkitMcpClient` instance.
+ * This defines how the client connects to a single MCP server and how it behaves.
+ */
+type McpClientOptions = {
+    /** Client name to advertise to the server. */
+    name: string;
+    /** Name for the server, defaults to the server's advertised name. */
+    serverName?: string;
+    /**
+     * An optional version number for this client. This is primarily for logging
+     * and identification purposes. Defaults to '1.0.0'.
+     */
+    version?: string;
+    /**
+     * If true, tool responses from the MCP server will be returned in their raw
+     * MCP format. Otherwise (default), they are processed and potentially
+     * simplified for better compatibility with Genkit's typical data structures.
+     */
+    rawToolResponses?: boolean;
+    /** The server configuration to connect. */
+    mcpServer: McpServerConfig;
+    /** Manually supply a session id for HTTP streaming clients if desired. */
+    sessionId?: string;
+};
+/**
+ * Represents a client connection to a single MCP (Model Context Protocol) server.
+ * It handles the lifecycle of the connection (connect, disconnect, disable, re-enable, reconnect)
+ * and provides methods to fetch tools from the connected server.
+ *
+ * An instance of `GenkitMcpClient` is typically managed by a `GenkitMcpHost`
+ * when dealing with multiple MCP server connections.
+ */
+declare class GenkitMcpClient {
+    _server?: McpServerRef;
+    sessionId?: string;
+    readonly name: string;
+    readonly suppliedServerName?: string;
+    private version;
+    private serverConfig;
+    private rawToolResponses?;
+    private disabled;
+    private roots?;
+    private _readyListeners;
+    private _ready;
+    constructor(options: McpClientOptions);
+    get serverName(): string;
+    updateRoots(roots: Root[]): Promise<void>;
+    /**
+     * Sets up a connection based on a provided map of server configurations.
+     * - Reconnects existing servers if their configuration appears to have
+     *   changed (implicitly handled by `connectServer`).
+     * - Sets the client's ready state once all connection attempts are complete.
+     * @param mcpServers A record mapping server names to their configurations.
+     */
+    private _initializeConnection;
+    /**
+     * Returns a Promise that resolves when the client has attempted to connect
+     * to all configured servers, or rejects if a critical error occurs during
+     * the initial connection phase.
+     */
+    ready(): Promise<void>;
+    /**
+     * Connects to a single MCP server defined by the provided configuration.
+     * @param config The configuration object for the server.
+     */
+    private _connect;
+    /**
+     * Disconnects the MCP server and removes its registration
+     * from this client instance.
+     */
+    _disconnect(): Promise<void>;
+    /**
+     * Disables a server. Closes the underlying transport and server's configuration. Does nothing if the server is
+     * already disabled.
+     */
+    disable(): Promise<void>;
+    /**
+     * Whether this client-server connection is enabled or not.
+     */
+    isEnabled(): boolean;
+    /**
+     * Enables a server connection, including previously disabled ones. Does nothing if the
+     * server is not disabled.
+     */
+    enable(): Promise<void>;
+    /**
+     * Closes and then restarts the transport connection for the specified server.
+     * Useful for attempting to recover from connection issues without full
+     * reconfiguration.
+     */
+    restart(): Promise<void>;
+    /**
+     * Fetches all tools available through this client, if the server
+     * configuration is not disabled.
+     */
+    getActiveTools(ai: Genkit): Promise<ToolAction[]>;
+    /**
+     * Fetches all resources available through this client, if the server
+     * configuration is not disabled.
+     */
+    getActiveResources(ai: Genkit): Promise<DynamicResourceAction[]>;
+    /**
+     * Fetches all active prompts available through this client, if the server
+     * configuration supports prompts.
+     * @param ai The Genkit instance.
+     * @param options Optional prompt generation options.
+     * @returns A promise that resolves to an array of ExecutablePrompt.
+     */
+    getActivePrompts(ai: Genkit, options?: PromptGenerateOptions): Promise<ExecutablePrompt[]>;
+    /**
+     * Get the specified prompt as an `ExecutablePrompt` available through this
+     * client. If no such prompt is found, return undefined.
+     */
+    getPrompt(ai: Genkit, promptName: string, opts?: PromptGenerateOptions): Promise<ExecutablePrompt | undefined>;
+    /** Returns the underlying MCP SDK client if one has been initialized. */
+    get mcpClient(): Client | undefined;
+}
+
+export { GenkitMcpClient, type McpClientOptions, type McpServerConfig, type McpServerControls, type McpStdioServerConfig, type McpStreamableHttpConfig, type McpTransportServerConfig };

package/lib/client/client.js

@@ -0,0 +1,282 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var client_exports = {};
+__export(client_exports, {
+  GenkitMcpClient: () => GenkitMcpClient
+});
+module.exports = __toCommonJS(client_exports);
+var import_client = require("@modelcontextprotocol/sdk/client/index.js");
+var import_types = require("@modelcontextprotocol/sdk/types.js");
+var import_genkit = require("genkit");
+var import_logging = require("genkit/logging");
+var import_util = require("../util");
+var import_resource = require("../util/resource");
+class GenkitMcpClient {
+  _server;
+  sessionId;
+  name;
+  suppliedServerName;
+  version;
+  serverConfig;
+  rawToolResponses;
+  disabled;
+  roots;
+  _readyListeners = [];
+  _ready = false;
+  constructor(options) {
+    this.name = options.name;
+    this.suppliedServerName = options.serverName;
+    this.version = options.version || "1.0.0";
+    this.serverConfig = options.mcpServer;
+    this.rawToolResponses = !!options.rawToolResponses;
+    this.disabled = !!options.mcpServer.disabled;
+    this.roots = options.mcpServer.roots;
+    this.sessionId = options.sessionId;
+    this._initializeConnection();
+  }
+  get serverName() {
+    return this.suppliedServerName ?? this._server?.client.getServerVersion()?.name ?? "unknown-server";
+  }
+  async updateRoots(roots) {
+    this.roots = roots;
+    await this._server?.client.sendRootsListChanged();
+  }
+  /**
+   * Sets up a connection based on a provided map of server configurations.
+   * - Reconnects existing servers if their configuration appears to have
+   *   changed (implicitly handled by `connectServer`).
+   * - Sets the client's ready state once all connection attempts are complete.
+   * @param mcpServers A record mapping server names to their configurations.
+   */
+  async _initializeConnection() {
+    this._ready = false;
+    try {
+      await this._connect();
+      this._ready = true;
+      while (this._readyListeners.length) {
+        this._readyListeners.pop()?.resolve();
+      }
+    } catch (err) {
+      while (this._readyListeners.length) {
+        this._readyListeners.pop()?.reject(err);
+      }
+    }
+    if (this.roots) {
+      await this.updateRoots(this.roots);
+    }
+  }
+  /**
+   * Returns a Promise that resolves when the client has attempted to connect
+   * to all configured servers, or rejects if a critical error occurs during
+   * the initial connection phase.
+   */
+  async ready() {
+    if (this._ready) return;
+    return new Promise((resolve, reject) => {
+      this._readyListeners.push({ resolve, reject });
+    });
+  }
+  /**
+   * Connects to a single MCP server defined by the provided configuration.
+   * @param config The configuration object for the server.
+   */
+  async _connect() {
+    if (this._server) await this._server.transport.close();
+    import_logging.logger.debug(
+      `[MCP Client] Connecting MCP server '${this.serverName}' in client '${this.name}'.`
+    );
+    const { transport, type: transportType } = await (0, import_util.transportFrom)(
+      this.serverConfig,
+      this.sessionId
+    );
+    if (!transport) {
+      throw new import_genkit.GenkitError({
+        status: "INVALID_ARGUMENT",
+        message: `[MCP Client] Could not determine valid transport config from supplied options.`
+      });
+    }
+    let error;
+    const client = new import_client.Client(
+      { name: this.name, version: this.version },
+      { capabilities: { roots: { listChanged: true } } }
+    );
+    client.setRequestHandler(import_types.ListRootsRequestSchema, () => {
+      import_logging.logger.debug(`[MCP Client] fetching roots for ${this.name}`);
+      return { roots: this.roots || [] };
+    });
+    try {
+      await client.connect(transport);
+    } catch (e) {
+      import_logging.logger.warn(
+        `[MCP Client] Error connecting server via ${transportType} transport: ${e}`
+      );
+      this.disabled = true;
+      error = e.toString();
+    }
+    this._server = {
+      client,
+      transport,
+      error
+    };
+  }
+  /**
+   * Disconnects the MCP server and removes its registration
+   * from this client instance.
+   */
+  async _disconnect() {
+    if (this._server) {
+      import_logging.logger.debug(
+        `[MCP Client] Disconnecting MCP server in client '${this.name}'.`
+      );
+      await this._server.client.close();
+      this._server = void 0;
+    }
+  }
+  /**
+   * Disables a server. Closes the underlying transport and server's configuration. Does nothing if the server is
+   * already disabled.
+   */
+  async disable() {
+    if (!this.isEnabled()) return;
+    if (this._server) {
+      import_logging.logger.debug(
+        `[MCP Client] Disabling MCP server in client '${this.name}'`
+      );
+      await this._disconnect();
+      this.disabled = true;
+    }
+  }
+  /**
+   * Whether this client-server connection is enabled or not.
+   */
+  isEnabled() {
+    return !this.disabled;
+  }
+  /**
+   * Enables a server connection, including previously disabled ones. Does nothing if the
+   * server is not disabled.
+   */
+  async enable() {
+    if (this.isEnabled()) return;
+    import_logging.logger.debug(`[MCP Client] Reenabling MCP server in client '${this.name}'`);
+    await this._initializeConnection();
+    this.disabled = !!this._server.error;
+  }
+  /**
+   * Closes and then restarts the transport connection for the specified server.
+   * Useful for attempting to recover from connection issues without full
+   * reconfiguration.
+   */
+  async restart() {
+    if (this._server) {
+      import_logging.logger.debug(
+        `[MCP Client] Restarting connection to MCP server in client '${this.name}'`
+      );
+      await this._disconnect();
+      await this._initializeConnection();
+    }
+  }
+  /**
+   * Fetches all tools available through this client, if the server
+   * configuration is not disabled.
+   */
+  async getActiveTools(ai) {
+    await this.ready();
+    let tools = [];
+    if (this._server) {
+      const capabilities = this._server.client.getServerCapabilities();
+      if (capabilities?.tools)
+        tools.push(
+          ...await (0, import_util.fetchDynamicTools)(ai, this._server.client, {
+            rawToolResponses: this.rawToolResponses,
+            serverName: this.serverName,
+            name: this.name
+          })
+        );
+    }
+    return tools;
+  }
+  /**
+   * Fetches all resources available through this client, if the server
+   * configuration is not disabled.
+   */
+  async getActiveResources(ai) {
+    await this.ready();
+    let resources = [];
+    if (this._server) {
+      const capabilities = this._server.client.getServerCapabilities();
+      if (capabilities?.resources)
+        resources.push(
+          ...await (0, import_resource.fetchDynamicResources)(ai, this._server.client, {
+            serverName: this.serverName,
+            name: this.name
+          })
+        );
+    }
+    return resources;
+  }
+  /**
+   * Fetches all active prompts available through this client, if the server
+   * configuration supports prompts.
+   * @param ai The Genkit instance.
+   * @param options Optional prompt generation options.
+   * @returns A promise that resolves to an array of ExecutablePrompt.
+   */
+  async getActivePrompts(ai, options) {
+    if (this._server?.client.getServerCapabilities()?.prompts) {
+      return (0, import_util.fetchAllPrompts)(this._server.client, {
+        ai,
+        serverName: this.serverName,
+        name: this.name,
+        options
+      });
+    }
+    return [];
+  }
+  /**
+   * Get the specified prompt as an `ExecutablePrompt` available through this
+   * client. If no such prompt is found, return undefined.
+   */
+  async getPrompt(ai, promptName, opts) {
+    await this.ready();
+    if (this._server) {
+      const capabilities = await this._server.client.getServerCapabilities();
+      if (capabilities?.prompts) {
+        return await (0, import_util.getExecutablePrompt)(this._server.client, {
+          ai,
+          serverName: this.name,
+          promptName,
+          name: this.name,
+          options: opts
+        });
+      }
+      import_logging.logger.debug(`[MCP Client] No prompts are found in this MCP server.`);
+    }
+    return;
+  }
+  /** Returns the underlying MCP SDK client if one has been initialized. */
+  get mcpClient() {
+    return this._server?.client;
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  GenkitMcpClient
+});
+//# sourceMappingURL=client.js.map

package/lib/client/client.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/client/client.ts"],"sourcesContent":["/**\n * Copyright 2025 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Client } from '@modelcontextprotocol/sdk/client/index.js';\nimport type { StdioServerParameters } from '@modelcontextprotocol/sdk/client/stdio.js';\nimport type { StreamableHTTPClientTransportOptions } from '@modelcontextprotocol/sdk/client/streamableHttp.js';\nimport { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';\nimport {\n  ListRootsRequestSchema,\n  Root,\n} from '@modelcontextprotocol/sdk/types.js';\nimport {\n  DynamicResourceAction,\n  ExecutablePrompt,\n  Genkit,\n  GenkitError,\n  PromptGenerateOptions,\n  ToolAction,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport {\n  fetchAllPrompts,\n  fetchDynamicTools,\n  getExecutablePrompt,\n  transportFrom,\n} from '../util';\nimport { fetchDynamicResources } from '../util/resource';\n\ninterface McpServerRef {\n  client: Client;\n  transport: Transport;\n  error?: string;\n}\n\nexport interface McpServerControls {\n  /** when true, the server will be stopped and its registered components will\n   * not appear in lists/plugins/etc */\n  disabled?: boolean;\n\n  // MCP roots configuration. See: https://modelcontextprotocol.io/docs/concepts/roots\n  roots?: Root[];\n}\n\nexport type McpStdioServerConfig = StdioServerParameters & {\n  url?: never;\n  transport?: never;\n};\n\nexport type McpStreamableHttpConfig = {\n  url: string;\n  command?: never;\n  transport?: never;\n} & Omit<StreamableHTTPClientTransportOptions, 'sessionId'>;\n\nexport type McpTransportServerConfig = {\n  transport: Transport;\n  command?: never;\n  url?: never;\n};\n\n/**\n * Configuration for an individual MCP server. The interface should be familiar\n * and compatible with existing tool configurations e.g. Cursor or Claude\n * Desktop.\n *\n * In addition to stdio servers, remote servers are supported via URL and\n * custom/arbitary transports are supported as well.\n */\nexport type McpServerConfig = (\n  | McpStdioServerConfig\n  | McpStreamableHttpConfig\n  | McpTransportServerConfig\n) &\n  McpServerControls;\n\n/**\n * Configuration options for an individual `GenkitMcpClient` instance.\n * This defines how the client connects to a single MCP server and how it behaves.\n */\nexport type McpClientOptions = {\n  /** Client name to advertise to the server. */\n  name: string;\n  /** Name for the server, defaults to the server's advertised name. */\n  serverName?: string;\n\n  /**\n   * An optional version number for this client. This is primarily for logging\n   * and identification purposes. Defaults to '1.0.0'.\n   */\n  version?: string;\n  /**\n   * If true, tool responses from the MCP server will be returned in their raw\n   * MCP format. Otherwise (default), they are processed and potentially\n   * simplified for better compatibility with Genkit's typical data structures.\n   */\n  rawToolResponses?: boolean;\n  /** The server configuration to connect. */\n  mcpServer: McpServerConfig;\n  /** Manually supply a session id for HTTP streaming clients if desired. */\n  sessionId?: string;\n};\n\n/**\n * Represents a client connection to a single MCP (Model Context Protocol) server.\n * It handles the lifecycle of the connection (connect, disconnect, disable, re-enable, reconnect)\n * and provides methods to fetch tools from the connected server.\n *\n * An instance of `GenkitMcpClient` is typically managed by a `GenkitMcpHost`\n * when dealing with multiple MCP server connections.\n */\nexport class GenkitMcpClient {\n  _server?: McpServerRef;\n\n  sessionId?: string;\n  readonly name: string;\n  readonly suppliedServerName?: string;\n  private version: string;\n  private serverConfig: McpServerConfig;\n  private rawToolResponses?: boolean;\n  private disabled: boolean;\n  private roots?: Root[];\n\n  private _readyListeners: {\n    resolve: () => void;\n    reject: (err: Error) => void;\n  }[] = [];\n  private _ready = false;\n\n  constructor(options: McpClientOptions) {\n    this.name = options.name;\n    this.suppliedServerName = options.serverName;\n    this.version = options.version || '1.0.0';\n    this.serverConfig = options.mcpServer;\n    this.rawToolResponses = !!options.rawToolResponses;\n    this.disabled = !!options.mcpServer.disabled;\n    this.roots = options.mcpServer.roots;\n    this.sessionId = options.sessionId;\n\n    this._initializeConnection();\n  }\n\n  get serverName(): string {\n    return (\n      this.suppliedServerName ??\n      this._server?.client.getServerVersion()?.name ??\n      'unknown-server'\n    );\n  }\n\n  async updateRoots(roots: Root[]) {\n    this.roots = roots;\n    await this._server?.client.sendRootsListChanged();\n  }\n\n  /**\n   * Sets up a connection based on a provided map of server configurations.\n   * - Reconnects existing servers if their configuration appears to have\n   *   changed (implicitly handled by `connectServer`).\n   * - Sets the client's ready state once all connection attempts are complete.\n   * @param mcpServers A record mapping server names to their configurations.\n   */\n  private async _initializeConnection() {\n    this._ready = false;\n    try {\n      await this._connect();\n      this._ready = true;\n      while (this._readyListeners.length) {\n        this._readyListeners.pop()?.resolve();\n      }\n    } catch (err) {\n      while (this._readyListeners.length) {\n        this._readyListeners.pop()?.reject(err as Error);\n      }\n    }\n    if (this.roots) {\n      await this.updateRoots(this.roots);\n    }\n  }\n\n  /**\n   * Returns a Promise that resolves when the client has attempted to connect\n   * to all configured servers, or rejects if a critical error occurs during\n   * the initial connection phase.\n   */\n  async ready() {\n    if (this._ready) return;\n    return new Promise<void>((resolve, reject) => {\n      this._readyListeners.push({ resolve, reject });\n    });\n  }\n\n  /**\n   * Connects to a single MCP server defined by the provided configuration.\n   * @param config The configuration object for the server.\n   */\n  private async _connect() {\n    if (this._server) await this._server.transport.close();\n    logger.debug(\n      `[MCP Client] Connecting MCP server '${this.serverName}' in client '${this.name}'.`\n    );\n\n    const { transport, type: transportType } = await transportFrom(\n      this.serverConfig,\n      this.sessionId\n    );\n    if (!transport) {\n      throw new GenkitError({\n        status: 'INVALID_ARGUMENT',\n        message: `[MCP Client] Could not determine valid transport config from supplied options.`,\n      });\n    }\n\n    let error: string | undefined;\n\n    const client = new Client(\n      { name: this.name, version: this.version },\n      { capabilities: { roots: { listChanged: true } } }\n    );\n    client.setRequestHandler(ListRootsRequestSchema, () => {\n      logger.debug(`[MCP Client] fetching roots for ${this.name}`);\n      return { roots: this.roots || [] };\n    });\n\n    try {\n      await client.connect(transport);\n    } catch (e) {\n      logger.warn(\n        `[MCP Client] Error connecting server via ${transportType} transport: ${e}`\n      );\n      this.disabled = true;\n      error = (e as Error).toString();\n    }\n\n    this._server = {\n      client,\n      transport,\n      error,\n    } as McpServerRef;\n  }\n\n  /**\n   * Disconnects the MCP server and removes its registration\n   * from this client instance.\n   */\n  async _disconnect() {\n    if (this._server) {\n      logger.debug(\n        `[MCP Client] Disconnecting MCP server in client '${this.name}'.`\n      );\n      await this._server.client.close();\n      this._server = undefined;\n    }\n  }\n\n  /**\n   * Disables a server. Closes the underlying transport and server's configuration. Does nothing if the server is\n   * already disabled.\n   */\n  async disable() {\n    if (!this.isEnabled()) return;\n    if (this._server) {\n      logger.debug(\n        `[MCP Client] Disabling MCP server in client '${this.name}'`\n      );\n      await this._disconnect();\n      this.disabled = true;\n    }\n  }\n\n  /**\n   * Whether this client-server connection is enabled or not.\n   */\n  isEnabled() {\n    return !this.disabled;\n  }\n\n  /**\n   * Enables a server connection, including previously disabled ones. Does nothing if the\n   * server is not disabled.\n   */\n  async enable() {\n    if (this.isEnabled()) return;\n    logger.debug(`[MCP Client] Reenabling MCP server in client '${this.name}'`);\n    await this._initializeConnection();\n    this.disabled = !!this._server!.error;\n  }\n\n  /**\n   * Closes and then restarts the transport connection for the specified server.\n   * Useful for attempting to recover from connection issues without full\n   * reconfiguration.\n   */\n  async restart() {\n    if (this._server) {\n      logger.debug(\n        `[MCP Client] Restarting connection to MCP server in client '${this.name}'`\n      );\n      await this._disconnect();\n      await this._initializeConnection();\n    }\n  }\n\n  /**\n   * Fetches all tools available through this client, if the server\n   * configuration is not disabled.\n   */\n  async getActiveTools(ai: Genkit): Promise<ToolAction[]> {\n    await this.ready();\n    let tools: ToolAction[] = [];\n\n    if (this._server) {\n      const capabilities = this._server.client.getServerCapabilities();\n      if (capabilities?.tools)\n        tools.push(\n          ...(await fetchDynamicTools(ai, this._server.client, {\n            rawToolResponses: this.rawToolResponses,\n            serverName: this.serverName,\n            name: this.name,\n          }))\n        );\n    }\n\n    return tools;\n  }\n\n  /**\n   * Fetches all resources available through this client, if the server\n   * configuration is not disabled.\n   */\n  async getActiveResources(ai: Genkit): Promise<DynamicResourceAction[]> {\n    await this.ready();\n    let resources: DynamicResourceAction[] = [];\n\n    if (this._server) {\n      const capabilities = this._server.client.getServerCapabilities();\n      if (capabilities?.resources)\n        resources.push(\n          ...(await fetchDynamicResources(ai, this._server.client, {\n            serverName: this.serverName,\n            name: this.name,\n          }))\n        );\n    }\n\n    return resources;\n  }\n\n  /**\n   * Fetches all active prompts available through this client, if the server\n   * configuration supports prompts.\n   * @param ai The Genkit instance.\n   * @param options Optional prompt generation options.\n   * @returns A promise that resolves to an array of ExecutablePrompt.\n   */\n  async getActivePrompts(\n    ai: Genkit,\n    options?: PromptGenerateOptions\n  ): Promise<ExecutablePrompt[]> {\n    if (this._server?.client.getServerCapabilities()?.prompts) {\n      return fetchAllPrompts(this._server.client, {\n        ai,\n        serverName: this.serverName,\n        name: this.name,\n        options,\n      });\n    }\n    return [];\n  }\n\n  /**\n   * Get the specified prompt as an `ExecutablePrompt` available through this\n   * client. If no such prompt is found, return undefined.\n   */\n  async getPrompt(\n    ai: Genkit,\n    promptName: string,\n    opts?: PromptGenerateOptions\n  ): Promise<ExecutablePrompt | undefined> {\n    await this.ready();\n\n    if (this._server) {\n      const capabilities = await this._server.client.getServerCapabilities();\n      if (capabilities?.prompts) {\n        return await getExecutablePrompt(this._server.client, {\n          ai,\n          serverName: this.name,\n          promptName,\n          name: this.name,\n          options: opts,\n        });\n      }\n      logger.debug(`[MCP Client] No prompts are found in this MCP server.`);\n    }\n    return;\n  }\n\n  /** Returns the underlying MCP SDK client if one has been initialized. */\n  get mcpClient(): Client | undefined {\n    return this._server?.client;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAgBA,oBAAuB;AAIvB,mBAGO;AACP,oBAOO;AACP,qBAAuB;AACvB,kBAKO;AACP,sBAAsC;AAoF/B,MAAM,gBAAgB;AAAA,EAC3B;AAAA,EAEA;AAAA,EACS;AAAA,EACA;AAAA,EACD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAEA,kBAGF,CAAC;AAAA,EACC,SAAS;AAAA,EAEjB,YAAY,SAA2B;AACrC,SAAK,OAAO,QAAQ;AACpB,SAAK,qBAAqB,QAAQ;AAClC,SAAK,UAAU,QAAQ,WAAW;AAClC,SAAK,eAAe,QAAQ;AAC5B,SAAK,mBAAmB,CAAC,CAAC,QAAQ;AAClC,SAAK,WAAW,CAAC,CAAC,QAAQ,UAAU;AACpC,SAAK,QAAQ,QAAQ,UAAU;AAC/B,SAAK,YAAY,QAAQ;AAEzB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,IAAI,aAAqB;AACvB,WACE,KAAK,sBACL,KAAK,SAAS,OAAO,iBAAiB,GAAG,QACzC;AAAA,EAEJ;AAAA,EAEA,MAAM,YAAY,OAAe;AAC/B,SAAK,QAAQ;AACb,UAAM,KAAK,SAAS,OAAO,qBAAqB;AAAA,EAClD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,wBAAwB;AACpC,SAAK,SAAS;AACd,QAAI;AACF,YAAM,KAAK,SAAS;AACpB,WAAK,SAAS;AACd,aAAO,KAAK,gBAAgB,QAAQ;AAClC,aAAK,gBAAgB,IAAI,GAAG,QAAQ;AAAA,MACtC;AAAA,IACF,SAAS,KAAK;AACZ,aAAO,KAAK,gBAAgB,QAAQ;AAClC,aAAK,gBAAgB,IAAI,GAAG,OAAO,GAAY;AAAA,MACjD;AAAA,IACF;AACA,QAAI,KAAK,OAAO;AACd,YAAM,KAAK,YAAY,KAAK,KAAK;AAAA,IACnC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,QAAQ;AACZ,QAAI,KAAK,OAAQ;AACjB,WAAO,IAAI,QAAc,CAAC,SAAS,WAAW;AAC5C,WAAK,gBAAgB,KAAK,EAAE,SAAS,OAAO,CAAC;AAAA,IAC/C,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAc,WAAW;AACvB,QAAI,KAAK,QAAS,OAAM,KAAK,QAAQ,UAAU,MAAM;AACrD,0BAAO;AAAA,MACL,uCAAuC,KAAK,UAAU,gBAAgB,KAAK,IAAI;AAAA,IACjF;AAEA,UAAM,EAAE,WAAW,MAAM,cAAc,IAAI,UAAM;AAAA,MAC/C,KAAK;AAAA,MACL,KAAK;AAAA,IACP;AACA,QAAI,CAAC,WAAW;AACd,YAAM,IAAI,0BAAY;AAAA,QACpB,QAAQ;AAAA,QACR,SAAS;AAAA,MACX,CAAC;AAAA,IACH;AAEA,QAAI;AAEJ,UAAM,SAAS,IAAI;AAAA,MACjB,EAAE,MAAM,KAAK,MAAM,SAAS,KAAK,QAAQ;AAAA,MACzC,EAAE,cAAc,EAAE,OAAO,EAAE,aAAa,KAAK,EAAE,EAAE;AAAA,IACnD;AACA,WAAO,kBAAkB,qCAAwB,MAAM;AACrD,4BAAO,MAAM,mCAAmC,KAAK,IAAI,EAAE;AAC3D,aAAO,EAAE,OAAO,KAAK,SAAS,CAAC,EAAE;AAAA,IACnC,CAAC;AAED,QAAI;AACF,YAAM,OAAO,QAAQ,SAAS;AAAA,IAChC,SAAS,GAAG;AACV,4BAAO;AAAA,QACL,4CAA4C,aAAa,eAAe,CAAC;AAAA,MAC3E;AACA,WAAK,WAAW;AAChB,cAAS,EAAY,SAAS;AAAA,IAChC;AAEA,SAAK,UAAU;AAAA,MACb;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,cAAc;AAClB,QAAI,KAAK,SAAS;AAChB,4BAAO;AAAA,QACL,oDAAoD,KAAK,IAAI;AAAA,MAC/D;AACA,YAAM,KAAK,QAAQ,OAAO,MAAM;AAChC,WAAK,UAAU;AAAA,IACjB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,UAAU;AACd,QAAI,CAAC,KAAK,UAAU,EAAG;AACvB,QAAI,KAAK,SAAS;AAChB,4BAAO;AAAA,QACL,gDAAgD,KAAK,IAAI;AAAA,MAC3D;AACA,YAAM,KAAK,YAAY;AACvB,WAAK,WAAW;AAAA,IAClB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,YAAY;AACV,WAAO,CAAC,KAAK;AAAA,EACf;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,SAAS;AACb,QAAI,KAAK,UAAU,EAAG;AACtB,0BAAO,MAAM,iDAAiD,KAAK,IAAI,GAAG;AAC1E,UAAM,KAAK,sBAAsB;AACjC,SAAK,WAAW,CAAC,CAAC,KAAK,QAAS;AAAA,EAClC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,UAAU;AACd,QAAI,KAAK,SAAS;AAChB,4BAAO;AAAA,QACL,+DAA+D,KAAK,IAAI;AAAA,MAC1E;AACA,YAAM,KAAK,YAAY;AACvB,YAAM,KAAK,sBAAsB;AAAA,IACnC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,eAAe,IAAmC;AACtD,UAAM,KAAK,MAAM;AACjB,QAAI,QAAsB,CAAC;AAE3B,QAAI,KAAK,SAAS;AAChB,YAAM,eAAe,KAAK,QAAQ,OAAO,sBAAsB;AAC/D,UAAI,cAAc;AAChB,cAAM;AAAA,UACJ,GAAI,UAAM,+BAAkB,IAAI,KAAK,QAAQ,QAAQ;AAAA,YACnD,kBAAkB,KAAK;AAAA,YACvB,YAAY,KAAK;AAAA,YACjB,MAAM,KAAK;AAAA,UACb,CAAC;AAAA,QACH;AAAA,IACJ;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,mBAAmB,IAA8C;AACrE,UAAM,KAAK,MAAM;AACjB,QAAI,YAAqC,CAAC;AAE1C,QAAI,KAAK,SAAS;AAChB,YAAM,eAAe,KAAK,QAAQ,OAAO,sBAAsB;AAC/D,UAAI,cAAc;AAChB,kBAAU;AAAA,UACR,GAAI,UAAM,uCAAsB,IAAI,KAAK,QAAQ,QAAQ;AAAA,YACvD,YAAY,KAAK;AAAA,YACjB,MAAM,KAAK;AAAA,UACb,CAAC;AAAA,QACH;AAAA,IACJ;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAM,iBACJ,IACA,SAC6B;AAC7B,QAAI,KAAK,SAAS,OAAO,sBAAsB,GAAG,SAAS;AACzD,iBAAO,6BAAgB,KAAK,QAAQ,QAAQ;AAAA,QAC1C;AAAA,QACA,YAAY,KAAK;AAAA,QACjB,MAAM,KAAK;AAAA,QACX;AAAA,MACF,CAAC;AAAA,IACH;AACA,WAAO,CAAC;AAAA,EACV;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,UACJ,IACA,YACA,MACuC;AACvC,UAAM,KAAK,MAAM;AAEjB,QAAI,KAAK,SAAS;AAChB,YAAM,eAAe,MAAM,KAAK,QAAQ,OAAO,sBAAsB;AACrE,UAAI,cAAc,SAAS;AACzB,eAAO,UAAM,iCAAoB,KAAK,QAAQ,QAAQ;AAAA,UACpD;AAAA,UACA,YAAY,KAAK;AAAA,UACjB;AAAA,UACA,MAAM,KAAK;AAAA,UACX,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AACA,4BAAO,MAAM,uDAAuD;AAAA,IACtE;AACA;AAAA,EACF;AAAA;AAAA,EAGA,IAAI,YAAgC;AAClC,WAAO,KAAK,SAAS;AAAA,EACvB;AACF;","names":[]}