Found an OhMyZsh theme that shows the current version of Ruby, and which Gemset is selected. To enable this theme, you need to configure it in ~/.zshrc file

ZSH_THEME="suvash"

And reload the config

source ~/.zshrc