[zsh] Update git-in-prompt functionality