Resolvido com a seguinte configuração.
{ pkgs ? import <nixpkgs> {} }:
with pkgs;
let py3s = pkgs.python3.buildEnv.override {
extraLibs = with pkgs.python3Packages; [ jedi pyyaml ];
};
in stdenv.mkDerivation {
name = "python3-shell";
buildInputs = [ py3s ];
shellHook = ''
export PS1="\[3[1;32m\][nix-shell:\w]\n$ \[3[0m\]"
'';
}
Ainda não entendi completamente o código, mas de alguma forma recebi o prompt de desejo e o jedi-vim
agora pode ver yaml