diff --git a/hosts/levi/ai.nix b/hosts/levi/ai.nix
new file mode 100644
index 0000000..8c95ba8
--- /dev/null
+++ b/hosts/levi/ai.nix
@@ -0,0 +1,10 @@
+{
+  services.ollama = {
+    enable = true;
+    acceleration = "cuda";
+  };
+  programs.bash.interactiveShellInit = ''
+    alias ai-code='ollama run qwen2.5-coder:14b'
+    alias ai-text='ollama run llama3.1:8b'
+  '';
+}
diff --git a/hosts/levi/configuration.nix b/hosts/levi/configuration.nix
index 84cc7d2..efc7b10 100644
--- a/hosts/levi/configuration.nix
+++ b/hosts/levi/configuration.nix
@@ -14,14 +14,6 @@ in
     }
   ];
 
-  # services.ollama = {
-  #   enable = true;
-  #   acceleration = "cuda";
-  # };
-  # programs.bash.interactiveShellInit = ''
-  #   alias ai='ollama run llama3.1:8b'
-  # '';
-
   sconfig = {
     gnome = true;
     desktop.enable = true;
diff --git a/hosts/levi/default.nix b/hosts/levi/default.nix
index b88de26..c1b47b8 100644
--- a/hosts/levi/default.nix
+++ b/hosts/levi/default.nix
@@ -1,6 +1,7 @@
 {
   system = "x86_64-linux";
   modules = [
+    ./ai.nix
     ./configuration.nix
     ./xorg.nix
   ];