Adding streaming API support with WebSocket