{
  pkgs,
  lib,
  config,
  ...
}:
with lib; let
  cfg = config.ollama;
in {
  options.ollama = {enable = mkEnableOption "is ollama host";};
  config = mkIf cfg.enable {
    services.ollama = {
      enable = true;
      package = pkgs.ollama-rocm;
      # Optional: preload models, see https://ollama.com/library
      loadModels = ["qwen2.5-coder:1.5b"];
    };
  };
}