@genkit-ai/mcp 1.14.1-rc.0

package/lib/server.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/server.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 {\n  GenkitError,\n  Message,\n  type Genkit,\n  type MessageData,\n  type PromptAction,\n} from 'genkit';\nimport type { McpServerOptions } from './index.js';\n\nimport type { Server } from '@modelcontextprotocol/sdk/server/index.js' with { 'resolution-mode': 'import' };\nimport type { Transport } from '@modelcontextprotocol/sdk/shared/transport.js' with { 'resolution-mode': 'import' };\nimport type {\n  CallToolRequest,\n  CallToolResult,\n  GetPromptRequest,\n  GetPromptResult,\n  ListPromptsRequest,\n  ListPromptsResult,\n  ListToolsRequest,\n  ListToolsResult,\n  Prompt,\n  PromptMessage,\n  Tool,\n} from '@modelcontextprotocol/sdk/types.js' with { 'resolution-mode': 'import' };\nimport { logger } from 'genkit/logging';\nimport { toJsonSchema } from 'genkit/schema';\nimport { ToolAction, toToolDefinition } from 'genkit/tool';\n\n/**\n * Represents an MCP (Model Context Protocol) server that exposes Genkit tools\n * and prompts. This class wraps a Genkit instance and makes its registered\n * actions (tools, prompts) available to MCP clients. It handles the translation\n * between Genkit's action definitions and MCP's expected formats.\n */\nexport class GenkitMcpServer {\n  ai: Genkit;\n  options: McpServerOptions;\n  server?: Server;\n  actionsResolved = false;\n  toolActions: ToolAction[] = [];\n  promptActions: PromptAction[] = [];\n\n  /**\n   * Creates an instance of GenkitMcpServer.\n   * @param ai The Genkit instance whose actions will be exposed.\n   * @param options Configuration options for the MCP server, like its name and version.\n   */\n  constructor(ai: Genkit, options: McpServerOptions) {\n    this.ai = ai;\n    this.options = options;\n  }\n\n  /**\n   * Initializes the MCP server instance and registers request handlers. It\n   * dynamically imports MCP SDK components and sets up handlers for listing\n   * tools, calling tools, listing prompts, and getting prompts. It also\n   * resolves and stores all tool and prompt actions from the Genkit instance.\n   *\n   * This method is called by the constructor and ensures the server is ready\n   * before any requests are handled. It's idempotent.\n   */\n  async setup(): Promise<void> {\n    if (this.actionsResolved) return;\n    const { Server } = await import(\n      '@modelcontextprotocol/sdk/server/index.js'\n    );\n\n    this.server = new Server(\n      { name: this.options.name, version: this.options.version || '1.0.0' },\n      {\n        capabilities: {\n          prompts: {},\n          tools: {},\n        },\n      }\n    );\n\n    const {\n      CallToolRequestSchema,\n      GetPromptRequestSchema,\n      ListPromptsRequestSchema,\n      ListToolsRequestSchema,\n    } = await import('@modelcontextprotocol/sdk/types.js');\n\n    this.server.setRequestHandler(\n      ListToolsRequestSchema,\n      this.listTools.bind(this)\n    );\n    this.server.setRequestHandler(\n      CallToolRequestSchema,\n      this.callTool.bind(this)\n    );\n    this.server.setRequestHandler(\n      ListPromptsRequestSchema,\n      this.listPrompts.bind(this)\n    );\n    this.server.setRequestHandler(\n      GetPromptRequestSchema,\n      this.getPrompt.bind(this)\n    );\n\n    // TODO -- use listResolvableActions.\n    const allActions = await this.ai.registry.listActions();\n    const toolList: ToolAction[] = [];\n    const promptList: PromptAction[] = [];\n    for (const k in allActions) {\n      if (k.startsWith('/tool/')) {\n        toolList.push(allActions[k] as ToolAction);\n      } else if (k.startsWith('/prompt/')) {\n        promptList.push(allActions[k] as PromptAction);\n      }\n    }\n    this.toolActions = toolList;\n    this.promptActions = promptList;\n    this.actionsResolved = true;\n  }\n\n  /**\n   * Handles MCP requests to list available tools.\n   * It maps the resolved Genkit tool actions to the MCP Tool format.\n   * @param req The MCP ListToolsRequest.\n   * @returns A Promise resolving to an MCP ListToolsResult.\n   */\n  async listTools(req: ListToolsRequest): Promise<ListToolsResult> {\n    await this.setup();\n    return {\n      tools: this.toolActions.map((t): Tool => {\n        const def = toToolDefinition(t);\n        return {\n          name: def.name,\n          inputSchema: (def.inputSchema as any) || { type: 'object' },\n          description: def.description,\n          _meta: t.__action.metadata?.mcp?._meta,\n        };\n      }),\n    };\n  }\n\n  /**\n   * Handles MCP requests to call a specific tool. It finds the corresponding\n   * Genkit tool action and executes it with the provided arguments. The result\n   * is then formatted as an MCP CallToolResult.\n   * @param req The MCP CallToolRequest containing the tool name and arguments.\n   * @returns A Promise resolving to an MCP CallToolResult.\n   * @throws GenkitError if the requested tool is not found.\n   */\n  async callTool(req: CallToolRequest): Promise<CallToolResult> {\n    await this.setup();\n    const tool = this.toolActions.find(\n      (t) => t.__action.name === req.params.name\n    );\n    if (!tool)\n      throw new GenkitError({\n        status: 'NOT_FOUND',\n        message: `Tried to call tool '${req.params.name}' but it could not be found.`,\n      });\n    const result = await tool(req.params.arguments);\n    return {\n      content: [\n        {\n          type: 'text',\n          text: typeof result === 'string' ? result : JSON.stringify(result),\n        },\n      ],\n    };\n  }\n\n  /**\n   * Handles MCP requests to list available prompts.\n   * It maps the resolved Genkit prompt actions to the MCP Prompt format,\n   * including converting input schemas to MCP argument definitions.\n   * @param req The MCP ListPromptsRequest.\n   * @returns A Promise resolving to an MCP ListPromptsResult.\n   */\n  async listPrompts(req: ListPromptsRequest): Promise<ListPromptsResult> {\n    await this.setup();\n    return {\n      prompts: this.promptActions.map((p): Prompt => {\n        return {\n          name: p.__action.name,\n          description: p.__action.description,\n          arguments: toMcpPromptArguments(p),\n          _meta: p.__action.metadata?.mcp?._meta,\n        };\n      }),\n    };\n  }\n\n  /**\n   * Handles MCP requests to get (render) a specific prompt. It finds the\n   * corresponding Genkit prompt action, executes it with the provided\n   * arguments, and then formats the resulting messages into the MCP\n   * PromptMessage format.\n   * @param req The MCP GetPromptRequest containing the prompt name and\n   * arguments.\n   * @returns A Promise resolving to an MCP GetPromptResult.\n   * @throws GenkitError if the requested prompt is not found.\n   */\n  async getPrompt(req: GetPromptRequest): Promise<GetPromptResult> {\n    await this.setup();\n    const prompt = this.promptActions.find(\n      (p) => p.__action.name === req.params.name\n    );\n    if (!prompt)\n      throw new GenkitError({\n        status: 'NOT_FOUND',\n        message: `[MCP Server] Tried to call prompt '${req.params.name}' but it could not be found.`,\n      });\n    const result = await prompt(req.params.arguments);\n    return {\n      description: prompt.__action.description,\n      messages: result.messages.map(toMcpPromptMessage),\n    };\n  }\n\n  /**\n   * Starts the MCP server with the specified transport or a default\n   * StdioServerTransport. Ensures the server is set up before connecting the\n   * transport.\n   * @param transport Optional MCP transport instance. If not provided, a\n   * StdioServerTransport will be created and used.\n   */\n  async start(transport?: Transport) {\n    if (!transport) {\n      const { StdioServerTransport } = await import(\n        '@modelcontextprotocol/sdk/server/stdio.js'\n      );\n      transport = new StdioServerTransport();\n    }\n    await this.setup();\n    await this.server!.connect(transport);\n    logger.info(\n      `[MCP Server] MCP server '${this.options.name}' started successfully.`\n    );\n  }\n}\n\n/**\n * Converts a Genkit prompt's input schema to an array of MCP prompt arguments.\n * MCP prompts currently only support string arguments.\n * @param p The Genkit PromptAction.\n * @returns An array of MCP prompt arguments, or undefined if the schema is not defined.\n * @throws GenkitError if the input schema is not an object or if any property is not a string.\n */\nfunction toMcpPromptArguments(\n  p: PromptAction\n): Prompt['arguments'] | undefined {\n  const jsonSchema = toJsonSchema({\n    schema: p.__action.inputSchema,\n    jsonSchema: p.__action.inputJsonSchema,\n  });\n\n  if (!jsonSchema) return undefined;\n  if (!jsonSchema.properties)\n    throw new GenkitError({\n      status: 'FAILED_PRECONDITION',\n      message: '[MCP Server] MCP prompts must take objects as input schema.',\n    });\n\n  const args: Prompt['arguments'] = [];\n  for (const k in jsonSchema.properties) {\n    const { type, description } = jsonSchema.properties[k];\n    if (\n      type !== 'string' &&\n      (!Array.isArray(type) || !type.includes('string'))\n    ) {\n      throw new GenkitError({\n        status: 'FAILED_PRECONDITION',\n        message: `[MCP Server] MCP prompts may only take string arguments, but ${p.__action.name} has property '${k}' of type '${type}'.`,\n      });\n    }\n    args.push({\n      name: k,\n      description,\n      required: jsonSchema.required?.includes(k),\n    });\n  }\n  return args;\n}\n\nconst ROLE_MAP = { model: 'assistant', user: 'user' } as const;\n\n/**\n * Converts a Genkit MessageData object to an MCP PromptMessage.\n * Handles mapping of roles and content types (text, image).\n * MCP only supports 'user' and 'assistant' (model) roles and base64 data images.\n * @param messageData The Genkit MessageData object.\n * @returns An MCP PromptMessage.\n * @throws GenkitError if the role is unsupported or if media is not a data URL.\n */\nfunction toMcpPromptMessage(messageData: MessageData): PromptMessage {\n  if (messageData.role !== 'model' && messageData.role !== 'user') {\n    throw new GenkitError({\n      status: 'UNIMPLEMENTED',\n      message: `[MCP Server] MCP prompt messages do not support role '${messageData.role}'. Only 'user' and 'model' messages are supported.`,\n    });\n  }\n  const message = new Message(messageData);\n  const common = { role: ROLE_MAP[messageData.role] };\n  if (message.media) {\n    const { url, contentType } = message.media;\n    if (!url.startsWith('data:'))\n      throw new GenkitError({\n        status: 'UNIMPLEMENTED',\n        message: `[MCP Server] MCP prompt messages only support base64 data images.`,\n      });\n    const mimeType =\n      contentType || url.substring(url.indexOf(':')! + 1, url.indexOf(';'));\n    const data = url.substring(url.indexOf(',') + 1);\n    return { ...common, content: { type: 'image', mimeType, data } };\n  } else {\n    return { ...common, content: { type: 'text', text: message.text } };\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAgBA,oBAMO;AAkBP,qBAAuB;AACvB,oBAA6B;AAC7B,kBAA6C;AAQtC,MAAM,gBAAgB;AAAA,EAC3B;AAAA,EACA;AAAA,EACA;AAAA,EACA,kBAAkB;AAAA,EAClB,cAA4B,CAAC;AAAA,EAC7B,gBAAgC,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOjC,YAAY,IAAY,SAA2B;AACjD,SAAK,KAAK;AACV,SAAK,UAAU;AAAA,EACjB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,QAAuB;AAC3B,QAAI,KAAK,gBAAiB;AAC1B,UAAM,EAAE,OAAO,IAAI,MAAM,OACvB,2CACF;AAEA,SAAK,SAAS,IAAI;AAAA,MAChB,EAAE,MAAM,KAAK,QAAQ,MAAM,SAAS,KAAK,QAAQ,WAAW,QAAQ;AAAA,MACpE;AAAA,QACE,cAAc;AAAA,UACZ,SAAS,CAAC;AAAA,UACV,OAAO,CAAC;AAAA,QACV;AAAA,MACF;AAAA,IACF;AAEA,UAAM;AAAA,MACJ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,IAAI,MAAM,OAAO,oCAAoC;AAErD,SAAK,OAAO;AAAA,MACV;AAAA,MACA,KAAK,UAAU,KAAK,IAAI;AAAA,IAC1B;AACA,SAAK,OAAO;AAAA,MACV;AAAA,MACA,KAAK,SAAS,KAAK,IAAI;AAAA,IACzB;AACA,SAAK,OAAO;AAAA,MACV;AAAA,MACA,KAAK,YAAY,KAAK,IAAI;AAAA,IAC5B;AACA,SAAK,OAAO;AAAA,MACV;AAAA,MACA,KAAK,UAAU,KAAK,IAAI;AAAA,IAC1B;AAGA,UAAM,aAAa,MAAM,KAAK,GAAG,SAAS,YAAY;AACtD,UAAM,WAAyB,CAAC;AAChC,UAAM,aAA6B,CAAC;AACpC,eAAW,KAAK,YAAY;AAC1B,UAAI,EAAE,WAAW,QAAQ,GAAG;AAC1B,iBAAS,KAAK,WAAW,CAAC,CAAe;AAAA,MAC3C,WAAW,EAAE,WAAW,UAAU,GAAG;AACnC,mBAAW,KAAK,WAAW,CAAC,CAAiB;AAAA,MAC/C;AAAA,IACF;AACA,SAAK,cAAc;AACnB,SAAK,gBAAgB;AACrB,SAAK,kBAAkB;AAAA,EACzB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,UAAU,KAAiD;AAC/D,UAAM,KAAK,MAAM;AACjB,WAAO;AAAA,MACL,OAAO,KAAK,YAAY,IAAI,CAAC,MAAY;AACvC,cAAM,UAAM,8BAAiB,CAAC;AAC9B,eAAO;AAAA,UACL,MAAM,IAAI;AAAA,UACV,aAAc,IAAI,eAAuB,EAAE,MAAM,SAAS;AAAA,UAC1D,aAAa,IAAI;AAAA,UACjB,OAAO,EAAE,SAAS,UAAU,KAAK;AAAA,QACnC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,SAAS,KAA+C;AAC5D,UAAM,KAAK,MAAM;AACjB,UAAM,OAAO,KAAK,YAAY;AAAA,MAC5B,CAAC,MAAM,EAAE,SAAS,SAAS,IAAI,OAAO;AAAA,IACxC;AACA,QAAI,CAAC;AACH,YAAM,IAAI,0BAAY;AAAA,QACpB,QAAQ;AAAA,QACR,SAAS,uBAAuB,IAAI,OAAO,IAAI;AAAA,MACjD,CAAC;AACH,UAAM,SAAS,MAAM,KAAK,IAAI,OAAO,SAAS;AAC9C,WAAO;AAAA,MACL,SAAS;AAAA,QACP;AAAA,UACE,MAAM;AAAA,UACN,MAAM,OAAO,WAAW,WAAW,SAAS,KAAK,UAAU,MAAM;AAAA,QACnE;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAM,YAAY,KAAqD;AACrE,UAAM,KAAK,MAAM;AACjB,WAAO;AAAA,MACL,SAAS,KAAK,cAAc,IAAI,CAAC,MAAc;AAC7C,eAAO;AAAA,UACL,MAAM,EAAE,SAAS;AAAA,UACjB,aAAa,EAAE,SAAS;AAAA,UACxB,WAAW,qBAAqB,CAAC;AAAA,UACjC,OAAO,EAAE,SAAS,UAAU,KAAK;AAAA,QACnC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYA,MAAM,UAAU,KAAiD;AAC/D,UAAM,KAAK,MAAM;AACjB,UAAM,SAAS,KAAK,cAAc;AAAA,MAChC,CAAC,MAAM,EAAE,SAAS,SAAS,IAAI,OAAO;AAAA,IACxC;AACA,QAAI,CAAC;AACH,YAAM,IAAI,0BAAY;AAAA,QACpB,QAAQ;AAAA,QACR,SAAS,sCAAsC,IAAI,OAAO,IAAI;AAAA,MAChE,CAAC;AACH,UAAM,SAAS,MAAM,OAAO,IAAI,OAAO,SAAS;AAChD,WAAO;AAAA,MACL,aAAa,OAAO,SAAS;AAAA,MAC7B,UAAU,OAAO,SAAS,IAAI,kBAAkB;AAAA,IAClD;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAM,MAAM,WAAuB;AACjC,QAAI,CAAC,WAAW;AACd,YAAM,EAAE,qBAAqB,IAAI,MAAM,OACrC,2CACF;AACA,kBAAY,IAAI,qBAAqB;AAAA,IACvC;AACA,UAAM,KAAK,MAAM;AACjB,UAAM,KAAK,OAAQ,QAAQ,SAAS;AACpC,0BAAO;AAAA,MACL,4BAA4B,KAAK,QAAQ,IAAI;AAAA,IAC/C;AAAA,EACF;AACF;AASA,SAAS,qBACP,GACiC;AACjC,QAAM,iBAAa,4BAAa;AAAA,IAC9B,QAAQ,EAAE,SAAS;AAAA,IACnB,YAAY,EAAE,SAAS;AAAA,EACzB,CAAC;AAED,MAAI,CAAC,WAAY,QAAO;AACxB,MAAI,CAAC,WAAW;AACd,UAAM,IAAI,0BAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SAAS;AAAA,IACX,CAAC;AAEH,QAAM,OAA4B,CAAC;AACnC,aAAW,KAAK,WAAW,YAAY;AACrC,UAAM,EAAE,MAAM,YAAY,IAAI,WAAW,WAAW,CAAC;AACrD,QACE,SAAS,aACR,CAAC,MAAM,QAAQ,IAAI,KAAK,CAAC,KAAK,SAAS,QAAQ,IAChD;AACA,YAAM,IAAI,0BAAY;AAAA,QACpB,QAAQ;AAAA,QACR,SAAS,gEAAgE,EAAE,SAAS,IAAI,kBAAkB,CAAC,cAAc,IAAI;AAAA,MAC/H,CAAC;AAAA,IACH;AACA,SAAK,KAAK;AAAA,MACR,MAAM;AAAA,MACN;AAAA,MACA,UAAU,WAAW,UAAU,SAAS,CAAC;AAAA,IAC3C,CAAC;AAAA,EACH;AACA,SAAO;AACT;AAEA,MAAM,WAAW,EAAE,OAAO,aAAa,MAAM,OAAO;AAUpD,SAAS,mBAAmB,aAAyC;AACnE,MAAI,YAAY,SAAS,WAAW,YAAY,SAAS,QAAQ;AAC/D,UAAM,IAAI,0BAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SAAS,yDAAyD,YAAY,IAAI;AAAA,IACpF,CAAC;AAAA,EACH;AACA,QAAM,UAAU,IAAI,sBAAQ,WAAW;AACvC,QAAM,SAAS,EAAE,MAAM,SAAS,YAAY,IAAI,EAAE;AAClD,MAAI,QAAQ,OAAO;AACjB,UAAM,EAAE,KAAK,YAAY,IAAI,QAAQ;AACrC,QAAI,CAAC,IAAI,WAAW,OAAO;AACzB,YAAM,IAAI,0BAAY;AAAA,QACpB,QAAQ;AAAA,QACR,SAAS;AAAA,MACX,CAAC;AACH,UAAM,WACJ,eAAe,IAAI,UAAU,IAAI,QAAQ,GAAG,IAAK,GAAG,IAAI,QAAQ,GAAG,CAAC;AACtE,UAAM,OAAO,IAAI,UAAU,IAAI,QAAQ,GAAG,IAAI,CAAC;AAC/C,WAAO,EAAE,GAAG,QAAQ,SAAS,EAAE,MAAM,SAAS,UAAU,KAAK,EAAE;AAAA,EACjE,OAAO;AACL,WAAO,EAAE,GAAG,QAAQ,SAAS,EAAE,MAAM,QAAQ,MAAM,QAAQ,KAAK,EAAE;AAAA,EACpE;AACF;","names":[]}

package/lib/server.mjs

@@ -0,0 +1,249 @@
+import {
+  GenkitError,
+  Message
+} from "genkit";
+import { logger } from "genkit/logging";
+import { toJsonSchema } from "genkit/schema";
+import { toToolDefinition } from "genkit/tool";
+class GenkitMcpServer {
+  ai;
+  options;
+  server;
+  actionsResolved = false;
+  toolActions = [];
+  promptActions = [];
+  /**
+   * Creates an instance of GenkitMcpServer.
+   * @param ai The Genkit instance whose actions will be exposed.
+   * @param options Configuration options for the MCP server, like its name and version.
+   */
+  constructor(ai, options) {
+    this.ai = ai;
+    this.options = options;
+  }
+  /**
+   * Initializes the MCP server instance and registers request handlers. It
+   * dynamically imports MCP SDK components and sets up handlers for listing
+   * tools, calling tools, listing prompts, and getting prompts. It also
+   * resolves and stores all tool and prompt actions from the Genkit instance.
+   *
+   * This method is called by the constructor and ensures the server is ready
+   * before any requests are handled. It's idempotent.
+   */
+  async setup() {
+    if (this.actionsResolved) return;
+    const { Server } = await import("@modelcontextprotocol/sdk/server/index.js");
+    this.server = new Server(
+      { name: this.options.name, version: this.options.version || "1.0.0" },
+      {
+        capabilities: {
+          prompts: {},
+          tools: {}
+        }
+      }
+    );
+    const {
+      CallToolRequestSchema,
+      GetPromptRequestSchema,
+      ListPromptsRequestSchema,
+      ListToolsRequestSchema
+    } = await import("@modelcontextprotocol/sdk/types.js");
+    this.server.setRequestHandler(
+      ListToolsRequestSchema,
+      this.listTools.bind(this)
+    );
+    this.server.setRequestHandler(
+      CallToolRequestSchema,
+      this.callTool.bind(this)
+    );
+    this.server.setRequestHandler(
+      ListPromptsRequestSchema,
+      this.listPrompts.bind(this)
+    );
+    this.server.setRequestHandler(
+      GetPromptRequestSchema,
+      this.getPrompt.bind(this)
+    );
+    const allActions = await this.ai.registry.listActions();
+    const toolList = [];
+    const promptList = [];
+    for (const k in allActions) {
+      if (k.startsWith("/tool/")) {
+        toolList.push(allActions[k]);
+      } else if (k.startsWith("/prompt/")) {
+        promptList.push(allActions[k]);
+      }
+    }
+    this.toolActions = toolList;
+    this.promptActions = promptList;
+    this.actionsResolved = true;
+  }
+  /**
+   * Handles MCP requests to list available tools.
+   * It maps the resolved Genkit tool actions to the MCP Tool format.
+   * @param req The MCP ListToolsRequest.
+   * @returns A Promise resolving to an MCP ListToolsResult.
+   */
+  async listTools(req) {
+    await this.setup();
+    return {
+      tools: this.toolActions.map((t) => {
+        const def = toToolDefinition(t);
+        return {
+          name: def.name,
+          inputSchema: def.inputSchema || { type: "object" },
+          description: def.description,
+          _meta: t.__action.metadata?.mcp?._meta
+        };
+      })
+    };
+  }
+  /**
+   * Handles MCP requests to call a specific tool. It finds the corresponding
+   * Genkit tool action and executes it with the provided arguments. The result
+   * is then formatted as an MCP CallToolResult.
+   * @param req The MCP CallToolRequest containing the tool name and arguments.
+   * @returns A Promise resolving to an MCP CallToolResult.
+   * @throws GenkitError if the requested tool is not found.
+   */
+  async callTool(req) {
+    await this.setup();
+    const tool = this.toolActions.find(
+      (t) => t.__action.name === req.params.name
+    );
+    if (!tool)
+      throw new GenkitError({
+        status: "NOT_FOUND",
+        message: `Tried to call tool '${req.params.name}' but it could not be found.`
+      });
+    const result = await tool(req.params.arguments);
+    return {
+      content: [
+        {
+          type: "text",
+          text: typeof result === "string" ? result : JSON.stringify(result)
+        }
+      ]
+    };
+  }
+  /**
+   * Handles MCP requests to list available prompts.
+   * It maps the resolved Genkit prompt actions to the MCP Prompt format,
+   * including converting input schemas to MCP argument definitions.
+   * @param req The MCP ListPromptsRequest.
+   * @returns A Promise resolving to an MCP ListPromptsResult.
+   */
+  async listPrompts(req) {
+    await this.setup();
+    return {
+      prompts: this.promptActions.map((p) => {
+        return {
+          name: p.__action.name,
+          description: p.__action.description,
+          arguments: toMcpPromptArguments(p),
+          _meta: p.__action.metadata?.mcp?._meta
+        };
+      })
+    };
+  }
+  /**
+   * Handles MCP requests to get (render) a specific prompt. It finds the
+   * corresponding Genkit prompt action, executes it with the provided
+   * arguments, and then formats the resulting messages into the MCP
+   * PromptMessage format.
+   * @param req The MCP GetPromptRequest containing the prompt name and
+   * arguments.
+   * @returns A Promise resolving to an MCP GetPromptResult.
+   * @throws GenkitError if the requested prompt is not found.
+   */
+  async getPrompt(req) {
+    await this.setup();
+    const prompt = this.promptActions.find(
+      (p) => p.__action.name === req.params.name
+    );
+    if (!prompt)
+      throw new GenkitError({
+        status: "NOT_FOUND",
+        message: `[MCP Server] Tried to call prompt '${req.params.name}' but it could not be found.`
+      });
+    const result = await prompt(req.params.arguments);
+    return {
+      description: prompt.__action.description,
+      messages: result.messages.map(toMcpPromptMessage)
+    };
+  }
+  /**
+   * Starts the MCP server with the specified transport or a default
+   * StdioServerTransport. Ensures the server is set up before connecting the
+   * transport.
+   * @param transport Optional MCP transport instance. If not provided, a
+   * StdioServerTransport will be created and used.
+   */
+  async start(transport) {
+    if (!transport) {
+      const { StdioServerTransport } = await import("@modelcontextprotocol/sdk/server/stdio.js");
+      transport = new StdioServerTransport();
+    }
+    await this.setup();
+    await this.server.connect(transport);
+    logger.info(
+      `[MCP Server] MCP server '${this.options.name}' started successfully.`
+    );
+  }
+}
+function toMcpPromptArguments(p) {
+  const jsonSchema = toJsonSchema({
+    schema: p.__action.inputSchema,
+    jsonSchema: p.__action.inputJsonSchema
+  });
+  if (!jsonSchema) return void 0;
+  if (!jsonSchema.properties)
+    throw new GenkitError({
+      status: "FAILED_PRECONDITION",
+      message: "[MCP Server] MCP prompts must take objects as input schema."
+    });
+  const args = [];
+  for (const k in jsonSchema.properties) {
+    const { type, description } = jsonSchema.properties[k];
+    if (type !== "string" && (!Array.isArray(type) || !type.includes("string"))) {
+      throw new GenkitError({
+        status: "FAILED_PRECONDITION",
+        message: `[MCP Server] MCP prompts may only take string arguments, but ${p.__action.name} has property '${k}' of type '${type}'.`
+      });
+    }
+    args.push({
+      name: k,
+      description,
+      required: jsonSchema.required?.includes(k)
+    });
+  }
+  return args;
+}
+const ROLE_MAP = { model: "assistant", user: "user" };
+function toMcpPromptMessage(messageData) {
+  if (messageData.role !== "model" && messageData.role !== "user") {
+    throw new GenkitError({
+      status: "UNIMPLEMENTED",
+      message: `[MCP Server] MCP prompt messages do not support role '${messageData.role}'. Only 'user' and 'model' messages are supported.`
+    });
+  }
+  const message = new Message(messageData);
+  const common = { role: ROLE_MAP[messageData.role] };
+  if (message.media) {
+    const { url, contentType } = message.media;
+    if (!url.startsWith("data:"))
+      throw new GenkitError({
+        status: "UNIMPLEMENTED",
+        message: `[MCP Server] MCP prompt messages only support base64 data images.`
+      });
+    const mimeType = contentType || url.substring(url.indexOf(":") + 1, url.indexOf(";"));
+    const data = url.substring(url.indexOf(",") + 1);
+    return { ...common, content: { type: "image", mimeType, data } };
+  } else {
+    return { ...common, content: { type: "text", text: message.text } };
+  }
+}
+export {
+  GenkitMcpServer
+};
+//# sourceMappingURL=server.mjs.map

package/lib/server.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/server.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 {\n  GenkitError,\n  Message,\n  type Genkit,\n  type MessageData,\n  type PromptAction,\n} from 'genkit';\nimport type { McpServerOptions } from './index.js';\n\nimport type { Server } from '@modelcontextprotocol/sdk/server/index.js' with { 'resolution-mode': 'import' };\nimport type { Transport } from '@modelcontextprotocol/sdk/shared/transport.js' with { 'resolution-mode': 'import' };\nimport type {\n  CallToolRequest,\n  CallToolResult,\n  GetPromptRequest,\n  GetPromptResult,\n  ListPromptsRequest,\n  ListPromptsResult,\n  ListToolsRequest,\n  ListToolsResult,\n  Prompt,\n  PromptMessage,\n  Tool,\n} from '@modelcontextprotocol/sdk/types.js' with { 'resolution-mode': 'import' };\nimport { logger } from 'genkit/logging';\nimport { toJsonSchema } from 'genkit/schema';\nimport { ToolAction, toToolDefinition } from 'genkit/tool';\n\n/**\n * Represents an MCP (Model Context Protocol) server that exposes Genkit tools\n * and prompts. This class wraps a Genkit instance and makes its registered\n * actions (tools, prompts) available to MCP clients. It handles the translation\n * between Genkit's action definitions and MCP's expected formats.\n */\nexport class GenkitMcpServer {\n  ai: Genkit;\n  options: McpServerOptions;\n  server?: Server;\n  actionsResolved = false;\n  toolActions: ToolAction[] = [];\n  promptActions: PromptAction[] = [];\n\n  /**\n   * Creates an instance of GenkitMcpServer.\n   * @param ai The Genkit instance whose actions will be exposed.\n   * @param options Configuration options for the MCP server, like its name and version.\n   */\n  constructor(ai: Genkit, options: McpServerOptions) {\n    this.ai = ai;\n    this.options = options;\n  }\n\n  /**\n   * Initializes the MCP server instance and registers request handlers. It\n   * dynamically imports MCP SDK components and sets up handlers for listing\n   * tools, calling tools, listing prompts, and getting prompts. It also\n   * resolves and stores all tool and prompt actions from the Genkit instance.\n   *\n   * This method is called by the constructor and ensures the server is ready\n   * before any requests are handled. It's idempotent.\n   */\n  async setup(): Promise<void> {\n    if (this.actionsResolved) return;\n    const { Server } = await import(\n      '@modelcontextprotocol/sdk/server/index.js'\n    );\n\n    this.server = new Server(\n      { name: this.options.name, version: this.options.version || '1.0.0' },\n      {\n        capabilities: {\n          prompts: {},\n          tools: {},\n        },\n      }\n    );\n\n    const {\n      CallToolRequestSchema,\n      GetPromptRequestSchema,\n      ListPromptsRequestSchema,\n      ListToolsRequestSchema,\n    } = await import('@modelcontextprotocol/sdk/types.js');\n\n    this.server.setRequestHandler(\n      ListToolsRequestSchema,\n      this.listTools.bind(this)\n    );\n    this.server.setRequestHandler(\n      CallToolRequestSchema,\n      this.callTool.bind(this)\n    );\n    this.server.setRequestHandler(\n      ListPromptsRequestSchema,\n      this.listPrompts.bind(this)\n    );\n    this.server.setRequestHandler(\n      GetPromptRequestSchema,\n      this.getPrompt.bind(this)\n    );\n\n    // TODO -- use listResolvableActions.\n    const allActions = await this.ai.registry.listActions();\n    const toolList: ToolAction[] = [];\n    const promptList: PromptAction[] = [];\n    for (const k in allActions) {\n      if (k.startsWith('/tool/')) {\n        toolList.push(allActions[k] as ToolAction);\n      } else if (k.startsWith('/prompt/')) {\n        promptList.push(allActions[k] as PromptAction);\n      }\n    }\n    this.toolActions = toolList;\n    this.promptActions = promptList;\n    this.actionsResolved = true;\n  }\n\n  /**\n   * Handles MCP requests to list available tools.\n   * It maps the resolved Genkit tool actions to the MCP Tool format.\n   * @param req The MCP ListToolsRequest.\n   * @returns A Promise resolving to an MCP ListToolsResult.\n   */\n  async listTools(req: ListToolsRequest): Promise<ListToolsResult> {\n    await this.setup();\n    return {\n      tools: this.toolActions.map((t): Tool => {\n        const def = toToolDefinition(t);\n        return {\n          name: def.name,\n          inputSchema: (def.inputSchema as any) || { type: 'object' },\n          description: def.description,\n          _meta: t.__action.metadata?.mcp?._meta,\n        };\n      }),\n    };\n  }\n\n  /**\n   * Handles MCP requests to call a specific tool. It finds the corresponding\n   * Genkit tool action and executes it with the provided arguments. The result\n   * is then formatted as an MCP CallToolResult.\n   * @param req The MCP CallToolRequest containing the tool name and arguments.\n   * @returns A Promise resolving to an MCP CallToolResult.\n   * @throws GenkitError if the requested tool is not found.\n   */\n  async callTool(req: CallToolRequest): Promise<CallToolResult> {\n    await this.setup();\n    const tool = this.toolActions.find(\n      (t) => t.__action.name === req.params.name\n    );\n    if (!tool)\n      throw new GenkitError({\n        status: 'NOT_FOUND',\n        message: `Tried to call tool '${req.params.name}' but it could not be found.`,\n      });\n    const result = await tool(req.params.arguments);\n    return {\n      content: [\n        {\n          type: 'text',\n          text: typeof result === 'string' ? result : JSON.stringify(result),\n        },\n      ],\n    };\n  }\n\n  /**\n   * Handles MCP requests to list available prompts.\n   * It maps the resolved Genkit prompt actions to the MCP Prompt format,\n   * including converting input schemas to MCP argument definitions.\n   * @param req The MCP ListPromptsRequest.\n   * @returns A Promise resolving to an MCP ListPromptsResult.\n   */\n  async listPrompts(req: ListPromptsRequest): Promise<ListPromptsResult> {\n    await this.setup();\n    return {\n      prompts: this.promptActions.map((p): Prompt => {\n        return {\n          name: p.__action.name,\n          description: p.__action.description,\n          arguments: toMcpPromptArguments(p),\n          _meta: p.__action.metadata?.mcp?._meta,\n        };\n      }),\n    };\n  }\n\n  /**\n   * Handles MCP requests to get (render) a specific prompt. It finds the\n   * corresponding Genkit prompt action, executes it with the provided\n   * arguments, and then formats the resulting messages into the MCP\n   * PromptMessage format.\n   * @param req The MCP GetPromptRequest containing the prompt name and\n   * arguments.\n   * @returns A Promise resolving to an MCP GetPromptResult.\n   * @throws GenkitError if the requested prompt is not found.\n   */\n  async getPrompt(req: GetPromptRequest): Promise<GetPromptResult> {\n    await this.setup();\n    const prompt = this.promptActions.find(\n      (p) => p.__action.name === req.params.name\n    );\n    if (!prompt)\n      throw new GenkitError({\n        status: 'NOT_FOUND',\n        message: `[MCP Server] Tried to call prompt '${req.params.name}' but it could not be found.`,\n      });\n    const result = await prompt(req.params.arguments);\n    return {\n      description: prompt.__action.description,\n      messages: result.messages.map(toMcpPromptMessage),\n    };\n  }\n\n  /**\n   * Starts the MCP server with the specified transport or a default\n   * StdioServerTransport. Ensures the server is set up before connecting the\n   * transport.\n   * @param transport Optional MCP transport instance. If not provided, a\n   * StdioServerTransport will be created and used.\n   */\n  async start(transport?: Transport) {\n    if (!transport) {\n      const { StdioServerTransport } = await import(\n        '@modelcontextprotocol/sdk/server/stdio.js'\n      );\n      transport = new StdioServerTransport();\n    }\n    await this.setup();\n    await this.server!.connect(transport);\n    logger.info(\n      `[MCP Server] MCP server '${this.options.name}' started successfully.`\n    );\n  }\n}\n\n/**\n * Converts a Genkit prompt's input schema to an array of MCP prompt arguments.\n * MCP prompts currently only support string arguments.\n * @param p The Genkit PromptAction.\n * @returns An array of MCP prompt arguments, or undefined if the schema is not defined.\n * @throws GenkitError if the input schema is not an object or if any property is not a string.\n */\nfunction toMcpPromptArguments(\n  p: PromptAction\n): Prompt['arguments'] | undefined {\n  const jsonSchema = toJsonSchema({\n    schema: p.__action.inputSchema,\n    jsonSchema: p.__action.inputJsonSchema,\n  });\n\n  if (!jsonSchema) return undefined;\n  if (!jsonSchema.properties)\n    throw new GenkitError({\n      status: 'FAILED_PRECONDITION',\n      message: '[MCP Server] MCP prompts must take objects as input schema.',\n    });\n\n  const args: Prompt['arguments'] = [];\n  for (const k in jsonSchema.properties) {\n    const { type, description } = jsonSchema.properties[k];\n    if (\n      type !== 'string' &&\n      (!Array.isArray(type) || !type.includes('string'))\n    ) {\n      throw new GenkitError({\n        status: 'FAILED_PRECONDITION',\n        message: `[MCP Server] MCP prompts may only take string arguments, but ${p.__action.name} has property '${k}' of type '${type}'.`,\n      });\n    }\n    args.push({\n      name: k,\n      description,\n      required: jsonSchema.required?.includes(k),\n    });\n  }\n  return args;\n}\n\nconst ROLE_MAP = { model: 'assistant', user: 'user' } as const;\n\n/**\n * Converts a Genkit MessageData object to an MCP PromptMessage.\n * Handles mapping of roles and content types (text, image).\n * MCP only supports 'user' and 'assistant' (model) roles and base64 data images.\n * @param messageData The Genkit MessageData object.\n * @returns An MCP PromptMessage.\n * @throws GenkitError if the role is unsupported or if media is not a data URL.\n */\nfunction toMcpPromptMessage(messageData: MessageData): PromptMessage {\n  if (messageData.role !== 'model' && messageData.role !== 'user') {\n    throw new GenkitError({\n      status: 'UNIMPLEMENTED',\n      message: `[MCP Server] MCP prompt messages do not support role '${messageData.role}'. Only 'user' and 'model' messages are supported.`,\n    });\n  }\n  const message = new Message(messageData);\n  const common = { role: ROLE_MAP[messageData.role] };\n  if (message.media) {\n    const { url, contentType } = message.media;\n    if (!url.startsWith('data:'))\n      throw new GenkitError({\n        status: 'UNIMPLEMENTED',\n        message: `[MCP Server] MCP prompt messages only support base64 data images.`,\n      });\n    const mimeType =\n      contentType || url.substring(url.indexOf(':')! + 1, url.indexOf(';'));\n    const data = url.substring(url.indexOf(',') + 1);\n    return { ...common, content: { type: 'image', mimeType, data } };\n  } else {\n    return { ...common, content: { type: 'text', text: message.text } };\n  }\n}\n"],"mappings":"AAgBA;AAAA,EACE;AAAA,EACA;AAAA,OAIK;AAkBP,SAAS,cAAc;AACvB,SAAS,oBAAoB;AAC7B,SAAqB,wBAAwB;AAQtC,MAAM,gBAAgB;AAAA,EAC3B;AAAA,EACA;AAAA,EACA;AAAA,EACA,kBAAkB;AAAA,EAClB,cAA4B,CAAC;AAAA,EAC7B,gBAAgC,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOjC,YAAY,IAAY,SAA2B;AACjD,SAAK,KAAK;AACV,SAAK,UAAU;AAAA,EACjB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,QAAuB;AAC3B,QAAI,KAAK,gBAAiB;AAC1B,UAAM,EAAE,OAAO,IAAI,MAAM,OACvB,2CACF;AAEA,SAAK,SAAS,IAAI;AAAA,MAChB,EAAE,MAAM,KAAK,QAAQ,MAAM,SAAS,KAAK,QAAQ,WAAW,QAAQ;AAAA,MACpE;AAAA,QACE,cAAc;AAAA,UACZ,SAAS,CAAC;AAAA,UACV,OAAO,CAAC;AAAA,QACV;AAAA,MACF;AAAA,IACF;AAEA,UAAM;AAAA,MACJ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,IAAI,MAAM,OAAO,oCAAoC;AAErD,SAAK,OAAO;AAAA,MACV;AAAA,MACA,KAAK,UAAU,KAAK,IAAI;AAAA,IAC1B;AACA,SAAK,OAAO;AAAA,MACV;AAAA,MACA,KAAK,SAAS,KAAK,IAAI;AAAA,IACzB;AACA,SAAK,OAAO;AAAA,MACV;AAAA,MACA,KAAK,YAAY,KAAK,IAAI;AAAA,IAC5B;AACA,SAAK,OAAO;AAAA,MACV;AAAA,MACA,KAAK,UAAU,KAAK,IAAI;AAAA,IAC1B;AAGA,UAAM,aAAa,MAAM,KAAK,GAAG,SAAS,YAAY;AACtD,UAAM,WAAyB,CAAC;AAChC,UAAM,aAA6B,CAAC;AACpC,eAAW,KAAK,YAAY;AAC1B,UAAI,EAAE,WAAW,QAAQ,GAAG;AAC1B,iBAAS,KAAK,WAAW,CAAC,CAAe;AAAA,MAC3C,WAAW,EAAE,WAAW,UAAU,GAAG;AACnC,mBAAW,KAAK,WAAW,CAAC,CAAiB;AAAA,MAC/C;AAAA,IACF;AACA,SAAK,cAAc;AACnB,SAAK,gBAAgB;AACrB,SAAK,kBAAkB;AAAA,EACzB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,UAAU,KAAiD;AAC/D,UAAM,KAAK,MAAM;AACjB,WAAO;AAAA,MACL,OAAO,KAAK,YAAY,IAAI,CAAC,MAAY;AACvC,cAAM,MAAM,iBAAiB,CAAC;AAC9B,eAAO;AAAA,UACL,MAAM,IAAI;AAAA,UACV,aAAc,IAAI,eAAuB,EAAE,MAAM,SAAS;AAAA,UAC1D,aAAa,IAAI;AAAA,UACjB,OAAO,EAAE,SAAS,UAAU,KAAK;AAAA,QACnC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,SAAS,KAA+C;AAC5D,UAAM,KAAK,MAAM;AACjB,UAAM,OAAO,KAAK,YAAY;AAAA,MAC5B,CAAC,MAAM,EAAE,SAAS,SAAS,IAAI,OAAO;AAAA,IACxC;AACA,QAAI,CAAC;AACH,YAAM,IAAI,YAAY;AAAA,QACpB,QAAQ;AAAA,QACR,SAAS,uBAAuB,IAAI,OAAO,IAAI;AAAA,MACjD,CAAC;AACH,UAAM,SAAS,MAAM,KAAK,IAAI,OAAO,SAAS;AAC9C,WAAO;AAAA,MACL,SAAS;AAAA,QACP;AAAA,UACE,MAAM;AAAA,UACN,MAAM,OAAO,WAAW,WAAW,SAAS,KAAK,UAAU,MAAM;AAAA,QACnE;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAM,YAAY,KAAqD;AACrE,UAAM,KAAK,MAAM;AACjB,WAAO;AAAA,MACL,SAAS,KAAK,cAAc,IAAI,CAAC,MAAc;AAC7C,eAAO;AAAA,UACL,MAAM,EAAE,SAAS;AAAA,UACjB,aAAa,EAAE,SAAS;AAAA,UACxB,WAAW,qBAAqB,CAAC;AAAA,UACjC,OAAO,EAAE,SAAS,UAAU,KAAK;AAAA,QACnC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYA,MAAM,UAAU,KAAiD;AAC/D,UAAM,KAAK,MAAM;AACjB,UAAM,SAAS,KAAK,cAAc;AAAA,MAChC,CAAC,MAAM,EAAE,SAAS,SAAS,IAAI,OAAO;AAAA,IACxC;AACA,QAAI,CAAC;AACH,YAAM,IAAI,YAAY;AAAA,QACpB,QAAQ;AAAA,QACR,SAAS,sCAAsC,IAAI,OAAO,IAAI;AAAA,MAChE,CAAC;AACH,UAAM,SAAS,MAAM,OAAO,IAAI,OAAO,SAAS;AAChD,WAAO;AAAA,MACL,aAAa,OAAO,SAAS;AAAA,MAC7B,UAAU,OAAO,SAAS,IAAI,kBAAkB;AAAA,IAClD;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAM,MAAM,WAAuB;AACjC,QAAI,CAAC,WAAW;AACd,YAAM,EAAE,qBAAqB,IAAI,MAAM,OACrC,2CACF;AACA,kBAAY,IAAI,qBAAqB;AAAA,IACvC;AACA,UAAM,KAAK,MAAM;AACjB,UAAM,KAAK,OAAQ,QAAQ,SAAS;AACpC,WAAO;AAAA,MACL,4BAA4B,KAAK,QAAQ,IAAI;AAAA,IAC/C;AAAA,EACF;AACF;AASA,SAAS,qBACP,GACiC;AACjC,QAAM,aAAa,aAAa;AAAA,IAC9B,QAAQ,EAAE,SAAS;AAAA,IACnB,YAAY,EAAE,SAAS;AAAA,EACzB,CAAC;AAED,MAAI,CAAC,WAAY,QAAO;AACxB,MAAI,CAAC,WAAW;AACd,UAAM,IAAI,YAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SAAS;AAAA,IACX,CAAC;AAEH,QAAM,OAA4B,CAAC;AACnC,aAAW,KAAK,WAAW,YAAY;AACrC,UAAM,EAAE,MAAM,YAAY,IAAI,WAAW,WAAW,CAAC;AACrD,QACE,SAAS,aACR,CAAC,MAAM,QAAQ,IAAI,KAAK,CAAC,KAAK,SAAS,QAAQ,IAChD;AACA,YAAM,IAAI,YAAY;AAAA,QACpB,QAAQ;AAAA,QACR,SAAS,gEAAgE,EAAE,SAAS,IAAI,kBAAkB,CAAC,cAAc,IAAI;AAAA,MAC/H,CAAC;AAAA,IACH;AACA,SAAK,KAAK;AAAA,MACR,MAAM;AAAA,MACN;AAAA,MACA,UAAU,WAAW,UAAU,SAAS,CAAC;AAAA,IAC3C,CAAC;AAAA,EACH;AACA,SAAO;AACT;AAEA,MAAM,WAAW,EAAE,OAAO,aAAa,MAAM,OAAO;AAUpD,SAAS,mBAAmB,aAAyC;AACnE,MAAI,YAAY,SAAS,WAAW,YAAY,SAAS,QAAQ;AAC/D,UAAM,IAAI,YAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SAAS,yDAAyD,YAAY,IAAI;AAAA,IACpF,CAAC;AAAA,EACH;AACA,QAAM,UAAU,IAAI,QAAQ,WAAW;AACvC,QAAM,SAAS,EAAE,MAAM,SAAS,YAAY,IAAI,EAAE;AAClD,MAAI,QAAQ,OAAO;AACjB,UAAM,EAAE,KAAK,YAAY,IAAI,QAAQ;AACrC,QAAI,CAAC,IAAI,WAAW,OAAO;AACzB,YAAM,IAAI,YAAY;AAAA,QACpB,QAAQ;AAAA,QACR,SAAS;AAAA,MACX,CAAC;AACH,UAAM,WACJ,eAAe,IAAI,UAAU,IAAI,QAAQ,GAAG,IAAK,GAAG,IAAI,QAAQ,GAAG,CAAC;AACtE,UAAM,OAAO,IAAI,UAAU,IAAI,QAAQ,GAAG,IAAI,CAAC;AAC/C,WAAO,EAAE,GAAG,QAAQ,SAAS,EAAE,MAAM,SAAS,UAAU,KAAK,EAAE;AAAA,EACjE,OAAO;AACL,WAAO,EAAE,GAAG,QAAQ,SAAS,EAAE,MAAM,QAAQ,MAAM,QAAQ,KAAK,EAAE;AAAA,EACpE;AACF;","names":[]}

package/lib/util/index.d.mts

@@ -0,0 +1,11 @@
+export { fromMcpPart, fromMcpPromptMessage } from './message.mjs';
+export { fetchAllPrompts, getExecutablePrompt, registerAllPrompts } from './prompts.mjs';
+export { fetchDynamicTools, registerAllTools } from './tools.mjs';
+export { transportFrom } from './transport.mjs';
+import '@modelcontextprotocol/sdk/types.js';
+import 'genkit';
+import '@modelcontextprotocol/sdk/client/index.js';
+import '../client/client.mjs';
+import '@modelcontextprotocol/sdk/client/stdio.js';
+import '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import '@modelcontextprotocol/sdk/shared/transport.js';

package/lib/util/index.d.ts

@@ -0,0 +1,11 @@
+export { fromMcpPart, fromMcpPromptMessage } from './message.js';
+export { fetchAllPrompts, getExecutablePrompt, registerAllPrompts } from './prompts.js';
+export { fetchDynamicTools, registerAllTools } from './tools.js';
+export { transportFrom } from './transport.js';
+import '@modelcontextprotocol/sdk/types.js';
+import 'genkit';
+import '@modelcontextprotocol/sdk/client/index.js';
+import '../client/client.js';
+import '@modelcontextprotocol/sdk/client/stdio.js';
+import '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import '@modelcontextprotocol/sdk/shared/transport.js';

package/lib/util/index.js

@@ -0,0 +1,29 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+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 __reExport = (target, mod, secondTarget) => (__copyProps(target, mod, "default"), secondTarget && __copyProps(secondTarget, mod, "default"));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var util_exports = {};
+module.exports = __toCommonJS(util_exports);
+__reExport(util_exports, require("./message.js"), module.exports);
+__reExport(util_exports, require("./prompts.js"), module.exports);
+__reExport(util_exports, require("./tools.js"), module.exports);
+__reExport(util_exports, require("./transport.js"), module.exports);
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ...require("./message.js"),
+  ...require("./prompts.js"),
+  ...require("./tools.js"),
+  ...require("./transport.js")
+});
+//# sourceMappingURL=index.js.map

package/lib/util/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/index.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\nexport * from './message.js';\nexport * from './prompts.js';\nexport * from './tools.js';\nexport * from './transport.js';\n"],"mappings":";;;;;;;;;;;;;;;AAAA;AAAA;AAgBA,yBAAc,yBAhBd;AAiBA,yBAAc,yBAjBd;AAkBA,yBAAc,uBAlBd;AAmBA,yBAAc,2BAnBd;","names":[]}

package/lib/util/index.mjs

@@ -0,0 +1,5 @@
+export * from "./message.js";
+export * from "./prompts.js";
+export * from "./tools.js";
+export * from "./transport.js";
+//# sourceMappingURL=index.mjs.map

package/lib/util/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/index.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\nexport * from './message.js';\nexport * from './prompts.js';\nexport * from './tools.js';\nexport * from './transport.js';\n"],"mappings":"AAgBA,cAAc;AACd,cAAc;AACd,cAAc;AACd,cAAc;","names":[]}

package/lib/util/message.d.mts

@@ -0,0 +1,43 @@
+import { PromptMessage } from '@modelcontextprotocol/sdk/types.js';
+import { MessageData, Part } 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.
+ */
+
+/**
+ * Converts an MCP (Model Context Protocol) PromptMessage into Genkit's
+ * MessageData format. This involves mapping MCP roles (user, assistant) to
+ * Genkit roles (user, model) and transforming the MCP content part into a
+ * Genkit Part.
+ *
+ * @param message The MCP PromptMessage to convert.
+ * @returns The corresponding Genkit MessageData object.
+ */
+declare function fromMcpPromptMessage(message: PromptMessage): MessageData;
+/**
+ * Converts an MCP message content part (text, image, or resource) into a Genkit
+ * Part.
+ * - Text parts are directly mapped.
+ * - Image parts are converted to Genkit media parts with a data URL.
+ * - Resource parts currently result in an empty Genkit Part (further
+ *   implementation may be needed).
+ *
+ * @param part The MCP PromptMessage content part to convert.
+ * @returns The corresponding Genkit Part.
+ */
+declare function fromMcpPart(part: PromptMessage['content']): Part;
+
+export { fromMcpPart, fromMcpPromptMessage };

package/lib/util/message.d.ts

@@ -0,0 +1,43 @@
+import { PromptMessage } from '@modelcontextprotocol/sdk/types.js';
+import { MessageData, Part } 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.
+ */
+
+/**
+ * Converts an MCP (Model Context Protocol) PromptMessage into Genkit's
+ * MessageData format. This involves mapping MCP roles (user, assistant) to
+ * Genkit roles (user, model) and transforming the MCP content part into a
+ * Genkit Part.
+ *
+ * @param message The MCP PromptMessage to convert.
+ * @returns The corresponding Genkit MessageData object.
+ */
+declare function fromMcpPromptMessage(message: PromptMessage): MessageData;
+/**
+ * Converts an MCP message content part (text, image, or resource) into a Genkit
+ * Part.
+ * - Text parts are directly mapped.
+ * - Image parts are converted to Genkit media parts with a data URL.
+ * - Resource parts currently result in an empty Genkit Part (further
+ *   implementation may be needed).
+ *
+ * @param part The MCP PromptMessage content part to convert.
+ * @returns The corresponding Genkit Part.
+ */
+declare function fromMcpPart(part: PromptMessage['content']): Part;
+
+export { fromMcpPart, fromMcpPromptMessage };

package/lib/util/message.js

@@ -0,0 +1,61 @@
+"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 message_exports = {};
+__export(message_exports, {
+  fromMcpPart: () => fromMcpPart,
+  fromMcpPromptMessage: () => fromMcpPromptMessage
+});
+module.exports = __toCommonJS(message_exports);
+const ROLE_MAP = {
+  user: "user",
+  assistant: "model"
+};
+function fromMcpPromptMessage(message) {
+  return {
+    role: ROLE_MAP[message.role],
+    content: [fromMcpPart(message.content)]
+  };
+}
+function fromMcpPart(part) {
+  switch (part.type) {
+    case "text":
+      return { text: part.text };
+    case "image":
+      return {
+        media: {
+          contentType: part.mimeType,
+          url: `data:${part.mimeType};base64,${part.data}`
+        }
+      };
+    case "resource":
+      return {
+        resource: {
+          uri: part.uri
+        }
+      };
+    default:
+      return {};
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  fromMcpPart,
+  fromMcpPromptMessage
+});
+//# sourceMappingURL=message.js.map

package/lib/util/message.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/message.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 type { PromptMessage } from '@modelcontextprotocol/sdk/types.js' with { 'resolution-mode': 'import' };\nimport { type MessageData, type Part } from 'genkit';\n\nconst ROLE_MAP = {\n  user: 'user',\n  assistant: 'model',\n} as const;\n\n/**\n * Converts an MCP (Model Context Protocol) PromptMessage into Genkit's\n * MessageData format. This involves mapping MCP roles (user, assistant) to\n * Genkit roles (user, model) and transforming the MCP content part into a\n * Genkit Part.\n *\n * @param message The MCP PromptMessage to convert.\n * @returns The corresponding Genkit MessageData object.\n */\nexport function fromMcpPromptMessage(message: PromptMessage): MessageData {\n  return {\n    role: ROLE_MAP[message.role],\n    content: [fromMcpPart(message.content)],\n  };\n}\n\n/**\n * Converts an MCP message content part (text, image, or resource) into a Genkit\n * Part.\n * - Text parts are directly mapped.\n * - Image parts are converted to Genkit media parts with a data URL.\n * - Resource parts currently result in an empty Genkit Part (further\n *   implementation may be needed).\n *\n * @param part The MCP PromptMessage content part to convert.\n * @returns The corresponding Genkit Part.\n */\nexport function fromMcpPart(part: PromptMessage['content']): Part {\n  switch (part.type) {\n    case 'text':\n      return { text: part.text };\n    case 'image':\n      return {\n        media: {\n          contentType: part.mimeType,\n          url: `data:${part.mimeType};base64,${part.data}`,\n        },\n      };\n    case 'resource':\n      return {\n        resource: {\n          uri: part.uri as string,\n        },\n      };\n    default:\n      return {};\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAmBA,MAAM,WAAW;AAAA,EACf,MAAM;AAAA,EACN,WAAW;AACb;AAWO,SAAS,qBAAqB,SAAqC;AACxE,SAAO;AAAA,IACL,MAAM,SAAS,QAAQ,IAAI;AAAA,IAC3B,SAAS,CAAC,YAAY,QAAQ,OAAO,CAAC;AAAA,EACxC;AACF;AAaO,SAAS,YAAY,MAAsC;AAChE,UAAQ,KAAK,MAAM;AAAA,IACjB,KAAK;AACH,aAAO,EAAE,MAAM,KAAK,KAAK;AAAA,IAC3B,KAAK;AACH,aAAO;AAAA,QACL,OAAO;AAAA,UACL,aAAa,KAAK;AAAA,UAClB,KAAK,QAAQ,KAAK,QAAQ,WAAW,KAAK,IAAI;AAAA,QAChD;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,UAAU;AAAA,UACR,KAAK,KAAK;AAAA,QACZ;AAAA,MACF;AAAA,IACF;AACE,aAAO,CAAC;AAAA,EACZ;AACF;","names":[]}

package/lib/util/message.mjs

@@ -0,0 +1,36 @@
+const ROLE_MAP = {
+  user: "user",
+  assistant: "model"
+};
+function fromMcpPromptMessage(message) {
+  return {
+    role: ROLE_MAP[message.role],
+    content: [fromMcpPart(message.content)]
+  };
+}
+function fromMcpPart(part) {
+  switch (part.type) {
+    case "text":
+      return { text: part.text };
+    case "image":
+      return {
+        media: {
+          contentType: part.mimeType,
+          url: `data:${part.mimeType};base64,${part.data}`
+        }
+      };
+    case "resource":
+      return {
+        resource: {
+          uri: part.uri
+        }
+      };
+    default:
+      return {};
+  }
+}
+export {
+  fromMcpPart,
+  fromMcpPromptMessage
+};
+//# sourceMappingURL=message.mjs.map

package/lib/util/message.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/message.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 type { PromptMessage } from '@modelcontextprotocol/sdk/types.js' with { 'resolution-mode': 'import' };\nimport { type MessageData, type Part } from 'genkit';\n\nconst ROLE_MAP = {\n  user: 'user',\n  assistant: 'model',\n} as const;\n\n/**\n * Converts an MCP (Model Context Protocol) PromptMessage into Genkit's\n * MessageData format. This involves mapping MCP roles (user, assistant) to\n * Genkit roles (user, model) and transforming the MCP content part into a\n * Genkit Part.\n *\n * @param message The MCP PromptMessage to convert.\n * @returns The corresponding Genkit MessageData object.\n */\nexport function fromMcpPromptMessage(message: PromptMessage): MessageData {\n  return {\n    role: ROLE_MAP[message.role],\n    content: [fromMcpPart(message.content)],\n  };\n}\n\n/**\n * Converts an MCP message content part (text, image, or resource) into a Genkit\n * Part.\n * - Text parts are directly mapped.\n * - Image parts are converted to Genkit media parts with a data URL.\n * - Resource parts currently result in an empty Genkit Part (further\n *   implementation may be needed).\n *\n * @param part The MCP PromptMessage content part to convert.\n * @returns The corresponding Genkit Part.\n */\nexport function fromMcpPart(part: PromptMessage['content']): Part {\n  switch (part.type) {\n    case 'text':\n      return { text: part.text };\n    case 'image':\n      return {\n        media: {\n          contentType: part.mimeType,\n          url: `data:${part.mimeType};base64,${part.data}`,\n        },\n      };\n    case 'resource':\n      return {\n        resource: {\n          uri: part.uri as string,\n        },\n      };\n    default:\n      return {};\n  }\n}\n"],"mappings":"AAmBA,MAAM,WAAW;AAAA,EACf,MAAM;AAAA,EACN,WAAW;AACb;AAWO,SAAS,qBAAqB,SAAqC;AACxE,SAAO;AAAA,IACL,MAAM,SAAS,QAAQ,IAAI;AAAA,IAC3B,SAAS,CAAC,YAAY,QAAQ,OAAO,CAAC;AAAA,EACxC;AACF;AAaO,SAAS,YAAY,MAAsC;AAChE,UAAQ,KAAK,MAAM;AAAA,IACjB,KAAK;AACH,aAAO,EAAE,MAAM,KAAK,KAAK;AAAA,IAC3B,KAAK;AACH,aAAO;AAAA,QACL,OAAO;AAAA,UACL,aAAa,KAAK;AAAA,UAClB,KAAK,QAAQ,KAAK,QAAQ,WAAW,KAAK,IAAI;AAAA,QAChD;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,UAAU;AAAA,UACR,KAAK,KAAK;AAAA,QACZ;AAAA,MACF;AAAA,IACF;AACE,aAAO,CAAC;AAAA,EACZ;AACF;","names":[]}

package/lib/util/prompts.d.mts

@@ -0,0 +1,45 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { Genkit, PromptGenerateOptions, ExecutablePrompt } 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 prompts available in the server and register each as a Genkit
+ * prompt.
+ */
+declare function registerAllPrompts(ai: Genkit, client: Client, params: {
+    name: string;
+    serverName: string;
+}): Promise<void>;
+/**
+ * Lookup a specified prompt from the server and return as an ExecutablePrompt.
+ */
+declare function getExecutablePrompt(client: Client, params: {
+    name: string;
+    serverName: string;
+    promptName: string;
+    ai: Genkit;
+    options?: PromptGenerateOptions;
+}): Promise<ExecutablePrompt | undefined>;
+declare function fetchAllPrompts(client: Client, params: {
+    name: string;
+    serverName: string;
+    ai: Genkit;
+    options?: PromptGenerateOptions;
+}): Promise<ExecutablePrompt[]>;
+
+export { fetchAllPrompts, getExecutablePrompt, registerAllPrompts };