Update server.py

server.py

@@ -107,7 +107,7 @@ async def handle_post(request: Request):
             response_data = json.dumps(response)
             try:
                 await sse_writer.send({"event": "message", "data": response_data})
-                logger.info(f"Sent tools/list response for session {session_id} via SSE")
+                logger.info(f"Successfully 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)
@@ -117,8 +117,11 @@ async def handle_post(request: Request):
             logger.error(f"No write_stream or SSE writer found for session_id: {session_id}")
             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")
+        try:
+            await transport.handle_post_message(request.scope, request.receive, request._send)
+            logger.info(f"Handled post message for session {session_id} via transport")
+        except Exception as e:
+            logger.error(f"Failed to handle post message via transport for session {session_id}: {str(e)}")
         return Response(status_code=202)
     except Exception as e:
         logger.error(f"Error handling POST message: {str(e)}")