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

small refactor for send logic

Browse source
This commit is contained in:
ihrpr 2025-04-09 10:14:21 +01:00
commit 6a8fa5ff2b
No known key found for this signature in database
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/server/streamableHttp.ts
View file

@ -359,13 +359,9 @@ export class StreamableHTTPServerTransport implements Transport {
async send(message: JSONRPCMessage, options?: { relatedRequestId?: RequestId }): Promise<void> {
let requestId = options?.relatedRequestId;
let shouldCloseConnection = false;
if ('result' in message || 'error' in message) {
// If the message is a response, use the request ID from the message
requestId = message.id;
// This is a response to the original request, we can close the stream
// after sending all related responses
shouldCloseConnection = true;
}
if (requestId === undefined) {
throw new Error("No request ID provided for the message");
@ -380,8 +376,8 @@ export class StreamableHTTPServerTransport implements Transport {
sseResponse.write(
`event: message\ndata: ${JSON.stringify(message)}\n\n`,
);
if (shouldCloseConnection) {
// After all JSON-RPC responses have been sent, the server SHOULD close the SSE stream.
if ('result' in message || 'error' in message) {
this._sseResponseMapping.delete(requestId);
// Only close the connection if it's not needed by other requests
const canCloseConnection = ![...this._sseResponseMapping.entries()].some(([id, res]) => res === sseResponse && id !== requestId);