Getting error in websocket

I'm new to fyers API, I installed API through pip install fyers-apiv2 and going through the documentation but I'm getting an error : ModuleNotFoundError: No module named 'fyers_api.Websocket'

when I tried from fyers_api.websocket import ws then it is working but I'm getting an error in the custom message function

pong received

Exception is custom_message() missing 1 required positional argument: 'msg'

Thanks