NixOS config for LLM-API-Key-Proxy

shell.nix

# NixOS config for LLM-API-Key-Proxy
#   https://gist.github.com/rstacruz/3a7e9b4a2cabdd9c21947eeb3e191f2c
#   https://github.com/Mirrowel/LLM-API-Key-Proxy
#
# First start:
#   git clone https://github.com/Mirrowel/LLM-API-Key-Proxy.git
#   cd LLM-API-Key-Proxy
#   # ...then copy this shell.nix in
#   nix shell --run "pip install -r requirements.txt"
#
# To run:
#   nix shell --run "python src/proxy_app/main.py"

{ pkgs ? import <nixpkgs> {} }:

pkgs.mkShell {
  buildInputs = with pkgs; [
    python311
    python311Packages.pip
    python311Packages.virtualenv
    stdenv.cc.cc.lib
    zlib
  ];

  shellHook = ''
    # Set up the library path for nix-ld and standard loading
    export LD_LIBRARY_PATH="${pkgs.lib.makeLibraryPath [ pkgs.stdenv.cc.cc.lib pkgs.zlib ]}:$LD_LIBRARY_PATH"
    
    # Optional: Automatically set up virtualenv
    if [ ! -d ".venv" ]; then
      echo "Creating virtual environment..."
      python -m venv .venv
    fi
    source .venv/bin/activate
  '';
}