@genkit-ai/mcp 1.14.1-rc.0

package/lib/util/prompts.d.ts

@@ -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 };

package/lib/util/prompts.js

@@ -0,0 +1,147 @@
+"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 prompts_exports = {};
+__export(prompts_exports, {
+  fetchAllPrompts: () => fetchAllPrompts,
+  getExecutablePrompt: () => getExecutablePrompt,
+  registerAllPrompts: () => registerAllPrompts
+});
+module.exports = __toCommonJS(prompts_exports);
+var import_genkit = require("genkit");
+var import_logging = require("genkit/logging");
+var import_message = require("./message.js");
+function toSchema(args) {
+  if (!args) return {};
+  const schema = { type: "object", properties: {}, required: [] };
+  for (const arg of args) {
+    schema.properties[arg.name] = {
+      type: "string",
+      description: arg.description
+    };
+    if (arg.required) schema.required.push(arg.name);
+  }
+  return schema;
+}
+function registerPrompt(ai, client, prompt, params) {
+  import_logging.logger.debug(`[MCP] Registering MCP prompt ${params.name}/${prompt.name}`);
+  ai.definePrompt({
+    name: prompt.name,
+    description: prompt.description || "",
+    input: { jsonSchema: toSchema(prompt.arguments) },
+    output: { format: "text" },
+    metadata: { mcp: { _meta: prompt._meta || {} } },
+    messages: async (args, { context }) => {
+      import_logging.logger.debug(
+        `[MCP] Calling MCP prompt ${params.name}/${prompt.name} with arguments`,
+        JSON.stringify(args)
+      );
+      const result = await client.getPrompt({
+        name: prompt.name,
+        arguments: args,
+        _meta: context?.mcp?._meta
+      });
+      return result.messages.map(import_message.fromMcpPromptMessage);
+    }
+  });
+}
+function createExecutablePrompt(client, prompt, params) {
+  const callPrompt = async (input, opts) => {
+    import_logging.logger.debug(`[MCP] Calling MCP prompt ${params.name}/${prompt.name}`);
+    return params.ai.generate(callPrompt.render(input, opts));
+  };
+  callPrompt.ref = {
+    name: prompt.name,
+    metadata: {
+      description: prompt.description,
+      arguments: prompt.arguments,
+      mcp: { _meta: prompt._meta || {} }
+    }
+  };
+  callPrompt.stream = (input, opts) => {
+    import_logging.logger.debug(`[MCP] Streaming MCP prompt ${params.name}/${prompt.name}`);
+    return params.ai.generateStream(callPrompt.render(input, opts));
+  };
+  callPrompt.render = async (input, opts) => {
+    import_logging.logger.debug(`[MCP] Rendering MCP prompt ${params.name}/${prompt.name}`);
+    const result = await client.getPrompt({
+      name: prompt.name,
+      arguments: input,
+      _meta: opts?.context?.mcp?._meta
+    });
+    const messages = result.messages.map(import_message.fromMcpPromptMessage);
+    return {
+      ...params.options,
+      ...opts,
+      messages
+    };
+  };
+  callPrompt.asTool = async () => {
+    throw new import_genkit.GenkitError({
+      status: "UNIMPLEMENTED",
+      message: `[MCP] prompt.asTool not supported with MCP`
+    });
+  };
+  return callPrompt;
+}
+async function registerAllPrompts(ai, client, params) {
+  let cursor;
+  while (true) {
+    const { nextCursor, prompts } = await client.listPrompts({ cursor });
+    prompts.forEach((p) => registerPrompt(ai, client, p, params));
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+}
+async function getExecutablePrompt(client, params) {
+  let cursor;
+  while (true) {
+    const { nextCursor, prompts } = await client.listPrompts({ cursor });
+    const maybePrompt = prompts.find(
+      (p) => p.name === params.promptName
+    );
+    if (maybePrompt) {
+      return createExecutablePrompt(client, maybePrompt, params);
+    }
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+  return void 0;
+}
+async function fetchAllPrompts(client, params) {
+  let cursor;
+  const out = [];
+  while (true) {
+    const { nextCursor, prompts } = await client.listPrompts({ cursor });
+    for (const p of prompts) {
+      out.push(
+        createExecutablePrompt(client, p, { ...params, promptName: p.name })
+      );
+    }
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+  return out;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  fetchAllPrompts,
+  getExecutablePrompt,
+  registerAllPrompts
+});
+//# sourceMappingURL=prompts.js.map

package/lib/util/prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/prompts.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 { Prompt } from '@modelcontextprotocol/sdk/types.js';\nimport {\n  ExecutablePrompt,\n  GenerateOptions,\n  GenerateResponse,\n  GenerateStreamResponse,\n  Genkit,\n  GenkitError,\n  JSONSchema,\n  ToolAction,\n  z,\n  type PromptGenerateOptions,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { fromMcpPromptMessage } from './message.js';\n\nfunction toSchema(args: Prompt['arguments']) {\n  if (!args) return {};\n  const schema: JSONSchema = { type: 'object', properties: {}, required: [] };\n  for (const arg of args) {\n    schema.properties[arg.name] = {\n      type: 'string',\n      description: arg.description,\n    };\n    if (arg.required) schema.required.push(arg.name);\n  }\n  return schema;\n}\n\n/**\n * Registers a single MCP prompt as a Genkit prompt.\n * It defines a new Genkit prompt action that, when called, will\n * interact with the MCP client to fetch and render the corresponding MCP prompt.\n *\n * @param ai The Genkit instance to define the prompt on.\n * @param client The MCP client instance used to interact with the MCP server.\n * @param prompt The MCP Prompt object to register.\n * @param params Contains the Genkit client name and the MCP server name for namespacing and logging.\n */\nfunction registerPrompt(\n  ai: Genkit,\n  client: Client,\n  prompt: Prompt,\n  params: { name: string; serverName: string }\n) {\n  logger.debug(`[MCP] Registering MCP prompt ${params.name}/${prompt.name}`);\n  ai.definePrompt({\n    name: prompt.name,\n    description: prompt.description || '',\n    input: { jsonSchema: toSchema(prompt.arguments) },\n    output: { format: 'text' },\n    metadata: { mcp: { _meta: prompt._meta || {} } },\n    messages: async (args, { context }) => {\n      logger.debug(\n        `[MCP] Calling MCP prompt ${params.name}/${prompt.name} with arguments`,\n        JSON.stringify(args)\n      );\n      const result = await client.getPrompt({\n        name: prompt.name,\n        arguments: args,\n        _meta: context?.mcp?._meta,\n      });\n      return result.messages.map(fromMcpPromptMessage);\n    },\n  });\n}\n\nfunction createExecutablePrompt<\n  I extends z.ZodTypeAny = z.ZodTypeAny,\n  O extends z.ZodTypeAny = z.ZodTypeAny,\n  CustomOptions extends z.ZodTypeAny = z.ZodTypeAny,\n>(\n  client: Client,\n  prompt: Prompt,\n  params: {\n    ai: Genkit;\n    name: string;\n    serverName: string;\n    promptName: string;\n    options?: PromptGenerateOptions<any, any>;\n  }\n): ExecutablePrompt<z.infer<I>, O, CustomOptions> {\n  const callPrompt = (async (\n    input?: z.infer<I>,\n    opts?: PromptGenerateOptions<O, CustomOptions>\n  ): Promise<GenerateResponse<z.infer<O>>> => {\n    logger.debug(`[MCP] Calling MCP prompt ${params.name}/${prompt.name}`);\n    return params.ai.generate(callPrompt.render(input, opts));\n  }) as ExecutablePrompt<z.infer<I>, O, CustomOptions>;\n\n  callPrompt.ref = {\n    name: prompt.name,\n    metadata: {\n      description: prompt.description,\n      arguments: prompt.arguments,\n      mcp: { _meta: prompt._meta || {} },\n    },\n  };\n\n  callPrompt.stream = (\n    input?: z.infer<I>,\n    opts?: PromptGenerateOptions<O, CustomOptions>\n  ): GenerateStreamResponse<z.infer<O>> => {\n    logger.debug(`[MCP] Streaming MCP prompt ${params.name}/${prompt.name}`);\n    return params.ai.generateStream(callPrompt.render(input, opts));\n  };\n\n  callPrompt.render = async (\n    input?: I,\n    opts?: PromptGenerateOptions<O, CustomOptions>\n  ): Promise<GenerateOptions<O, CustomOptions>> => {\n    logger.debug(`[MCP] Rendering MCP prompt ${params.name}/${prompt.name}`);\n    const result = await client.getPrompt({\n      name: prompt.name,\n      arguments: input as any,\n      _meta: opts?.context?.mcp?._meta,\n    });\n    const messages = result.messages.map(fromMcpPromptMessage);\n    return {\n      ...params.options,\n      ...opts,\n      messages,\n    };\n  };\n\n  callPrompt.asTool = async (): Promise<ToolAction> => {\n    throw new GenkitError({\n      status: 'UNIMPLEMENTED',\n      message: `[MCP] prompt.asTool not supported with MCP`,\n    });\n  };\n\n  return callPrompt;\n}\n\n/**\n * Lookup all prompts available in the server and register each as a Genkit\n * prompt.\n */\nexport async function registerAllPrompts(\n  ai: Genkit,\n  client: Client,\n  params: { name: string; serverName: string }\n): Promise<void> {\n  let cursor: string | undefined;\n  while (true) {\n    const { nextCursor, prompts } = await client.listPrompts({ cursor });\n    prompts.forEach((p) => registerPrompt(ai, client, p, params));\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n}\n\n/**\n * Lookup a specified prompt from the server and return as an ExecutablePrompt.\n */\nexport async function getExecutablePrompt(\n  client: Client,\n  params: {\n    name: string;\n    serverName: string;\n    promptName: string;\n    ai: Genkit;\n    options?: PromptGenerateOptions;\n  }\n): Promise<ExecutablePrompt | undefined> {\n  let cursor: string | undefined;\n\n  while (true) {\n    const { nextCursor, prompts } = await client.listPrompts({ cursor });\n    const maybePrompt = prompts.find(\n      (p: Prompt) => p.name === params.promptName\n    );\n    if (maybePrompt) {\n      return createExecutablePrompt(client, maybePrompt, params);\n    }\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n  return undefined;\n}\n\nexport async function fetchAllPrompts(\n  client: Client,\n  params: {\n    name: string;\n    serverName: string;\n    ai: Genkit;\n    options?: PromptGenerateOptions;\n  }\n): Promise<ExecutablePrompt[]> {\n  let cursor: string | undefined;\n  const out: ExecutablePrompt[] = [];\n\n  while (true) {\n    const { nextCursor, prompts } = await client.listPrompts({ cursor });\n    for (const p of prompts) {\n      out.push(\n        createExecutablePrompt(client, p, { ...params, promptName: p.name })\n      );\n    }\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n  return out;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAkBA,oBAWO;AACP,qBAAuB;AACvB,qBAAqC;AAErC,SAAS,SAAS,MAA2B;AAC3C,MAAI,CAAC,KAAM,QAAO,CAAC;AACnB,QAAM,SAAqB,EAAE,MAAM,UAAU,YAAY,CAAC,GAAG,UAAU,CAAC,EAAE;AAC1E,aAAW,OAAO,MAAM;AACtB,WAAO,WAAW,IAAI,IAAI,IAAI;AAAA,MAC5B,MAAM;AAAA,MACN,aAAa,IAAI;AAAA,IACnB;AACA,QAAI,IAAI,SAAU,QAAO,SAAS,KAAK,IAAI,IAAI;AAAA,EACjD;AACA,SAAO;AACT;AAYA,SAAS,eACP,IACA,QACA,QACA,QACA;AACA,wBAAO,MAAM,gCAAgC,OAAO,IAAI,IAAI,OAAO,IAAI,EAAE;AACzE,KAAG,aAAa;AAAA,IACd,MAAM,OAAO;AAAA,IACb,aAAa,OAAO,eAAe;AAAA,IACnC,OAAO,EAAE,YAAY,SAAS,OAAO,SAAS,EAAE;AAAA,IAChD,QAAQ,EAAE,QAAQ,OAAO;AAAA,IACzB,UAAU,EAAE,KAAK,EAAE,OAAO,OAAO,SAAS,CAAC,EAAE,EAAE;AAAA,IAC/C,UAAU,OAAO,MAAM,EAAE,QAAQ,MAAM;AACrC,4BAAO;AAAA,QACL,4BAA4B,OAAO,IAAI,IAAI,OAAO,IAAI;AAAA,QACtD,KAAK,UAAU,IAAI;AAAA,MACrB;AACA,YAAM,SAAS,MAAM,OAAO,UAAU;AAAA,QACpC,MAAM,OAAO;AAAA,QACb,WAAW;AAAA,QACX,OAAO,SAAS,KAAK;AAAA,MACvB,CAAC;AACD,aAAO,OAAO,SAAS,IAAI,mCAAoB;AAAA,IACjD;AAAA,EACF,CAAC;AACH;AAEA,SAAS,uBAKP,QACA,QACA,QAOgD;AAChD,QAAM,aAAc,OAClB,OACA,SAC0C;AAC1C,0BAAO,MAAM,4BAA4B,OAAO,IAAI,IAAI,OAAO,IAAI,EAAE;AACrE,WAAO,OAAO,GAAG,SAAS,WAAW,OAAO,OAAO,IAAI,CAAC;AAAA,EAC1D;AAEA,aAAW,MAAM;AAAA,IACf,MAAM,OAAO;AAAA,IACb,UAAU;AAAA,MACR,aAAa,OAAO;AAAA,MACpB,WAAW,OAAO;AAAA,MAClB,KAAK,EAAE,OAAO,OAAO,SAAS,CAAC,EAAE;AAAA,IACnC;AAAA,EACF;AAEA,aAAW,SAAS,CAClB,OACA,SACuC;AACvC,0BAAO,MAAM,8BAA8B,OAAO,IAAI,IAAI,OAAO,IAAI,EAAE;AACvE,WAAO,OAAO,GAAG,eAAe,WAAW,OAAO,OAAO,IAAI,CAAC;AAAA,EAChE;AAEA,aAAW,SAAS,OAClB,OACA,SAC+C;AAC/C,0BAAO,MAAM,8BAA8B,OAAO,IAAI,IAAI,OAAO,IAAI,EAAE;AACvE,UAAM,SAAS,MAAM,OAAO,UAAU;AAAA,MACpC,MAAM,OAAO;AAAA,MACb,WAAW;AAAA,MACX,OAAO,MAAM,SAAS,KAAK;AAAA,IAC7B,CAAC;AACD,UAAM,WAAW,OAAO,SAAS,IAAI,mCAAoB;AACzD,WAAO;AAAA,MACL,GAAG,OAAO;AAAA,MACV,GAAG;AAAA,MACH;AAAA,IACF;AAAA,EACF;AAEA,aAAW,SAAS,YAAiC;AACnD,UAAM,IAAI,0BAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SAAS;AAAA,IACX,CAAC;AAAA,EACH;AAEA,SAAO;AACT;AAMA,eAAsB,mBACpB,IACA,QACA,QACe;AACf,MAAI;AACJ,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,QAAQ,IAAI,MAAM,OAAO,YAAY,EAAE,OAAO,CAAC;AACnE,YAAQ,QAAQ,CAAC,MAAM,eAAe,IAAI,QAAQ,GAAG,MAAM,CAAC;AAC5D,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACF;AAKA,eAAsB,oBACpB,QACA,QAOuC;AACvC,MAAI;AAEJ,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,QAAQ,IAAI,MAAM,OAAO,YAAY,EAAE,OAAO,CAAC;AACnE,UAAM,cAAc,QAAQ;AAAA,MAC1B,CAAC,MAAc,EAAE,SAAS,OAAO;AAAA,IACnC;AACA,QAAI,aAAa;AACf,aAAO,uBAAuB,QAAQ,aAAa,MAAM;AAAA,IAC3D;AACA,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACA,SAAO;AACT;AAEA,eAAsB,gBACpB,QACA,QAM6B;AAC7B,MAAI;AACJ,QAAM,MAA0B,CAAC;AAEjC,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,QAAQ,IAAI,MAAM,OAAO,YAAY,EAAE,OAAO,CAAC;AACnE,eAAW,KAAK,SAAS;AACvB,UAAI;AAAA,QACF,uBAAuB,QAAQ,GAAG,EAAE,GAAG,QAAQ,YAAY,EAAE,KAAK,CAAC;AAAA,MACrE;AAAA,IACF;AACA,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACA,SAAO;AACT;","names":[]}

package/lib/util/prompts.mjs

@@ -0,0 +1,123 @@
+import {
+  GenkitError
+} from "genkit";
+import { logger } from "genkit/logging";
+import { fromMcpPromptMessage } from "./message.js";
+function toSchema(args) {
+  if (!args) return {};
+  const schema = { type: "object", properties: {}, required: [] };
+  for (const arg of args) {
+    schema.properties[arg.name] = {
+      type: "string",
+      description: arg.description
+    };
+    if (arg.required) schema.required.push(arg.name);
+  }
+  return schema;
+}
+function registerPrompt(ai, client, prompt, params) {
+  logger.debug(`[MCP] Registering MCP prompt ${params.name}/${prompt.name}`);
+  ai.definePrompt({
+    name: prompt.name,
+    description: prompt.description || "",
+    input: { jsonSchema: toSchema(prompt.arguments) },
+    output: { format: "text" },
+    metadata: { mcp: { _meta: prompt._meta || {} } },
+    messages: async (args, { context }) => {
+      logger.debug(
+        `[MCP] Calling MCP prompt ${params.name}/${prompt.name} with arguments`,
+        JSON.stringify(args)
+      );
+      const result = await client.getPrompt({
+        name: prompt.name,
+        arguments: args,
+        _meta: context?.mcp?._meta
+      });
+      return result.messages.map(fromMcpPromptMessage);
+    }
+  });
+}
+function createExecutablePrompt(client, prompt, params) {
+  const callPrompt = async (input, opts) => {
+    logger.debug(`[MCP] Calling MCP prompt ${params.name}/${prompt.name}`);
+    return params.ai.generate(callPrompt.render(input, opts));
+  };
+  callPrompt.ref = {
+    name: prompt.name,
+    metadata: {
+      description: prompt.description,
+      arguments: prompt.arguments,
+      mcp: { _meta: prompt._meta || {} }
+    }
+  };
+  callPrompt.stream = (input, opts) => {
+    logger.debug(`[MCP] Streaming MCP prompt ${params.name}/${prompt.name}`);
+    return params.ai.generateStream(callPrompt.render(input, opts));
+  };
+  callPrompt.render = async (input, opts) => {
+    logger.debug(`[MCP] Rendering MCP prompt ${params.name}/${prompt.name}`);
+    const result = await client.getPrompt({
+      name: prompt.name,
+      arguments: input,
+      _meta: opts?.context?.mcp?._meta
+    });
+    const messages = result.messages.map(fromMcpPromptMessage);
+    return {
+      ...params.options,
+      ...opts,
+      messages
+    };
+  };
+  callPrompt.asTool = async () => {
+    throw new GenkitError({
+      status: "UNIMPLEMENTED",
+      message: `[MCP] prompt.asTool not supported with MCP`
+    });
+  };
+  return callPrompt;
+}
+async function registerAllPrompts(ai, client, params) {
+  let cursor;
+  while (true) {
+    const { nextCursor, prompts } = await client.listPrompts({ cursor });
+    prompts.forEach((p) => registerPrompt(ai, client, p, params));
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+}
+async function getExecutablePrompt(client, params) {
+  let cursor;
+  while (true) {
+    const { nextCursor, prompts } = await client.listPrompts({ cursor });
+    const maybePrompt = prompts.find(
+      (p) => p.name === params.promptName
+    );
+    if (maybePrompt) {
+      return createExecutablePrompt(client, maybePrompt, params);
+    }
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+  return void 0;
+}
+async function fetchAllPrompts(client, params) {
+  let cursor;
+  const out = [];
+  while (true) {
+    const { nextCursor, prompts } = await client.listPrompts({ cursor });
+    for (const p of prompts) {
+      out.push(
+        createExecutablePrompt(client, p, { ...params, promptName: p.name })
+      );
+    }
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+  return out;
+}
+export {
+  fetchAllPrompts,
+  getExecutablePrompt,
+  registerAllPrompts
+};
+//# sourceMappingURL=prompts.mjs.map

package/lib/util/prompts.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/prompts.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 { Prompt } from '@modelcontextprotocol/sdk/types.js';\nimport {\n  ExecutablePrompt,\n  GenerateOptions,\n  GenerateResponse,\n  GenerateStreamResponse,\n  Genkit,\n  GenkitError,\n  JSONSchema,\n  ToolAction,\n  z,\n  type PromptGenerateOptions,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { fromMcpPromptMessage } from './message.js';\n\nfunction toSchema(args: Prompt['arguments']) {\n  if (!args) return {};\n  const schema: JSONSchema = { type: 'object', properties: {}, required: [] };\n  for (const arg of args) {\n    schema.properties[arg.name] = {\n      type: 'string',\n      description: arg.description,\n    };\n    if (arg.required) schema.required.push(arg.name);\n  }\n  return schema;\n}\n\n/**\n * Registers a single MCP prompt as a Genkit prompt.\n * It defines a new Genkit prompt action that, when called, will\n * interact with the MCP client to fetch and render the corresponding MCP prompt.\n *\n * @param ai The Genkit instance to define the prompt on.\n * @param client The MCP client instance used to interact with the MCP server.\n * @param prompt The MCP Prompt object to register.\n * @param params Contains the Genkit client name and the MCP server name for namespacing and logging.\n */\nfunction registerPrompt(\n  ai: Genkit,\n  client: Client,\n  prompt: Prompt,\n  params: { name: string; serverName: string }\n) {\n  logger.debug(`[MCP] Registering MCP prompt ${params.name}/${prompt.name}`);\n  ai.definePrompt({\n    name: prompt.name,\n    description: prompt.description || '',\n    input: { jsonSchema: toSchema(prompt.arguments) },\n    output: { format: 'text' },\n    metadata: { mcp: { _meta: prompt._meta || {} } },\n    messages: async (args, { context }) => {\n      logger.debug(\n        `[MCP] Calling MCP prompt ${params.name}/${prompt.name} with arguments`,\n        JSON.stringify(args)\n      );\n      const result = await client.getPrompt({\n        name: prompt.name,\n        arguments: args,\n        _meta: context?.mcp?._meta,\n      });\n      return result.messages.map(fromMcpPromptMessage);\n    },\n  });\n}\n\nfunction createExecutablePrompt<\n  I extends z.ZodTypeAny = z.ZodTypeAny,\n  O extends z.ZodTypeAny = z.ZodTypeAny,\n  CustomOptions extends z.ZodTypeAny = z.ZodTypeAny,\n>(\n  client: Client,\n  prompt: Prompt,\n  params: {\n    ai: Genkit;\n    name: string;\n    serverName: string;\n    promptName: string;\n    options?: PromptGenerateOptions<any, any>;\n  }\n): ExecutablePrompt<z.infer<I>, O, CustomOptions> {\n  const callPrompt = (async (\n    input?: z.infer<I>,\n    opts?: PromptGenerateOptions<O, CustomOptions>\n  ): Promise<GenerateResponse<z.infer<O>>> => {\n    logger.debug(`[MCP] Calling MCP prompt ${params.name}/${prompt.name}`);\n    return params.ai.generate(callPrompt.render(input, opts));\n  }) as ExecutablePrompt<z.infer<I>, O, CustomOptions>;\n\n  callPrompt.ref = {\n    name: prompt.name,\n    metadata: {\n      description: prompt.description,\n      arguments: prompt.arguments,\n      mcp: { _meta: prompt._meta || {} },\n    },\n  };\n\n  callPrompt.stream = (\n    input?: z.infer<I>,\n    opts?: PromptGenerateOptions<O, CustomOptions>\n  ): GenerateStreamResponse<z.infer<O>> => {\n    logger.debug(`[MCP] Streaming MCP prompt ${params.name}/${prompt.name}`);\n    return params.ai.generateStream(callPrompt.render(input, opts));\n  };\n\n  callPrompt.render = async (\n    input?: I,\n    opts?: PromptGenerateOptions<O, CustomOptions>\n  ): Promise<GenerateOptions<O, CustomOptions>> => {\n    logger.debug(`[MCP] Rendering MCP prompt ${params.name}/${prompt.name}`);\n    const result = await client.getPrompt({\n      name: prompt.name,\n      arguments: input as any,\n      _meta: opts?.context?.mcp?._meta,\n    });\n    const messages = result.messages.map(fromMcpPromptMessage);\n    return {\n      ...params.options,\n      ...opts,\n      messages,\n    };\n  };\n\n  callPrompt.asTool = async (): Promise<ToolAction> => {\n    throw new GenkitError({\n      status: 'UNIMPLEMENTED',\n      message: `[MCP] prompt.asTool not supported with MCP`,\n    });\n  };\n\n  return callPrompt;\n}\n\n/**\n * Lookup all prompts available in the server and register each as a Genkit\n * prompt.\n */\nexport async function registerAllPrompts(\n  ai: Genkit,\n  client: Client,\n  params: { name: string; serverName: string }\n): Promise<void> {\n  let cursor: string | undefined;\n  while (true) {\n    const { nextCursor, prompts } = await client.listPrompts({ cursor });\n    prompts.forEach((p) => registerPrompt(ai, client, p, params));\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n}\n\n/**\n * Lookup a specified prompt from the server and return as an ExecutablePrompt.\n */\nexport async function getExecutablePrompt(\n  client: Client,\n  params: {\n    name: string;\n    serverName: string;\n    promptName: string;\n    ai: Genkit;\n    options?: PromptGenerateOptions;\n  }\n): Promise<ExecutablePrompt | undefined> {\n  let cursor: string | undefined;\n\n  while (true) {\n    const { nextCursor, prompts } = await client.listPrompts({ cursor });\n    const maybePrompt = prompts.find(\n      (p: Prompt) => p.name === params.promptName\n    );\n    if (maybePrompt) {\n      return createExecutablePrompt(client, maybePrompt, params);\n    }\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n  return undefined;\n}\n\nexport async function fetchAllPrompts(\n  client: Client,\n  params: {\n    name: string;\n    serverName: string;\n    ai: Genkit;\n    options?: PromptGenerateOptions;\n  }\n): Promise<ExecutablePrompt[]> {\n  let cursor: string | undefined;\n  const out: ExecutablePrompt[] = [];\n\n  while (true) {\n    const { nextCursor, prompts } = await client.listPrompts({ cursor });\n    for (const p of prompts) {\n      out.push(\n        createExecutablePrompt(client, p, { ...params, promptName: p.name })\n      );\n    }\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n  return out;\n}\n"],"mappings":"AAkBA;AAAA,EAME;AAAA,OAKK;AACP,SAAS,cAAc;AACvB,SAAS,4BAA4B;AAErC,SAAS,SAAS,MAA2B;AAC3C,MAAI,CAAC,KAAM,QAAO,CAAC;AACnB,QAAM,SAAqB,EAAE,MAAM,UAAU,YAAY,CAAC,GAAG,UAAU,CAAC,EAAE;AAC1E,aAAW,OAAO,MAAM;AACtB,WAAO,WAAW,IAAI,IAAI,IAAI;AAAA,MAC5B,MAAM;AAAA,MACN,aAAa,IAAI;AAAA,IACnB;AACA,QAAI,IAAI,SAAU,QAAO,SAAS,KAAK,IAAI,IAAI;AAAA,EACjD;AACA,SAAO;AACT;AAYA,SAAS,eACP,IACA,QACA,QACA,QACA;AACA,SAAO,MAAM,gCAAgC,OAAO,IAAI,IAAI,OAAO,IAAI,EAAE;AACzE,KAAG,aAAa;AAAA,IACd,MAAM,OAAO;AAAA,IACb,aAAa,OAAO,eAAe;AAAA,IACnC,OAAO,EAAE,YAAY,SAAS,OAAO,SAAS,EAAE;AAAA,IAChD,QAAQ,EAAE,QAAQ,OAAO;AAAA,IACzB,UAAU,EAAE,KAAK,EAAE,OAAO,OAAO,SAAS,CAAC,EAAE,EAAE;AAAA,IAC/C,UAAU,OAAO,MAAM,EAAE,QAAQ,MAAM;AACrC,aAAO;AAAA,QACL,4BAA4B,OAAO,IAAI,IAAI,OAAO,IAAI;AAAA,QACtD,KAAK,UAAU,IAAI;AAAA,MACrB;AACA,YAAM,SAAS,MAAM,OAAO,UAAU;AAAA,QACpC,MAAM,OAAO;AAAA,QACb,WAAW;AAAA,QACX,OAAO,SAAS,KAAK;AAAA,MACvB,CAAC;AACD,aAAO,OAAO,SAAS,IAAI,oBAAoB;AAAA,IACjD;AAAA,EACF,CAAC;AACH;AAEA,SAAS,uBAKP,QACA,QACA,QAOgD;AAChD,QAAM,aAAc,OAClB,OACA,SAC0C;AAC1C,WAAO,MAAM,4BAA4B,OAAO,IAAI,IAAI,OAAO,IAAI,EAAE;AACrE,WAAO,OAAO,GAAG,SAAS,WAAW,OAAO,OAAO,IAAI,CAAC;AAAA,EAC1D;AAEA,aAAW,MAAM;AAAA,IACf,MAAM,OAAO;AAAA,IACb,UAAU;AAAA,MACR,aAAa,OAAO;AAAA,MACpB,WAAW,OAAO;AAAA,MAClB,KAAK,EAAE,OAAO,OAAO,SAAS,CAAC,EAAE;AAAA,IACnC;AAAA,EACF;AAEA,aAAW,SAAS,CAClB,OACA,SACuC;AACvC,WAAO,MAAM,8BAA8B,OAAO,IAAI,IAAI,OAAO,IAAI,EAAE;AACvE,WAAO,OAAO,GAAG,eAAe,WAAW,OAAO,OAAO,IAAI,CAAC;AAAA,EAChE;AAEA,aAAW,SAAS,OAClB,OACA,SAC+C;AAC/C,WAAO,MAAM,8BAA8B,OAAO,IAAI,IAAI,OAAO,IAAI,EAAE;AACvE,UAAM,SAAS,MAAM,OAAO,UAAU;AAAA,MACpC,MAAM,OAAO;AAAA,MACb,WAAW;AAAA,MACX,OAAO,MAAM,SAAS,KAAK;AAAA,IAC7B,CAAC;AACD,UAAM,WAAW,OAAO,SAAS,IAAI,oBAAoB;AACzD,WAAO;AAAA,MACL,GAAG,OAAO;AAAA,MACV,GAAG;AAAA,MACH;AAAA,IACF;AAAA,EACF;AAEA,aAAW,SAAS,YAAiC;AACnD,UAAM,IAAI,YAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SAAS;AAAA,IACX,CAAC;AAAA,EACH;AAEA,SAAO;AACT;AAMA,eAAsB,mBACpB,IACA,QACA,QACe;AACf,MAAI;AACJ,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,QAAQ,IAAI,MAAM,OAAO,YAAY,EAAE,OAAO,CAAC;AACnE,YAAQ,QAAQ,CAAC,MAAM,eAAe,IAAI,QAAQ,GAAG,MAAM,CAAC;AAC5D,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACF;AAKA,eAAsB,oBACpB,QACA,QAOuC;AACvC,MAAI;AAEJ,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,QAAQ,IAAI,MAAM,OAAO,YAAY,EAAE,OAAO,CAAC;AACnE,UAAM,cAAc,QAAQ;AAAA,MAC1B,CAAC,MAAc,EAAE,SAAS,OAAO;AAAA,IACnC;AACA,QAAI,aAAa;AACf,aAAO,uBAAuB,QAAQ,aAAa,MAAM;AAAA,IAC3D;AACA,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACA,SAAO;AACT;AAEA,eAAsB,gBACpB,QACA,QAM6B;AAC7B,MAAI;AACJ,QAAM,MAA0B,CAAC;AAEjC,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,QAAQ,IAAI,MAAM,OAAO,YAAY,EAAE,OAAO,CAAC;AACnE,eAAW,KAAK,SAAS;AACvB,UAAI;AAAA,QACF,uBAAuB,QAAQ,GAAG,EAAE,GAAG,QAAQ,YAAY,EAAE,KAAK,CAAC;AAAA,MACrE;AAAA,IACF;AACA,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACA,SAAO;AACT;","names":[]}

package/lib/util/resource.d.mts

@@ -0,0 +1,28 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { Genkit, DynamicResourceAction } 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 resources available in the server and fetches as a Genkit dynamic resource.
+ */
+declare function fetchDynamicResources(ai: Genkit, client: Client, params: {
+    name: string;
+    serverName: string;
+}): Promise<DynamicResourceAction[]>;
+
+export { fetchDynamicResources };

package/lib/util/resource.d.ts

@@ -0,0 +1,28 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { Genkit, DynamicResourceAction } 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 resources available in the server and fetches as a Genkit dynamic resource.
+ */
+declare function fetchDynamicResources(ai: Genkit, client: Client, params: {
+    name: string;
+    serverName: string;
+}): Promise<DynamicResourceAction[]>;
+
+export { fetchDynamicResources };

package/lib/util/resource.js

@@ -0,0 +1,116 @@
+"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 resource_exports = {};
+__export(resource_exports, {
+  fetchDynamicResources: () => fetchDynamicResources
+});
+module.exports = __toCommonJS(resource_exports);
+var import_genkit = require("genkit");
+var import_logging = require("genkit/logging");
+function createDynamicResource(client, resource, params) {
+  return (0, import_genkit.dynamicResource)(
+    {
+      name: `${params.serverName}/${resource.name}`,
+      description: resource.description || void 0,
+      metadata: { mcp: { _meta: resource._meta || {} } },
+      uri: resource.uri
+    },
+    async (args, { context }) => {
+      import_logging.logger.debug(
+        `[MCP] calling resource '${params.serverName}/${resource.name}' in host '${params.name}'`
+      );
+      const result = await client.readResource({
+        uri: args.uri,
+        _meta: context?.mcp?._meta
+      });
+      return {
+        content: result.contents.map((p) => fromMcpResourcePart(p))
+      };
+    }
+  );
+}
+function createDynamicResourceTemplate(client, template, params) {
+  return (0, import_genkit.dynamicResource)(
+    {
+      name: `${params.serverName}/${template.name}`,
+      description: template.description || void 0,
+      metadata: { mcp: { _meta: template._meta || {} } },
+      template: template.uriTemplate
+    },
+    async (args, { context }) => {
+      import_logging.logger.debug(
+        `[MCP] calling resource template '${params.serverName}/${template.name}' in host '${params.name}'`
+      );
+      const result = await client.readResource({
+        uri: args.uri,
+        _meta: context?.mcp?._meta
+      });
+      return {
+        content: result.contents.map((p) => fromMcpResourcePart(p)),
+        metadata: result._meta
+      };
+    }
+  );
+}
+function fromMcpResourcePart(part) {
+  if (part.text) {
+    return { text: part.text, metadata: part._meta };
+  }
+  if (part.blob) {
+    return {
+      media: {
+        contentType: part.mimeType,
+        url: `data:${part.mimeType};base64,${part.blob}`
+      },
+      metadata: part._meta
+    };
+  }
+  throw new import_genkit.GenkitError({
+    status: "UNIMPLEMENTED",
+    message: `Part type ${part.type} is not currently supported.`
+  });
+}
+async function fetchDynamicResources(ai, client, params) {
+  let cursor;
+  let allResources = [];
+  while (true) {
+    const { nextCursor, resources } = await client.listResources({ cursor });
+    allResources.push(
+      ...resources.map((r) => createDynamicResource(client, r, params))
+    );
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+  while (true) {
+    const { nextCursor, resourceTemplates } = await client.listResourceTemplates({ cursor });
+    allResources.push(
+      ...resourceTemplates.map(
+        (r) => createDynamicResourceTemplate(client, r, params)
+      )
+    );
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+  return allResources;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  fetchDynamicResources
+});
+//# sourceMappingURL=resource.js.map

package/lib/util/resource.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/resource.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 {\n  ReadResourceResult,\n  Resource,\n  ResourceTemplate,\n} from '@modelcontextprotocol/sdk/types.js';\nimport {\n  DynamicResourceAction,\n  Genkit,\n  GenkitError,\n  Part,\n  dynamicResource,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\n\nfunction createDynamicResource(\n  client: Client,\n  resource: Resource,\n  params: { serverName: string; name: string }\n): DynamicResourceAction {\n  return dynamicResource(\n    {\n      name: `${params.serverName}/${resource.name}`,\n      description: resource.description || undefined,\n      metadata: { mcp: { _meta: resource._meta || {} } },\n      uri: resource.uri,\n    },\n    async (args, { context }) => {\n      logger.debug(\n        `[MCP] calling resource '${params.serverName}/${resource.name}' in host '${params.name}'`\n      );\n      const result = await client.readResource({\n        uri: args.uri,\n        _meta: context?.mcp?._meta,\n      });\n      return {\n        content: result.contents.map((p) => fromMcpResourcePart(p)),\n      };\n    }\n  );\n}\n\nfunction createDynamicResourceTemplate(\n  client: Client,\n  template: ResourceTemplate,\n  params: { serverName: string; name: string }\n): DynamicResourceAction {\n  return dynamicResource(\n    {\n      name: `${params.serverName}/${template.name}`,\n      description: template.description || undefined,\n      metadata: { mcp: { _meta: template._meta || {} } },\n      template: template.uriTemplate,\n    },\n    async (args, { context }) => {\n      logger.debug(\n        `[MCP] calling resource template '${params.serverName}/${template.name}' in host '${params.name}'`\n      );\n      const result = await client.readResource({\n        uri: args.uri,\n        _meta: context?.mcp?._meta,\n      });\n      return {\n        content: result.contents.map((p) => fromMcpResourcePart(p)),\n        metadata: result._meta,\n      };\n    }\n  );\n}\n\ntype ArrayElement<ArrayType extends readonly unknown[]> =\n  ArrayType extends readonly (infer ElementType)[] ? ElementType : never;\n\nfunction fromMcpResourcePart(\n  part: ArrayElement<ReadResourceResult['contents']>\n): Part {\n  if (part.text) {\n    return { text: part.text as string, metadata: part._meta };\n  }\n  if (part.blob) {\n    return {\n      media: {\n        contentType: part.mimeType,\n        url: `data:${part.mimeType};base64,${part.blob}`,\n      },\n      metadata: part._meta,\n    };\n  }\n  throw new GenkitError({\n    status: 'UNIMPLEMENTED',\n    message: `Part type ${part.type} is not currently supported.`,\n  });\n}\n\n/**\n * Lookup all resources available in the server and fetches as a Genkit dynamic resource.\n */\nexport async function fetchDynamicResources(\n  ai: Genkit,\n  client: Client,\n  params: { name: string; serverName: string }\n): Promise<DynamicResourceAction[]> {\n  let cursor: string | undefined;\n  let allResources: DynamicResourceAction[] = [];\n  while (true) {\n    const { nextCursor, resources } = await client.listResources({ cursor });\n    allResources.push(\n      ...resources.map((r) => createDynamicResource(client, r, params))\n    );\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n  while (true) {\n    const { nextCursor, resourceTemplates } =\n      await client.listResourceTemplates({ cursor });\n    allResources.push(\n      ...resourceTemplates.map((r) =>\n        createDynamicResourceTemplate(client, r, params)\n      )\n    );\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n  return allResources;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAsBA,oBAMO;AACP,qBAAuB;AAEvB,SAAS,sBACP,QACA,UACA,QACuB;AACvB,aAAO;AAAA,IACL;AAAA,MACE,MAAM,GAAG,OAAO,UAAU,IAAI,SAAS,IAAI;AAAA,MAC3C,aAAa,SAAS,eAAe;AAAA,MACrC,UAAU,EAAE,KAAK,EAAE,OAAO,SAAS,SAAS,CAAC,EAAE,EAAE;AAAA,MACjD,KAAK,SAAS;AAAA,IAChB;AAAA,IACA,OAAO,MAAM,EAAE,QAAQ,MAAM;AAC3B,4BAAO;AAAA,QACL,2BAA2B,OAAO,UAAU,IAAI,SAAS,IAAI,cAAc,OAAO,IAAI;AAAA,MACxF;AACA,YAAM,SAAS,MAAM,OAAO,aAAa;AAAA,QACvC,KAAK,KAAK;AAAA,QACV,OAAO,SAAS,KAAK;AAAA,MACvB,CAAC;AACD,aAAO;AAAA,QACL,SAAS,OAAO,SAAS,IAAI,CAAC,MAAM,oBAAoB,CAAC,CAAC;AAAA,MAC5D;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,8BACP,QACA,UACA,QACuB;AACvB,aAAO;AAAA,IACL;AAAA,MACE,MAAM,GAAG,OAAO,UAAU,IAAI,SAAS,IAAI;AAAA,MAC3C,aAAa,SAAS,eAAe;AAAA,MACrC,UAAU,EAAE,KAAK,EAAE,OAAO,SAAS,SAAS,CAAC,EAAE,EAAE;AAAA,MACjD,UAAU,SAAS;AAAA,IACrB;AAAA,IACA,OAAO,MAAM,EAAE,QAAQ,MAAM;AAC3B,4BAAO;AAAA,QACL,oCAAoC,OAAO,UAAU,IAAI,SAAS,IAAI,cAAc,OAAO,IAAI;AAAA,MACjG;AACA,YAAM,SAAS,MAAM,OAAO,aAAa;AAAA,QACvC,KAAK,KAAK;AAAA,QACV,OAAO,SAAS,KAAK;AAAA,MACvB,CAAC;AACD,aAAO;AAAA,QACL,SAAS,OAAO,SAAS,IAAI,CAAC,MAAM,oBAAoB,CAAC,CAAC;AAAA,QAC1D,UAAU,OAAO;AAAA,MACnB;AAAA,IACF;AAAA,EACF;AACF;AAKA,SAAS,oBACP,MACM;AACN,MAAI,KAAK,MAAM;AACb,WAAO,EAAE,MAAM,KAAK,MAAgB,UAAU,KAAK,MAAM;AAAA,EAC3D;AACA,MAAI,KAAK,MAAM;AACb,WAAO;AAAA,MACL,OAAO;AAAA,QACL,aAAa,KAAK;AAAA,QAClB,KAAK,QAAQ,KAAK,QAAQ,WAAW,KAAK,IAAI;AAAA,MAChD;AAAA,MACA,UAAU,KAAK;AAAA,IACjB;AAAA,EACF;AACA,QAAM,IAAI,0BAAY;AAAA,IACpB,QAAQ;AAAA,IACR,SAAS,aAAa,KAAK,IAAI;AAAA,EACjC,CAAC;AACH;AAKA,eAAsB,sBACpB,IACA,QACA,QACkC;AAClC,MAAI;AACJ,MAAI,eAAwC,CAAC;AAC7C,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,UAAU,IAAI,MAAM,OAAO,cAAc,EAAE,OAAO,CAAC;AACvE,iBAAa;AAAA,MACX,GAAG,UAAU,IAAI,CAAC,MAAM,sBAAsB,QAAQ,GAAG,MAAM,CAAC;AAAA,IAClE;AACA,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACA,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,kBAAkB,IACpC,MAAM,OAAO,sBAAsB,EAAE,OAAO,CAAC;AAC/C,iBAAa;AAAA,MACX,GAAG,kBAAkB;AAAA,QAAI,CAAC,MACxB,8BAA8B,QAAQ,GAAG,MAAM;AAAA,MACjD;AAAA,IACF;AACA,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACA,SAAO;AACT;","names":[]}

package/lib/util/resource.mjs

@@ -0,0 +1,95 @@
+import {
+  GenkitError,
+  dynamicResource
+} from "genkit";
+import { logger } from "genkit/logging";
+function createDynamicResource(client, resource, params) {
+  return dynamicResource(
+    {
+      name: `${params.serverName}/${resource.name}`,
+      description: resource.description || void 0,
+      metadata: { mcp: { _meta: resource._meta || {} } },
+      uri: resource.uri
+    },
+    async (args, { context }) => {
+      logger.debug(
+        `[MCP] calling resource '${params.serverName}/${resource.name}' in host '${params.name}'`
+      );
+      const result = await client.readResource({
+        uri: args.uri,
+        _meta: context?.mcp?._meta
+      });
+      return {
+        content: result.contents.map((p) => fromMcpResourcePart(p))
+      };
+    }
+  );
+}
+function createDynamicResourceTemplate(client, template, params) {
+  return dynamicResource(
+    {
+      name: `${params.serverName}/${template.name}`,
+      description: template.description || void 0,
+      metadata: { mcp: { _meta: template._meta || {} } },
+      template: template.uriTemplate
+    },
+    async (args, { context }) => {
+      logger.debug(
+        `[MCP] calling resource template '${params.serverName}/${template.name}' in host '${params.name}'`
+      );
+      const result = await client.readResource({
+        uri: args.uri,
+        _meta: context?.mcp?._meta
+      });
+      return {
+        content: result.contents.map((p) => fromMcpResourcePart(p)),
+        metadata: result._meta
+      };
+    }
+  );
+}
+function fromMcpResourcePart(part) {
+  if (part.text) {
+    return { text: part.text, metadata: part._meta };
+  }
+  if (part.blob) {
+    return {
+      media: {
+        contentType: part.mimeType,
+        url: `data:${part.mimeType};base64,${part.blob}`
+      },
+      metadata: part._meta
+    };
+  }
+  throw new GenkitError({
+    status: "UNIMPLEMENTED",
+    message: `Part type ${part.type} is not currently supported.`
+  });
+}
+async function fetchDynamicResources(ai, client, params) {
+  let cursor;
+  let allResources = [];
+  while (true) {
+    const { nextCursor, resources } = await client.listResources({ cursor });
+    allResources.push(
+      ...resources.map((r) => createDynamicResource(client, r, params))
+    );
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+  while (true) {
+    const { nextCursor, resourceTemplates } = await client.listResourceTemplates({ cursor });
+    allResources.push(
+      ...resourceTemplates.map(
+        (r) => createDynamicResourceTemplate(client, r, params)
+      )
+    );
+    cursor = nextCursor;
+    if (!cursor) break;
+  }
+  return allResources;
+}
+export {
+  fetchDynamicResources
+};
+//# sourceMappingURL=resource.mjs.map

package/lib/util/resource.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/util/resource.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 {\n  ReadResourceResult,\n  Resource,\n  ResourceTemplate,\n} from '@modelcontextprotocol/sdk/types.js';\nimport {\n  DynamicResourceAction,\n  Genkit,\n  GenkitError,\n  Part,\n  dynamicResource,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\n\nfunction createDynamicResource(\n  client: Client,\n  resource: Resource,\n  params: { serverName: string; name: string }\n): DynamicResourceAction {\n  return dynamicResource(\n    {\n      name: `${params.serverName}/${resource.name}`,\n      description: resource.description || undefined,\n      metadata: { mcp: { _meta: resource._meta || {} } },\n      uri: resource.uri,\n    },\n    async (args, { context }) => {\n      logger.debug(\n        `[MCP] calling resource '${params.serverName}/${resource.name}' in host '${params.name}'`\n      );\n      const result = await client.readResource({\n        uri: args.uri,\n        _meta: context?.mcp?._meta,\n      });\n      return {\n        content: result.contents.map((p) => fromMcpResourcePart(p)),\n      };\n    }\n  );\n}\n\nfunction createDynamicResourceTemplate(\n  client: Client,\n  template: ResourceTemplate,\n  params: { serverName: string; name: string }\n): DynamicResourceAction {\n  return dynamicResource(\n    {\n      name: `${params.serverName}/${template.name}`,\n      description: template.description || undefined,\n      metadata: { mcp: { _meta: template._meta || {} } },\n      template: template.uriTemplate,\n    },\n    async (args, { context }) => {\n      logger.debug(\n        `[MCP] calling resource template '${params.serverName}/${template.name}' in host '${params.name}'`\n      );\n      const result = await client.readResource({\n        uri: args.uri,\n        _meta: context?.mcp?._meta,\n      });\n      return {\n        content: result.contents.map((p) => fromMcpResourcePart(p)),\n        metadata: result._meta,\n      };\n    }\n  );\n}\n\ntype ArrayElement<ArrayType extends readonly unknown[]> =\n  ArrayType extends readonly (infer ElementType)[] ? ElementType : never;\n\nfunction fromMcpResourcePart(\n  part: ArrayElement<ReadResourceResult['contents']>\n): Part {\n  if (part.text) {\n    return { text: part.text as string, metadata: part._meta };\n  }\n  if (part.blob) {\n    return {\n      media: {\n        contentType: part.mimeType,\n        url: `data:${part.mimeType};base64,${part.blob}`,\n      },\n      metadata: part._meta,\n    };\n  }\n  throw new GenkitError({\n    status: 'UNIMPLEMENTED',\n    message: `Part type ${part.type} is not currently supported.`,\n  });\n}\n\n/**\n * Lookup all resources available in the server and fetches as a Genkit dynamic resource.\n */\nexport async function fetchDynamicResources(\n  ai: Genkit,\n  client: Client,\n  params: { name: string; serverName: string }\n): Promise<DynamicResourceAction[]> {\n  let cursor: string | undefined;\n  let allResources: DynamicResourceAction[] = [];\n  while (true) {\n    const { nextCursor, resources } = await client.listResources({ cursor });\n    allResources.push(\n      ...resources.map((r) => createDynamicResource(client, r, params))\n    );\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n  while (true) {\n    const { nextCursor, resourceTemplates } =\n      await client.listResourceTemplates({ cursor });\n    allResources.push(\n      ...resourceTemplates.map((r) =>\n        createDynamicResourceTemplate(client, r, params)\n      )\n    );\n    cursor = nextCursor;\n    if (!cursor) break;\n  }\n  return allResources;\n}\n"],"mappings":"AAsBA;AAAA,EAGE;AAAA,EAEA;AAAA,OACK;AACP,SAAS,cAAc;AAEvB,SAAS,sBACP,QACA,UACA,QACuB;AACvB,SAAO;AAAA,IACL;AAAA,MACE,MAAM,GAAG,OAAO,UAAU,IAAI,SAAS,IAAI;AAAA,MAC3C,aAAa,SAAS,eAAe;AAAA,MACrC,UAAU,EAAE,KAAK,EAAE,OAAO,SAAS,SAAS,CAAC,EAAE,EAAE;AAAA,MACjD,KAAK,SAAS;AAAA,IAChB;AAAA,IACA,OAAO,MAAM,EAAE,QAAQ,MAAM;AAC3B,aAAO;AAAA,QACL,2BAA2B,OAAO,UAAU,IAAI,SAAS,IAAI,cAAc,OAAO,IAAI;AAAA,MACxF;AACA,YAAM,SAAS,MAAM,OAAO,aAAa;AAAA,QACvC,KAAK,KAAK;AAAA,QACV,OAAO,SAAS,KAAK;AAAA,MACvB,CAAC;AACD,aAAO;AAAA,QACL,SAAS,OAAO,SAAS,IAAI,CAAC,MAAM,oBAAoB,CAAC,CAAC;AAAA,MAC5D;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,8BACP,QACA,UACA,QACuB;AACvB,SAAO;AAAA,IACL;AAAA,MACE,MAAM,GAAG,OAAO,UAAU,IAAI,SAAS,IAAI;AAAA,MAC3C,aAAa,SAAS,eAAe;AAAA,MACrC,UAAU,EAAE,KAAK,EAAE,OAAO,SAAS,SAAS,CAAC,EAAE,EAAE;AAAA,MACjD,UAAU,SAAS;AAAA,IACrB;AAAA,IACA,OAAO,MAAM,EAAE,QAAQ,MAAM;AAC3B,aAAO;AAAA,QACL,oCAAoC,OAAO,UAAU,IAAI,SAAS,IAAI,cAAc,OAAO,IAAI;AAAA,MACjG;AACA,YAAM,SAAS,MAAM,OAAO,aAAa;AAAA,QACvC,KAAK,KAAK;AAAA,QACV,OAAO,SAAS,KAAK;AAAA,MACvB,CAAC;AACD,aAAO;AAAA,QACL,SAAS,OAAO,SAAS,IAAI,CAAC,MAAM,oBAAoB,CAAC,CAAC;AAAA,QAC1D,UAAU,OAAO;AAAA,MACnB;AAAA,IACF;AAAA,EACF;AACF;AAKA,SAAS,oBACP,MACM;AACN,MAAI,KAAK,MAAM;AACb,WAAO,EAAE,MAAM,KAAK,MAAgB,UAAU,KAAK,MAAM;AAAA,EAC3D;AACA,MAAI,KAAK,MAAM;AACb,WAAO;AAAA,MACL,OAAO;AAAA,QACL,aAAa,KAAK;AAAA,QAClB,KAAK,QAAQ,KAAK,QAAQ,WAAW,KAAK,IAAI;AAAA,MAChD;AAAA,MACA,UAAU,KAAK;AAAA,IACjB;AAAA,EACF;AACA,QAAM,IAAI,YAAY;AAAA,IACpB,QAAQ;AAAA,IACR,SAAS,aAAa,KAAK,IAAI;AAAA,EACjC,CAAC;AACH;AAKA,eAAsB,sBACpB,IACA,QACA,QACkC;AAClC,MAAI;AACJ,MAAI,eAAwC,CAAC;AAC7C,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,UAAU,IAAI,MAAM,OAAO,cAAc,EAAE,OAAO,CAAC;AACvE,iBAAa;AAAA,MACX,GAAG,UAAU,IAAI,CAAC,MAAM,sBAAsB,QAAQ,GAAG,MAAM,CAAC;AAAA,IAClE;AACA,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACA,SAAO,MAAM;AACX,UAAM,EAAE,YAAY,kBAAkB,IACpC,MAAM,OAAO,sBAAsB,EAAE,OAAO,CAAC;AAC/C,iBAAa;AAAA,MACX,GAAG,kBAAkB;AAAA,QAAI,CAAC,MACxB,8BAA8B,QAAQ,GAAG,MAAM;AAAA,MACjD;AAAA,IACF;AACA,aAAS;AACT,QAAI,CAAC,OAAQ;AAAA,EACf;AACA,SAAO;AACT;","names":[]}