| 726 | |
| 727 | # Chat WebSocket connection manager |
| 728 | class ConnectionManager: |
| 729 | def __init__(self): |
| 730 | self.active_connections: List[WebSocket] = [] |
| 731 | |
| 732 | async def connect(self, websocket: WebSocket): |
| 733 | await websocket.accept() |
| 734 | self.active_connections.append(websocket) |
| 735 | |
| 736 | def disconnect(self, websocket: WebSocket): |
| 737 | self.active_connections.remove(websocket) |
| 738 | |
| 739 | async def send_message(self, message: str, websocket: WebSocket): |
| 740 | await websocket.send_text(message) |
| 741 | |
| 742 | manager = ConnectionManager() |
| 743 | |