agents 0.0.0-85d8edd → 0.0.0-86cae6f

package/dist/mcp/index.js

@@ -1,12 +1,9 @@
 import {
   Agent
-} from "../chunk-XG52S6YY.js";
-import {
-  __privateAdd,
-  __privateGet,
-  __privateMethod,
-  __privateSet
-} from "../chunk-HMLY7DHA.js";
+} from "../chunk-VNSFDJYL.js";
+import "../chunk-Y67CHZBI.js";
+import "../chunk-BZXOAZUX.js";
+import "../chunk-VCSB47AK.js";
 
 // src/mcp/index.ts
 import { DurableObject } from "cloudflare:workers";
@@ -19,37 +16,38 @@ import {
   JSONRPCMessageSchema
 } from "@modelcontextprotocol/sdk/types.js";
 var MAXIMUM_MESSAGE_SIZE_BYTES = 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()
+function corsHeaders(request, corsOptions = {}) {
+  const origin = "*";
+  return {
+    "Access-Control-Allow-Origin": corsOptions.origin || origin,
+    "Access-Control-Allow-Methods": corsOptions.methods || "GET, POST, OPTIONS",
+    "Access-Control-Allow-Headers": corsOptions.headers || "Content-Type, mcp-session-id",
+    "Access-Control-Max-Age": (corsOptions.maxAge || 86400).toString(),
+    "Access-Control-Expose-Headers": corsOptions.exposeHeaders || "mcp-session-id"
   };
+}
+function handleCORS(request, corsOptions) {
   if (request.method === "OPTIONS") {
-    return new Response(null, { headers: corsHeaders });
+    return new Response(null, { headers: corsHeaders(request, corsOptions) });
   }
   return null;
 }
-var _getWebSocket, _started;
 var McpSSETransport = class {
   constructor(getWebSocket) {
-    __privateAdd(this, _getWebSocket);
-    __privateAdd(this, _started, false);
-    __privateSet(this, _getWebSocket, getWebSocket);
+    this._started = false;
+    this._getWebSocket = getWebSocket;
   }
   async start() {
-    if (__privateGet(this, _started)) {
+    if (this._started) {
       throw new Error("Transport already started");
     }
-    __privateSet(this, _started, true);
+    this._started = true;
   }
   async send(message) {
-    if (!__privateGet(this, _started)) {
+    if (!this._started) {
       throw new Error("Transport not started");
     }
-    const websocket = __privateGet(this, _getWebSocket).call(this);
+    const websocket = this._getWebSocket();
     if (!websocket) {
       throw new Error("WebSocket not connected");
     }
@@ -64,52 +62,40 @@ var McpSSETransport = class {
     this.onclose?.();
   }
 };
-_getWebSocket = new WeakMap();
-_started = new WeakMap();
-var _getWebSocketForGetRequest, _getWebSocketForMessageID, _notifyResponseIdSent, _started2;
 var McpStreamableHttpTransport = class {
   constructor(getWebSocketForMessageID, notifyResponseIdSent) {
-    // TODO: If there is an open connection to send server-initiated messages
-    // back, we should use that connection
-    __privateAdd(this, _getWebSocketForGetRequest);
-    // Get the appropriate websocket connection for a given message id
-    __privateAdd(this, _getWebSocketForMessageID);
-    // Notify the server that a response has been sent for a given message id
-    // so that it may clean up it's mapping of message ids to connections
-    // once they are no longer needed
-    __privateAdd(this, _notifyResponseIdSent);
-    __privateAdd(this, _started2, false);
-    __privateSet(this, _getWebSocketForMessageID, getWebSocketForMessageID);
-    __privateSet(this, _notifyResponseIdSent, notifyResponseIdSent);
-    __privateSet(this, _getWebSocketForGetRequest, () => null);
+    this._started = false;
+    this._getWebSocketForMessageID = getWebSocketForMessageID;
+    this._notifyResponseIdSent = notifyResponseIdSent;
+    this._getWebSocketForGetRequest = () => null;
   }
   async start() {
-    if (__privateGet(this, _started2)) {
+    if (this._started) {
       throw new Error("Transport already started");
     }
-    __privateSet(this, _started2, true);
+    this._started = true;
   }
   async send(message) {
-    if (!__privateGet(this, _started2)) {
+    if (!this._started) {
       throw new Error("Transport not started");
     }
     let websocket = null;
     if (isJSONRPCResponse(message) || isJSONRPCError(message)) {
-      websocket = __privateGet(this, _getWebSocketForMessageID).call(this, message.id.toString());
+      websocket = this._getWebSocketForMessageID(message.id.toString());
       if (!websocket) {
         throw new Error(
           `Could not find WebSocket for message id: ${message.id}`
         );
       }
     } else if (isJSONRPCRequest(message)) {
-      websocket = __privateGet(this, _getWebSocketForGetRequest).call(this);
+      websocket = this._getWebSocketForGetRequest();
     } else if (isJSONRPCNotification(message)) {
       websocket = null;
     }
     try {
       websocket?.send(JSON.stringify(message));
       if (isJSONRPCResponse(message)) {
-        __privateGet(this, _notifyResponseIdSent).call(this, message.id.toString());
+        this._notifyResponseIdSent(message.id.toString());
       }
     } catch (error) {
       this.onerror?.(error);
@@ -120,28 +106,16 @@ var McpStreamableHttpTransport = class {
     this.onclose?.();
   }
 };
-_getWebSocketForGetRequest = new WeakMap();
-_getWebSocketForMessageID = new WeakMap();
-_notifyResponseIdSent = new WeakMap();
-_started2 = new WeakMap();
-var _status, _transport, _transportType, _requestIdToConnectionId, _agent, _McpAgent_instances, initialize_fn;
-var _McpAgent = class _McpAgent extends DurableObject {
+var McpAgent = class _McpAgent extends DurableObject {
   constructor(ctx, env) {
     var _a;
     super(ctx, env);
-    __privateAdd(this, _McpAgent_instances);
-    __privateAdd(this, _status, "zero");
-    __privateAdd(this, _transport);
-    __privateAdd(this, _transportType, "unset");
-    __privateAdd(this, _requestIdToConnectionId, /* @__PURE__ */ new Map());
-    /**
-     * Since McpAgent's _aren't_ yet real "Agents", let's only expose a couple of the methods
-     * to the outer class: initialState/state/setState/onStateUpdate/sql
-     */
-    __privateAdd(this, _agent);
+    this._status = "zero";
+    this._transportType = "unset";
+    this._requestIdToConnectionId = /* @__PURE__ */ new Map();
     this.initRun = false;
     const self = this;
-    __privateSet(this, _agent, new (_a = class extends Agent {
+    this._agent = new (_a = class extends Agent {
       onStateUpdate(state, source) {
         return self.onStateUpdate(state, source);
       }
@@ -150,23 +124,26 @@ var _McpAgent = class _McpAgent extends DurableObject {
       }
     }, _a.options = {
       hibernate: true
-    }, _a)(ctx, env));
+    }, _a)(ctx, env);
+  }
+  get mcp() {
+    return this._agent.mcp;
   }
   get state() {
-    return __privateGet(this, _agent).state;
+    return this._agent.state;
   }
   sql(strings, ...values) {
-    return __privateGet(this, _agent).sql(strings, ...values);
+    return this._agent.sql(strings, ...values);
   }
   setState(state) {
-    return __privateGet(this, _agent).setState(state);
+    return this._agent.setState(state);
   }
   onStateUpdate(state, source) {
   }
   async onStart() {
     var _a;
     const self = this;
-    __privateSet(this, _agent, new (_a = class extends Agent {
+    this._agent = new (_a = class extends Agent {
       constructor() {
         super(...arguments);
         this.initialState = self.initialState;
@@ -179,21 +156,22 @@ var _McpAgent = class _McpAgent extends DurableObject {
       }
     }, _a.options = {
       hibernate: true
-    }, _a)(this.ctx, this.env));
+    }, _a)(this.ctx, this.env);
     this.props = await this.ctx.storage.get("props");
-    __privateSet(this, _transportType, await this.ctx.storage.get(
+    this._transportType = await this.ctx.storage.get(
       "transportType"
-    ));
+    );
     await this._init(this.props);
-    if (__privateGet(this, _transportType) === "sse") {
-      __privateSet(this, _transport, new McpSSETransport(() => this.getWebSocket()));
-      await this.server.connect(__privateGet(this, _transport));
-    } else if (__privateGet(this, _transportType) === "streamable-http") {
-      __privateSet(this, _transport, new McpStreamableHttpTransport(
+    const server = await this.server;
+    if (this._transportType === "sse") {
+      this._transport = new McpSSETransport(() => this.getWebSocket());
+      await server.connect(this._transport);
+    } else if (this._transportType === "streamable-http") {
+      this._transport = new McpStreamableHttpTransport(
         (id) => this.getWebSocketForResponseID(id),
-        (id) => __privateGet(this, _requestIdToConnectionId).delete(id)
-      ));
-      await this.server.connect(__privateGet(this, _transport));
+        (id) => this._requestIdToConnectionId.delete(id)
+      );
+      await server.connect(this._transport);
     }
   }
   async _init(props) {
@@ -213,10 +191,17 @@ var _McpAgent = class _McpAgent extends DurableObject {
   async isInitialized() {
     return await this.ctx.storage.get("initialized") === true;
   }
+  async _initialize() {
+    await this.ctx.blockConcurrencyWhile(async () => {
+      this._status = "starting";
+      await this.onStart();
+      this._status = "started";
+    });
+  }
   // 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 (this._status !== "started") {
+      await this._initialize();
     }
     if (request.headers.get("Upgrade") !== "websocket") {
       return new Response("Expected WebSocket Upgrade request", {
@@ -225,6 +210,7 @@ var _McpAgent = class _McpAgent extends DurableObject {
     }
     const url = new URL(request.url);
     const path = url.pathname;
+    const server = await this.server;
     switch (path) {
       case "/sse": {
         const websockets = this.ctx.getWebSockets();
@@ -232,24 +218,24 @@ var _McpAgent = class _McpAgent extends DurableObject {
           return new Response("Websocket already connected", { status: 400 });
         }
         await this.ctx.storage.put("transportType", "sse");
-        __privateSet(this, _transportType, "sse");
-        if (!__privateGet(this, _transport)) {
-          __privateSet(this, _transport, new McpSSETransport(() => this.getWebSocket()));
-          await this.server.connect(__privateGet(this, _transport));
+        this._transportType = "sse";
+        if (!this._transport) {
+          this._transport = new McpSSETransport(() => this.getWebSocket());
+          await server.connect(this._transport);
         }
-        return __privateGet(this, _agent).fetch(request);
+        return this._agent.fetch(request);
       }
       case "/streamable-http": {
-        if (!__privateGet(this, _transport)) {
-          __privateSet(this, _transport, new McpStreamableHttpTransport(
+        if (!this._transport) {
+          this._transport = new McpStreamableHttpTransport(
             (id) => this.getWebSocketForResponseID(id),
-            (id) => __privateGet(this, _requestIdToConnectionId).delete(id)
-          ));
-          await this.server.connect(__privateGet(this, _transport));
+            (id) => this._requestIdToConnectionId.delete(id)
+          );
+          await server.connect(this._transport);
         }
         await this.ctx.storage.put("transportType", "streamable-http");
-        __privateSet(this, _transportType, "streamable-http");
-        return __privateGet(this, _agent).fetch(request);
+        this._transportType = "streamable-http";
+        return this._agent.fetch(request);
       }
       default:
         return new Response(
@@ -268,19 +254,19 @@ var _McpAgent = class _McpAgent extends DurableObject {
     return websockets[0];
   }
   getWebSocketForResponseID(id) {
-    const connectionId = __privateGet(this, _requestIdToConnectionId).get(id);
+    const connectionId = this._requestIdToConnectionId.get(id);
     if (connectionId === void 0) {
       return null;
     }
-    return __privateGet(this, _agent).getConnection(connectionId) ?? null;
+    return this._agent.getConnection(connectionId) ?? null;
   }
   // All messages received here. This is currently never called
   async onMessage(connection, event) {
-    if (__privateGet(this, _transportType) !== "streamable-http") {
+    if (this._transportType !== "streamable-http") {
       const err = new Error(
         "Internal Server Error: Expected streamable-http protocol"
       );
-      __privateGet(this, _transport)?.onerror?.(err);
+      this._transport?.onerror?.(err);
       return;
     }
     let message;
@@ -288,21 +274,21 @@ var _McpAgent = class _McpAgent extends DurableObject {
       const data = typeof event === "string" ? event : new TextDecoder().decode(event);
       message = JSONRPCMessageSchema.parse(JSON.parse(data));
     } catch (error) {
-      __privateGet(this, _transport)?.onerror?.(error);
+      this._transport?.onerror?.(error);
       return;
     }
     if (isJSONRPCRequest(message)) {
-      __privateGet(this, _requestIdToConnectionId).set(message.id.toString(), connection.id);
+      this._requestIdToConnectionId.set(message.id.toString(), connection.id);
     }
-    __privateGet(this, _transport)?.onmessage?.(message);
+    this._transport?.onmessage?.(message);
   }
   // All messages received over SSE after the initial connection has been established
   // will be passed here
   async onSSEMcpMessage(sessionId, request) {
-    if (__privateGet(this, _status) !== "started") {
-      await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
+    if (this._status !== "started") {
+      await this._initialize();
     }
-    if (__privateGet(this, _transportType) !== "sse") {
+    if (this._transportType !== "sse") {
       return new Error("Internal Server Error: Expected SSE protocol");
     }
     try {
@@ -311,35 +297,36 @@ var _McpAgent = class _McpAgent extends DurableObject {
       try {
         parsedMessage = JSONRPCMessageSchema.parse(message);
       } catch (error) {
-        __privateGet(this, _transport)?.onerror?.(error);
+        this._transport?.onerror?.(error);
         throw error;
       }
-      __privateGet(this, _transport)?.onmessage?.(parsedMessage);
+      this._transport?.onmessage?.(parsedMessage);
       return null;
     } catch (error) {
-      __privateGet(this, _transport)?.onerror?.(error);
+      console.error("Error forwarding message to SSE:", error);
+      this._transport?.onerror?.(error);
       return error;
     }
   }
   // Delegate all websocket events to the underlying agent
   async webSocketMessage(ws, event) {
-    if (__privateGet(this, _status) !== "started") {
-      await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
+    if (this._status !== "started") {
+      await this._initialize();
     }
-    return await __privateGet(this, _agent).webSocketMessage(ws, event);
+    return await this._agent.webSocketMessage(ws, event);
   }
   // WebSocket event handlers for hibernation support
   async webSocketError(ws, error) {
-    if (__privateGet(this, _status) !== "started") {
-      await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
+    if (this._status !== "started") {
+      await this._initialize();
     }
-    return await __privateGet(this, _agent).webSocketError(ws, error);
+    return await this._agent.webSocketError(ws, error);
   }
   async webSocketClose(ws, code, reason, wasClean) {
-    if (__privateGet(this, _status) !== "started") {
-      await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
+    if (this._status !== "started") {
+      await this._initialize();
     }
-    return await __privateGet(this, _agent).webSocketClose(ws, code, reason, wasClean);
+    return await this._agent.webSocketClose(ws, code, reason, wasClean);
   }
   static mount(path, {
     binding = "MCP_OBJECT",
@@ -358,18 +345,32 @@ var _McpAgent = class _McpAgent extends DurableObject {
     const basePattern = new URLPattern({ pathname });
     const messagePattern = new URLPattern({ pathname: `${pathname}/message` });
     return {
-      fetch: async (request, env, ctx) => {
+      async fetch(request, env, ctx) {
         const corsResponse = handleCORS(request, corsOptions);
         if (corsResponse) return corsResponse;
         const url = new URL(request.url);
-        const namespace = env[binding];
+        const bindingValue = env[binding];
+        if (bindingValue == null || typeof bindingValue !== "object") {
+          console.error(
+            `Could not find McpAgent binding for ${binding}. Did you update your wrangler configuration?`
+          );
+          return new Response("Invalid binding", { status: 500 });
+        }
+        if (bindingValue.toString() !== "[object DurableObjectNamespace]") {
+          return new Response("Invalid binding", { status: 500 });
+        }
+        const namespace = bindingValue;
         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 endpointUrl = new URL(request.url);
+          endpointUrl.pathname = encodeURI(`${pathname}/message`);
+          endpointUrl.searchParams.set("sessionId", sessionId);
+          const relativeUrlWithSession = endpointUrl.pathname + endpointUrl.search + endpointUrl.hash;
           const endpointMessage = `event: endpoint
-data: ${encodeURI(`${pathname}/message`)}?sessionId=${sessionId}
+data: ${relativeUrlWithSession}
 
 `;
           writer.write(encoder.encode(endpointMessage));
@@ -439,7 +440,7 @@ data: ${JSON.stringify(result.data)}
               "Content-Type": "text/event-stream",
               "Cache-Control": "no-cache",
               Connection: "keep-alive",
-              "Access-Control-Allow-Origin": corsOptions?.origin || "*"
+              ...corsHeaders(request, corsOptions)
             }
           });
         }
@@ -479,7 +480,7 @@ data: ${JSON.stringify(result.data)}
                 "Content-Type": "text/event-stream",
                 "Cache-Control": "no-cache",
                 Connection: "keep-alive",
-                "Access-Control-Allow-Origin": corsOptions?.origin || "*"
+                ...corsHeaders(request, corsOptions)
               }
             });
           }
@@ -489,7 +490,7 @@ data: ${JSON.stringify(result.data)}
               "Content-Type": "text/event-stream",
               "Cache-Control": "no-cache",
               Connection: "keep-alive",
-              "Access-Control-Allow-Origin": corsOptions?.origin || "*"
+              ...corsHeaders(request, corsOptions)
             }
           });
         }
@@ -507,13 +508,23 @@ data: ${JSON.stringify(result.data)}
     }
     const basePattern = new URLPattern({ pathname });
     return {
-      fetch: async (request, env, ctx) => {
+      async fetch(request, env, ctx) {
         const corsResponse = handleCORS(request, corsOptions);
         if (corsResponse) {
           return corsResponse;
         }
         const url = new URL(request.url);
-        const namespace = env[binding];
+        const bindingValue = env[binding];
+        if (bindingValue == null || typeof bindingValue !== "object") {
+          console.error(
+            `Could not find McpAgent binding for ${binding}. Did you update your wrangler configuration?`
+          );
+          return new Response("Invalid binding", { status: 500 });
+        }
+        if (bindingValue.toString() !== "[object DurableObjectNamespace]") {
+          return new Response("Invalid binding", { status: 500 });
+        }
+        const namespace = bindingValue;
         if (request.method === "POST" && basePattern.test(url)) {
           const acceptHeader = request.headers.get("accept");
           if (!acceptHeader?.includes("application/json") || !acceptHeader.includes("text/event-stream")) {
@@ -631,6 +642,7 @@ data: ${JSON.stringify(result.data)}
           const doStub = namespace.get(id);
           const isInitialized = await doStub.isInitialized();
           if (isInitializationRequest) {
+            await doStub._init(ctx.props);
             await doStub.setInitialized();
           } else if (!isInitialized) {
             const body2 = JSON.stringify({
@@ -726,7 +738,10 @@ data: ${JSON.stringify(result.data)}
               ws.send(JSON.stringify(message));
             }
             ws.close();
-            return new Response(null, { status: 202 });
+            return new Response(null, {
+              status: 202,
+              headers: corsHeaders(request, corsOptions)
+            });
           }
           for (const message of messages) {
             if (isJSONRPCRequest(message)) {
@@ -740,7 +755,7 @@ data: ${JSON.stringify(result.data)}
               "Cache-Control": "no-cache",
               Connection: "keep-alive",
               "mcp-session-id": sessionId,
-              "Access-Control-Allow-Origin": corsOptions?.origin || "*"
+              ...corsHeaders(request, corsOptions)
             },
             status: 200
           });
@@ -758,20 +773,6 @@ data: ${JSON.stringify(result.data)}
     };
   }
 };
-_status = new WeakMap();
-_transport = new WeakMap();
-_transportType = new WeakMap();
-_requestIdToConnectionId = 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");
-  });
-};
-var McpAgent = _McpAgent;
 export {
   McpAgent
 };