Remove username@computer from Agonster ZSH Theme

agnoster.zsh-theme

# If just logged on normal user, display a short name like 'Daniel' instead of full user account name like 'dmeechan'
# But if logged in as a different user (like on SSH), display the full username@machine
prompt_context() {
  if [[ "$USER" != "$DEFAULT_USER" || -n "$SSH_CLIENT" ]]; then
    if [[ "$USER" == "YOUR-USER-ACCOUNT-NAME" ]]; then
      USER="YOUR-DESIRED-NAME"
      prompt_segment black default "%(!.%{%F{yellow}%}.)$USER"
    else
      prompt_segment black default "%(!.%{%F{yellow}%}.)$USER@%m"
    fi
  fi
}

# EXAMPLE:
# prompt_context() {
#   if [[ "$USER" != "$DEFAULT_USER" || -n "$SSH_CLIENT" ]]; then
#     if [[ "$USER" == "dmeechan" ]]; then
#       USER="Daniel"
#       prompt_segment black default "%(!.%{%F{yellow}%}.)$USER"
#     else
#       prompt_segment black default "%(!.%{%F{yellow}%}.)$USER@%m"
#     fi
#   fi
# }