I'd go ahead and just remove this ChatClient, as it's only for testing.  If you have this included in your code, the WebSocketSession dependency would have to be included, which doesn't belong in the service tier.  Just a little code cleanup and remove the code that isn't needed.