oizys/modules/llm/default.nix

25 lines
432 B
Nix
Raw Normal View History

2024-08-08 13:53:51 -05:00
{
pkgs,
config,
enabled,
mkOizysModule,
...
}:
let
inherit (pkgs) python3Packages;
2024-08-08 14:20:37 -05:00
llm-ollama = python3Packages.callPackage ./llm-plugins/llm-ollama { };
llm-claude3 = python3Packages.callPackage ./llm-plugins/llm-claude-3 { };
2024-08-08 13:53:51 -05:00
llm = (
pkgs.llm.withPlugins [
llm-ollama
llm-claude3
]
);
in
mkOizysModule config "llm" {
services.ollama = enabled;
environment.systemPackages = [ llm ];
}