Passing Header When Using Websockets

In this issue, I see that Dask can operate on websockets: https://github.com/dask/distributed/pull/4396

When using websockets for Dask, is it possible to set the Sec-WebSocket-Protocol header in some way for auth purposes? I would like to use this as a way to pass a JWT (which will be checked by the auth proxy) before the websocket can connect with the scheduler.