Skip to content

Commit e8e0491

Browse files
author
Hoa Lam
committed
Add close read write stream
1 parent f7d33ad commit e8e0491

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/mcp/server/sse.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,12 +123,16 @@ async def sse_writer():
123123
async def handle_see_disconnect(message: Message) -> None:
124124
logger.debug(f"Disconnect sse {session_id}")
125125
del self._read_stream_writers[session_id]
126+
await read_stream.aclose()
127+
await read_stream_writer.aclose()
128+
await write_stream.aclose()
129+
await write_stream_reader.aclose()
126130

127131
async with anyio.create_task_group() as tg:
128132
response = EventSourceResponse(
129133
content=sse_stream_reader,
130134
data_sender_callable=sse_writer,
131-
client_close_handler_callable=handle_see_disconnect, # type: ignore
135+
client_close_handler_callable=handle_see_disconnect # type: ignore
132136
)
133137
logger.debug("Starting SSE response task")
134138
tg.start_soon(response, scope, receive, send)

0 commit comments

Comments
 (0)