{ inputs, config, ... }:
let
  hmModules = config.flake.modules.homeManager;
in
{
  flake.modules.nixos.ai = {
    home-manager.sharedModules = [ hmModules.ai ];

    nixpkgs.overlays = [ inputs.llm-agents.overlays.default ];

    nix.settings = {
      extra-substituters = [ "https://cache.numtide.com" ];
      extra-trusted-public-keys = [
        "niks3.numtide.com-1:DTx8wZduET09hRmMtKdQDxNNthLQETkc/yaX7M4qK0g="
      ];
    };
  };

  flake.modules.homeManager.ai =
    { pkgs, ... }:
    {
      programs.mcp = {
        enable = true;
        servers.nixos = {
          command = "${pkgs.uv}/bin/uvx";
          args = [ "mcp-nixos" ];
        };
      };

      home.packages = [
        pkgs.llm-agents.codex
      ];
    };
}