possible to set timeout on websocket connection?

I need to make my software robust to loss in internet connection. Therefore I would like to set a shorter timeout, such as 2 seconds, to the websocket requests. Currently the timeout is very long, and this causes the whole program to halt while waiting. It is preferable to fail fast so I can handle it. 
I cannot see in the documentation how to alter the timeout.

Thank you