No consigo que los avisos personalizados funcionen en zsh

Quiero establecer un prompt personalizado en zsh. Quiero que mi prompt muestre el contexto actual de k8s.

Probé esto:

kubectx()
{
  export PROMPT="$(kubectl config current-context) "
}

precmd_functions=[ 'kubectx' ]

y esto:

setopt PROMPT_SUBST
kubectx()
{
  echo "running..."
  CURRENT_CONTEXT="$(kubectl config current-context)"
}

precmd_functions=[ 'kubectx' ]

PROMPT="${CURRENT_CONTEXT} $"

Y también esto

export PROMPT="$(kubectl config current-context) "

Pero en ambos casos, obtengo un aviso que se fija en cuando .zshrc y no se actualiza cuando, por ejemplo, cambio los contextos de kube.

Intenté leer la documentación de precmd pero no pude encontrar qué hacer para habilitar esta personalización en zsh shell.

¿Alguien ha mostrado cómo script o programar esto?