{ pkgs ? import <nixpkgs> {} }:

pkgs.mkShell {
  name = "ai-security-lab";

  buildInputs = with pkgs; [
    python311
    uv
    conda
    git
    stdenv.cc.cc.lib
  ];

  shellHook = ''
    export LD_LIBRARY_PATH="${pkgs.stdenv.cc.cc.lib}/lib:$LD_LIBRARY_PATH"
    export CONDA_ENVS_PATH="$PWD/.conda/envs"
    export CONDA_PKGS_DIRS="$PWD/.conda/pkgs"
    mkdir -p "$CONDA_ENVS_PATH" "$CONDA_PKGS_DIRS"
  '';
}