Update server.py

server.py

@@ -52,7 +52,7 @@ async def handle_sse(request: Request):
                             placeholder_id = f"placeholder_{id(write_stream)}"
                             if placeholder_id in write_streams:
                                 write_streams[session_id] = write_streams.pop(placeholder_id)
-                                sse_writers[session_id] = sse_writers.pop(placeholder_id)
+                                sse_writers[session_id] = sse_stream_writer
                                 logger.info(f"Updated placeholder {placeholder_id} to session_id {session_id}")
                     await sse_stream_writer.send({"event": event, "data": data})
 
@@ -114,6 +114,7 @@ async def handle_post(request: Request):
             return Response(status_code=202)
 
         await transport.handle_post_message(request.scope, request.receive, request._send)
+        logger.info(f"Handled post message for session {session_id} via transport")
         return Response(status_code=202)
     except Exception as e:
         logger.error(f"Error handling POST message: {str(e)}")