move ai stuffs into its own module

yuanw · Nov 1, 2023 · 0615364 · 0615364
1 parent 8577b0f
commit 0615364
Show file tree

Hide file tree

Showing 5 changed files with 22 additions and 4 deletions.
diff --git a/hosts/wk01174.nix b/hosts/wk01174.nix
@@ -35,6 +35,7 @@
     #   enable = true;
     #   supportLocalVirtualBuilder = true;
     # };
+    ai.enable = true;
     secrets.agenix = { enable = true; };
     brew = {
       enable = true;

diff --git a/modules/ai.nix b/modules/ai.nix
@@ -1,3 +1,21 @@
-{ ... }:
+{ config, lib, pkgs, ... }:
 
-{ }
+with lib;
+let
+  cfg = config.modules.ai;
+in
+{
+  options.modules.ai = { enable = mkEnableOption "ai"; };
+
+  config = mkIf cfg.enable {
+    home-manager.users.${config.my.username} = {
+      home.packages = [
+        (pkgs.python3.withPackages (ps:
+          with ps; [
+            huggingface-hub
+          ]))
+        ollama
+      ];
+    };
+  };
+}
diff --git a/modules/default.nix b/modules/default.nix
@@ -3,6 +3,7 @@
   flake.nixosModules = {
     common.imports = [
       ./agenix.nix
+      ./ai.nix
       ./common.nix
       ./browsers/firefox.nix
       ./dev/agda.nix

diff --git a/modules/dev/python.nix b/modules/dev/python.nix
@@ -25,7 +25,6 @@ in {
           # pylsp-mypy
           virtualenv
           virtualenvwrapper
-          huggingface-hub
           litellm
           tokenizers
         ]))

diff --git a/modules/packages.nix b/modules/packages.nix
@@ -69,7 +69,6 @@ with pkgs; [
   just
   pkg-config
   racket
-  ollama
   # open-interpreter
   # reiryoku-firmware
   # reiryoku-flash