wizardlink/linuxware
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fix(neovim): make ollama optional

Browse source
This commit is contained in:
Alexandre Cavalheiro 2025-01-16 21:03:29 -03:00
parent 925551ec26
commit 0ad475e6bf
Signed by: wizardlink
GPG key ID: A5767B54367CFBDF
2 changed files with 26 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
flake.nix
View file

@ -25,7 +25,11 @@
}; };
outputs = outputs =
{ home-manager, nixpkgs, ... }@inputs: {
home-manager,
nixpkgs,
...
}@inputs:
let let
system = "x86_64-linux"; system = "x86_64-linux";
pkgs = nixpkgs.legacyPackages.${system}; pkgs = nixpkgs.legacyPackages.${system};

18
modules/home-manager/programs/neovim/default.nix
View file

@ -6,14 +6,21 @@
}: }:
let let
inherit (lib) types mkOption; inherit (lib)
ollamaPackage = types
mkOption
mkIf
mkEnableOption
;
ollamaPackage = mkIf config.programs.neovim.ollama.enable (
if config.programs.neovim.ollama.type == "amd" then if config.programs.neovim.ollama.type == "amd" then
pkgs.ollama-rocm pkgs.ollama-rocm
else if config.programs.neovim.ollama.type == "nvidia" then else if config.programs.neovim.ollama.type == "nvidia" then
pkgs.ollama-cuda pkgs.ollama-cuda
else else
pkgs.ollama; pkgs.ollama
);
in in
{ {
options.programs.neovim = { options.programs.neovim = {
@ -33,7 +40,9 @@ in
}; };
}; };
ollama.type = mkOption { ollama = {
enable = mkEnableOption "enable";
type = mkOption {
default = "amd"; default = "amd";
description = "The type of ollama package to install, AMD GPU accelerated or NVIDIA GPU accelerated."; description = "The type of ollama package to install, AMD GPU accelerated or NVIDIA GPU accelerated.";
example = "amd"; example = "amd";
@ -43,6 +52,7 @@ in
]; ];
}; };
}; };
};
config = { config = {
programs.neovim = { programs.neovim = {