agents 0.0.0-dfa677f → 0.0.0-e376805

package/dist/mcp/client.js

@@ -0,0 +1,268 @@
+import "../chunk-HMLY7DHA.js";
+
+// src/mcp/sse-edge.ts
+import {
+  SSEClientTransport
+} from "@modelcontextprotocol/sdk/client/sse.js";
+var SSEEdgeClientTransport = class extends SSEClientTransport {
+  /**
+   * Creates a new EdgeSSEClientTransport, which overrides fetch to be compatible with the CF workers environment
+   */
+  constructor(url, options) {
+    const fetchOverride = (url2, options2 = {}) => {
+      const workerOptions = {
+        ...options2
+      };
+      delete workerOptions.mode;
+      return global.fetch(url2, workerOptions);
+    };
+    super(url, {
+      ...options,
+      eventSourceInit: {
+        fetch: fetchOverride
+      }
+    });
+    this.url = url;
+  }
+};
+
+// src/mcp/client-connection.ts
+import {
+  ToolListChangedNotificationSchema,
+  ResourceListChangedNotificationSchema,
+  PromptListChangedNotificationSchema
+} from "@modelcontextprotocol/sdk/types.js";
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+var MCPClientConnection = class {
+  constructor(url, info, opts = { transport: {}, client: {}, capabilities: {} }) {
+    this.info = info;
+    this.transport = new SSEEdgeClientTransport(url, opts.transport);
+    this.client = new Client(info, opts.client);
+    this.client.registerCapabilities(opts.capabilities);
+    this.connected = false;
+    this.tools = [];
+    this.prompts = [];
+    this.resources = [];
+    this.resourceTemplates = [];
+  }
+  async init() {
+    await this.client.connect(this.transport);
+    this.serverCapabilities = await this.client.getServerCapabilities();
+    if (!this.serverCapabilities) {
+      throw new Error(
+        `The MCP Server ${this.info.name} failed to return server capabilities`
+      );
+    }
+    const [instructions, tools, resources, prompts, resourceTemplates] = await Promise.all([
+      this.client.getInstructions(),
+      this.registerTools(),
+      this.registerResources(),
+      this.registerPrompts(),
+      this.registerResourceTemplates()
+    ]);
+    this.instructions = instructions;
+    this.tools = tools;
+    this.resources = resources;
+    this.prompts = prompts;
+    this.resourceTemplates = resourceTemplates;
+  }
+  /**
+   * Notification handler registration
+   */
+  async registerTools() {
+    if (!this.serverCapabilities || !this.serverCapabilities.tools) {
+      return [];
+    }
+    if (this.serverCapabilities.tools.listChanged) {
+      this.client.setNotificationHandler(
+        ToolListChangedNotificationSchema,
+        async (_notification) => {
+          this.tools = await this.fetchTools();
+        }
+      );
+    }
+    return this.fetchTools();
+  }
+  async registerResources() {
+    if (!this.serverCapabilities || !this.serverCapabilities.resources) {
+      return [];
+    }
+    if (this.serverCapabilities.resources.listChanged) {
+      this.client.setNotificationHandler(
+        ResourceListChangedNotificationSchema,
+        async (_notification) => {
+          this.resources = await this.fetchResources();
+        }
+      );
+    }
+    return this.fetchResources();
+  }
+  async registerPrompts() {
+    if (!this.serverCapabilities || !this.serverCapabilities.prompts) {
+      return [];
+    }
+    if (this.serverCapabilities.prompts.listChanged) {
+      this.client.setNotificationHandler(
+        PromptListChangedNotificationSchema,
+        async (_notification) => {
+          this.prompts = await this.fetchPrompts();
+        }
+      );
+    }
+    return this.fetchPrompts();
+  }
+  async registerResourceTemplates() {
+    if (!this.serverCapabilities || !this.serverCapabilities.resources) {
+      return [];
+    }
+    return this.fetchResourceTemplates();
+  }
+  async fetchTools() {
+    let toolsAgg = [];
+    let toolsResult = { tools: [] };
+    do {
+      toolsResult = await this.client.listTools({
+        cursor: toolsResult.nextCursor
+      });
+      toolsAgg = toolsAgg.concat(toolsResult.tools);
+    } while (toolsResult.nextCursor);
+    return toolsAgg;
+  }
+  async fetchResources() {
+    let resourcesAgg = [];
+    let resourcesResult = { resources: [] };
+    do {
+      resourcesResult = await this.client.listResources({
+        cursor: resourcesResult.nextCursor
+      });
+      resourcesAgg = resourcesAgg.concat(resourcesResult.resources);
+    } while (resourcesResult.nextCursor);
+    return resourcesAgg;
+  }
+  async fetchPrompts() {
+    let promptsAgg = [];
+    let promptsResult = { prompts: [] };
+    do {
+      promptsResult = await this.client.listPrompts({
+        cursor: promptsResult.nextCursor
+      });
+      promptsAgg = promptsAgg.concat(promptsResult.prompts);
+    } while (promptsResult.nextCursor);
+    return promptsAgg;
+  }
+  async fetchResourceTemplates() {
+    let templatesAgg = [];
+    let templatesResult = {
+      resourceTemplates: []
+    };
+    do {
+      templatesResult = await this.client.listResourceTemplates({
+        cursor: templatesResult.nextCursor
+      });
+      templatesAgg = templatesAgg.concat(templatesResult.resourceTemplates);
+    } while (templatesResult.nextCursor);
+    return templatesAgg;
+  }
+};
+
+// src/mcp/client.ts
+var MCPClientManager = class {
+  constructor() {
+    this.mcpConnections = {};
+  }
+  /**
+   * Connect to and register an MCP server
+   *
+   * @param transportConfig Transport config
+   * @param clientConfig Client config
+   * @param capabilities Client capabilities (i.e. if the client supports roots/sampling)
+   */
+  async connectToServer(url, info, opts = { transport: {}, client: {}, capabilities: {} }) {
+    if (info.name in this.mcpConnections) {
+      throw new Error(
+        `An existing MCP client has already been registered under the name "${info.name}". The MCP client name must be unique.`
+      );
+    }
+    this.mcpConnections[info.name] = new MCPClientConnection(url, info, opts);
+    await this.mcpConnections[info.name].init();
+  }
+  /**
+   * @returns namespaced list of tools
+   */
+  listTools() {
+    return getNamespacedData(this.mcpConnections, "tools");
+  }
+  /**
+   * @returns namespaced list of prompts
+   */
+  listPrompts() {
+    return getNamespacedData(this.mcpConnections, "prompts");
+  }
+  /**
+   * @returns namespaced list of tools
+   */
+  listResources() {
+    return getNamespacedData(this.mcpConnections, "resources");
+  }
+  /**
+   * @returns namespaced list of resource templates
+   */
+  listResourceTemplates() {
+    return getNamespacedData(this.mcpConnections, "resourceTemplates");
+  }
+  /**
+   * Namespaced version of callTool
+   */
+  callTool(params, resultSchema, options) {
+    const unqualifiedName = params.name.replace(`${params.serverName}.`, "");
+    return this.mcpConnections[params.serverName].client.callTool(
+      {
+        ...params,
+        name: unqualifiedName
+      },
+      resultSchema,
+      options
+    );
+  }
+  /**
+   * Namespaced version of readResource
+   */
+  readResource(params, options) {
+    return this.mcpConnections[params.serverName].client.readResource(
+      params,
+      options
+    );
+  }
+  /**
+   * Namespaced version of getPrompt
+   */
+  getPrompt(params, options) {
+    return this.mcpConnections[params.serverName].client.getPrompt(
+      params,
+      options
+    );
+  }
+};
+function getNamespacedData(mcpClients, type) {
+  const sets = Object.entries(mcpClients).map(([name, conn]) => {
+    return { name, data: conn[type] };
+  });
+  const namespacedData = sets.flatMap(({ name: serverName, data }) => {
+    return data.map((item) => {
+      return {
+        ...item,
+        // we add a servername so we can easily pull it out and convert between qualified<->unqualified name
+        // just in case the server name or item name includes a "."
+        serverName: `${serverName}`,
+        // qualified name
+        name: `${serverName}.${item.name}`
+      };
+    });
+  });
+  return namespacedData;
+}
+export {
+  MCPClientManager,
+  getNamespacedData
+};
+//# sourceMappingURL=client.js.map

package/dist/mcp/client.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/mcp/sse-edge.ts","../../src/mcp/client-connection.ts","../../src/mcp/client.ts"],"sourcesContent":["import {\n  SSEClientTransport,\n  type SSEClientTransportOptions,\n} from \"@modelcontextprotocol/sdk/client/sse.js\";\n\nexport class SSEEdgeClientTransport extends SSEClientTransport {\n  /**\n   * Creates a new EdgeSSEClientTransport, which overrides fetch to be compatible with the CF workers environment\n   */\n  constructor(\n    private url: URL,\n    options: SSEClientTransportOptions\n  ) {\n    // biome-ignore lint/suspicious/noExplicitAny: Overriding fetch, type doesn't matter here\n    const fetchOverride = (url: any, options = {}) => {\n      const workerOptions = {\n        ...options,\n      };\n      // Remove unsupported properties\n      // @ts-ignore\n      // biome-ignore lint/performance/noDelete: workaround for workers environment\n      delete workerOptions.mode;\n\n      // Call the original fetch with fixed options\n      return global.fetch(url, workerOptions);\n    };\n\n    super(url, {\n      ...options,\n      eventSourceInit: {\n        fetch: fetchOverride,\n      },\n    });\n  }\n}\n","import { SSEEdgeClientTransport } from \"./sse-edge\";\n\nimport {\n  ToolListChangedNotificationSchema,\n  type ClientCapabilities,\n  type Resource,\n  type Tool,\n  type Prompt,\n  ResourceListChangedNotificationSchema,\n  PromptListChangedNotificationSchema,\n  type ListToolsResult,\n  type ListResourcesResult,\n  type ListPromptsResult,\n  type ServerCapabilities,\n  type ResourceTemplate,\n  type ListResourceTemplatesResult,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport type {\n  SSEClientTransport,\n  SSEClientTransportOptions,\n} from \"@modelcontextprotocol/sdk/client/sse.js\";\n\nexport class MCPClientConnection {\n  client: Client;\n  transport: SSEClientTransport;\n  connected: boolean;\n  instructions?: string;\n  tools: Tool[];\n  prompts: Prompt[];\n  resources: Resource[];\n  resourceTemplates: ResourceTemplate[];\n  serverCapabilities: ServerCapabilities | undefined;\n\n  constructor(\n    url: URL,\n    private info: ConstructorParameters<typeof Client>[0],\n    opts: {\n      transport: SSEClientTransportOptions;\n      client: ConstructorParameters<typeof Client>[1];\n      capabilities: ClientCapabilities;\n    } = { transport: {}, client: {}, capabilities: {} }\n  ) {\n    this.transport = new SSEEdgeClientTransport(url, opts.transport);\n    this.client = new Client(info, opts.client);\n    this.client.registerCapabilities(opts.capabilities);\n    this.connected = false;\n    this.tools = [];\n    this.prompts = [];\n    this.resources = [];\n    this.resourceTemplates = [];\n  }\n\n  async init() {\n    await this.client.connect(this.transport);\n\n    this.serverCapabilities = await this.client.getServerCapabilities();\n    if (!this.serverCapabilities) {\n      throw new Error(\n        `The MCP Server ${this.info.name} failed to return server capabilities`\n      );\n    }\n\n    const [instructions, tools, resources, prompts, resourceTemplates] =\n      await Promise.all([\n        this.client.getInstructions(),\n        this.registerTools(),\n        this.registerResources(),\n        this.registerPrompts(),\n        this.registerResourceTemplates(),\n      ]);\n\n    this.instructions = instructions;\n    this.tools = tools;\n    this.resources = resources;\n    this.prompts = prompts;\n    this.resourceTemplates = resourceTemplates;\n  }\n\n  /**\n   * Notification handler registration\n   */\n  async registerTools(): Promise<Tool[]> {\n    if (!this.serverCapabilities || !this.serverCapabilities.tools) {\n      return [];\n    }\n\n    if (this.serverCapabilities.tools.listChanged) {\n      this.client.setNotificationHandler(\n        ToolListChangedNotificationSchema,\n        async (_notification) => {\n          this.tools = await this.fetchTools();\n        }\n      );\n    }\n\n    return this.fetchTools();\n  }\n\n  async registerResources(): Promise<Resource[]> {\n    if (!this.serverCapabilities || !this.serverCapabilities.resources) {\n      return [];\n    }\n\n    if (this.serverCapabilities.resources.listChanged) {\n      this.client.setNotificationHandler(\n        ResourceListChangedNotificationSchema,\n        async (_notification) => {\n          this.resources = await this.fetchResources();\n        }\n      );\n    }\n\n    return this.fetchResources();\n  }\n\n  async registerPrompts(): Promise<Prompt[]> {\n    if (!this.serverCapabilities || !this.serverCapabilities.prompts) {\n      return [];\n    }\n\n    if (this.serverCapabilities.prompts.listChanged) {\n      this.client.setNotificationHandler(\n        PromptListChangedNotificationSchema,\n        async (_notification) => {\n          this.prompts = await this.fetchPrompts();\n        }\n      );\n    }\n\n    return this.fetchPrompts();\n  }\n\n  async registerResourceTemplates(): Promise<ResourceTemplate[]> {\n    if (!this.serverCapabilities || !this.serverCapabilities.resources) {\n      return [];\n    }\n\n    return this.fetchResourceTemplates();\n  }\n\n  async fetchTools() {\n    let toolsAgg: Tool[] = [];\n    let toolsResult: ListToolsResult = { tools: [] };\n    do {\n      toolsResult = await this.client.listTools({\n        cursor: toolsResult.nextCursor,\n      });\n      toolsAgg = toolsAgg.concat(toolsResult.tools);\n    } while (toolsResult.nextCursor);\n    return toolsAgg;\n  }\n\n  async fetchResources() {\n    let resourcesAgg: Resource[] = [];\n    let resourcesResult: ListResourcesResult = { resources: [] };\n    do {\n      resourcesResult = await this.client.listResources({\n        cursor: resourcesResult.nextCursor,\n      });\n      resourcesAgg = resourcesAgg.concat(resourcesResult.resources);\n    } while (resourcesResult.nextCursor);\n    return resourcesAgg;\n  }\n\n  async fetchPrompts() {\n    let promptsAgg: Prompt[] = [];\n    let promptsResult: ListPromptsResult = { prompts: [] };\n    do {\n      promptsResult = await this.client.listPrompts({\n        cursor: promptsResult.nextCursor,\n      });\n      promptsAgg = promptsAgg.concat(promptsResult.prompts);\n    } while (promptsResult.nextCursor);\n    return promptsAgg;\n  }\n\n  async fetchResourceTemplates() {\n    let templatesAgg: ResourceTemplate[] = [];\n    let templatesResult: ListResourceTemplatesResult = {\n      resourceTemplates: [],\n    };\n    do {\n      templatesResult = await this.client.listResourceTemplates({\n        cursor: templatesResult.nextCursor,\n      });\n      templatesAgg = templatesAgg.concat(templatesResult.resourceTemplates);\n    } while (templatesResult.nextCursor);\n    return templatesAgg;\n  }\n}\n","import { MCPClientConnection } from \"./client-connection\";\n\nimport type {\n  ClientCapabilities,\n  CallToolRequest,\n  CallToolResultSchema,\n  CompatibilityCallToolResultSchema,\n  ReadResourceRequest,\n  GetPromptRequest,\n  Tool,\n  Resource,\n  ResourceTemplate,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport type { SSEClientTransportOptions } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport type { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport type { RequestOptions } from \"@modelcontextprotocol/sdk/shared/protocol.js\";\n\n/**\n * Utility class that aggregates multiple MCP clients into one\n */\nexport class MCPClientManager {\n  public mcpConnections: Record<string, MCPClientConnection> = {};\n\n  /**\n   * Connect to and register an MCP server\n   *\n   * @param transportConfig Transport config\n   * @param clientConfig Client config\n   * @param capabilities Client capabilities (i.e. if the client supports roots/sampling)\n   */\n  async connectToServer(\n    url: URL,\n    info: ConstructorParameters<typeof Client>[0],\n    opts: {\n      transport: SSEClientTransportOptions;\n      client: ConstructorParameters<typeof Client>[1];\n      capabilities: ClientCapabilities;\n    } = { transport: {}, client: {}, capabilities: {} }\n  ) {\n    if (info.name in this.mcpConnections) {\n      throw new Error(\n        `An existing MCP client has already been registered under the name \"${info.name}\". The MCP client name must be unique.`\n      );\n    }\n\n    this.mcpConnections[info.name] = new MCPClientConnection(url, info, opts);\n    await this.mcpConnections[info.name].init();\n  }\n\n  /**\n   * @returns namespaced list of tools\n   */\n  listTools(): NamespacedData[\"tools\"] {\n    return getNamespacedData(this.mcpConnections, \"tools\");\n  }\n\n  /**\n   * @returns namespaced list of prompts\n   */\n  listPrompts(): NamespacedData[\"prompts\"] {\n    return getNamespacedData(this.mcpConnections, \"prompts\");\n  }\n\n  /**\n   * @returns namespaced list of tools\n   */\n  listResources(): NamespacedData[\"resources\"] {\n    return getNamespacedData(this.mcpConnections, \"resources\");\n  }\n\n  /**\n   * @returns namespaced list of resource templates\n   */\n  listResourceTemplates(): NamespacedData[\"resourceTemplates\"] {\n    return getNamespacedData(this.mcpConnections, \"resourceTemplates\");\n  }\n\n  /**\n   * Namespaced version of callTool\n   */\n  callTool(\n    params: CallToolRequest[\"params\"] & { serverName: string },\n    resultSchema:\n      | typeof CallToolResultSchema\n      | typeof CompatibilityCallToolResultSchema,\n    options: RequestOptions\n  ) {\n    const unqualifiedName = params.name.replace(`${params.serverName}.`, \"\");\n    return this.mcpConnections[params.serverName].client.callTool(\n      {\n        ...params,\n        name: unqualifiedName,\n      },\n      resultSchema,\n      options\n    );\n  }\n\n  /**\n   * Namespaced version of readResource\n   */\n  readResource(\n    params: ReadResourceRequest[\"params\"] & { serverName: string },\n    options: RequestOptions\n  ) {\n    return this.mcpConnections[params.serverName].client.readResource(\n      params,\n      options\n    );\n  }\n\n  /**\n   * Namespaced version of getPrompt\n   */\n  getPrompt(\n    params: GetPromptRequest[\"params\"] & { serverName: string },\n    options: RequestOptions\n  ) {\n    return this.mcpConnections[params.serverName].client.getPrompt(\n      params,\n      options\n    );\n  }\n}\n\ntype NamespacedData = {\n  tools: (Tool & { serverName: string })[];\n  prompts: (Prompt & { serverName: string })[];\n  resources: (Resource & { serverName: string })[];\n  resourceTemplates: (ResourceTemplate & { serverName: string })[];\n};\n\nexport function getNamespacedData<T extends keyof NamespacedData>(\n  mcpClients: Record<string, MCPClientConnection>,\n  type: T\n): NamespacedData[T] {\n  const sets = Object.entries(mcpClients).map(([name, conn]) => {\n    return { name, data: conn[type] };\n  });\n\n  const namespacedData = sets.flatMap(({ name: serverName, data }) => {\n    return data.map((item) => {\n      return {\n        ...item,\n        // we add a servername so we can easily pull it out and convert between qualified<->unqualified name\n        // just in case the server name or item name includes a \".\"\n        serverName: `${serverName}`,\n        // qualified name\n        name: `${serverName}.${item.name}`,\n      };\n    });\n  });\n\n  return namespacedData as NamespacedData[T]; // Type assertion needed due to TS limitations with conditional return types\n}\n"],"mappings":";;;AAAA;AAAA,EACE;AAAA,OAEK;AAEA,IAAM,yBAAN,cAAqC,mBAAmB;AAAA;AAAA;AAAA;AAAA,EAI7D,YACU,KACR,SACA;AAEA,UAAM,gBAAgB,CAACA,MAAUC,WAAU,CAAC,MAAM;AAChD,YAAM,gBAAgB;AAAA,QACpB,GAAGA;AAAA,MACL;AAIA,aAAO,cAAc;AAGrB,aAAO,OAAO,MAAMD,MAAK,aAAa;AAAA,IACxC;AAEA,UAAM,KAAK;AAAA,MACT,GAAG;AAAA,MACH,iBAAiB;AAAA,QACf,OAAO;AAAA,MACT;AAAA,IACF,CAAC;AAtBO;AAAA,EAuBV;AACF;;;AChCA;AAAA,EACE;AAAA,EAKA;AAAA,EACA;AAAA,OAOK;AACP,SAAS,cAAc;AAMhB,IAAM,sBAAN,MAA0B;AAAA,EAW/B,YACE,KACQ,MACR,OAII,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC,GAAG,cAAc,CAAC,EAAE,GAClD;AANQ;AAOR,SAAK,YAAY,IAAI,uBAAuB,KAAK,KAAK,SAAS;AAC/D,SAAK,SAAS,IAAI,OAAO,MAAM,KAAK,MAAM;AAC1C,SAAK,OAAO,qBAAqB,KAAK,YAAY;AAClD,SAAK,YAAY;AACjB,SAAK,QAAQ,CAAC;AACd,SAAK,UAAU,CAAC;AAChB,SAAK,YAAY,CAAC;AAClB,SAAK,oBAAoB,CAAC;AAAA,EAC5B;AAAA,EAEA,MAAM,OAAO;AACX,UAAM,KAAK,OAAO,QAAQ,KAAK,SAAS;AAExC,SAAK,qBAAqB,MAAM,KAAK,OAAO,sBAAsB;AAClE,QAAI,CAAC,KAAK,oBAAoB;AAC5B,YAAM,IAAI;AAAA,QACR,kBAAkB,KAAK,KAAK,IAAI;AAAA,MAClC;AAAA,IACF;AAEA,UAAM,CAAC,cAAc,OAAO,WAAW,SAAS,iBAAiB,IAC/D,MAAM,QAAQ,IAAI;AAAA,MAChB,KAAK,OAAO,gBAAgB;AAAA,MAC5B,KAAK,cAAc;AAAA,MACnB,KAAK,kBAAkB;AAAA,MACvB,KAAK,gBAAgB;AAAA,MACrB,KAAK,0BAA0B;AAAA,IACjC,CAAC;AAEH,SAAK,eAAe;AACpB,SAAK,QAAQ;AACb,SAAK,YAAY;AACjB,SAAK,UAAU;AACf,SAAK,oBAAoB;AAAA,EAC3B;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,gBAAiC;AACrC,QAAI,CAAC,KAAK,sBAAsB,CAAC,KAAK,mBAAmB,OAAO;AAC9D,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,KAAK,mBAAmB,MAAM,aAAa;AAC7C,WAAK,OAAO;AAAA,QACV;AAAA,QACA,OAAO,kBAAkB;AACvB,eAAK,QAAQ,MAAM,KAAK,WAAW;AAAA,QACrC;AAAA,MACF;AAAA,IACF;AAEA,WAAO,KAAK,WAAW;AAAA,EACzB;AAAA,EAEA,MAAM,oBAAyC;AAC7C,QAAI,CAAC,KAAK,sBAAsB,CAAC,KAAK,mBAAmB,WAAW;AAClE,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,KAAK,mBAAmB,UAAU,aAAa;AACjD,WAAK,OAAO;AAAA,QACV;AAAA,QACA,OAAO,kBAAkB;AACvB,eAAK,YAAY,MAAM,KAAK,eAAe;AAAA,QAC7C;AAAA,MACF;AAAA,IACF;AAEA,WAAO,KAAK,eAAe;AAAA,EAC7B;AAAA,EAEA,MAAM,kBAAqC;AACzC,QAAI,CAAC,KAAK,sBAAsB,CAAC,KAAK,mBAAmB,SAAS;AAChE,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,KAAK,mBAAmB,QAAQ,aAAa;AAC/C,WAAK,OAAO;AAAA,QACV;AAAA,QACA,OAAO,kBAAkB;AACvB,eAAK,UAAU,MAAM,KAAK,aAAa;AAAA,QACzC;AAAA,MACF;AAAA,IACF;AAEA,WAAO,KAAK,aAAa;AAAA,EAC3B;AAAA,EAEA,MAAM,4BAAyD;AAC7D,QAAI,CAAC,KAAK,sBAAsB,CAAC,KAAK,mBAAmB,WAAW;AAClE,aAAO,CAAC;AAAA,IACV;AAEA,WAAO,KAAK,uBAAuB;AAAA,EACrC;AAAA,EAEA,MAAM,aAAa;AACjB,QAAI,WAAmB,CAAC;AACxB,QAAI,cAA+B,EAAE,OAAO,CAAC,EAAE;AAC/C,OAAG;AACD,oBAAc,MAAM,KAAK,OAAO,UAAU;AAAA,QACxC,QAAQ,YAAY;AAAA,MACtB,CAAC;AACD,iBAAW,SAAS,OAAO,YAAY,KAAK;AAAA,IAC9C,SAAS,YAAY;AACrB,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,iBAAiB;AACrB,QAAI,eAA2B,CAAC;AAChC,QAAI,kBAAuC,EAAE,WAAW,CAAC,EAAE;AAC3D,OAAG;AACD,wBAAkB,MAAM,KAAK,OAAO,cAAc;AAAA,QAChD,QAAQ,gBAAgB;AAAA,MAC1B,CAAC;AACD,qBAAe,aAAa,OAAO,gBAAgB,SAAS;AAAA,IAC9D,SAAS,gBAAgB;AACzB,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,eAAe;AACnB,QAAI,aAAuB,CAAC;AAC5B,QAAI,gBAAmC,EAAE,SAAS,CAAC,EAAE;AACrD,OAAG;AACD,sBAAgB,MAAM,KAAK,OAAO,YAAY;AAAA,QAC5C,QAAQ,cAAc;AAAA,MACxB,CAAC;AACD,mBAAa,WAAW,OAAO,cAAc,OAAO;AAAA,IACtD,SAAS,cAAc;AACvB,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,yBAAyB;AAC7B,QAAI,eAAmC,CAAC;AACxC,QAAI,kBAA+C;AAAA,MACjD,mBAAmB,CAAC;AAAA,IACtB;AACA,OAAG;AACD,wBAAkB,MAAM,KAAK,OAAO,sBAAsB;AAAA,QACxD,QAAQ,gBAAgB;AAAA,MAC1B,CAAC;AACD,qBAAe,aAAa,OAAO,gBAAgB,iBAAiB;AAAA,IACtE,SAAS,gBAAgB;AACzB,WAAO;AAAA,EACT;AACF;;;AC1KO,IAAM,mBAAN,MAAuB;AAAA,EAAvB;AACL,SAAO,iBAAsD,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAS9D,MAAM,gBACJ,KACA,MACA,OAII,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC,GAAG,cAAc,CAAC,EAAE,GAClD;AACA,QAAI,KAAK,QAAQ,KAAK,gBAAgB;AACpC,YAAM,IAAI;AAAA,QACR,sEAAsE,KAAK,IAAI;AAAA,MACjF;AAAA,IACF;AAEA,SAAK,eAAe,KAAK,IAAI,IAAI,IAAI,oBAAoB,KAAK,MAAM,IAAI;AACxE,UAAM,KAAK,eAAe,KAAK,IAAI,EAAE,KAAK;AAAA,EAC5C;AAAA;AAAA;AAAA;AAAA,EAKA,YAAqC;AACnC,WAAO,kBAAkB,KAAK,gBAAgB,OAAO;AAAA,EACvD;AAAA;AAAA;AAAA;AAAA,EAKA,cAAyC;AACvC,WAAO,kBAAkB,KAAK,gBAAgB,SAAS;AAAA,EACzD;AAAA;AAAA;AAAA;AAAA,EAKA,gBAA6C;AAC3C,WAAO,kBAAkB,KAAK,gBAAgB,WAAW;AAAA,EAC3D;AAAA;AAAA;AAAA;AAAA,EAKA,wBAA6D;AAC3D,WAAO,kBAAkB,KAAK,gBAAgB,mBAAmB;AAAA,EACnE;AAAA;AAAA;AAAA;AAAA,EAKA,SACE,QACA,cAGA,SACA;AACA,UAAM,kBAAkB,OAAO,KAAK,QAAQ,GAAG,OAAO,UAAU,KAAK,EAAE;AACvE,WAAO,KAAK,eAAe,OAAO,UAAU,EAAE,OAAO;AAAA,MACnD;AAAA,QACE,GAAG;AAAA,QACH,MAAM;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,aACE,QACA,SACA;AACA,WAAO,KAAK,eAAe,OAAO,UAAU,EAAE,OAAO;AAAA,MACnD;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,UACE,QACA,SACA;AACA,WAAO,KAAK,eAAe,OAAO,UAAU,EAAE,OAAO;AAAA,MACnD;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AASO,SAAS,kBACd,YACA,MACmB;AACnB,QAAM,OAAO,OAAO,QAAQ,UAAU,EAAE,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM;AAC5D,WAAO,EAAE,MAAM,MAAM,KAAK,IAAI,EAAE;AAAA,EAClC,CAAC;AAED,QAAM,iBAAiB,KAAK,QAAQ,CAAC,EAAE,MAAM,YAAY,KAAK,MAAM;AAClE,WAAO,KAAK,IAAI,CAAC,SAAS;AACxB,aAAO;AAAA,QACL,GAAG;AAAA;AAAA;AAAA,QAGH,YAAY,GAAG,UAAU;AAAA;AAAA,QAEzB,MAAM,GAAG,UAAU,IAAI,KAAK,IAAI;AAAA,MAClC;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAED,SAAO;AACT;","names":["url","options"]}

package/dist/mcp/index.d.ts

@@ -0,0 +1,45 @@
+import { DurableObject } from 'cloudflare:workers';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { Connection } from 'partyserver';
+
+interface CORSOptions {
+    origin?: string;
+    methods?: string;
+    headers?: string;
+    maxAge?: number;
+}
+declare abstract class McpAgent<Env = unknown, State = unknown, Props extends Record<string, unknown> = Record<string, unknown>> extends DurableObject<Env> {
+    #private;
+    protected constructor(ctx: DurableObjectState, env: Env);
+    /**
+     * Agents API allowlist
+     */
+    initialState: State;
+    get state(): State;
+    sql<T = Record<string, string | number | boolean | null>>(strings: TemplateStringsArray, ...values: (string | number | boolean | null)[]): T[];
+    setState(state: State): void;
+    onStateUpdate(state: State | undefined, source: Connection | "server"): void;
+    onStart(): Promise<void>;
+    /**
+     * McpAgent API
+     */
+    abstract server: McpServer;
+    props: Props;
+    initRun: boolean;
+    abstract init(): Promise<void>;
+    _init(props: Props): Promise<void>;
+    fetch(request: Request): Promise<Response>;
+    getWebSocket(): WebSocket | null;
+    onMCPMessage(sessionId: string, request: Request): Promise<Response>;
+    webSocketMessage(ws: WebSocket, event: ArrayBuffer | string): Promise<void>;
+    webSocketError(ws: WebSocket, error: unknown): Promise<void>;
+    webSocketClose(ws: WebSocket, code: number, reason: string, wasClean: boolean): Promise<void>;
+    static mount(path: string, { binding, corsOptions, }?: {
+        binding?: string;
+        corsOptions?: CORSOptions;
+    }): {
+        fetch: (request: Request, env: Record<string, DurableObjectNamespace<McpAgent>>, ctx: ExecutionContext) => Promise<Response | undefined>;
+    };
+}
+
+export { McpAgent };

package/dist/mcp/index.js

@@ -0,0 +1,343 @@
+import {
+  Agent
+} from "../chunk-YMUU7QHV.js";
+import {
+  __privateAdd,
+  __privateGet,
+  __privateMethod,
+  __privateSet
+} from "../chunk-HMLY7DHA.js";
+
+// src/mcp/index.ts
+import { DurableObject } from "cloudflare:workers";
+import { JSONRPCMessageSchema } from "@modelcontextprotocol/sdk/types.js";
+var MAXIMUM_MESSAGE_SIZE = 4 * 1024 * 1024;
+function handleCORS(request, corsOptions) {
+  const origin = request.headers.get("Origin") || "*";
+  const corsHeaders = {
+    "Access-Control-Allow-Origin": corsOptions?.origin || origin,
+    "Access-Control-Allow-Methods": corsOptions?.methods || "GET, POST, OPTIONS",
+    "Access-Control-Allow-Headers": corsOptions?.headers || "Content-Type",
+    "Access-Control-Max-Age": (corsOptions?.maxAge || 86400).toString()
+  };
+  if (request.method === "OPTIONS") {
+    return new Response(null, { headers: corsHeaders });
+  }
+  return null;
+}
+var _getWebSocket, _started;
+var McpTransport = class {
+  constructor(getWebSocket) {
+    __privateAdd(this, _getWebSocket);
+    __privateAdd(this, _started, false);
+    __privateSet(this, _getWebSocket, getWebSocket);
+  }
+  async start() {
+    if (__privateGet(this, _started)) {
+      throw new Error("Transport already started");
+    }
+    __privateSet(this, _started, true);
+  }
+  async send(message) {
+    if (!__privateGet(this, _started)) {
+      throw new Error("Transport not started");
+    }
+    const websocket = __privateGet(this, _getWebSocket).call(this);
+    if (!websocket) {
+      throw new Error("WebSocket not connected");
+    }
+    try {
+      websocket.send(JSON.stringify(message));
+    } catch (error) {
+      this.onerror?.(error);
+      throw error;
+    }
+  }
+  async close() {
+    this.onclose?.();
+  }
+};
+_getWebSocket = new WeakMap();
+_started = new WeakMap();
+var _status, _transport, _connected, _agent, _McpAgent_instances, initialize_fn;
+var McpAgent = class extends DurableObject {
+  constructor(ctx, env) {
+    var _a;
+    super(ctx, env);
+    __privateAdd(this, _McpAgent_instances);
+    __privateAdd(this, _status, "zero");
+    __privateAdd(this, _transport);
+    __privateAdd(this, _connected, false);
+    /**
+     * Since McpAgent's _aren't_ yet real "Agents" (they route differently, don't support
+     * websockets, don't support hibernation), let's only expose a couple of the methods
+     * to the outer class: initialState/state/setState/onStateUpdate/sql
+     */
+    __privateAdd(this, _agent);
+    this.initRun = false;
+    const self = this;
+    __privateSet(this, _agent, new (_a = class extends Agent {
+      onStateUpdate(state, source) {
+        return self.onStateUpdate(state, source);
+      }
+    }, _a.options = {
+      hibernate: true
+    }, _a)(ctx, env));
+  }
+  get state() {
+    if (this.initialState) __privateGet(this, _agent).initialState = this.initialState;
+    return __privateGet(this, _agent).state;
+  }
+  sql(strings, ...values) {
+    return __privateGet(this, _agent).sql(strings, ...values);
+  }
+  setState(state) {
+    return __privateGet(this, _agent).setState(state);
+  }
+  onStateUpdate(state, source) {
+  }
+  async onStart() {
+    this.props = await this.ctx.storage.get("props");
+    this.init?.();
+    __privateSet(this, _transport, new McpTransport(() => this.getWebSocket()));
+    await this.server.connect(__privateGet(this, _transport));
+  }
+  async _init(props) {
+    await this.ctx.storage.put("props", props);
+    this.props = props;
+    if (!this.initRun) {
+      this.initRun = true;
+      await this.init();
+    }
+  }
+  // Allow the worker to fetch a websocket connection to the agent
+  async fetch(request) {
+    if (__privateGet(this, _status) !== "started") {
+      await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
+    }
+    if (request.headers.get("Upgrade") !== "websocket") {
+      return new Response("Expected WebSocket Upgrade request", {
+        status: 400
+      });
+    }
+    const url = new URL(request.url);
+    const sessionId = url.searchParams.get("sessionId");
+    if (!sessionId) {
+      return new Response("Missing sessionId", { status: 400 });
+    }
+    const webSocketPair = new WebSocketPair();
+    const [client, server] = Object.values(webSocketPair);
+    if (__privateGet(this, _connected)) {
+      return new Response("WebSocket already connected", { status: 400 });
+    }
+    this.ctx.acceptWebSocket(server);
+    __privateSet(this, _connected, true);
+    __privateSet(this, _transport, new McpTransport(() => this.getWebSocket()));
+    await this.server.connect(__privateGet(this, _transport));
+    return new Response(null, {
+      status: 101,
+      webSocket: client
+    });
+  }
+  getWebSocket() {
+    const websockets = this.ctx.getWebSockets();
+    if (websockets.length === 0) {
+      return null;
+    }
+    return websockets[0];
+  }
+  async onMCPMessage(sessionId, request) {
+    if (__privateGet(this, _status) !== "started") {
+      await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
+    }
+    try {
+      const contentType = request.headers.get("content-type") || "";
+      if (!contentType.includes("application/json")) {
+        return new Response(`Unsupported content-type: ${contentType}`, {
+          status: 400
+        });
+      }
+      const contentLength = Number.parseInt(
+        request.headers.get("content-length") || "0",
+        10
+      );
+      if (contentLength > MAXIMUM_MESSAGE_SIZE) {
+        return new Response(`Request body too large: ${contentLength} bytes`, {
+          status: 400
+        });
+      }
+      const message = await request.json();
+      let parsedMessage;
+      try {
+        parsedMessage = JSONRPCMessageSchema.parse(message);
+      } catch (error) {
+        __privateGet(this, _transport)?.onerror?.(error);
+        throw error;
+      }
+      __privateGet(this, _transport)?.onmessage?.(parsedMessage);
+      return new Response("Accepted", { status: 202 });
+    } catch (error) {
+      __privateGet(this, _transport)?.onerror?.(error);
+      return new Response(String(error), { status: 400 });
+    }
+  }
+  // This is unused since there are no incoming websocket messages
+  async webSocketMessage(ws, event) {
+    let message;
+    try {
+      const data = typeof event === "string" ? event : new TextDecoder().decode(event);
+      message = JSONRPCMessageSchema.parse(JSON.parse(data));
+    } catch (error) {
+      __privateGet(this, _transport)?.onerror?.(error);
+      return;
+    }
+    if (__privateGet(this, _status) !== "started") {
+      await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
+    }
+    __privateGet(this, _transport)?.onmessage?.(message);
+  }
+  // WebSocket event handlers for hibernation support
+  async webSocketError(ws, error) {
+    if (__privateGet(this, _status) !== "started") {
+      await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
+    }
+    __privateGet(this, _transport)?.onerror?.(error);
+  }
+  async webSocketClose(ws, code, reason, wasClean) {
+    if (__privateGet(this, _status) !== "started") {
+      await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
+    }
+    __privateGet(this, _transport)?.onclose?.();
+    __privateSet(this, _connected, false);
+  }
+  static mount(path, {
+    binding = "MCP_OBJECT",
+    corsOptions
+  } = {}) {
+    const basePattern = new URLPattern({ pathname: path });
+    const messagePattern = new URLPattern({ pathname: `${path}/message` });
+    return {
+      fetch: async (request, env, ctx) => {
+        const corsResponse = handleCORS(request, corsOptions);
+        if (corsResponse) return corsResponse;
+        const url = new URL(request.url);
+        const namespace = env[binding];
+        if (request.method === "GET" && basePattern.test(url)) {
+          const sessionId = url.searchParams.get("sessionId") || namespace.newUniqueId().toString();
+          const { readable, writable } = new TransformStream();
+          const writer = writable.getWriter();
+          const encoder = new TextEncoder();
+          const endpointMessage = `event: endpoint
+data: ${encodeURI(`${path}/message`)}?sessionId=${sessionId}
+
+`;
+          writer.write(encoder.encode(endpointMessage));
+          const id = namespace.idFromString(sessionId);
+          const doStub = namespace.get(id);
+          await doStub._init(ctx.props);
+          const upgradeUrl = new URL(request.url);
+          upgradeUrl.searchParams.set("sessionId", sessionId);
+          const response = await doStub.fetch(
+            new Request(upgradeUrl, {
+              headers: {
+                Upgrade: "websocket"
+              }
+            })
+          );
+          const ws = response.webSocket;
+          if (!ws) {
+            console.error("Failed to establish WebSocket connection");
+            await writer.close();
+            return;
+          }
+          ws.accept();
+          ws.addEventListener("message", async (event) => {
+            try {
+              const message = JSON.parse(event.data);
+              if (!(typeof message.id === "number" || message.id === null)) {
+                throw new Error("Invalid jsonrpc message id");
+              }
+              if (message.jsonrpc !== "2.0") {
+                throw new Error("Invalid jsonrpc version");
+              }
+              if (!Object.hasOwn(message, "result") && !Object.hasOwn(message, "error")) {
+                throw new Error(
+                  "Invalid jsonrpc message. Must have either result or error field"
+                );
+              }
+              const messageText = `event: message
+data: ${event.data}
+
+`;
+              await writer.write(encoder.encode(messageText));
+            } catch (error) {
+              console.error("Error forwarding message to SSE:", error);
+            }
+          });
+          ws.addEventListener("error", async (error) => {
+            try {
+              await writer.close();
+            } catch (e) {
+            }
+          });
+          ws.addEventListener("close", async () => {
+            try {
+              await writer.close();
+            } catch (error) {
+              console.error("Error closing SSE connection:", error);
+            }
+          });
+          return new Response(readable, {
+            headers: {
+              "Content-Type": "text/event-stream",
+              "Cache-Control": "no-cache",
+              Connection: "keep-alive",
+              "Access-Control-Allow-Origin": corsOptions?.origin || "*"
+            }
+          });
+        }
+        if (request.method === "POST" && messagePattern.test(url)) {
+          const sessionId = url.searchParams.get("sessionId");
+          if (!sessionId) {
+            return new Response(
+              `Missing sessionId. Expected POST to ${path} to initiate new one`,
+              { status: 400 }
+            );
+          }
+          const object = namespace.get(namespace.idFromString(sessionId));
+          const response = await object.onMCPMessage(sessionId, request);
+          const headers = new Headers();
+          response.headers.forEach?.((value, key) => {
+            headers.set(key, value);
+          });
+          headers.set(
+            "Access-Control-Allow-Origin",
+            corsOptions?.origin || "*"
+          );
+          return new Response(response.body, {
+            status: response.status,
+            statusText: response.statusText,
+            headers
+          });
+        }
+        return new Response("Not Found", { status: 404 });
+      }
+    };
+  }
+};
+_status = new WeakMap();
+_transport = new WeakMap();
+_connected = new WeakMap();
+_agent = new WeakMap();
+_McpAgent_instances = new WeakSet();
+initialize_fn = async function() {
+  await this.ctx.blockConcurrencyWhile(async () => {
+    __privateSet(this, _status, "starting");
+    await this.onStart();
+    __privateSet(this, _status, "started");
+  });
+};
+export {
+  McpAgent
+};
+//# sourceMappingURL=index.js.map