{ pkgs ? import sources.nixpkgs { }
, sources ? import ./npins
}:

with pkgs;

let

  python =
    python312.withPackages (pythonPkgs: [
    ]);

    pull = writeShellApplication {
      name = "pull";
      text = ''
        ollama pull gemma3
        ollama pull llama3.1:8b
      '';
    };

    serve = writeShellApplication {
      name = "serve";
      text = ''
        ollama serve
      '';
    };

in

mkShell {
  packages = [
    ollama
    pull
    python
    serve
  ];
  shellHook = ''
    if [ ! -d .venv ]; then
      python -m venv .venv
      source .venv/bin/activate
      pip install --upgrade uv
    else
      source .venv/bin/activate
    fi

    uv sync
  '';
}