starship has been excellent, and gives me the right context all the time. Like I'm running inside a nix shell, the python venv is enabled or the git branch.

Configuring it with home-manager was as simple as:

programs.starship.enable = true;