modelcontextprotocol/typescript-sdk
Watch
1
0
Fork
You've already forked typescript-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Use eventsource package

Browse source
This commit is contained in:
Justin Spahr-Summers 2025-01-22 18:49:04 +00:00
commit c828067171
No known key found for this signature in database
1 changed files with 31 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/client/sse.ts
View file

@ -1,11 +1,10 @@
import { Transport } from "../shared/transport.js";
import { JSONRPCMessage, JSONRPCMessageSchema } from "../types.js";
import { EventSource, type EventSourceInit } from "eventsource";
/**
* Client transport for SSE: this will connect to a server using Server-Sent Events for receiving
* messages and make separate POST requests for sending messages.
*
* This uses the EventSource API in browsers. You can install the `eventsource` package for Node.js.
*/
export class SSEClientTransport implements Transport {
private _eventSource?: EventSource;
@ -19,7 +18,10 @@ export class SSEClientTransport implements Transport {
onerror?: (error: Error) => void;
onmessage?: (message: JSONRPCMessage) => void;
constructor(url: URL, opts?: { eventSourceInit?: EventSourceInit, requestInit?: RequestInit }) {
constructor(
url: URL,
opts?: { eventSourceInit?: EventSourceInit; requestInit?: RequestInit },
) {
this._url = url;
this._eventSourceInit = opts?.eventSourceInit;
this._requestInit = opts?.requestInit;
@ -33,7 +35,10 @@ export class SSEClientTransport implements Transport {
}
return new Promise((resolve, reject) => {
this._eventSource = new EventSource(this._url.href, this._eventSourceInit);
this._eventSource = new EventSource(
this._url.href,
this._eventSourceInit,
);
this._abortController = new AbortController();
this._eventSource.onerror = (event) => {
@ -101,7 +106,7 @@ export class SSEClientTransport implements Transport {
method: "POST",
headers,
body: JSON.stringify(message),
signal: this._abortController?.signal
signal: this._abortController?.signal,
};
const response = await fetch(this._endpoint, init);