agents 0.0.0-2801d35 → 0.0.0-2a0b252

package/dist/mcp/index.js

@@ -1,31 +1,41 @@
 import {
   Agent
-} from "../chunk-J6T74FUS.js";
-import "../chunk-BZXOAZUX.js";
-import "../chunk-QSGN3REV.js";
-import "../chunk-Y67CHZBI.js";
+} from "../chunk-PNF6ZMUA.js";
+import {
+  SSEEdgeClientTransport,
+  StreamableHTTPEdgeClientTransport
+} from "../chunk-VYENMKFS.js";
+import "../chunk-LL2AFX7V.js";
+import "../chunk-QEVM4BVL.js";
+import "../chunk-AVYJQSLW.js";
 
 // src/mcp/index.ts
 import { DurableObject } from "cloudflare:workers";
 import {
   InitializeRequestSchema,
+  JSONRPCMessageSchema,
   isJSONRPCError,
   isJSONRPCNotification,
   isJSONRPCRequest,
-  isJSONRPCResponse,
-  JSONRPCMessageSchema
+  isJSONRPCResponse
+} from "@modelcontextprotocol/sdk/types.js";
+import {
+  ElicitRequestSchema
 } from "@modelcontextprotocol/sdk/types.js";
 var MAXIMUM_MESSAGE_SIZE_BYTES = 4 * 1024 * 1024;
-function corsHeaders(request, corsOptions = {}) {
+function corsHeaders(_request, corsOptions = {}) {
   const origin = "*";
   return {
-    "Access-Control-Allow-Origin": corsOptions.origin || origin,
+    "Access-Control-Allow-Headers": corsOptions.headers || "Content-Type, mcp-session-id, mcp-protocol-version",
     "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"
+    "Access-Control-Allow-Origin": corsOptions.origin || origin,
+    "Access-Control-Expose-Headers": corsOptions.exposeHeaders || "mcp-session-id",
+    "Access-Control-Max-Age": (corsOptions.maxAge || 86400).toString()
   };
 }
+function isDurableObjectNamespace(namespace) {
+  return typeof namespace === "object" && namespace !== null && "newUniqueId" in namespace && typeof namespace.newUniqueId === "function" && "idFromName" in namespace && typeof namespace.idFromName === "function";
+}
 function handleCORS(request, corsOptions) {
   if (request.method === "OPTIONS") {
     return new Response(null, { headers: corsHeaders(request, corsOptions) });
@@ -138,6 +148,46 @@ var McpAgent = class _McpAgent extends DurableObject {
   setState(state) {
     return this._agent.setState(state);
   }
+  /**
+   * Elicit user input with a message and schema
+   */
+  async elicitInput(params) {
+    const requestId = `elicit_${Math.random().toString(36).substring(2, 11)}`;
+    await this.ctx.storage.put(`elicitation:${requestId}`, {
+      message: params.message,
+      requestedSchema: params.requestedSchema,
+      timestamp: Date.now()
+    });
+    const elicitRequest = {
+      jsonrpc: "2.0",
+      id: requestId,
+      method: "elicitation/create",
+      params: {
+        message: params.message,
+        requestedSchema: params.requestedSchema
+      }
+    };
+    if (this._transport) {
+      await this._transport.send(elicitRequest);
+    } else {
+      const connections = this._agent?.getConnections();
+      if (!connections || Array.from(connections).length === 0) {
+        await this.ctx.storage.delete(`elicitation:${requestId}`);
+        throw new Error("No active connections available for elicitation");
+      }
+      const connectionList = Array.from(connections);
+      for (const connection of connectionList) {
+        try {
+          connection.send(JSON.stringify(elicitRequest));
+        } catch (error) {
+          console.error("Failed to send elicitation request:", error);
+        }
+      }
+    }
+    return this._waitForElicitationResponse(requestId);
+  }
+  // we leave the variables as unused for autocomplete purposes
+  // biome-ignore lint/correctness/noUnusedFunctionParameters: overriden later
   onStateUpdate(state, source) {
   }
   async onStart() {
@@ -174,15 +224,32 @@ var McpAgent = class _McpAgent extends DurableObject {
       await server.connect(this._transport);
     }
   }
+  /**
+   * Handle errors that occur during initialization or operation.
+   * Override this method to provide custom error handling.
+   * @param error - The error that occurred
+   * @returns An error response object with status code and message
+   */
+  onError(error) {
+    console.error("McpAgent error:", error);
+    return {
+      status: 500,
+      message: error.message || "An unexpected error occurred during initialization"
+    };
+  }
   async _init(props) {
-    await this.ctx.storage.put("props", props ?? {});
+    await this.updateProps(props);
     if (!this.ctx.storage.get("transportType")) {
       await this.ctx.storage.put("transportType", "unset");
     }
-    this.props = props;
     if (!this.initRun) {
       this.initRun = true;
-      await this.init();
+      try {
+        await this.init();
+      } catch (error) {
+        const errorResponse = this.onError(error);
+        throw new Error(`Initialization failed: ${errorResponse.message}`);
+      }
     }
   }
   async setInitialized() {
@@ -191,6 +258,10 @@ var McpAgent = class _McpAgent extends DurableObject {
   async isInitialized() {
     return await this.ctx.storage.get("initialized") === true;
   }
+  async updateProps(props) {
+    await this.ctx.storage.put("props", props ?? {});
+    this.props = props;
+  }
   async _initialize() {
     await this.ctx.blockConcurrencyWhile(async () => {
       this._status = "starting";
@@ -277,14 +348,78 @@ var McpAgent = class _McpAgent extends DurableObject {
       this._transport?.onerror?.(error);
       return;
     }
+    if (await this._handleElicitationResponse(message)) {
+      return;
+    }
     if (isJSONRPCRequest(message)) {
       this._requestIdToConnectionId.set(message.id.toString(), connection.id);
     }
     this._transport?.onmessage?.(message);
   }
+  /**
+   * Wait for elicitation response through storage polling
+   */
+  async _waitForElicitationResponse(requestId) {
+    const startTime = Date.now();
+    const timeout = 6e4;
+    try {
+      while (Date.now() - startTime < timeout) {
+        const response = await this.ctx.storage.get(
+          `elicitation:response:${requestId}`
+        );
+        if (response) {
+          await this.ctx.storage.delete(`elicitation:${requestId}`);
+          await this.ctx.storage.delete(`elicitation:response:${requestId}`);
+          return response;
+        }
+        await new Promise((resolve) => setTimeout(resolve, 100));
+      }
+      throw new Error("Elicitation request timed out");
+    } finally {
+      await this.ctx.storage.delete(`elicitation:${requestId}`);
+      await this.ctx.storage.delete(`elicitation:response:${requestId}`);
+    }
+  }
+  /**
+   * Handle elicitation responses   */
+  async _handleElicitationResponse(message) {
+    if (isJSONRPCResponse(message) && message.result) {
+      const requestId = message.id?.toString();
+      if (!requestId || !requestId.startsWith("elicit_")) return false;
+      const pendingRequest = await this.ctx.storage.get(
+        `elicitation:${requestId}`
+      );
+      if (!pendingRequest) return false;
+      await this.ctx.storage.put(
+        `elicitation:response:${requestId}`,
+        message.result
+      );
+      return true;
+    }
+    if (isJSONRPCError(message)) {
+      const requestId = message.id?.toString();
+      if (!requestId || !requestId.startsWith("elicit_")) return false;
+      const pendingRequest = await this.ctx.storage.get(
+        `elicitation:${requestId}`
+      );
+      if (!pendingRequest) return false;
+      const errorResult = {
+        action: "cancel",
+        content: {
+          error: message.error.message || "Elicitation request failed"
+        }
+      };
+      await this.ctx.storage.put(
+        `elicitation:response:${requestId}`,
+        errorResult
+      );
+      return true;
+    }
+    return false;
+  }
   // All messages received over SSE after the initial connection has been established
   // will be passed here
-  async onSSEMcpMessage(sessionId, request) {
+  async onSSEMcpMessage(_sessionId, messageBody) {
     if (this._status !== "started") {
       await this._initialize();
     }
@@ -292,14 +427,16 @@ var McpAgent = class _McpAgent extends DurableObject {
       return new Error("Internal Server Error: Expected SSE protocol");
     }
     try {
-      const message = await request.json();
       let parsedMessage;
       try {
-        parsedMessage = JSONRPCMessageSchema.parse(message);
+        parsedMessage = JSONRPCMessageSchema.parse(messageBody);
       } catch (error) {
         this._transport?.onerror?.(error);
         throw error;
       }
+      if (await this._handleElicitationResponse(parsedMessage)) {
+        return null;
+      }
       this._transport?.onmessage?.(parsedMessage);
       return null;
     } catch (error) {
@@ -356,7 +493,7 @@ var McpAgent = class _McpAgent extends DurableObject {
           );
           return new Response("Invalid binding", { status: 500 });
         }
-        if (bindingValue.toString() !== "[object DurableObjectNamespace]") {
+        if (!isDurableObjectNamespace(bindingValue)) {
           return new Response("Invalid binding", { status: 500 });
         }
         const namespace = bindingValue;
@@ -376,12 +513,26 @@ data: ${relativeUrlWithSession}
           writer.write(encoder.encode(endpointMessage));
           const id = namespace.idFromName(`sse:${sessionId}`);
           const doStub = namespace.get(id);
-          await doStub._init(ctx.props);
+          try {
+            await doStub._init(ctx.props);
+          } catch (error) {
+            console.error("Failed to initialize McpAgent:", error);
+            await writer.close();
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            return new Response(`Initialization failed: ${errorMessage}`, {
+              status: 500
+            });
+          }
           const upgradeUrl = new URL(request.url);
           upgradeUrl.pathname = "/sse";
+          const existingHeaders = {};
+          request.headers.forEach((value, key) => {
+            existingHeaders[key] = value;
+          });
           const response = await doStub.fetch(
             new Request(upgradeUrl, {
               headers: {
+                ...existingHeaders,
                 Upgrade: "websocket",
                 // Required by PartyServer
                 "x-partykit-room": sessionId
@@ -417,10 +568,10 @@ data: ${JSON.stringify(result.data)}
             onMessage(event).catch(console.error);
           });
           ws.addEventListener("error", (error) => {
-            async function onError(error2) {
+            async function onError(_error) {
               try {
                 await writer.close();
-              } catch (e) {
+              } catch (_e) {
               }
             }
             onError(error).catch(console.error);
@@ -437,9 +588,9 @@ data: ${JSON.stringify(result.data)}
           });
           return new Response(readable, {
             headers: {
-              "Content-Type": "text/event-stream",
               "Cache-Control": "no-cache",
               Connection: "keep-alive",
+              "Content-Type": "text/event-stream",
               ...corsHeaders(request, corsOptions)
             }
           });
@@ -472,26 +623,28 @@ data: ${JSON.stringify(result.data)}
           }
           const id = namespace.idFromName(`sse:${sessionId}`);
           const doStub = namespace.get(id);
-          const error = await doStub.onSSEMcpMessage(sessionId, request);
+          const messageBody = await request.json();
+          await doStub.updateProps(ctx.props);
+          const error = await doStub.onSSEMcpMessage(sessionId, messageBody);
           if (error) {
             return new Response(error.message, {
-              status: 400,
               headers: {
-                "Content-Type": "text/event-stream",
                 "Cache-Control": "no-cache",
                 Connection: "keep-alive",
+                "Content-Type": "text/event-stream",
                 ...corsHeaders(request, corsOptions)
-              }
+              },
+              status: 400
             });
           }
           return new Response("Accepted", {
-            status: 202,
             headers: {
-              "Content-Type": "text/event-stream",
               "Cache-Control": "no-cache",
               Connection: "keep-alive",
+              "Content-Type": "text/event-stream",
               ...corsHeaders(request, corsOptions)
-            }
+            },
+            status: 202
           });
         }
         return new Response("Not Found", { status: 404 });
@@ -521,7 +674,7 @@ data: ${JSON.stringify(result.data)}
           );
           return new Response("Invalid binding", { status: 500 });
         }
-        if (bindingValue.toString() !== "[object DurableObjectNamespace]") {
+        if (!isDurableObjectNamespace(bindingValue)) {
           return new Response("Invalid binding", { status: 500 });
         }
         const namespace = bindingValue;
@@ -529,24 +682,24 @@ data: ${JSON.stringify(result.data)}
           const acceptHeader = request.headers.get("accept");
           if (!acceptHeader?.includes("application/json") || !acceptHeader.includes("text/event-stream")) {
             const body2 = JSON.stringify({
-              jsonrpc: "2.0",
               error: {
                 code: -32e3,
                 message: "Not Acceptable: Client must accept both application/json and text/event-stream"
               },
-              id: null
+              id: null,
+              jsonrpc: "2.0"
             });
             return new Response(body2, { status: 406 });
           }
           const ct = request.headers.get("content-type");
           if (!ct || !ct.includes("application/json")) {
             const body2 = JSON.stringify({
-              jsonrpc: "2.0",
               error: {
                 code: -32e3,
                 message: "Unsupported Media Type: Content-Type must be application/json"
               },
-              id: null
+              id: null,
+              jsonrpc: "2.0"
             });
             return new Response(body2, { status: 415 });
           }
@@ -556,12 +709,12 @@ data: ${JSON.stringify(result.data)}
           );
           if (contentLength > MAXIMUM_MESSAGE_SIZE_BYTES) {
             const body2 = JSON.stringify({
-              jsonrpc: "2.0",
               error: {
                 code: -32e3,
                 message: `Request body too large. Maximum size is ${MAXIMUM_MESSAGE_SIZE_BYTES} bytes`
               },
-              id: null
+              id: null,
+              jsonrpc: "2.0"
             });
             return new Response(body2, { status: 413 });
           }
@@ -569,14 +722,14 @@ data: ${JSON.stringify(result.data)}
           let rawMessage;
           try {
             rawMessage = await request.json();
-          } catch (error) {
+          } catch (_error) {
             const body2 = JSON.stringify({
-              jsonrpc: "2.0",
               error: {
                 code: -32700,
                 message: "Parse error: Invalid JSON"
               },
-              id: null
+              id: null,
+              jsonrpc: "2.0"
             });
             return new Response(body2, { status: 400 });
           }
@@ -590,12 +743,12 @@ data: ${JSON.stringify(result.data)}
           for (const msg of arrayMessage) {
             if (!JSONRPCMessageSchema.safeParse(msg).success) {
               const body2 = JSON.stringify({
-                jsonrpc: "2.0",
                 error: {
                   code: -32700,
                   message: "Parse error: Invalid JSON-RPC message"
                 },
-                id: null
+                id: null,
+                jsonrpc: "2.0"
               });
               return new Response(body2, { status: 400 });
             }
@@ -606,34 +759,34 @@ data: ${JSON.stringify(result.data)}
           );
           if (isInitializationRequest && sessionId) {
             const body2 = JSON.stringify({
-              jsonrpc: "2.0",
               error: {
                 code: -32600,
                 message: "Invalid Request: Initialization requests must not include a sessionId"
               },
-              id: null
+              id: null,
+              jsonrpc: "2.0"
             });
             return new Response(body2, { status: 400 });
           }
           if (isInitializationRequest && messages.length > 1) {
             const body2 = JSON.stringify({
-              jsonrpc: "2.0",
               error: {
                 code: -32600,
                 message: "Invalid Request: Only one initialization request is allowed"
               },
-              id: null
+              id: null,
+              jsonrpc: "2.0"
             });
             return new Response(body2, { status: 400 });
           }
           if (!isInitializationRequest && !sessionId) {
             const body2 = JSON.stringify({
-              jsonrpc: "2.0",
               error: {
                 code: -32e3,
                 message: "Bad Request: Mcp-Session-Id header is required"
               },
-              id: null
+              id: null,
+              jsonrpc: "2.0"
             });
             return new Response(body2, { status: 400 });
           }
@@ -642,27 +795,48 @@ 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();
+            try {
+              await doStub._init(ctx.props);
+              await doStub.setInitialized();
+            } catch (error) {
+              console.error("Failed to initialize McpAgent:", error);
+              const errorMessage = error instanceof Error ? error.message : String(error);
+              const body2 = JSON.stringify({
+                error: {
+                  code: -32001,
+                  message: `Initialization failed: ${errorMessage}`
+                },
+                id: null,
+                jsonrpc: "2.0"
+              });
+              return new Response(body2, { status: 500 });
+            }
           } else if (!isInitialized) {
             const body2 = JSON.stringify({
-              jsonrpc: "2.0",
               error: {
                 code: -32001,
                 message: "Session not found"
               },
-              id: null
+              id: null,
+              jsonrpc: "2.0"
             });
             return new Response(body2, { status: 404 });
+          } else {
+            await doStub.updateProps(ctx.props);
           }
           const { readable, writable } = new TransformStream();
           const writer = writable.getWriter();
           const encoder = new TextEncoder();
           const upgradeUrl = new URL(request.url);
           upgradeUrl.pathname = "/streamable-http";
+          const existingHeaders = {};
+          request.headers.forEach((value, key) => {
+            existingHeaders[key] = value;
+          });
           const response = await doStub.fetch(
             new Request(upgradeUrl, {
               headers: {
+                ...existingHeaders,
                 Upgrade: "websocket",
                 // Required by PartyServer
                 "x-partykit-room": sessionId
@@ -674,12 +848,12 @@ data: ${JSON.stringify(result.data)}
             console.error("Failed to establish WebSocket connection");
             await writer.close();
             const body2 = JSON.stringify({
-              jsonrpc: "2.0",
               error: {
                 code: -32001,
                 message: "Failed to establish WebSocket connection"
               },
-              id: null
+              id: null,
+              jsonrpc: "2.0"
             });
             return new Response(body2, { status: 500 });
           }
@@ -712,10 +886,10 @@ data: ${JSON.stringify(result.data)}
             onMessage(event).catch(console.error);
           });
           ws.addEventListener("error", (error) => {
-            async function onError(error2) {
+            async function onError(_error) {
               try {
                 await writer.close();
-              } catch (e) {
+              } catch (_e) {
               }
             }
             onError(error).catch(console.error);
@@ -739,8 +913,8 @@ data: ${JSON.stringify(result.data)}
             }
             ws.close();
             return new Response(null, {
-              status: 202,
-              headers: corsHeaders(request, corsOptions)
+              headers: corsHeaders(request, corsOptions),
+              status: 202
             });
           }
           for (const message of messages) {
@@ -751,9 +925,9 @@ data: ${JSON.stringify(result.data)}
           }
           return new Response(readable, {
             headers: {
-              "Content-Type": "text/event-stream",
               "Cache-Control": "no-cache",
               Connection: "keep-alive",
+              "Content-Type": "text/event-stream",
               "mcp-session-id": sessionId,
               ...corsHeaders(request, corsOptions)
             },
@@ -761,12 +935,12 @@ data: ${JSON.stringify(result.data)}
           });
         }
         const body = JSON.stringify({
-          jsonrpc: "2.0",
           error: {
             code: -32e3,
             message: "Method not allowed"
           },
-          id: null
+          id: null,
+          jsonrpc: "2.0"
         });
         return new Response(body, { status: 405 });
       }
@@ -774,6 +948,9 @@ data: ${JSON.stringify(result.data)}
   }
 };
 export {
-  McpAgent
+  ElicitRequestSchema,
+  McpAgent,
+  SSEEdgeClientTransport,
+  StreamableHTTPEdgeClientTransport
 };
 //# sourceMappingURL=index.js.map