{ lib, pkgs, config, inputs, ... }:

let
  nixpkgs-unstable = inputs.nixpkgs-unstable.
    legacyPackages.x86_64-linux;
  cfg = config.this.ollama;
in
{
  options = {
    this.ollama = {
      enable = lib.mkEnableOption "ollama";
      unstable = lib.mkEnableOption "unstable ollama package"
                 // { default = true; };
    };
  };

  config = lib.mkIf cfg.enable {
    services.ollama = {
      enable = true;
      package = lib.mkIf cfg.unstable nixpkgs-unstable.ollama;
      environmentVariables = {
        OLLAMA_CONTEXT_LENGTH = "8192";
      };
    };
  };
}