@genkit-ai/mcp 1.14.1-rc.0

package/lib/util/tools.d.mts

@@ -0,0 +1,37 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { Genkit, ToolAction } from 'genkit';
+
+/**
+ * Copyright 2024 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.
+ */
+
+/**
+ * Lookup all tools available in the server and register each as a Genkit tool.
+ */
+declare function registerAllTools(ai: Genkit, client: Client, params: {
+    name: string;
+    serverName: string;
+    rawToolResponses?: boolean;
+}): Promise<void>;
+/**
+ * Lookup all tools available in the server and fetches as a Genkit dynamic tool.
+ */
+declare function fetchDynamicTools(ai: Genkit, client: Client, params: {
+    name: string;
+    serverName: string;
+    rawToolResponses?: boolean;
+}): Promise<ToolAction[]>;
+
+export { fetchDynamicTools, registerAllTools };

package/lib/util/tools.d.ts

@@ -0,0 +1,37 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { Genkit, ToolAction } from 'genkit';
+
+/**
+ * Copyright 2024 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.
+ */
+
+/**
+ * Lookup all tools available in the server and register each as a Genkit tool.
+ */
+declare function registerAllTools(ai: Genkit, client: Client, params: {
+    name: string;
+    serverName: string;
+    rawToolResponses?: boolean;
+}): Promise<void>;
+/**
+ * Lookup all tools available in the server and fetches as a Genkit dynamic tool.
+ */
+declare function fetchDynamicTools(ai: Genkit, client: Client, params: {
+    name: string;
+    serverName: string;
+    rawToolResponses?: boolean;
+}): Promise<ToolAction[]>;
+
+export { fetchDynamicTools, registerAllTools };

package/lib/util/tools.js

@@ -0,0 +1,120 @@
+"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 tools_exports = {};
+__export(tools_exports, {
+  fetchDynamicTools: () => fetchDynamicTools,
+  registerAllTools: () => registerAllTools
+});
+module.exports = __toCommonJS(tools_exports);
+var import_genkit = require("genkit");
+var import_logging = require("genkit/logging");
+const toText = (c) => c.map((p) => p.text || "").join("");
+function processResult(result) {
+  if (result.isError) return { error: toText(result.content) };
+  if (result.content.every((c) => !!c.text)) {
+    const text = toText(result.content);
+    if (text.trim().startsWith("{") || text.trim().startsWith("[")) {
+      try {
+        return JSON.parse(text);
+      } catch (e) {
+        return text;
+      }
+    }
+    return text;
+  }
+  if (result.content.length === 1) return result.content[0];
+  return result;
+}
+function registerTool(ai, client, tool, params) {
+  import_logging.logger.debug(
+    `[MCP] Registering tool '${params.name}/${tool.name}'' from server '${params.serverName}'`
+  );
+  ai.defineTool(
+    {
+      name: `${params.serverName}/${tool.name}`,
+      description: tool.description || "",
+      inputJsonSchema: tool.inputSchema,
+      outputSchema: import_genkit.z.any(),
+      metadata: { mcp: { _meta: tool._meta || {} } }
+    },
+    async (args) => {
+      import_logging.logger.debug(
+        `[MCP] Calling MCP tool '${params.serverName}/${tool.name}' with arguments`,
+        JSON.stringify(args)
+      );
+      const result = await client.callTool({
+        name: tool.name,
+        arguments: args
+      });
+      if (params.rawToolResponses) return result;
+      return processResult(result);
+    }
+  );
+}
+function createDynamicTool(ai, client, tool, params) {
+  return ai.dynamicTool(
+    {
+      name: `${params.serverName}/${tool.name}`,
+      description: tool.description || "",
+      inputJsonSchema: tool.inputSchema,
+      outputSchema: import_genkit.z.any(),
+      metadata: { mcp: { _meta: tool._meta || {} } }
+    },
+    async (args, { context }) => {
+      import_logging.logger.debug(
+        `[MCP] calling tool '${params.serverName}/${tool.name}' in host '${params.name}'`
+      );
+      const result = await client.callTool({
+        name: tool.name,
+        arguments: args,
+        _meta: context?.mcp?._meta
+      });
+      if (params.rawToolResponses) return result;
+      return processResult(result);
+    }
+  );
+}
+async function registerAllTools(ai, client, params) {
+  let cursor;
+  while (true) {
+    const { nextCursor, tools } = await client.listTools({ cursor });
+    tools.forEach((t) => registerTool(ai, client, t, params));
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+}
+async function fetchDynamicTools(ai, client, params) {
+  let cursor;
+  let allTools = [];
+  while (true) {
+    const { nextCursor, tools } = await client.listTools({ cursor });
+    allTools.push(
+      ...tools.map((t) => createDynamicTool(ai, client, t, params))
+    );
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+  return allTools;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  fetchDynamicTools,
+  registerAllTools
+});
+//# sourceMappingURL=tools.js.map

package/lib/util/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/tools.ts"],"sourcesContent":["/**\n * Copyright 2024 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 {\n  CallToolResult,\n  Tool,\n} from '@modelcontextprotocol/sdk/types.js' with { 'resolution-mode': 'import' };\nimport { Genkit, JSONSchema7, ToolAction, z } from 'genkit';\nimport { logger } from 'genkit/logging';\n\nconst toText = (c: CallToolResult['content']) =>\n  c.map((p) => p.text || '').join('');\n\nfunction processResult(result: CallToolResult) {\n  if (result.isError) return { error: toText(result.content) };\n  if (result.content.every((c) => !!c.text)) {\n    const text = toText(result.content);\n    if (text.trim().startsWith('{') || text.trim().startsWith('[')) {\n      try {\n        return JSON.parse(text);\n      } catch (e) {\n        return text;\n      }\n    }\n    return text;\n  }\n  if (result.content.length === 1) return result.content[0];\n  return result;\n}\n\n/**\n * Registers a single MCP tool as a Genkit tool.\n * It defines a new Genkit tool action that, when called, will\n * interact with the MCP client to execute the corresponding MCP tool.\n *\n * @param ai The Genkit instance to define the tool on.\n * @param client The MCP client instance used to interact with the MCP server.\n * @param tool The MCP Tool object to register.\n * @param params Contains the Genkit client name, MCP server name for namespacing,\n *               and a flag for raw tool responses.\n */\nfunction registerTool(\n  ai: Genkit,\n  client: Client,\n  tool: Tool,\n  params: { serverName: string; name: string; rawToolResponses?: boolean }\n) {\n  logger.debug(\n    `[MCP] Registering tool '${params.name}/${tool.name}'' from server '${params.serverName}'`\n  );\n  ai.defineTool(\n    {\n      name: `${params.serverName}/${tool.name}`,\n      description: tool.description || '',\n      inputJsonSchema: tool.inputSchema as JSONSchema7,\n      outputSchema: z.any(),\n      metadata: { mcp: { _meta: tool._meta || {} } },\n    },\n    async (args) => {\n      logger.debug(\n        `[MCP] Calling MCP tool '${params.serverName}/${tool.name}' with arguments`,\n        JSON.stringify(args)\n      );\n      const result = await client.callTool({\n        name: tool.name,\n        arguments: args,\n      });\n      if (params.rawToolResponses) return result;\n      return processResult(result as CallToolResult);\n    }\n  );\n}\n\n/**\n * Creates a Genkit dynamic tool action for a given MCP tool.\n * This is similar to `registerTool` but returns the `ToolAction` directly\n * instead of defining it on the Genkit instance.\n *\n * @param ai The Genkit instance, used for creating the dynamic tool.\n * @param client The MCP client instance.\n * @param tool The MCP Tool object.\n * @param params Configuration parameters including namespacing and raw response flag.\n * @returns A Genkit `ToolAction` representing the MCP tool.\n */\nfunction createDynamicTool(\n  ai: Genkit,\n  client: Client,\n  tool: Tool,\n  params: { serverName: string; name: string; rawToolResponses?: boolean }\n): ToolAction {\n  return ai.dynamicTool(\n    {\n      name: `${params.serverName}/${tool.name}`,\n      description: tool.description || '',\n      inputJsonSchema: tool.inputSchema as JSONSchema7,\n      outputSchema: z.any(),\n      metadata: { mcp: { _meta: tool._meta || {} } },\n    },\n    async (args, { context }) => {\n      logger.debug(\n        `[MCP] calling tool '${params.serverName}/${tool.name}' in host '${params.name}'`\n      );\n      const result = await client.callTool({\n        name: tool.name,\n        arguments: args,\n        _meta: context?.mcp?._meta,\n      });\n      if (params.rawToolResponses) return result;\n      return processResult(result as CallToolResult);\n    }\n  );\n}\n\n/**\n * Lookup all tools available in the server and register each as a Genkit tool.\n */\nexport async function registerAllTools(\n  ai: Genkit,\n  client: Client,\n  params: { name: string; serverName: string; rawToolResponses?: boolean }\n): Promise<void> {\n  let cursor: string | undefined;\n  while (true) {\n    const { nextCursor, tools } = await client.listTools({ cursor });\n    tools.forEach((t) => registerTool(ai, client, t, params));\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n}\n\n/**\n * Lookup all tools available in the server and fetches as a Genkit dynamic tool.\n */\nexport async function fetchDynamicTools(\n  ai: Genkit,\n  client: Client,\n  params: { name: string; serverName: string; rawToolResponses?: boolean }\n): Promise<ToolAction[]> {\n  let cursor: string | undefined;\n  let allTools: ToolAction[] = [];\n  while (true) {\n    const { nextCursor, tools } = await client.listTools({ cursor });\n    allTools.push(\n      ...tools.map((t) => createDynamicTool(ai, client, t, params))\n    );\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n  return allTools;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAqBA,oBAAmD;AACnD,qBAAuB;AAEvB,MAAM,SAAS,CAAC,MACd,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE;AAEpC,SAAS,cAAc,QAAwB;AAC7C,MAAI,OAAO,QAAS,QAAO,EAAE,OAAO,OAAO,OAAO,OAAO,EAAE;AAC3D,MAAI,OAAO,QAAQ,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,GAAG;AACzC,UAAM,OAAO,OAAO,OAAO,OAAO;AAClC,QAAI,KAAK,KAAK,EAAE,WAAW,GAAG,KAAK,KAAK,KAAK,EAAE,WAAW,GAAG,GAAG;AAC9D,UAAI;AACF,eAAO,KAAK,MAAM,IAAI;AAAA,MACxB,SAAS,GAAG;AACV,eAAO;AAAA,MACT;AAAA,IACF;AACA,WAAO;AAAA,EACT;AACA,MAAI,OAAO,QAAQ,WAAW,EAAG,QAAO,OAAO,QAAQ,CAAC;AACxD,SAAO;AACT;AAaA,SAAS,aACP,IACA,QACA,MACA,QACA;AACA,wBAAO;AAAA,IACL,2BAA2B,OAAO,IAAI,IAAI,KAAK,IAAI,mBAAmB,OAAO,UAAU;AAAA,EACzF;AACA,KAAG;AAAA,IACD;AAAA,MACE,MAAM,GAAG,OAAO,UAAU,IAAI,KAAK,IAAI;AAAA,MACvC,aAAa,KAAK,eAAe;AAAA,MACjC,iBAAiB,KAAK;AAAA,MACtB,cAAc,gBAAE,IAAI;AAAA,MACpB,UAAU,EAAE,KAAK,EAAE,OAAO,KAAK,SAAS,CAAC,EAAE,EAAE;AAAA,IAC/C;AAAA,IACA,OAAO,SAAS;AACd,4BAAO;AAAA,QACL,2BAA2B,OAAO,UAAU,IAAI,KAAK,IAAI;AAAA,QACzD,KAAK,UAAU,IAAI;AAAA,MACrB;AACA,YAAM,SAAS,MAAM,OAAO,SAAS;AAAA,QACnC,MAAM,KAAK;AAAA,QACX,WAAW;AAAA,MACb,CAAC;AACD,UAAI,OAAO,iBAAkB,QAAO;AACpC,aAAO,cAAc,MAAwB;AAAA,IAC/C;AAAA,EACF;AACF;AAaA,SAAS,kBACP,IACA,QACA,MACA,QACY;AACZ,SAAO,GAAG;AAAA,IACR;AAAA,MACE,MAAM,GAAG,OAAO,UAAU,IAAI,KAAK,IAAI;AAAA,MACvC,aAAa,KAAK,eAAe;AAAA,MACjC,iBAAiB,KAAK;AAAA,MACtB,cAAc,gBAAE,IAAI;AAAA,MACpB,UAAU,EAAE,KAAK,EAAE,OAAO,KAAK,SAAS,CAAC,EAAE,EAAE;AAAA,IAC/C;AAAA,IACA,OAAO,MAAM,EAAE,QAAQ,MAAM;AAC3B,4BAAO;AAAA,QACL,uBAAuB,OAAO,UAAU,IAAI,KAAK,IAAI,cAAc,OAAO,IAAI;AAAA,MAChF;AACA,YAAM,SAAS,MAAM,OAAO,SAAS;AAAA,QACnC,MAAM,KAAK;AAAA,QACX,WAAW;AAAA,QACX,OAAO,SAAS,KAAK;AAAA,MACvB,CAAC;AACD,UAAI,OAAO,iBAAkB,QAAO;AACpC,aAAO,cAAc,MAAwB;AAAA,IAC/C;AAAA,EACF;AACF;AAKA,eAAsB,iBACpB,IACA,QACA,QACe;AACf,MAAI;AACJ,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,MAAM,IAAI,MAAM,OAAO,UAAU,EAAE,OAAO,CAAC;AAC/D,UAAM,QAAQ,CAAC,MAAM,aAAa,IAAI,QAAQ,GAAG,MAAM,CAAC;AACxD,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACF;AAKA,eAAsB,kBACpB,IACA,QACA,QACuB;AACvB,MAAI;AACJ,MAAI,WAAyB,CAAC;AAC9B,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,MAAM,IAAI,MAAM,OAAO,UAAU,EAAE,OAAO,CAAC;AAC/D,aAAS;AAAA,MACP,GAAG,MAAM,IAAI,CAAC,MAAM,kBAAkB,IAAI,QAAQ,GAAG,MAAM,CAAC;AAAA,IAC9D;AACA,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACA,SAAO;AACT;","names":[]}

package/lib/util/tools.mjs

@@ -0,0 +1,95 @@
+import { z } from "genkit";
+import { logger } from "genkit/logging";
+const toText = (c) => c.map((p) => p.text || "").join("");
+function processResult(result) {
+  if (result.isError) return { error: toText(result.content) };
+  if (result.content.every((c) => !!c.text)) {
+    const text = toText(result.content);
+    if (text.trim().startsWith("{") || text.trim().startsWith("[")) {
+      try {
+        return JSON.parse(text);
+      } catch (e) {
+        return text;
+      }
+    }
+    return text;
+  }
+  if (result.content.length === 1) return result.content[0];
+  return result;
+}
+function registerTool(ai, client, tool, params) {
+  logger.debug(
+    `[MCP] Registering tool '${params.name}/${tool.name}'' from server '${params.serverName}'`
+  );
+  ai.defineTool(
+    {
+      name: `${params.serverName}/${tool.name}`,
+      description: tool.description || "",
+      inputJsonSchema: tool.inputSchema,
+      outputSchema: z.any(),
+      metadata: { mcp: { _meta: tool._meta || {} } }
+    },
+    async (args) => {
+      logger.debug(
+        `[MCP] Calling MCP tool '${params.serverName}/${tool.name}' with arguments`,
+        JSON.stringify(args)
+      );
+      const result = await client.callTool({
+        name: tool.name,
+        arguments: args
+      });
+      if (params.rawToolResponses) return result;
+      return processResult(result);
+    }
+  );
+}
+function createDynamicTool(ai, client, tool, params) {
+  return ai.dynamicTool(
+    {
+      name: `${params.serverName}/${tool.name}`,
+      description: tool.description || "",
+      inputJsonSchema: tool.inputSchema,
+      outputSchema: z.any(),
+      metadata: { mcp: { _meta: tool._meta || {} } }
+    },
+    async (args, { context }) => {
+      logger.debug(
+        `[MCP] calling tool '${params.serverName}/${tool.name}' in host '${params.name}'`
+      );
+      const result = await client.callTool({
+        name: tool.name,
+        arguments: args,
+        _meta: context?.mcp?._meta
+      });
+      if (params.rawToolResponses) return result;
+      return processResult(result);
+    }
+  );
+}
+async function registerAllTools(ai, client, params) {
+  let cursor;
+  while (true) {
+    const { nextCursor, tools } = await client.listTools({ cursor });
+    tools.forEach((t) => registerTool(ai, client, t, params));
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+}
+async function fetchDynamicTools(ai, client, params) {
+  let cursor;
+  let allTools = [];
+  while (true) {
+    const { nextCursor, tools } = await client.listTools({ cursor });
+    allTools.push(
+      ...tools.map((t) => createDynamicTool(ai, client, t, params))
+    );
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+  return allTools;
+}
+export {
+  fetchDynamicTools,
+  registerAllTools
+};
+//# sourceMappingURL=tools.mjs.map

package/lib/util/tools.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/tools.ts"],"sourcesContent":["/**\n * Copyright 2024 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 {\n  CallToolResult,\n  Tool,\n} from '@modelcontextprotocol/sdk/types.js' with { 'resolution-mode': 'import' };\nimport { Genkit, JSONSchema7, ToolAction, z } from 'genkit';\nimport { logger } from 'genkit/logging';\n\nconst toText = (c: CallToolResult['content']) =>\n  c.map((p) => p.text || '').join('');\n\nfunction processResult(result: CallToolResult) {\n  if (result.isError) return { error: toText(result.content) };\n  if (result.content.every((c) => !!c.text)) {\n    const text = toText(result.content);\n    if (text.trim().startsWith('{') || text.trim().startsWith('[')) {\n      try {\n        return JSON.parse(text);\n      } catch (e) {\n        return text;\n      }\n    }\n    return text;\n  }\n  if (result.content.length === 1) return result.content[0];\n  return result;\n}\n\n/**\n * Registers a single MCP tool as a Genkit tool.\n * It defines a new Genkit tool action that, when called, will\n * interact with the MCP client to execute the corresponding MCP tool.\n *\n * @param ai The Genkit instance to define the tool on.\n * @param client The MCP client instance used to interact with the MCP server.\n * @param tool The MCP Tool object to register.\n * @param params Contains the Genkit client name, MCP server name for namespacing,\n *               and a flag for raw tool responses.\n */\nfunction registerTool(\n  ai: Genkit,\n  client: Client,\n  tool: Tool,\n  params: { serverName: string; name: string; rawToolResponses?: boolean }\n) {\n  logger.debug(\n    `[MCP] Registering tool '${params.name}/${tool.name}'' from server '${params.serverName}'`\n  );\n  ai.defineTool(\n    {\n      name: `${params.serverName}/${tool.name}`,\n      description: tool.description || '',\n      inputJsonSchema: tool.inputSchema as JSONSchema7,\n      outputSchema: z.any(),\n      metadata: { mcp: { _meta: tool._meta || {} } },\n    },\n    async (args) => {\n      logger.debug(\n        `[MCP] Calling MCP tool '${params.serverName}/${tool.name}' with arguments`,\n        JSON.stringify(args)\n      );\n      const result = await client.callTool({\n        name: tool.name,\n        arguments: args,\n      });\n      if (params.rawToolResponses) return result;\n      return processResult(result as CallToolResult);\n    }\n  );\n}\n\n/**\n * Creates a Genkit dynamic tool action for a given MCP tool.\n * This is similar to `registerTool` but returns the `ToolAction` directly\n * instead of defining it on the Genkit instance.\n *\n * @param ai The Genkit instance, used for creating the dynamic tool.\n * @param client The MCP client instance.\n * @param tool The MCP Tool object.\n * @param params Configuration parameters including namespacing and raw response flag.\n * @returns A Genkit `ToolAction` representing the MCP tool.\n */\nfunction createDynamicTool(\n  ai: Genkit,\n  client: Client,\n  tool: Tool,\n  params: { serverName: string; name: string; rawToolResponses?: boolean }\n): ToolAction {\n  return ai.dynamicTool(\n    {\n      name: `${params.serverName}/${tool.name}`,\n      description: tool.description || '',\n      inputJsonSchema: tool.inputSchema as JSONSchema7,\n      outputSchema: z.any(),\n      metadata: { mcp: { _meta: tool._meta || {} } },\n    },\n    async (args, { context }) => {\n      logger.debug(\n        `[MCP] calling tool '${params.serverName}/${tool.name}' in host '${params.name}'`\n      );\n      const result = await client.callTool({\n        name: tool.name,\n        arguments: args,\n        _meta: context?.mcp?._meta,\n      });\n      if (params.rawToolResponses) return result;\n      return processResult(result as CallToolResult);\n    }\n  );\n}\n\n/**\n * Lookup all tools available in the server and register each as a Genkit tool.\n */\nexport async function registerAllTools(\n  ai: Genkit,\n  client: Client,\n  params: { name: string; serverName: string; rawToolResponses?: boolean }\n): Promise<void> {\n  let cursor: string | undefined;\n  while (true) {\n    const { nextCursor, tools } = await client.listTools({ cursor });\n    tools.forEach((t) => registerTool(ai, client, t, params));\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n}\n\n/**\n * Lookup all tools available in the server and fetches as a Genkit dynamic tool.\n */\nexport async function fetchDynamicTools(\n  ai: Genkit,\n  client: Client,\n  params: { name: string; serverName: string; rawToolResponses?: boolean }\n): Promise<ToolAction[]> {\n  let cursor: string | undefined;\n  let allTools: ToolAction[] = [];\n  while (true) {\n    const { nextCursor, tools } = await client.listTools({ cursor });\n    allTools.push(\n      ...tools.map((t) => createDynamicTool(ai, client, t, params))\n    );\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n  return allTools;\n}\n"],"mappings":"AAqBA,SAA0C,SAAS;AACnD,SAAS,cAAc;AAEvB,MAAM,SAAS,CAAC,MACd,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE;AAEpC,SAAS,cAAc,QAAwB;AAC7C,MAAI,OAAO,QAAS,QAAO,EAAE,OAAO,OAAO,OAAO,OAAO,EAAE;AAC3D,MAAI,OAAO,QAAQ,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,GAAG;AACzC,UAAM,OAAO,OAAO,OAAO,OAAO;AAClC,QAAI,KAAK,KAAK,EAAE,WAAW,GAAG,KAAK,KAAK,KAAK,EAAE,WAAW,GAAG,GAAG;AAC9D,UAAI;AACF,eAAO,KAAK,MAAM,IAAI;AAAA,MACxB,SAAS,GAAG;AACV,eAAO;AAAA,MACT;AAAA,IACF;AACA,WAAO;AAAA,EACT;AACA,MAAI,OAAO,QAAQ,WAAW,EAAG,QAAO,OAAO,QAAQ,CAAC;AACxD,SAAO;AACT;AAaA,SAAS,aACP,IACA,QACA,MACA,QACA;AACA,SAAO;AAAA,IACL,2BAA2B,OAAO,IAAI,IAAI,KAAK,IAAI,mBAAmB,OAAO,UAAU;AAAA,EACzF;AACA,KAAG;AAAA,IACD;AAAA,MACE,MAAM,GAAG,OAAO,UAAU,IAAI,KAAK,IAAI;AAAA,MACvC,aAAa,KAAK,eAAe;AAAA,MACjC,iBAAiB,KAAK;AAAA,MACtB,cAAc,EAAE,IAAI;AAAA,MACpB,UAAU,EAAE,KAAK,EAAE,OAAO,KAAK,SAAS,CAAC,EAAE,EAAE;AAAA,IAC/C;AAAA,IACA,OAAO,SAAS;AACd,aAAO;AAAA,QACL,2BAA2B,OAAO,UAAU,IAAI,KAAK,IAAI;AAAA,QACzD,KAAK,UAAU,IAAI;AAAA,MACrB;AACA,YAAM,SAAS,MAAM,OAAO,SAAS;AAAA,QACnC,MAAM,KAAK;AAAA,QACX,WAAW;AAAA,MACb,CAAC;AACD,UAAI,OAAO,iBAAkB,QAAO;AACpC,aAAO,cAAc,MAAwB;AAAA,IAC/C;AAAA,EACF;AACF;AAaA,SAAS,kBACP,IACA,QACA,MACA,QACY;AACZ,SAAO,GAAG;AAAA,IACR;AAAA,MACE,MAAM,GAAG,OAAO,UAAU,IAAI,KAAK,IAAI;AAAA,MACvC,aAAa,KAAK,eAAe;AAAA,MACjC,iBAAiB,KAAK;AAAA,MACtB,cAAc,EAAE,IAAI;AAAA,MACpB,UAAU,EAAE,KAAK,EAAE,OAAO,KAAK,SAAS,CAAC,EAAE,EAAE;AAAA,IAC/C;AAAA,IACA,OAAO,MAAM,EAAE,QAAQ,MAAM;AAC3B,aAAO;AAAA,QACL,uBAAuB,OAAO,UAAU,IAAI,KAAK,IAAI,cAAc,OAAO,IAAI;AAAA,MAChF;AACA,YAAM,SAAS,MAAM,OAAO,SAAS;AAAA,QACnC,MAAM,KAAK;AAAA,QACX,WAAW;AAAA,QACX,OAAO,SAAS,KAAK;AAAA,MACvB,CAAC;AACD,UAAI,OAAO,iBAAkB,QAAO;AACpC,aAAO,cAAc,MAAwB;AAAA,IAC/C;AAAA,EACF;AACF;AAKA,eAAsB,iBACpB,IACA,QACA,QACe;AACf,MAAI;AACJ,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,MAAM,IAAI,MAAM,OAAO,UAAU,EAAE,OAAO,CAAC;AAC/D,UAAM,QAAQ,CAAC,MAAM,aAAa,IAAI,QAAQ,GAAG,MAAM,CAAC;AACxD,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACF;AAKA,eAAsB,kBACpB,IACA,QACA,QACuB;AACvB,MAAI;AACJ,MAAI,WAAyB,CAAC;AAC9B,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,MAAM,IAAI,MAAM,OAAO,UAAU,EAAE,OAAO,CAAC;AAC/D,aAAS;AAAA,MACP,GAAG,MAAM,IAAI,CAAC,MAAM,kBAAkB,IAAI,QAAQ,GAAG,MAAM,CAAC;AAAA,IAC9D;AACA,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACA,SAAO;AACT;","names":[]}

package/lib/util/transport.d.mts

@@ -0,0 +1,39 @@
+import { McpServerConfig } from '../client/client.mjs';
+import { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
+import '@modelcontextprotocol/sdk/client/index.js';
+import '@modelcontextprotocol/sdk/client/stdio.js';
+import '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import '@modelcontextprotocol/sdk/types.js';
+import '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.
+ */
+
+/**
+ * Creates an MCP transport instance based on the provided server configuration.
+ * It supports creating SSE, Stdio, or using a pre-configured custom transport.
+ *
+ * @param config The configuration for the MCP server, determining the type of transport to create.
+ * @returns A Promise resolving to an object containing the created `Transport` instance
+ *          (or `null` if configuration is invalid) and a string indicating the `type` of transport.
+ * @throws May throw an error if essential MCP SDK components cannot be imported.
+ */
+declare function transportFrom(config: McpServerConfig, sessionId?: string): Promise<{
+    transport: Transport | null;
+    type: string;
+}>;
+
+export { transportFrom };

package/lib/util/transport.d.ts

@@ -0,0 +1,39 @@
+import { McpServerConfig } from '../client/client.js';
+import { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
+import '@modelcontextprotocol/sdk/client/index.js';
+import '@modelcontextprotocol/sdk/client/stdio.js';
+import '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import '@modelcontextprotocol/sdk/types.js';
+import '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.
+ */
+
+/**
+ * Creates an MCP transport instance based on the provided server configuration.
+ * It supports creating SSE, Stdio, or using a pre-configured custom transport.
+ *
+ * @param config The configuration for the MCP server, determining the type of transport to create.
+ * @returns A Promise resolving to an object containing the created `Transport` instance
+ *          (or `null` if configuration is invalid) and a string indicating the `type` of transport.
+ * @throws May throw an error if essential MCP SDK components cannot be imported.
+ */
+declare function transportFrom(config: McpServerConfig, sessionId?: string): Promise<{
+    transport: Transport | null;
+    type: string;
+}>;
+
+export { transportFrom };

package/lib/util/transport.js

@@ -0,0 +1,63 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+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 __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var transport_exports = {};
+__export(transport_exports, {
+  transportFrom: () => transportFrom
+});
+module.exports = __toCommonJS(transport_exports);
+async function transportFrom(config, sessionId) {
+  if ("transport" in config && config.transport) {
+    return { transport: config.transport, type: "custom" };
+  }
+  if ("url" in config && config.url) {
+    const { url, ...httpConfig } = config;
+    const { StreamableHTTPClientTransport } = await import("@modelcontextprotocol/sdk/client/streamableHttp.js");
+    return {
+      transport: new StreamableHTTPClientTransport(new URL(url), {
+        ...httpConfig,
+        sessionId
+      }),
+      type: "http"
+    };
+  }
+  if ("command" in config && config.command) {
+    const stdioConfig = { ...config };
+    const { StdioClientTransport } = await import("@modelcontextprotocol/sdk/client/stdio.js");
+    return {
+      transport: new StdioClientTransport(stdioConfig),
+      type: "stdio"
+    };
+  }
+  return { transport: null, type: "unknown" };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  transportFrom
+});
+//# sourceMappingURL=transport.js.map

package/lib/util/transport.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/transport.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 { McpServerConfig } from '../client/client.js';\nimport type { StdioServerParameters, Transport } from '../client/index.js';\n\n/**\n * Creates an MCP transport instance based on the provided server configuration.\n * It supports creating SSE, Stdio, or using a pre-configured custom transport.\n *\n * @param config The configuration for the MCP server, determining the type of transport to create.\n * @returns A Promise resolving to an object containing the created `Transport` instance\n *          (or `null` if configuration is invalid) and a string indicating the `type` of transport.\n * @throws May throw an error if essential MCP SDK components cannot be imported.\n */\nexport async function transportFrom(\n  config: McpServerConfig,\n  sessionId?: string\n): Promise<{\n  transport: Transport | null;\n  type: string;\n}> {\n  // Handle pre-configured transport first\n  if ('transport' in config && config.transport) {\n    return { transport: config.transport, type: 'custom' };\n  }\n  // Handle SSE config\n  if ('url' in config && config.url) {\n    const { url, ...httpConfig } = config;\n    const { StreamableHTTPClientTransport } = await import(\n      '@modelcontextprotocol/sdk/client/streamableHttp.js'\n    );\n    return {\n      transport: new StreamableHTTPClientTransport(new URL(url), {\n        ...httpConfig,\n        sessionId,\n      }),\n      type: 'http',\n    };\n  }\n  // Handle Stdio config\n  if ('command' in config && config.command) {\n    // Create a copy and remove McpServerControls properties\n    const stdioConfig = { ...config };\n    const { StdioClientTransport } = await import(\n      '@modelcontextprotocol/sdk/client/stdio.js'\n    );\n    return {\n      transport: new StdioClientTransport(stdioConfig as StdioServerParameters),\n      type: 'stdio',\n    };\n  }\n  return { transport: null, type: 'unknown' };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AA4BA,eAAsB,cACpB,QACA,WAIC;AAED,MAAI,eAAe,UAAU,OAAO,WAAW;AAC7C,WAAO,EAAE,WAAW,OAAO,WAAW,MAAM,SAAS;AAAA,EACvD;AAEA,MAAI,SAAS,UAAU,OAAO,KAAK;AACjC,UAAM,EAAE,KAAK,GAAG,WAAW,IAAI;AAC/B,UAAM,EAAE,8BAA8B,IAAI,MAAM,OAC9C,oDACF;AACA,WAAO;AAAA,MACL,WAAW,IAAI,8BAA8B,IAAI,IAAI,GAAG,GAAG;AAAA,QACzD,GAAG;AAAA,QACH;AAAA,MACF,CAAC;AAAA,MACD,MAAM;AAAA,IACR;AAAA,EACF;AAEA,MAAI,aAAa,UAAU,OAAO,SAAS;AAEzC,UAAM,cAAc,EAAE,GAAG,OAAO;AAChC,UAAM,EAAE,qBAAqB,IAAI,MAAM,OACrC,2CACF;AACA,WAAO;AAAA,MACL,WAAW,IAAI,qBAAqB,WAAoC;AAAA,MACxE,MAAM;AAAA,IACR;AAAA,EACF;AACA,SAAO,EAAE,WAAW,MAAM,MAAM,UAAU;AAC5C;","names":[]}

package/lib/util/transport.mjs

@@ -0,0 +1,29 @@
+async function transportFrom(config, sessionId) {
+  if ("transport" in config && config.transport) {
+    return { transport: config.transport, type: "custom" };
+  }
+  if ("url" in config && config.url) {
+    const { url, ...httpConfig } = config;
+    const { StreamableHTTPClientTransport } = await import("@modelcontextprotocol/sdk/client/streamableHttp.js");
+    return {
+      transport: new StreamableHTTPClientTransport(new URL(url), {
+        ...httpConfig,
+        sessionId
+      }),
+      type: "http"
+    };
+  }
+  if ("command" in config && config.command) {
+    const stdioConfig = { ...config };
+    const { StdioClientTransport } = await import("@modelcontextprotocol/sdk/client/stdio.js");
+    return {
+      transport: new StdioClientTransport(stdioConfig),
+      type: "stdio"
+    };
+  }
+  return { transport: null, type: "unknown" };
+}
+export {
+  transportFrom
+};
+//# sourceMappingURL=transport.mjs.map

package/lib/util/transport.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/transport.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 { McpServerConfig } from '../client/client.js';\nimport type { StdioServerParameters, Transport } from '../client/index.js';\n\n/**\n * Creates an MCP transport instance based on the provided server configuration.\n * It supports creating SSE, Stdio, or using a pre-configured custom transport.\n *\n * @param config The configuration for the MCP server, determining the type of transport to create.\n * @returns A Promise resolving to an object containing the created `Transport` instance\n *          (or `null` if configuration is invalid) and a string indicating the `type` of transport.\n * @throws May throw an error if essential MCP SDK components cannot be imported.\n */\nexport async function transportFrom(\n  config: McpServerConfig,\n  sessionId?: string\n): Promise<{\n  transport: Transport | null;\n  type: string;\n}> {\n  // Handle pre-configured transport first\n  if ('transport' in config && config.transport) {\n    return { transport: config.transport, type: 'custom' };\n  }\n  // Handle SSE config\n  if ('url' in config && config.url) {\n    const { url, ...httpConfig } = config;\n    const { StreamableHTTPClientTransport } = await import(\n      '@modelcontextprotocol/sdk/client/streamableHttp.js'\n    );\n    return {\n      transport: new StreamableHTTPClientTransport(new URL(url), {\n        ...httpConfig,\n        sessionId,\n      }),\n      type: 'http',\n    };\n  }\n  // Handle Stdio config\n  if ('command' in config && config.command) {\n    // Create a copy and remove McpServerControls properties\n    const stdioConfig = { ...config };\n    const { StdioClientTransport } = await import(\n      '@modelcontextprotocol/sdk/client/stdio.js'\n    );\n    return {\n      transport: new StdioClientTransport(stdioConfig as StdioServerParameters),\n      type: 'stdio',\n    };\n  }\n  return { transport: null, type: 'unknown' };\n}\n"],"mappings":"AA4BA,eAAsB,cACpB,QACA,WAIC;AAED,MAAI,eAAe,UAAU,OAAO,WAAW;AAC7C,WAAO,EAAE,WAAW,OAAO,WAAW,MAAM,SAAS;AAAA,EACvD;AAEA,MAAI,SAAS,UAAU,OAAO,KAAK;AACjC,UAAM,EAAE,KAAK,GAAG,WAAW,IAAI;AAC/B,UAAM,EAAE,8BAA8B,IAAI,MAAM,OAC9C,oDACF;AACA,WAAO;AAAA,MACL,WAAW,IAAI,8BAA8B,IAAI,IAAI,GAAG,GAAG;AAAA,QACzD,GAAG;AAAA,QACH;AAAA,MACF,CAAC;AAAA,MACD,MAAM;AAAA,IACR;AAAA,EACF;AAEA,MAAI,aAAa,UAAU,OAAO,SAAS;AAEzC,UAAM,cAAc,EAAE,GAAG,OAAO;AAChC,UAAM,EAAE,qBAAqB,IAAI,MAAM,OACrC,2CACF;AACA,WAAO;AAAA,MACL,WAAW,IAAI,qBAAqB,WAAoC;AAAA,MACxE,MAAM;AAAA,IACR;AAAA,EACF;AACA,SAAO,EAAE,WAAW,MAAM,MAAM,UAAU;AAC5C;","names":[]}

package/package.json

@@ -0,0 +1,57 @@
+{
+  "name": "@genkit-ai/mcp",
+  "keywords": [
+    "genkit",
+    "genkit-plugin",
+    "mcp",
+    "model-context-protocol",
+    "ai",
+    "genai",
+    "generative-ai"
+  ],
+  "version": "1.14.1-rc.0",
+  "description": "A Genkit plugin that provides interoperability between Genkit and Model Context Protocol (MCP). Both client and server use cases are supported.",
+  "main": "./lib/index.js",
+  "types": "./lib/index.d.ts",
+  "exports": {
+    ".": {
+      "require": "./lib/index.js",
+      "import": "./lib/index.mjs",
+      "types": "./lib/index.d.ts",
+      "default": "./lib/index.js"
+    }
+  },
+  "type": "commonjs",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/firebase/genkit.git",
+    "directory": "js/plugins/mcp"
+  },
+  "author": "genkit",
+  "license": "Apache-2.0",
+  "peerDependencies": {
+    "@modelcontextprotocol/sdk": "^1.13.0",
+    "genkit": "^1.14.1"
+  },
+  "devDependencies": {
+    "get-port": "^5.1.0",
+    "express": "^5.1.0",
+    "@jest/globals": "^29.7.0",
+    "@types/node": "^20.11.16",
+    "jest": "^29.7.0",
+    "npm-run-all": "^4.1.5",
+    "rimraf": "^6.0.1",
+    "ts-jest": "^29.1.2",
+    "tsup": "^8.3.5",
+    "tsx": "^4.19.2",
+    "typescript": "^5.3.0"
+  },
+  "scripts": {
+    "check": "tsc",
+    "compile": "tsup-node",
+    "build:clean": "rimraf ./lib",
+    "build": "npm-run-all build:clean check compile",
+    "build:watch": "tsup-node --watch",
+    "test": "tsx --test ./tests/*_test.ts"
+  }
+}