how to update query state from websocket? - TanStack