Skip to content

Commit c880cc1

Browse files
committed
remove try finally block for testing and just add sse_stream_reader.aclose()
1 parent 7a83024 commit c880cc1

File tree

1 file changed

+2
-13
lines changed

1 file changed

+2
-13
lines changed

src/mcp/server/sse.py

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -190,25 +190,14 @@ async def response_wrapper(scope: Scope, receive: Receive, send: Send):
190190
)
191191
await read_stream_writer.aclose()
192192
await write_stream_reader.aclose()
193+
await sse_stream_reader.aclose()
193194
logging.debug(f"Client session disconnected {session_id}")
194195

195196
logger.debug("Starting SSE response task")
196197
tg.start_soon(response_wrapper, scope, receive, send)
197198

198199
logger.debug("Yielding read and write streams")
199-
try:
200-
yield (read_stream, write_stream)
201-
finally:
202-
# Close all remaining stream ends
203-
for stream, name in [
204-
(read_stream, "read_stream"),
205-
(write_stream, "write_stream"),
206-
(sse_stream_reader, "sse_stream_reader"),
207-
]:
208-
try:
209-
await stream.aclose()
210-
except Exception as e:
211-
logger.debug(f"Error closing {name}: {e}")
200+
yield (read_stream, write_stream)
212201

213202
async def handle_post_message(self, scope: Scope, receive: Receive, send: Send) -> None:
214203
logger.debug("Handling POST message")

0 commit comments

Comments
 (0)