Update server.py

server.py

@@ -106,11 +106,22 @@ async def handle_post(request: Request):
             }
             response_data = json.dumps(response)
             try:
-                await sse_writer.send({"event": "message", "data": response_data})
+                # Use a timeout to prevent hanging
+                async with anyio.fail_after(5):  # 5-second timeout
+                    await sse_writer.send({"event": "message", "data": response_data})
                 logger.info(f"Successfully sent tools/list response for session {session_id} via SSE")
+            except anyio.get_cancelled_exc_class():
+                logger.error(f"Timeout while sending SSE response for session {session_id}")
+                return Response(status_code=202)
             except Exception as e:
                 logger.error(f"Failed to send SSE response for session {session_id}: {str(e)}")
-                return Response(status_code=202)
+                # Fallback: Try sending directly through write_stream
+                try:
+                    await write_stream.send({"event": "message", "data": response_data})
+                    logger.info(f"Fallback: Successfully sent tools/list response for session {session_id} via write_stream")
+                except Exception as fallback_e:
+                    logger.error(f"Fallback failed for session {session_id}: {str(fallback_e)}")
+                    return Response(status_code=202)
             return Response(status_code=202)
 
         if not write_stream or not sse_writer: