feat: more AI-related tools

users/modules/dev/ai/claude.nix

@@ -0,0 +1,18 @@
+{
+  config,
+  lib,
+  inputs,
+  system,
+  ...
+}:
+with lib; let
+  cfg = config.home.dev.ai.claude;
+in {
+  options.home.dev.ai.claude.enable = mkEnableOption "Enables Claude-related packages";
+  config = mkIf cfg.enable {
+    home.packages = [inputs.claude-desktop.packages.${system}.claude-desktop-with-fhs];
+    programs.claude-code = {
+      inherit (cfg) enable;
+    };
+  };
+}

users/modules/dev/ai/default.nix

@@ -0,0 +1,19 @@
+{
+  config,
+  lib,
+  ...
+}:
+with lib; let
+  cfg = config.home.dev.ai;
+in {
+  imports = [
+    ./ollama.nix
+    ./claude.nix
+  ];
+
+  options.home.dev.ai.enable = mkEnableOption "Enables AI features";
+  config.home.dev.ai = mkIf cfg.enable {
+    ollama.enable = mkDefault cfg.enable;
+    claude.enable = mkDefault cfg.enable;
+  };
+}

users/modules/dev/ai/ollama.nix

@@ -0,0 +1,26 @@
+{
+  config,
+  lib,
+  ...
+}:
+with lib; let
+  cfg = config.home.dev.ai.ollama;
+in {
+  options.home.dev.ai.ollama = {
+    enable = mkEnableOption "Enables Ollama";
+    gpu = mkOption {
+      type = types.nullOr (types.enum [false "rocm" "cuda"]);
+      example = "rocm";
+      default = null;
+      description = "Which type of GPU should be used for hardware acceleration";
+    };
+  };
+
+  config.services.ollama = mkIf cfg.enable {
+    inherit (cfg) enable;
+    acceleration = cfg.gpu;
+    environmentVariables = {
+      OLLAMA_CONTEXT_LENGTH = "8192";
+    };
+  };
+}