style: format code with cargo fmt

Author: echobt

src/cortex-common/src/model_presets/mod.rs

@@ -18,7 +18,10 @@ pub use types::{ModelAlias, ModelPreset, ModelResolution};
 pub use constants::{DEFAULT_MODEL, DEFAULT_MODELS, DEFAULT_PROVIDER};
 
 // Re-export preset data and helpers
-pub use presets::{MODEL_PRESETS, get_model_preset, get_models_for_provider, validate_chutes_model, provider_allows_custom_models, CHUTES_ALLOWED_MODELS};
+pub use presets::{
+    CHUTES_ALLOWED_MODELS, MODEL_PRESETS, get_model_preset, get_models_for_provider,
+    provider_allows_custom_models, validate_chutes_model,
+};
 
 // Re-export alias data and helpers
 pub use aliases::{MODEL_ALIASES, list_model_aliases, resolve_model_alias};

src/cortex-common/src/model_presets/presets.rs

@@ -841,12 +841,12 @@ pub fn get_models_for_provider(provider: &str) -> Vec<&'static ModelPreset> {
 /// Returns Ok(()) if valid, Err with message if invalid.
 pub fn validate_chutes_model(model: &str) -> Result<(), String> {
     let model = model.trim();
-    
+
     // Check for empty model
     if model.is_empty() {
         return Err("Model name cannot be empty for Chutes provider".to_string());
     }
-    
+
     // Check suffix (case-insensitive)
     if !model.to_uppercase().ends_with("-TEE") {
         return Err(format!(
@@ -856,17 +856,20 @@ pub fn validate_chutes_model(model: &str) -> Result<(), String> {
             CHUTES_ALLOWED_MODELS.join(", ")
         ));
     }
-    
+
     // SECURITY: Also verify model is in the allowed whitelist
-    if !CHUTES_ALLOWED_MODELS.iter().any(|&allowed| allowed.eq_ignore_ascii_case(model)) {
+    if !CHUTES_ALLOWED_MODELS
+        .iter()
+        .any(|&allowed| allowed.eq_ignore_ascii_case(model))
+    {
         return Err(format!(
             "Model '{}' is not in the allowed Chutes models list. \
              Available models: {}",
             model,
             CHUTES_ALLOWED_MODELS.join(", ")
         ));
     }
-    
+
     Ok(())
 }
 
@@ -898,14 +901,14 @@ mod tests {
         // Not a TEE model
         assert!(validate_chutes_model("gpt-4").is_err());
         assert!(validate_chutes_model("claude-3").is_err());
-        
+
         // Has TEE suffix but NOT in whitelist
         assert!(validate_chutes_model("fake-model-TEE").is_err());
         assert!(validate_chutes_model("some-model-TEE").is_err());
-        
+
         // TEE in wrong position
         assert!(validate_chutes_model("model-TEE-v2").is_err());
-        
+
         // Empty string
         let result = validate_chutes_model("");
         assert!(result.is_err());
@@ -918,13 +921,13 @@ mod tests {
         assert!(!provider_allows_custom_models("chutes"));
         assert!(!provider_allows_custom_models("Chutes"));
         assert!(!provider_allows_custom_models("CHUTES"));
-        
+
         // Other providers allow custom models
         assert!(provider_allows_custom_models("cortex"));
         assert!(provider_allows_custom_models("openai"));
         assert!(provider_allows_custom_models("anthropic"));
     }
-    
+
     #[test]
     fn test_chutes_allowed_models_list() {
         // Verify the whitelist contains expected model

src/cortex-tui/src/providers/manager.rs

@@ -11,7 +11,7 @@ use cortex_engine::client::{
 
 use super::config::CortexConfig;
 use super::models::{ModelInfo, get_models_for_provider, get_popular_models};
-use cortex_common::model_presets::{validate_chutes_model, CHUTES_ALLOWED_MODELS};
+use cortex_common::model_presets::{CHUTES_ALLOWED_MODELS, validate_chutes_model};
 
 // ============================================================
 // PROVIDER MANAGER
@@ -227,7 +227,7 @@ impl ProviderManager {
     /// Sets the current provider.
     pub fn set_provider(&mut self, provider: &str) -> Result<()> {
         self.current_provider = provider.to_string();
-        
+
         // SECURITY: Validate current model when switching to Chutes provider
         if provider.eq_ignore_ascii_case("chutes") {
             if validate_chutes_model(&self.current_model).is_err() {
@@ -239,7 +239,7 @@ impl ProviderManager {
                 self.current_model = CHUTES_ALLOWED_MODELS[0].to_string();
             }
         }
-        
+
         self.client = None;
         Ok(())
     }
@@ -251,8 +251,7 @@ impl ProviderManager {
         // Validate model for Chutes provider (TEE-only security requirement)
         // Use case-insensitive comparison for provider check
         if self.current_provider.eq_ignore_ascii_case("chutes") {
-            validate_chutes_model(&resolved)
-                .map_err(|e| anyhow::anyhow!(e))?;
+            validate_chutes_model(&resolved).map_err(|e| anyhow::anyhow!(e))?;
         }
 
         self.current_model = resolved;
@@ -398,13 +397,12 @@ impl ProviderManager {
         if self.client.is_some() {
             return Ok(());
         }
-        
+
         // SECURITY: Defense-in-depth - validate TEE model before API calls
         if self.current_provider.eq_ignore_ascii_case("chutes") {
-            validate_chutes_model(&self.current_model)
-                .map_err(|e| anyhow::anyhow!("{}", e))?;
+            validate_chutes_model(&self.current_model).map_err(|e| anyhow::anyhow!("{}", e))?;
         }
-        
+
         let token = self.get_token()?;
         self.client = Some(create_client("cortex", &self.current_model, &token, None)?);
         Ok(())

src/cortex-tui/src/runner/event_loop/commands.rs

@@ -556,21 +556,22 @@ impl EventLoop {
                 } else {
                     Ok(())
                 };
-                
+
                 // Handle validation result after releasing the borrow
                 if let Err(e) = validation_result {
                     self.add_system_message(&format!("Cannot set model: {}", e));
                     return;
                 }
-                
+
                 // Only update state if validation passed
                 self.app_state.model = value.to_string();
                 self.update_session_model(value);
                 self.add_system_message(&format!("Model set to: {}", value));
             }
             "provider" => {
                 // Check validation result first, storing any error and new model
-                let validation_result: Result<Option<String>, String> = if let Some(pm) = &self.provider_manager
+                let validation_result: Result<Option<String>, String> = if let Some(pm) =
+                    &self.provider_manager
                     && let Ok(mut manager) = pm.try_write()
                 {
                     match manager.set_provider(value) {
@@ -583,7 +584,7 @@ impl EventLoop {
                 } else {
                     Ok(None)
                 };
-                
+
                 // Handle validation result after releasing the borrow
                 match validation_result {
                     Err(e) => {
@@ -596,7 +597,7 @@ impl EventLoop {
                     }
                     Ok(None) => {}
                 }
-                
+
                 self.app_state.provider = value.to_string();
                 self.add_system_message(&format!("Provider set to: {}", value));
             }

src/cortex-tui/src/runner/event_loop/modal.rs

@@ -43,7 +43,9 @@ impl EventLoop {
                                 && let Ok(mut manager) = pm.try_write()
                             {
                                 if let Err(e) = manager.set_model(&model_id) {
-                                    self.app_state.toasts.error(format!("Cannot use model: {}", e));
+                                    self.app_state
+                                        .toasts
+                                        .error(format!("Cannot use model: {}", e));
                                     self.app_state.close_modal();
                                     return Ok(true);
                                 }
@@ -229,7 +231,9 @@ impl EventLoop {
                 {
                     // Check result of set_model instead of ignoring it
                     if let Err(e) = manager.set_model(&model_id) {
-                        self.app_state.toasts.error(format!("Cannot use model: {}", e));
+                        self.app_state
+                            .toasts
+                            .error(format!("Cannot use model: {}", e));
                         return;
                     }
                     // Only update app state if validation passed
@@ -512,7 +516,9 @@ impl EventLoop {
                     && let Ok(mut manager) = pm.try_write()
                 {
                     if let Err(e) = manager.set_provider(&item_id) {
-                        self.app_state.toasts.error(format!("Cannot switch provider: {}", e));
+                        self.app_state
+                            .toasts
+                            .error(format!("Cannot switch provider: {}", e));
                         return false;
                     }
                     // Update model to reflect any changes made during provider switch
@@ -527,7 +533,9 @@ impl EventLoop {
                 {
                     // Check validation result
                     if let Err(e) = manager.set_model(&item_id) {
-                        self.app_state.toasts.error(format!("Cannot use model: {}", e));
+                        self.app_state
+                            .toasts
+                            .error(format!("Cannot use model: {}", e));
                         return false;
                     }
                     self.app_state.model = item_id.clone();