Update server.py

server.py

@@ -61,7 +61,7 @@ async def handle_sse(request: Request):
             write_stream_reader = write_stream
             placeholder_id = f"placeholder_{id(write_stream)}"
             write_streams[placeholder_id] = write_stream
-            sse_writers[placeholder_id] = sse_stream_writer  # Store the SSE writer with placeholder ID
+            sse_writers[placeholder_id] = sse_stream_writer
             logger.info("Starting MCP server with streams")
             await server.run(read_stream, write_stream, server.create_initialization_options())
             logger.info("MCP server running")
@@ -78,7 +78,7 @@ async def handle_post(request: Request):
         session_id = request.query_params.get("session_id")
         logger.info(f"Received POST with session_id: {session_id}, message: {message}")
 
-        # Try to find the write_stream and SSE writer, including checking for placeholder IDs
+        # Try to find the write_stream and SSE writer
         write_stream = write_streams.get(session_id)
         sse_writer = sse_writers.get(session_id)
         if not write_stream or not sse_writer:
@@ -105,8 +105,12 @@ async def handle_post(request: Request):
                 }
             }
             response_data = json.dumps(response)
-            await sse_writer.send({"event": "message", "data": response_data})
-            logger.info(f"Sent tools/list response for session {session_id} via SSE")
+            try:
+                await sse_writer.send({"event": "message", "data": response_data})
+                logger.info(f"Sent tools/list response for session {session_id} via SSE")
+            except Exception as e:
+                logger.error(f"Failed to send SSE response for session {session_id}: {str(e)}")
+                return Response(status_code=202)
             return Response(status_code=202)
 
         if not write_stream or not sse_writer: