Handle SSE Disconnects Properly (#612)

This commit is contained in:
Akash D 2025-05-02 09:32:46 -07:00 • committed by GitHub
commit 83968b5b2f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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]
):