depot/third_party/nixpkgs/pkgs/by-name/pr/private-gpt/package.nix
Default email 159e378cbb Project import generated by Copybara.
GitOrigin-RevId: c04d5652cfa9742b1d519688f65d1bbccea9eb7e
2024-09-19 17:19:46 +03:00

19 lines
538 B
Nix

{ python3Packages
, makeBinaryWrapper
}:
python3Packages.toPythonApplication (python3Packages.private-gpt.overrideAttrs (oldAttrs: {
nativeBuildInputs = oldAttrs.nativeBuildInputs ++ [ makeBinaryWrapper ];
passthru = (oldAttrs.passthru or {}) // {
cl100k_base = {
inherit (python3Packages.private-gpt.cl100k_base) tiktoken;
};
};
postInstall = ''
makeWrapper ${python3Packages.python.interpreter} $out/bin/private-gpt \
--prefix PYTHONPATH : "$PYTHONPATH" \
--add-flags "-m private_gpt"
'';
}))