update + ollama

src/install/mod.rs

@@ -10,6 +10,7 @@ use boot::setup_bootloader;
 use drives::{format_drives, mount_drives};
 use first_boot::{first_boot_values, genfstab};
 use kernel::setup_mkinitcpio;
+use ollama::setup_ollama;
 use security::{setup_secure_boot, setup_tpm_unlock};
 use ssh::setup_ssh;
 use user::setup_users;
@@ -24,6 +25,7 @@ pub mod security;
 pub mod ssh;
 pub mod user;
 pub mod zram;
+pub mod ollama;
 
 use crate::{
     config::InstallConfig,
@@ -118,6 +120,10 @@ pub fn install(conf: InstallConfig) {
         // TODO : Enable docker
     }
 
+    if let Some(ai) = conf.ai {
+        setup_ollama(&ai);
+    }
+
     setup_zram();
     setup_mkinitcpio(&conf.drive);
     setup_secure_boot();