from fastapi import APIRouter, WebSocket router = APIRouter() @router.websocket("/ws/workspace/{session_id}") async def websocket_endpoint(websocket: WebSocket, session_id: str): # Your WebSocket logic here pass