depot/third_party/nixpkgs/pkgs/development/python-modules/llm/default.nix

{ lib
, buildPythonApplication
, buildPythonPackage
, fetchFromGitHub
, makeWrapper
, pytestCheckHook
, python3
, pythonOlder
, ruff
, setuptools
}:
let
  llm = buildPythonPackage rec {
    pname = "llm";
    version = "0.13.1";
    pyproject = true;

    disabled = pythonOlder "3.8";

    src = fetchFromGitHub {
      owner = "simonw";
      repo = "llm";
      rev = "refs/tags/${version}";
      hash = "sha256-Nq6pduzl8IK+nA3pctst/W4ux7+P6mBFTEHMF+vtBQw=";
    };

    patches = [
      ./001-disable-install-uninstall-commands.patch
    ];

    nativeBuildInputs = [
      setuptools
    ];

    propagatedBuildInputs = with python3.pkgs; [
      click-default-group
      numpy
      openai
      pip
      pluggy
      pydantic
      python-ulid
      pyyaml
      setuptools # for pkg_resources
      sqlite-migrate
      sqlite-utils
    ];

    nativeCheckInputs = with python3.pkgs; [
      cogapp
      numpy
      pytest-httpx
      pytestCheckHook
    ];

    doCheck = true;

    pytestFlagsArray = [
      "-svv"
      "tests/"
    ];

    pythonImportsCheck = [
      "llm"
    ];

    passthru = {inherit withPlugins;};

    meta = with lib; {
      homepage = "https://github.com/simonw/llm";
      description = "Access large language models from the command-line";
      changelog = "https://github.com/simonw/llm/releases/tag/${version}";
      license = licenses.asl20;
      mainProgram = "llm";
      maintainers = with maintainers; [aldoborrero];
    };
  };

  withPlugins = plugins: buildPythonApplication {
    inherit (llm) pname version;
    format = "other";

    disabled = pythonOlder "3.8";

    dontUnpack = true;
    dontBuild = true;
    doCheck = false;

    nativeBuildInputs = [
      makeWrapper
    ];

    installPhase = ''
      makeWrapper ${llm}/bin/llm $out/bin/llm \
        --prefix PYTHONPATH : "${llm}/${python3.sitePackages}:$PYTHONPATH"
      ln -sfv ${llm}/lib $out/lib
    '';

    propagatedBuildInputs = llm.propagatedBuildInputs ++ plugins;

    passthru = llm.passthru // {
      withPlugins = morePlugins: withPlugins (morePlugins ++ plugins);
    };

    inherit (llm) meta;
  };
in
  llm