aljaz/mcp-python-sdk
1
0
Fork 0
mirror of https://github.com/aljazceru/mcp-python-sdk.git synced 2025-12-19 23:04:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Handle SSE Disconnects Properly (#612)

Browse Source
This commit is contained in:
Akash D
2025-05-02 09:32:46 -07:00
committed by GitHub
parent 5d8eaf77be
commit 83968b5b2f
7 changed files with 38 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/mcp/server/session.py
View File

@@ -104,9 +104,6 @@ class ServerSession(
self._exit_stack.push_async_callback(
lambda: self._incoming_message_stream_reader.aclose()
)
self._exit_stack.push_async_callback(
lambda: self._incoming_message_stream_writer.aclose()
)
@property
def client_params(self) -> types.InitializeRequestParams | None:
@@ -144,6 +141,10 @@ class ServerSession(
return True
async def _receive_loop(self) -> None:
async with self._incoming_message_stream_writer:
await super()._receive_loop()
async def _received_request(
self, responder: RequestResponder[types.ClientRequest, types.ServerResult]
):