update + ollama

src/config.rs

@@ -13,6 +13,14 @@ pub struct InstallConfig {
     pub user: Vec<UserConfig>,
     /// SSH Configuration
     pub ssh: Option<SSHConfig>,
+    /// Ollama AI Config
+    pub ai: Option<OllamaConfig>
+}
+
+#[derive(Debug, Deserialize)]
+pub struct OllamaConfig {
+    pub models: Option<Vec<String>>,
+    pub gpu: bool
 }
 
 #[derive(Debug, Deserialize)]