nix-config/users/modules/dev/ollama.nix

{
  config,
  lib,
  ...
}:
with lib; let
  cfg = config.home.dev.ollama;
in {
  options.home.dev.ollama = {
    enable = mkEnableOption "Enables Ollama";
    gpu = mkOption {
      type = types.nullOr types.enum ["none" "amd" "nvidia"];
      example = "amd";
      default = "none";
      description = "Which type of GPU should be used for hardware acceleration";
    };
  };

  config.services.ollama = mkIf cfg.enable {
    inherit (cfg) enable;
    environmentVariables = {
      OLLAMA_CONTEXT_LENGTH = "8192";
    };
  };
}