oizys/pkgs/llm/llm-with-plugins/default.nix

19 lines
350 B
Nix

{
pkgs,
...
}:
let
inherit (pkgs) python3Packages;
llm = python3Packages.callPackage ../llm { };
llm-claude-3 = python3Packages.callPackage ../llm-plugins/llm-claude-3 { };
pyWithLlm = (
pkgs.python3.withPackages (_: [
llm
llm-claude-3
])
);
in
pkgs.writeShellScriptBin "llm" ''
exec ${pyWithLlm}/bin/llm "$@"
''