agents 0.0.0-ee727ca → 0.0.0-f5b5854

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,778 @@
1
+ import {
2
+ Agent
3
+ } from "../chunk-XG52S6YY.js";
4
+ import {
5
+ __privateAdd,
6
+ __privateGet,
7
+ __privateMethod,
8
+ __privateSet
9
+ } from "../chunk-HMLY7DHA.js";
10
+
11
+ // src/mcp/index.ts
12
+ import { DurableObject } from "cloudflare:workers";
13
+ import {
14
+ InitializeRequestSchema,
15
+ isJSONRPCError,
16
+ isJSONRPCNotification,
17
+ isJSONRPCRequest,
18
+ isJSONRPCResponse,
19
+ JSONRPCMessageSchema
20
+ } from "@modelcontextprotocol/sdk/types.js";
21
+ var MAXIMUM_MESSAGE_SIZE_BYTES = 4 * 1024 * 1024;
22
+ function handleCORS(request, corsOptions) {
23
+ const origin = request.headers.get("Origin") || "*";
24
+ const corsHeaders = {
25
+ "Access-Control-Allow-Origin": corsOptions?.origin || origin,
26
+ "Access-Control-Allow-Methods": corsOptions?.methods || "GET, POST, OPTIONS",
27
+ "Access-Control-Allow-Headers": corsOptions?.headers || "Content-Type",
28
+ "Access-Control-Max-Age": (corsOptions?.maxAge || 86400).toString()
29
+ };
30
+ if (request.method === "OPTIONS") {
31
+ return new Response(null, { headers: corsHeaders });
32
+ }
33
+ return null;
34
+ }
35
+ var _getWebSocket, _started;
36
+ var McpSSETransport = class {
37
+ constructor(getWebSocket) {
38
+ __privateAdd(this, _getWebSocket);
39
+ __privateAdd(this, _started, false);
40
+ __privateSet(this, _getWebSocket, getWebSocket);
41
+ }
42
+ async start() {
43
+ if (__privateGet(this, _started)) {
44
+ throw new Error("Transport already started");
45
+ }
46
+ __privateSet(this, _started, true);
47
+ }
48
+ async send(message) {
49
+ if (!__privateGet(this, _started)) {
50
+ throw new Error("Transport not started");
51
+ }
52
+ const websocket = __privateGet(this, _getWebSocket).call(this);
53
+ if (!websocket) {
54
+ throw new Error("WebSocket not connected");
55
+ }
56
+ try {
57
+ websocket.send(JSON.stringify(message));
58
+ } catch (error) {
59
+ this.onerror?.(error);
60
+ throw error;
61
+ }
62
+ }
63
+ async close() {
64
+ this.onclose?.();
65
+ }
66
+ };
67
+ _getWebSocket = new WeakMap();
68
+ _started = new WeakMap();
69
+ var _getWebSocketForGetRequest, _getWebSocketForMessageID, _notifyResponseIdSent, _started2;
70
+ var McpStreamableHttpTransport = class {
71
+ constructor(getWebSocketForMessageID, notifyResponseIdSent) {
72
+ // TODO: If there is an open connection to send server-initiated messages
73
+ // back, we should use that connection
74
+ __privateAdd(this, _getWebSocketForGetRequest);
75
+ // Get the appropriate websocket connection for a given message id
76
+ __privateAdd(this, _getWebSocketForMessageID);
77
+ // Notify the server that a response has been sent for a given message id
78
+ // so that it may clean up it's mapping of message ids to connections
79
+ // once they are no longer needed
80
+ __privateAdd(this, _notifyResponseIdSent);
81
+ __privateAdd(this, _started2, false);
82
+ __privateSet(this, _getWebSocketForMessageID, getWebSocketForMessageID);
83
+ __privateSet(this, _notifyResponseIdSent, notifyResponseIdSent);
84
+ __privateSet(this, _getWebSocketForGetRequest, () => null);
85
+ }
86
+ async start() {
87
+ if (__privateGet(this, _started2)) {
88
+ throw new Error("Transport already started");
89
+ }
90
+ __privateSet(this, _started2, true);
91
+ }
92
+ async send(message) {
93
+ if (!__privateGet(this, _started2)) {
94
+ throw new Error("Transport not started");
95
+ }
96
+ let websocket = null;
97
+ if (isJSONRPCResponse(message) || isJSONRPCError(message)) {
98
+ websocket = __privateGet(this, _getWebSocketForMessageID).call(this, message.id.toString());
99
+ if (!websocket) {
100
+ throw new Error(
101
+ `Could not find WebSocket for message id: ${message.id}`
102
+ );
103
+ }
104
+ } else if (isJSONRPCRequest(message)) {
105
+ websocket = __privateGet(this, _getWebSocketForGetRequest).call(this);
106
+ } else if (isJSONRPCNotification(message)) {
107
+ websocket = null;
108
+ }
109
+ try {
110
+ websocket?.send(JSON.stringify(message));
111
+ if (isJSONRPCResponse(message)) {
112
+ __privateGet(this, _notifyResponseIdSent).call(this, message.id.toString());
113
+ }
114
+ } catch (error) {
115
+ this.onerror?.(error);
116
+ throw error;
117
+ }
118
+ }
119
+ async close() {
120
+ this.onclose?.();
121
+ }
122
+ };
123
+ _getWebSocketForGetRequest = new WeakMap();
124
+ _getWebSocketForMessageID = new WeakMap();
125
+ _notifyResponseIdSent = new WeakMap();
126
+ _started2 = new WeakMap();
127
+ var _status, _transport, _transportType, _requestIdToConnectionId, _agent, _McpAgent_instances, initialize_fn;
128
+ var _McpAgent = class _McpAgent extends DurableObject {
129
+ constructor(ctx, env) {
130
+ var _a;
131
+ super(ctx, env);
132
+ __privateAdd(this, _McpAgent_instances);
133
+ __privateAdd(this, _status, "zero");
134
+ __privateAdd(this, _transport);
135
+ __privateAdd(this, _transportType, "unset");
136
+ __privateAdd(this, _requestIdToConnectionId, /* @__PURE__ */ new Map());
137
+ /**
138
+ * Since McpAgent's _aren't_ yet real "Agents", let's only expose a couple of the methods
139
+ * to the outer class: initialState/state/setState/onStateUpdate/sql
140
+ */
141
+ __privateAdd(this, _agent);
142
+ this.initRun = false;
143
+ const self = this;
144
+ __privateSet(this, _agent, new (_a = class extends Agent {
145
+ onStateUpdate(state, source) {
146
+ return self.onStateUpdate(state, source);
147
+ }
148
+ async onMessage(connection, message) {
149
+ return self.onMessage(connection, message);
150
+ }
151
+ }, _a.options = {
152
+ hibernate: true
153
+ }, _a)(ctx, env));
154
+ }
155
+ get state() {
156
+ return __privateGet(this, _agent).state;
157
+ }
158
+ sql(strings, ...values) {
159
+ return __privateGet(this, _agent).sql(strings, ...values);
160
+ }
161
+ setState(state) {
162
+ return __privateGet(this, _agent).setState(state);
163
+ }
164
+ onStateUpdate(state, source) {
165
+ }
166
+ async onStart() {
167
+ var _a;
168
+ const self = this;
169
+ __privateSet(this, _agent, new (_a = class extends Agent {
170
+ constructor() {
171
+ super(...arguments);
172
+ this.initialState = self.initialState;
173
+ }
174
+ onStateUpdate(state, source) {
175
+ return self.onStateUpdate(state, source);
176
+ }
177
+ async onMessage(connection, event) {
178
+ return self.onMessage(connection, event);
179
+ }
180
+ }, _a.options = {
181
+ hibernate: true
182
+ }, _a)(this.ctx, this.env));
183
+ this.props = await this.ctx.storage.get("props");
184
+ __privateSet(this, _transportType, await this.ctx.storage.get(
185
+ "transportType"
186
+ ));
187
+ await this._init(this.props);
188
+ if (__privateGet(this, _transportType) === "sse") {
189
+ __privateSet(this, _transport, new McpSSETransport(() => this.getWebSocket()));
190
+ await this.server.connect(__privateGet(this, _transport));
191
+ } else if (__privateGet(this, _transportType) === "streamable-http") {
192
+ __privateSet(this, _transport, new McpStreamableHttpTransport(
193
+ (id) => this.getWebSocketForResponseID(id),
194
+ (id) => __privateGet(this, _requestIdToConnectionId).delete(id)
195
+ ));
196
+ await this.server.connect(__privateGet(this, _transport));
197
+ }
198
+ }
199
+ async _init(props) {
200
+ await this.ctx.storage.put("props", props ?? {});
201
+ if (!this.ctx.storage.get("transportType")) {
202
+ await this.ctx.storage.put("transportType", "unset");
203
+ }
204
+ this.props = props;
205
+ if (!this.initRun) {
206
+ this.initRun = true;
207
+ await this.init();
208
+ }
209
+ }
210
+ async setInitialized() {
211
+ await this.ctx.storage.put("initialized", true);
212
+ }
213
+ async isInitialized() {
214
+ return await this.ctx.storage.get("initialized") === true;
215
+ }
216
+ // Allow the worker to fetch a websocket connection to the agent
217
+ async fetch(request) {
218
+ if (__privateGet(this, _status) !== "started") {
219
+ await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
220
+ }
221
+ if (request.headers.get("Upgrade") !== "websocket") {
222
+ return new Response("Expected WebSocket Upgrade request", {
223
+ status: 400
224
+ });
225
+ }
226
+ const url = new URL(request.url);
227
+ const path = url.pathname;
228
+ switch (path) {
229
+ case "/sse": {
230
+ const websockets = this.ctx.getWebSockets();
231
+ if (websockets.length > 0) {
232
+ return new Response("Websocket already connected", { status: 400 });
233
+ }
234
+ await this.ctx.storage.put("transportType", "sse");
235
+ __privateSet(this, _transportType, "sse");
236
+ if (!__privateGet(this, _transport)) {
237
+ __privateSet(this, _transport, new McpSSETransport(() => this.getWebSocket()));
238
+ await this.server.connect(__privateGet(this, _transport));
239
+ }
240
+ return __privateGet(this, _agent).fetch(request);
241
+ }
242
+ case "/streamable-http": {
243
+ if (!__privateGet(this, _transport)) {
244
+ __privateSet(this, _transport, new McpStreamableHttpTransport(
245
+ (id) => this.getWebSocketForResponseID(id),
246
+ (id) => __privateGet(this, _requestIdToConnectionId).delete(id)
247
+ ));
248
+ await this.server.connect(__privateGet(this, _transport));
249
+ }
250
+ await this.ctx.storage.put("transportType", "streamable-http");
251
+ __privateSet(this, _transportType, "streamable-http");
252
+ return __privateGet(this, _agent).fetch(request);
253
+ }
254
+ default:
255
+ return new Response(
256
+ "Internal Server Error: Expected /sse or /streamable-http path",
257
+ {
258
+ status: 500
259
+ }
260
+ );
261
+ }
262
+ }
263
+ getWebSocket() {
264
+ const websockets = this.ctx.getWebSockets();
265
+ if (websockets.length === 0) {
266
+ return null;
267
+ }
268
+ return websockets[0];
269
+ }
270
+ getWebSocketForResponseID(id) {
271
+ const connectionId = __privateGet(this, _requestIdToConnectionId).get(id);
272
+ if (connectionId === void 0) {
273
+ return null;
274
+ }
275
+ return __privateGet(this, _agent).getConnection(connectionId) ?? null;
276
+ }
277
+ // All messages received here. This is currently never called
278
+ async onMessage(connection, event) {
279
+ if (__privateGet(this, _transportType) !== "streamable-http") {
280
+ const err = new Error(
281
+ "Internal Server Error: Expected streamable-http protocol"
282
+ );
283
+ __privateGet(this, _transport)?.onerror?.(err);
284
+ return;
285
+ }
286
+ let message;
287
+ try {
288
+ const data = typeof event === "string" ? event : new TextDecoder().decode(event);
289
+ message = JSONRPCMessageSchema.parse(JSON.parse(data));
290
+ } catch (error) {
291
+ __privateGet(this, _transport)?.onerror?.(error);
292
+ return;
293
+ }
294
+ if (isJSONRPCRequest(message)) {
295
+ __privateGet(this, _requestIdToConnectionId).set(message.id.toString(), connection.id);
296
+ }
297
+ __privateGet(this, _transport)?.onmessage?.(message);
298
+ }
299
+ // All messages received over SSE after the initial connection has been established
300
+ // will be passed here
301
+ async onSSEMcpMessage(sessionId, request) {
302
+ if (__privateGet(this, _status) !== "started") {
303
+ await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
304
+ }
305
+ if (__privateGet(this, _transportType) !== "sse") {
306
+ return new Error("Internal Server Error: Expected SSE protocol");
307
+ }
308
+ try {
309
+ const message = await request.json();
310
+ let parsedMessage;
311
+ try {
312
+ parsedMessage = JSONRPCMessageSchema.parse(message);
313
+ } catch (error) {
314
+ __privateGet(this, _transport)?.onerror?.(error);
315
+ throw error;
316
+ }
317
+ __privateGet(this, _transport)?.onmessage?.(parsedMessage);
318
+ return null;
319
+ } catch (error) {
320
+ __privateGet(this, _transport)?.onerror?.(error);
321
+ return error;
322
+ }
323
+ }
324
+ // Delegate all websocket events to the underlying agent
325
+ async webSocketMessage(ws, event) {
326
+ if (__privateGet(this, _status) !== "started") {
327
+ await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
328
+ }
329
+ return await __privateGet(this, _agent).webSocketMessage(ws, event);
330
+ }
331
+ // WebSocket event handlers for hibernation support
332
+ async webSocketError(ws, error) {
333
+ if (__privateGet(this, _status) !== "started") {
334
+ await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
335
+ }
336
+ return await __privateGet(this, _agent).webSocketError(ws, error);
337
+ }
338
+ async webSocketClose(ws, code, reason, wasClean) {
339
+ if (__privateGet(this, _status) !== "started") {
340
+ await __privateMethod(this, _McpAgent_instances, initialize_fn).call(this);
341
+ }
342
+ return await __privateGet(this, _agent).webSocketClose(ws, code, reason, wasClean);
343
+ }
344
+ static mount(path, {
345
+ binding = "MCP_OBJECT",
346
+ corsOptions
347
+ } = {}) {
348
+ return _McpAgent.serveSSE(path, { binding, corsOptions });
349
+ }
350
+ static serveSSE(path, {
351
+ binding = "MCP_OBJECT",
352
+ corsOptions
353
+ } = {}) {
354
+ let pathname = path;
355
+ if (path === "/") {
356
+ pathname = "/*";
357
+ }
358
+ const basePattern = new URLPattern({ pathname });
359
+ const messagePattern = new URLPattern({ pathname: `${pathname}/message` });
360
+ return {
361
+ fetch: async (request, env, ctx) => {
362
+ const corsResponse = handleCORS(request, corsOptions);
363
+ if (corsResponse) return corsResponse;
364
+ const url = new URL(request.url);
365
+ const namespace = env[binding];
366
+ if (request.method === "GET" && basePattern.test(url)) {
367
+ const sessionId = url.searchParams.get("sessionId") || namespace.newUniqueId().toString();
368
+ const { readable, writable } = new TransformStream();
369
+ const writer = writable.getWriter();
370
+ const encoder = new TextEncoder();
371
+ const endpointMessage = `event: endpoint
372
+ data: ${encodeURI(`${pathname}/message`)}?sessionId=${sessionId}
373
+
374
+ `;
375
+ writer.write(encoder.encode(endpointMessage));
376
+ const id = namespace.idFromName(`sse:${sessionId}`);
377
+ const doStub = namespace.get(id);
378
+ await doStub._init(ctx.props);
379
+ const upgradeUrl = new URL(request.url);
380
+ upgradeUrl.pathname = "/sse";
381
+ const response = await doStub.fetch(
382
+ new Request(upgradeUrl, {
383
+ headers: {
384
+ Upgrade: "websocket",
385
+ // Required by PartyServer
386
+ "x-partykit-room": sessionId
387
+ }
388
+ })
389
+ );
390
+ const ws = response.webSocket;
391
+ if (!ws) {
392
+ console.error("Failed to establish WebSocket connection");
393
+ await writer.close();
394
+ return new Response("Failed to establish WebSocket connection", {
395
+ status: 500
396
+ });
397
+ }
398
+ ws.accept();
399
+ ws.addEventListener("message", (event) => {
400
+ async function onMessage(event2) {
401
+ try {
402
+ const message = JSON.parse(event2.data);
403
+ const result = JSONRPCMessageSchema.safeParse(message);
404
+ if (!result.success) {
405
+ return;
406
+ }
407
+ const messageText = `event: message
408
+ data: ${JSON.stringify(result.data)}
409
+
410
+ `;
411
+ await writer.write(encoder.encode(messageText));
412
+ } catch (error) {
413
+ console.error("Error forwarding message to SSE:", error);
414
+ }
415
+ }
416
+ onMessage(event).catch(console.error);
417
+ });
418
+ ws.addEventListener("error", (error) => {
419
+ async function onError(error2) {
420
+ try {
421
+ await writer.close();
422
+ } catch (e) {
423
+ }
424
+ }
425
+ onError(error).catch(console.error);
426
+ });
427
+ ws.addEventListener("close", () => {
428
+ async function onClose() {
429
+ try {
430
+ await writer.close();
431
+ } catch (error) {
432
+ console.error("Error closing SSE connection:", error);
433
+ }
434
+ }
435
+ onClose().catch(console.error);
436
+ });
437
+ return new Response(readable, {
438
+ headers: {
439
+ "Content-Type": "text/event-stream",
440
+ "Cache-Control": "no-cache",
441
+ Connection: "keep-alive",
442
+ "Access-Control-Allow-Origin": corsOptions?.origin || "*"
443
+ }
444
+ });
445
+ }
446
+ if (request.method === "POST" && messagePattern.test(url)) {
447
+ const sessionId = url.searchParams.get("sessionId");
448
+ if (!sessionId) {
449
+ return new Response(
450
+ `Missing sessionId. Expected POST to ${pathname} to initiate new one`,
451
+ { status: 400 }
452
+ );
453
+ }
454
+ const contentType = request.headers.get("content-type") || "";
455
+ if (!contentType.includes("application/json")) {
456
+ return new Response(`Unsupported content-type: ${contentType}`, {
457
+ status: 400
458
+ });
459
+ }
460
+ const contentLength = Number.parseInt(
461
+ request.headers.get("content-length") || "0",
462
+ 10
463
+ );
464
+ if (contentLength > MAXIMUM_MESSAGE_SIZE_BYTES) {
465
+ return new Response(
466
+ `Request body too large: ${contentLength} bytes`,
467
+ {
468
+ status: 400
469
+ }
470
+ );
471
+ }
472
+ const id = namespace.idFromName(`sse:${sessionId}`);
473
+ const doStub = namespace.get(id);
474
+ const error = await doStub.onSSEMcpMessage(sessionId, request);
475
+ if (error) {
476
+ return new Response(error.message, {
477
+ status: 400,
478
+ headers: {
479
+ "Content-Type": "text/event-stream",
480
+ "Cache-Control": "no-cache",
481
+ Connection: "keep-alive",
482
+ "Access-Control-Allow-Origin": corsOptions?.origin || "*"
483
+ }
484
+ });
485
+ }
486
+ return new Response("Accepted", {
487
+ status: 202,
488
+ headers: {
489
+ "Content-Type": "text/event-stream",
490
+ "Cache-Control": "no-cache",
491
+ Connection: "keep-alive",
492
+ "Access-Control-Allow-Origin": corsOptions?.origin || "*"
493
+ }
494
+ });
495
+ }
496
+ return new Response("Not Found", { status: 404 });
497
+ }
498
+ };
499
+ }
500
+ static serve(path, {
501
+ binding = "MCP_OBJECT",
502
+ corsOptions
503
+ } = {}) {
504
+ let pathname = path;
505
+ if (path === "/") {
506
+ pathname = "/*";
507
+ }
508
+ const basePattern = new URLPattern({ pathname });
509
+ return {
510
+ fetch: async (request, env, ctx) => {
511
+ const corsResponse = handleCORS(request, corsOptions);
512
+ if (corsResponse) {
513
+ return corsResponse;
514
+ }
515
+ const url = new URL(request.url);
516
+ const namespace = env[binding];
517
+ if (request.method === "POST" && basePattern.test(url)) {
518
+ const acceptHeader = request.headers.get("accept");
519
+ if (!acceptHeader?.includes("application/json") || !acceptHeader.includes("text/event-stream")) {
520
+ const body2 = JSON.stringify({
521
+ jsonrpc: "2.0",
522
+ error: {
523
+ code: -32e3,
524
+ message: "Not Acceptable: Client must accept both application/json and text/event-stream"
525
+ },
526
+ id: null
527
+ });
528
+ return new Response(body2, { status: 406 });
529
+ }
530
+ const ct = request.headers.get("content-type");
531
+ if (!ct || !ct.includes("application/json")) {
532
+ const body2 = JSON.stringify({
533
+ jsonrpc: "2.0",
534
+ error: {
535
+ code: -32e3,
536
+ message: "Unsupported Media Type: Content-Type must be application/json"
537
+ },
538
+ id: null
539
+ });
540
+ return new Response(body2, { status: 415 });
541
+ }
542
+ const contentLength = Number.parseInt(
543
+ request.headers.get("content-length") ?? "0",
544
+ 10
545
+ );
546
+ if (contentLength > MAXIMUM_MESSAGE_SIZE_BYTES) {
547
+ const body2 = JSON.stringify({
548
+ jsonrpc: "2.0",
549
+ error: {
550
+ code: -32e3,
551
+ message: `Request body too large. Maximum size is ${MAXIMUM_MESSAGE_SIZE_BYTES} bytes`
552
+ },
553
+ id: null
554
+ });
555
+ return new Response(body2, { status: 413 });
556
+ }
557
+ let sessionId = request.headers.get("mcp-session-id");
558
+ let rawMessage;
559
+ try {
560
+ rawMessage = await request.json();
561
+ } catch (error) {
562
+ const body2 = JSON.stringify({
563
+ jsonrpc: "2.0",
564
+ error: {
565
+ code: -32700,
566
+ message: "Parse error: Invalid JSON"
567
+ },
568
+ id: null
569
+ });
570
+ return new Response(body2, { status: 400 });
571
+ }
572
+ let arrayMessage;
573
+ if (Array.isArray(rawMessage)) {
574
+ arrayMessage = rawMessage;
575
+ } else {
576
+ arrayMessage = [rawMessage];
577
+ }
578
+ let messages = [];
579
+ for (const msg of arrayMessage) {
580
+ if (!JSONRPCMessageSchema.safeParse(msg).success) {
581
+ const body2 = JSON.stringify({
582
+ jsonrpc: "2.0",
583
+ error: {
584
+ code: -32700,
585
+ message: "Parse error: Invalid JSON-RPC message"
586
+ },
587
+ id: null
588
+ });
589
+ return new Response(body2, { status: 400 });
590
+ }
591
+ }
592
+ messages = arrayMessage.map((msg) => JSONRPCMessageSchema.parse(msg));
593
+ const isInitializationRequest = messages.some(
594
+ (msg) => InitializeRequestSchema.safeParse(msg).success
595
+ );
596
+ if (isInitializationRequest && sessionId) {
597
+ const body2 = JSON.stringify({
598
+ jsonrpc: "2.0",
599
+ error: {
600
+ code: -32600,
601
+ message: "Invalid Request: Initialization requests must not include a sessionId"
602
+ },
603
+ id: null
604
+ });
605
+ return new Response(body2, { status: 400 });
606
+ }
607
+ if (isInitializationRequest && messages.length > 1) {
608
+ const body2 = JSON.stringify({
609
+ jsonrpc: "2.0",
610
+ error: {
611
+ code: -32600,
612
+ message: "Invalid Request: Only one initialization request is allowed"
613
+ },
614
+ id: null
615
+ });
616
+ return new Response(body2, { status: 400 });
617
+ }
618
+ if (!isInitializationRequest && !sessionId) {
619
+ const body2 = JSON.stringify({
620
+ jsonrpc: "2.0",
621
+ error: {
622
+ code: -32e3,
623
+ message: "Bad Request: Mcp-Session-Id header is required"
624
+ },
625
+ id: null
626
+ });
627
+ return new Response(body2, { status: 400 });
628
+ }
629
+ sessionId = sessionId ?? namespace.newUniqueId().toString();
630
+ const id = namespace.idFromName(`streamable-http:${sessionId}`);
631
+ const doStub = namespace.get(id);
632
+ const isInitialized = await doStub.isInitialized();
633
+ if (isInitializationRequest) {
634
+ await doStub.setInitialized();
635
+ } else if (!isInitialized) {
636
+ const body2 = JSON.stringify({
637
+ jsonrpc: "2.0",
638
+ error: {
639
+ code: -32001,
640
+ message: "Session not found"
641
+ },
642
+ id: null
643
+ });
644
+ return new Response(body2, { status: 404 });
645
+ }
646
+ const { readable, writable } = new TransformStream();
647
+ const writer = writable.getWriter();
648
+ const encoder = new TextEncoder();
649
+ const upgradeUrl = new URL(request.url);
650
+ upgradeUrl.pathname = "/streamable-http";
651
+ const response = await doStub.fetch(
652
+ new Request(upgradeUrl, {
653
+ headers: {
654
+ Upgrade: "websocket",
655
+ // Required by PartyServer
656
+ "x-partykit-room": sessionId
657
+ }
658
+ })
659
+ );
660
+ const ws = response.webSocket;
661
+ if (!ws) {
662
+ console.error("Failed to establish WebSocket connection");
663
+ await writer.close();
664
+ const body2 = JSON.stringify({
665
+ jsonrpc: "2.0",
666
+ error: {
667
+ code: -32001,
668
+ message: "Failed to establish WebSocket connection"
669
+ },
670
+ id: null
671
+ });
672
+ return new Response(body2, { status: 500 });
673
+ }
674
+ const requestIds = /* @__PURE__ */ new Set();
675
+ ws.accept();
676
+ ws.addEventListener("message", (event) => {
677
+ async function onMessage(event2) {
678
+ try {
679
+ const data = typeof event2.data === "string" ? event2.data : new TextDecoder().decode(event2.data);
680
+ const message = JSON.parse(data);
681
+ const result = JSONRPCMessageSchema.safeParse(message);
682
+ if (!result.success) {
683
+ return;
684
+ }
685
+ if (isJSONRPCResponse(result.data) || isJSONRPCError(result.data)) {
686
+ requestIds.delete(result.data.id);
687
+ }
688
+ const messageText = `event: message
689
+ data: ${JSON.stringify(result.data)}
690
+
691
+ `;
692
+ await writer.write(encoder.encode(messageText));
693
+ if (requestIds.size === 0) {
694
+ ws.close();
695
+ }
696
+ } catch (error) {
697
+ console.error("Error forwarding message to SSE:", error);
698
+ }
699
+ }
700
+ onMessage(event).catch(console.error);
701
+ });
702
+ ws.addEventListener("error", (error) => {
703
+ async function onError(error2) {
704
+ try {
705
+ await writer.close();
706
+ } catch (e) {
707
+ }
708
+ }
709
+ onError(error).catch(console.error);
710
+ });
711
+ ws.addEventListener("close", () => {
712
+ async function onClose() {
713
+ try {
714
+ await writer.close();
715
+ } catch (error) {
716
+ console.error("Error closing SSE connection:", error);
717
+ }
718
+ }
719
+ onClose().catch(console.error);
720
+ });
721
+ const hasOnlyNotificationsOrResponses = messages.every(
722
+ (msg) => isJSONRPCNotification(msg) || isJSONRPCResponse(msg)
723
+ );
724
+ if (hasOnlyNotificationsOrResponses) {
725
+ for (const message of messages) {
726
+ ws.send(JSON.stringify(message));
727
+ }
728
+ ws.close();
729
+ return new Response(null, { status: 202 });
730
+ }
731
+ for (const message of messages) {
732
+ if (isJSONRPCRequest(message)) {
733
+ requestIds.add(message.id);
734
+ }
735
+ ws.send(JSON.stringify(message));
736
+ }
737
+ return new Response(readable, {
738
+ headers: {
739
+ "Content-Type": "text/event-stream",
740
+ "Cache-Control": "no-cache",
741
+ Connection: "keep-alive",
742
+ "mcp-session-id": sessionId,
743
+ "Access-Control-Allow-Origin": corsOptions?.origin || "*"
744
+ },
745
+ status: 200
746
+ });
747
+ }
748
+ const body = JSON.stringify({
749
+ jsonrpc: "2.0",
750
+ error: {
751
+ code: -32e3,
752
+ message: "Method not allowed"
753
+ },
754
+ id: null
755
+ });
756
+ return new Response(body, { status: 405 });
757
+ }
758
+ };
759
+ }
760
+ };
761
+ _status = new WeakMap();
762
+ _transport = new WeakMap();
763
+ _transportType = new WeakMap();
764
+ _requestIdToConnectionId = new WeakMap();
765
+ _agent = new WeakMap();
766
+ _McpAgent_instances = new WeakSet();
767
+ initialize_fn = async function() {
768
+ await this.ctx.blockConcurrencyWhile(async () => {
769
+ __privateSet(this, _status, "starting");
770
+ await this.onStart();
771
+ __privateSet(this, _status, "started");
772
+ });
773
+ };
774
+ var McpAgent = _McpAgent;
775
+ export {
776
+ McpAgent
777
+ };
778
+ //# sourceMappingURL=index.js.map