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