agents 0.0.0-b57e1d9 → 0.0.0-b5aee9f

package/dist/mcp/client.js

@@ -1,7 +1,7 @@
 import {
   MCPClientManager,
   getNamespacedData
-} from "../chunk-OYJXQRRH.js";
+} from "../chunk-UNG3FXYX.js";
 export {
   MCPClientManager,
   getNamespacedData

package/dist/mcp/do-oauth-client-provider.js

@@ -1,6 +1,6 @@
 import {
   DurableObjectOAuthClientProvider
-} from "../chunk-BZXOAZUX.js";
+} from "../chunk-PVQZBKN7.js";
 export {
   DurableObjectOAuthClientProvider
 };

package/dist/mcp/index.d.ts

@@ -1,17 +1,40 @@
 import { MCPClientManager } from './client.js';
 import { DurableObject } from 'cloudflare:workers';
-import { Connection, WSMessage } from 'partyserver';
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { Connection, WSMessage } from 'partyserver';
+import { SSEClientTransport, SSEClientTransportOptions } from '@modelcontextprotocol/sdk/client/sse.js';
+import { StreamableHTTPClientTransport, StreamableHTTPClientTransportOptions } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import 'zod';
-import '@modelcontextprotocol/sdk/types.js';
 import '@modelcontextprotocol/sdk/client/index.js';
-import '@modelcontextprotocol/sdk/client/sse.js';
+import '@modelcontextprotocol/sdk/shared/protocol.js';
+import '@modelcontextprotocol/sdk/types.js';
+import 'ai';
 import './do-oauth-client-provider.js';
 import '@modelcontextprotocol/sdk/client/auth.js';
 import '@modelcontextprotocol/sdk/shared/auth.js';
-import '@modelcontextprotocol/sdk/shared/protocol.js';
-import 'ai';
+
+declare class SSEEdgeClientTransport extends SSEClientTransport {
+    private authProvider;
+    /**
+     * Creates a new EdgeSSEClientTransport, which overrides fetch to be compatible with the CF workers environment
+     */
+    constructor(url: URL, options: SSEClientTransportOptions);
+    authHeaders(): Promise<{
+        Authorization: string;
+    } | undefined>;
+}
+
+declare class StreamableHTTPEdgeClientTransport extends StreamableHTTPClientTransport {
+    private authProvider;
+    /**
+     * Creates a new StreamableHTTPEdgeClientTransport, which overrides fetch to be compatible with the CF workers environment
+     */
+    constructor(url: URL, options: StreamableHTTPClientTransportOptions);
+    authHeaders(): Promise<{
+        Authorization: string;
+    } | undefined>;
+}
 
 interface CORSOptions {
     origin?: string;
@@ -57,23 +80,23 @@ declare abstract class McpAgent<Env = unknown, State = unknown, Props extends Re
     getWebSocket(): WebSocket | null;
     getWebSocketForResponseID(id: string): WebSocket | null;
     onMessage(connection: Connection, event: WSMessage): Promise<void>;
-    onSSEMcpMessage(sessionId: string, request: Request): Promise<Error | null>;
+    onSSEMcpMessage(_sessionId: string, request: Request): Promise<Error | null>;
     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, }?: {
+    static mount(path: string, { binding, corsOptions }?: {
         binding?: string;
         corsOptions?: CORSOptions;
     }): {
         fetch<Env>(this: void, request: Request, env: Env, ctx: ExecutionContext): Promise<Response>;
     };
-    static serveSSE(path: string, { binding, corsOptions, }?: {
+    static serveSSE(path: string, { binding, corsOptions }?: {
         binding?: string;
         corsOptions?: CORSOptions;
     }): {
         fetch<Env>(this: void, request: Request, env: Env, ctx: ExecutionContext): Promise<Response>;
     };
-    static serve(path: string, { binding, corsOptions, }?: {
+    static serve(path: string, { binding, corsOptions }?: {
         binding?: string;
         corsOptions?: CORSOptions;
     }): {
@@ -81,4 +104,4 @@ declare abstract class McpAgent<Env = unknown, State = unknown, Props extends Re
     };
 }
 
-export { McpAgent };
+export { McpAgent, SSEEdgeClientTransport, StreamableHTTPEdgeClientTransport };

package/dist/mcp/index.js

@@ -1,29 +1,32 @@
 import {
   Agent
-} from "../chunk-MXJNY43J.js";
-import "../chunk-OYJXQRRH.js";
-import "../chunk-BZXOAZUX.js";
-import "../chunk-VCSB47AK.js";
+} from "../chunk-3IQQY2UH.js";
+import {
+  SSEEdgeClientTransport,
+  StreamableHTTPEdgeClientTransport
+} from "../chunk-UNG3FXYX.js";
+import "../chunk-PVQZBKN7.js";
+import "../chunk-KUH345EY.js";
 
 // src/mcp/index.ts
 import { DurableObject } from "cloudflare:workers";
 import {
   InitializeRequestSchema,
+  JSONRPCMessageSchema,
   isJSONRPCError,
   isJSONRPCNotification,
   isJSONRPCRequest,
-  isJSONRPCResponse,
-  JSONRPCMessageSchema
+  isJSONRPCResponse
 } 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) {
@@ -141,6 +144,7 @@ var McpAgent = class _McpAgent extends DurableObject {
   setState(state) {
     return this._agent.setState(state);
   }
+  // biome-ignore lint/correctness/noUnusedFunctionParameters: overriden later
   onStateUpdate(state, source) {
   }
   async onStart() {
@@ -287,7 +291,7 @@ var McpAgent = class _McpAgent extends DurableObject {
   }
   // All messages received over SSE after the initial connection has been established
   // will be passed here
-  async onSSEMcpMessage(sessionId, request) {
+  async onSSEMcpMessage(_sessionId, request) {
     if (this._status !== "started") {
       await this._initialize();
     }
@@ -382,9 +386,14 @@ data: ${relativeUrlWithSession}
           await doStub._init(ctx.props);
           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
@@ -420,10 +429,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);
@@ -440,9 +449,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)
             }
           });
@@ -478,23 +487,23 @@ data: ${JSON.stringify(result.data)}
           const error = await doStub.onSSEMcpMessage(sessionId, request);
           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 });
@@ -532,24 +541,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 });
           }
@@ -559,12 +568,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 });
           }
@@ -572,14 +581,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 });
           }
@@ -593,12 +602,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 });
             }
@@ -609,34 +618,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 });
           }
@@ -649,12 +658,12 @@ data: ${JSON.stringify(result.data)}
             await doStub.setInitialized();
           } 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 });
           }
@@ -663,9 +672,14 @@ data: ${JSON.stringify(result.data)}
           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
@@ -677,12 +691,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 });
           }
@@ -715,10 +729,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);
@@ -742,8 +756,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) {
@@ -754,9 +768,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)
             },
@@ -764,12 +778,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 });
       }
@@ -777,6 +791,8 @@ data: ${JSON.stringify(result.data)}
   }
 };
 export {
-  McpAgent
+  McpAgent,
+  SSEEdgeClientTransport,
+  StreamableHTTPEdgeClientTransport
 };
 //# sourceMappingURL=index.js.map