* Move imports to top for websocket_api * Move back an import because of circular dependency, add annotations