reflex/semantic/

configure.rs

//! Interactive TUI configuration wizard for AI provider setup

use anyhow::{Context, Result};
use crossterm::{
    event::{self, Event, KeyCode, KeyEvent, KeyModifiers},
    execute,
    terminal::{EnterAlternateScreen, LeaveAlternateScreen, disable_raw_mode, enable_raw_mode},
};
use ratatui::{
    Frame, Terminal,
    backend::CrosstermBackend,
    layout::{Alignment, Constraint, Direction, Layout},
    style::{Color, Modifier, Style},
    widgets::{Block, Borders, List, ListItem, ListState, Paragraph, Wrap},
};
use std::collections::HashMap;
use std::io::{self, Stdout};

/// Available AI providers
const PROVIDERS: &[&str] = &["openai", "anthropic", "openrouter", "openai-compatible"];

// OpenAI and Anthropic model lists are fetched dynamically from each provider's
// /v1/models endpoint at wizard runtime (see fetch_openai_models_blocking,
// fetch_anthropic_models_blocking). These small fallback lists are only used
// when the live fetch fails (offline, key revoked, regional outage), so the
// wizard remains usable without a network connection.
const OPENAI_FALLBACK_MODELS: &[&str] = &[
    "gpt-5.1",
    "gpt-5.1-mini",
    "gpt-5",
    "gpt-5-mini",
    "gpt-4.1",
    "gpt-4.1-mini",
    "gpt-4o",
    "gpt-4o-mini",
];
const ANTHROPIC_FALLBACK_MODELS: &[&str] =
    &["claude-sonnet-4-5", "claude-haiku-4-5", "claude-sonnet-4"];
use crate::semantic::providers::openrouter::OpenRouterModel;

/// Sort strategies for OpenRouter provider routing
const OPENROUTER_SORT_STRATEGIES: &[(&str, &str)] = &[
    ("price", "Cheapest provider for the model"),
    ("latency", "Fastest response time (lowest latency)"),
    ("throughput", "Highest tokens per second"),
];

/// Wizard screen states
#[derive(Debug, Clone, PartialEq)]
enum WizardScreen {
    ProviderSelection,
    BaseUrlInput,
    ApiKeyInput,
    FetchingModels,
    ModelSelection,
    ModelTextInput,
    SortStrategySelection,
    ConnectivityTest,
    Result { success: bool, message: String },
}

/// Load existing API key for a provider from ~/.reflex/config.toml
fn load_existing_api_key(provider: &str) -> Option<String> {
    match crate::semantic::config::get_api_key(provider) {
        Ok(key) if !key.is_empty() => {
            log::debug!("Found existing API key for {}", provider);
            Some(key)
        }
        _ => {
            log::debug!("No existing API key found for {}", provider);
            None
        }
    }
}

/// Load existing base URL for the openai-compatible provider
fn load_existing_base_url() -> Option<String> {
    crate::semantic::config::get_provider_options("openai-compatible")
        .and_then(|opts| opts.get("base_url").cloned())
        .filter(|s| !s.is_empty())
}

/// Load existing model preference for the openai-compatible provider
fn load_existing_compatible_model() -> Option<String> {
    crate::semantic::config::get_user_model("openai-compatible")
}

/// Mask API key for display (show first 7 and last 4 characters)
fn mask_api_key(key: &str) -> String {
    if key.len() <= 11 {
        // Too short to mask meaningfully
        return "*".repeat(key.len());
    }

    let start = &key[..7];
    let end = &key[key.len() - 4..];
    format!("{}...{}", start, end)
}

/// Main configuration wizard state
pub struct ConfigWizard {
    screen: WizardScreen,
    selected_provider_idx: usize,
    api_key: String,
    api_key_cursor: usize,
    selected_model_idx: usize,
    selected_sort_idx: usize,
    error_message: Option<String>,
    existing_api_key: Option<String>,
    /// Dynamically fetched models (OpenRouter)
    fetched_models: Vec<OpenRouterModel>,
    /// Dynamically fetched plain model IDs (OpenAI, Anthropic)
    fetched_dynamic_models: Vec<String>,
    /// Current search/filter text for model selection
    model_filter: String,
    /// Base URL for openai-compatible endpoints
    base_url: String,
    base_url_cursor: usize,
    /// Free-text model name for openai-compatible (since we cannot enumerate)
    model_text: String,
    model_text_cursor: usize,
    /// Previously-saved base URL (used to pre-populate the wizard on re-run)
    existing_base_url: Option<String>,
    /// Previously-saved openai-compatible model name
    existing_compatible_model: Option<String>,
}

impl ConfigWizard {
    pub fn new() -> Self {
        Self {
            screen: WizardScreen::ProviderSelection,
            selected_provider_idx: 0,
            api_key: String::new(),
            api_key_cursor: 0,
            selected_model_idx: 0,
            selected_sort_idx: 0,
            error_message: None,
            existing_api_key: None,
            fetched_models: Vec::new(),
            fetched_dynamic_models: Vec::new(),
            model_filter: String::new(),
            base_url: String::new(),
            base_url_cursor: 0,
            model_text: String::new(),
            model_text_cursor: 0,
            existing_base_url: None,
            existing_compatible_model: None,
        }
    }

    /// Get the currently selected provider
    fn selected_provider(&self) -> &str {
        PROVIDERS[self.selected_provider_idx]
    }

    /// True for providers whose model list supports a typed text filter.
    /// OpenAI/Anthropic/OpenRouter are dynamically fetched and may be long;
    /// type-to-filter narrows the displayed list.
    fn supports_filter(&self) -> bool {
        matches!(
            self.selected_provider(),
            "openrouter" | "openai" | "anthropic"
        )
    }

    /// Get available static models for the current provider.
    ///
    /// Returns empty for every provider since openai/anthropic/openrouter all
    /// fetch dynamically and openai-compatible uses free-text input. Kept as a
    /// hook in case a future provider ships with a static catalog.
    fn static_models(&self) -> &'static [&'static str] {
        &[]
    }

    /// Get filtered model IDs for display (applies search filter for dynamic providers).
    fn filtered_model_ids(&self) -> Vec<String> {
        let filter = self.model_filter.to_lowercase();
        match self.selected_provider() {
            "openrouter" => self
                .fetched_models
                .iter()
                .filter(|m| {
                    if filter.is_empty() {
                        return true;
                    }
                    m.id.to_lowercase().contains(&filter) || m.name.to_lowercase().contains(&filter)
                })
                .map(|m| m.id.clone())
                .collect(),
            "openai" | "anthropic" => self
                .fetched_dynamic_models
                .iter()
                .filter(|id| filter.is_empty() || id.to_lowercase().contains(&filter))
                .cloned()
                .collect(),
            _ => self.static_models().iter().map(|s| s.to_string()).collect(),
        }
    }

    /// Get the currently selected sort strategy (OpenRouter only)
    fn selected_sort(&self) -> &str {
        OPENROUTER_SORT_STRATEGIES[self.selected_sort_idx].0
    }

    /// Get the currently selected model
    fn selected_model(&self) -> String {
        let models = self.filtered_model_ids();
        if self.selected_model_idx < models.len() {
            models[self.selected_model_idx].clone()
        } else if !models.is_empty() {
            models[0].clone()
        } else {
            String::new()
        }
    }

    /// Get the OpenRouterModel info for a model ID in the filtered list
    fn filtered_openrouter_model(&self, idx: usize) -> Option<&OpenRouterModel> {
        let filter = self.model_filter.to_lowercase();
        self.fetched_models
            .iter()
            .filter(|m| {
                if filter.is_empty() {
                    return true;
                }
                m.id.to_lowercase().contains(&filter) || m.name.to_lowercase().contains(&filter)
            })
            .nth(idx)
    }

    /// Handle keyboard input based on current screen
    fn handle_key(&mut self, key: KeyEvent) -> Result<bool> {
        // Handle Ctrl+C globally to exit wizard
        if key.code == KeyCode::Char('c') && key.modifiers.contains(KeyModifiers::CONTROL) {
            return Ok(true);
        }

        match &self.screen {
            WizardScreen::ProviderSelection => self.handle_provider_selection_key(key),
            WizardScreen::BaseUrlInput => self.handle_base_url_input_key(key),
            WizardScreen::ApiKeyInput => self.handle_api_key_input_key(key),
            WizardScreen::FetchingModels => Ok(false), // No input during fetch
            WizardScreen::ModelSelection => self.handle_model_selection_key(key),
            WizardScreen::ModelTextInput => self.handle_model_text_input_key(key),
            WizardScreen::SortStrategySelection => self.handle_sort_strategy_key(key),
            WizardScreen::ConnectivityTest => Ok(false), // No input during test
            WizardScreen::Result { .. } => {
                // Any key exits on result screen
                if key.code == KeyCode::Enter || key.code == KeyCode::Char('q') {
                    return Ok(true);
                }
                Ok(false)
            }
        }
    }

    /// Handle keys for provider selection screen
    fn handle_provider_selection_key(&mut self, key: KeyEvent) -> Result<bool> {
        match key.code {
            KeyCode::Up | KeyCode::Char('k') => {
                if self.selected_provider_idx > 0 {
                    self.selected_provider_idx -= 1;
                }
            }
            KeyCode::Down | KeyCode::Char('j') => {
                if self.selected_provider_idx < PROVIDERS.len() - 1 {
                    self.selected_provider_idx += 1;
                }
            }
            KeyCode::Enter => {
                // Check if API key already exists for this provider
                self.existing_api_key = load_existing_api_key(self.selected_provider());

                if self.selected_provider() == "openai-compatible" {
                    // Pre-populate base URL and model from any prior config
                    self.existing_base_url = load_existing_base_url();
                    self.existing_compatible_model = load_existing_compatible_model();
                    self.base_url = self.existing_base_url.clone().unwrap_or_default();
                    self.base_url_cursor = self.base_url.len();
                    self.model_text = self.existing_compatible_model.clone().unwrap_or_default();
                    self.model_text_cursor = self.model_text.len();
                    self.error_message = None;
                    self.screen = WizardScreen::BaseUrlInput;
                } else {
                    // Move to API key input for the standard providers
                    self.screen = WizardScreen::ApiKeyInput;
                    self.api_key.clear();
                    self.api_key_cursor = 0;
                }
            }
            KeyCode::Esc | KeyCode::Char('q') => {
                return Ok(true); // Exit wizard
            }
            _ => {}
        }
        Ok(false)
    }

    /// Handle keys for base URL input screen (openai-compatible only)
    fn handle_base_url_input_key(&mut self, key: KeyEvent) -> Result<bool> {
        match key.code {
            KeyCode::Char(c) if !key.modifiers.contains(KeyModifiers::CONTROL) => {
                self.base_url.insert(self.base_url_cursor, c);
                self.base_url_cursor += 1;
            }
            KeyCode::Backspace => {
                if self.base_url_cursor > 0 {
                    self.base_url_cursor -= 1;
                    self.base_url.remove(self.base_url_cursor);
                }
            }
            KeyCode::Delete => {
                if self.base_url_cursor < self.base_url.len() {
                    self.base_url.remove(self.base_url_cursor);
                }
            }
            KeyCode::Left => {
                if self.base_url_cursor > 0 {
                    self.base_url_cursor -= 1;
                }
            }
            KeyCode::Right => {
                if self.base_url_cursor < self.base_url.len() {
                    self.base_url_cursor += 1;
                }
            }
            KeyCode::Home => {
                self.base_url_cursor = 0;
            }
            KeyCode::End => {
                self.base_url_cursor = self.base_url.len();
            }
            KeyCode::Enter => {
                let trimmed = self.base_url.trim().trim_end_matches('/');
                if trimmed.is_empty() {
                    self.error_message = Some("Base URL cannot be empty".to_string());
                } else if !trimmed.starts_with("http://") && !trimmed.starts_with("https://") {
                    self.error_message =
                        Some("Base URL must start with http:// or https://".to_string());
                } else {
                    self.base_url = trimmed.to_string();
                    self.base_url_cursor = self.base_url.len();
                    self.error_message = None;
                    self.screen = WizardScreen::ApiKeyInput;
                    self.api_key.clear();
                    self.api_key_cursor = 0;
                }
            }
            KeyCode::Esc => {
                self.error_message = None;
                self.screen = WizardScreen::ProviderSelection;
            }
            _ => {}
        }
        Ok(false)
    }

    /// Handle keys for API key input screen
    fn handle_api_key_input_key(&mut self, key: KeyEvent) -> Result<bool> {
        match key.code {
            KeyCode::Char(c) if !key.modifiers.contains(KeyModifiers::CONTROL) => {
                self.api_key.insert(self.api_key_cursor, c);
                self.api_key_cursor += 1;
            }
            KeyCode::Backspace => {
                if self.api_key_cursor > 0 {
                    self.api_key_cursor -= 1;
                    self.api_key.remove(self.api_key_cursor);
                }
            }
            KeyCode::Delete => {
                if self.api_key_cursor < self.api_key.len() {
                    self.api_key.remove(self.api_key_cursor);
                }
            }
            KeyCode::Left => {
                if self.api_key_cursor > 0 {
                    self.api_key_cursor -= 1;
                }
            }
            KeyCode::Right => {
                if self.api_key_cursor < self.api_key.len() {
                    self.api_key_cursor += 1;
                }
            }
            KeyCode::Home => {
                self.api_key_cursor = 0;
            }
            KeyCode::End => {
                self.api_key_cursor = self.api_key.len();
            }
            KeyCode::Enter => {
                let provider = self.selected_provider();
                let is_compatible = provider == "openai-compatible";

                // Determine the next screen for the chosen provider
                let next_screen = match provider {
                    "openrouter" | "openai" | "anthropic" => WizardScreen::FetchingModels,
                    "openai-compatible" => WizardScreen::ModelTextInput,
                    _ => WizardScreen::ModelSelection,
                };

                if self.api_key.is_empty() {
                    if let Some(ref existing_key) = self.existing_api_key {
                        log::debug!("Keeping existing API key for {}", provider);
                        self.api_key = existing_key.clone();
                        self.error_message = None;
                        self.selected_model_idx = 0;
                        self.model_filter.clear();
                        self.screen = next_screen;
                    } else if is_compatible {
                        // Local servers (LMStudio, Ollama, llama.cpp) often don't need a key
                        log::debug!("Proceeding without API key for openai-compatible");
                        self.error_message = None;
                        self.selected_model_idx = 0;
                        self.model_filter.clear();
                        self.screen = next_screen;
                    } else {
                        self.error_message = Some("API key cannot be empty".to_string());
                    }
                } else {
                    self.error_message = None;
                    self.selected_model_idx = 0;
                    self.model_filter.clear();
                    self.screen = next_screen;
                }
            }
            KeyCode::Esc => {
                // openai-compatible has BaseUrlInput as the previous screen
                if self.selected_provider() == "openai-compatible" {
                    self.screen = WizardScreen::BaseUrlInput;
                } else {
                    self.screen = WizardScreen::ProviderSelection;
                }
            }
            _ => {}
        }
        Ok(false)
    }

    /// Handle keys for model selection screen
    fn handle_model_selection_key(&mut self, key: KeyEvent) -> Result<bool> {
        let is_openrouter = self.selected_provider() == "openrouter";
        let supports_filter = self.supports_filter();
        let model_count = self.filtered_model_ids().len();

        match key.code {
            KeyCode::Up => {
                if self.selected_model_idx > 0 {
                    self.selected_model_idx -= 1;
                }
            }
            KeyCode::Down => {
                if model_count > 0 && self.selected_model_idx < model_count - 1 {
                    self.selected_model_idx += 1;
                }
            }
            // j/k vim navigation only works when typing-to-filter is disabled,
            // otherwise those characters would always be swallowed as filter input.
            KeyCode::Char('k') if !supports_filter => {
                if self.selected_model_idx > 0 {
                    self.selected_model_idx -= 1;
                }
            }
            KeyCode::Char('j') if !supports_filter => {
                if model_count > 0 && self.selected_model_idx < model_count - 1 {
                    self.selected_model_idx += 1;
                }
            }
            KeyCode::Char(c)
                if supports_filter && !key.modifiers.contains(KeyModifiers::CONTROL) =>
            {
                self.model_filter.push(c);
                self.selected_model_idx = 0;
            }
            KeyCode::Backspace if supports_filter => {
                self.model_filter.pop();
                self.selected_model_idx = 0;
            }
            KeyCode::Enter => {
                if model_count == 0 {
                    // No models to select
                    return Ok(false);
                }
                if is_openrouter {
                    self.selected_sort_idx = 0;
                    self.screen = WizardScreen::SortStrategySelection;
                } else {
                    self.screen = WizardScreen::ConnectivityTest;
                }
            }
            KeyCode::Esc => {
                self.model_filter.clear();
                self.screen = WizardScreen::ApiKeyInput;
            }
            _ => {}
        }
        Ok(false)
    }

    /// Handle keys for free-text model input (openai-compatible only)
    fn handle_model_text_input_key(&mut self, key: KeyEvent) -> Result<bool> {
        match key.code {
            KeyCode::Char(c) if !key.modifiers.contains(KeyModifiers::CONTROL) => {
                self.model_text.insert(self.model_text_cursor, c);
                self.model_text_cursor += 1;
            }
            KeyCode::Backspace => {
                if self.model_text_cursor > 0 {
                    self.model_text_cursor -= 1;
                    self.model_text.remove(self.model_text_cursor);
                }
            }
            KeyCode::Delete => {
                if self.model_text_cursor < self.model_text.len() {
                    self.model_text.remove(self.model_text_cursor);
                }
            }
            KeyCode::Left => {
                if self.model_text_cursor > 0 {
                    self.model_text_cursor -= 1;
                }
            }
            KeyCode::Right => {
                if self.model_text_cursor < self.model_text.len() {
                    self.model_text_cursor += 1;
                }
            }
            KeyCode::Home => {
                self.model_text_cursor = 0;
            }
            KeyCode::End => {
                self.model_text_cursor = self.model_text.len();
            }
            KeyCode::Enter => {
                if self.model_text.trim().is_empty() {
                    self.error_message = Some("Model name cannot be empty".to_string());
                } else {
                    self.error_message = None;
                    self.screen = WizardScreen::ConnectivityTest;
                }
            }
            KeyCode::Esc => {
                self.error_message = None;
                self.screen = WizardScreen::ApiKeyInput;
            }
            _ => {}
        }
        Ok(false)
    }

    /// Handle keys for sort strategy selection screen (OpenRouter only)
    fn handle_sort_strategy_key(&mut self, key: KeyEvent) -> Result<bool> {
        match key.code {
            KeyCode::Up | KeyCode::Char('k') => {
                if self.selected_sort_idx > 0 {
                    self.selected_sort_idx -= 1;
                }
            }
            KeyCode::Down | KeyCode::Char('j') => {
                if self.selected_sort_idx < OPENROUTER_SORT_STRATEGIES.len() - 1 {
                    self.selected_sort_idx += 1;
                }
            }
            KeyCode::Enter => {
                self.screen = WizardScreen::ConnectivityTest;
            }
            KeyCode::Esc => {
                // Go back to model selection
                self.screen = WizardScreen::ModelSelection;
            }
            _ => {}
        }
        Ok(false)
    }

    /// Render the current screen
    fn render(&mut self, frame: &mut Frame) {
        // Clone screen to avoid borrow conflict with &mut self render methods
        let screen = self.screen.clone();
        match &screen {
            WizardScreen::ProviderSelection => self.render_provider_selection(frame),
            WizardScreen::BaseUrlInput => self.render_base_url_input(frame),
            WizardScreen::ApiKeyInput => self.render_api_key_input(frame),
            WizardScreen::FetchingModels => self.render_fetching_models(frame),
            WizardScreen::ModelSelection => self.render_model_selection(frame),
            WizardScreen::ModelTextInput => self.render_model_text_input(frame),
            WizardScreen::SortStrategySelection => self.render_sort_strategy_selection(frame),
            WizardScreen::ConnectivityTest => self.render_connectivity_test(frame),
            WizardScreen::Result { success, message } => {
                self.render_result(frame, *success, message)
            }
        }
    }

    /// Render provider selection screen
    fn render_provider_selection(&mut self, frame: &mut Frame) {
        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .margin(2)
            .constraints([
                Constraint::Length(3),
                Constraint::Min(0),
                Constraint::Length(3),
            ])
            .split(frame.area());

        // Title
        let title = Paragraph::new("Reflex AI Configuration Wizard")
            .style(
                Style::default()
                    .fg(Color::Cyan)
                    .add_modifier(Modifier::BOLD),
            )
            .alignment(Alignment::Center)
            .block(Block::default().borders(Borders::ALL));
        frame.render_widget(title, chunks[0]);

        // Provider list
        let providers: Vec<ListItem> = PROVIDERS
            .iter()
            .map(|provider| {
                let provider_display = match *provider {
                    "openrouter" => format!("{} (200+ models)", provider),
                    _ => provider.to_string(),
                };

                ListItem::new(provider_display)
            })
            .collect();

        let list = List::new(providers)
            .block(Block::default().borders(Borders::ALL).title(
                "Select AI Provider (↑/↓ to navigate, Enter to select, Esc/q/Ctrl+C to quit)",
            ))
            .highlight_style(
                Style::default()
                    .fg(Color::Yellow)
                    .add_modifier(Modifier::BOLD),
            )
            .highlight_symbol("> ");

        let mut list_state = ListState::default().with_selected(Some(self.selected_provider_idx));
        frame.render_stateful_widget(list, chunks[1], &mut list_state);

        // Help text
        let help = Paragraph::new(
            "Use arrow keys or j/k to navigate, Enter to select, Esc/q/Ctrl+C to quit",
        )
        .style(Style::default().fg(Color::DarkGray))
        .alignment(Alignment::Center);
        frame.render_widget(help, chunks[2]);
    }

    /// Render API key input screen
    fn render_api_key_input(&mut self, frame: &mut Frame) {
        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .margin(2)
            .constraints([
                Constraint::Length(3),
                Constraint::Length(5),
                Constraint::Min(0),
                Constraint::Length(3),
            ])
            .split(frame.area());

        // Title
        let title = Paragraph::new(format!("Configure {} API Key", self.selected_provider()))
            .style(
                Style::default()
                    .fg(Color::Cyan)
                    .add_modifier(Modifier::BOLD),
            )
            .alignment(Alignment::Center)
            .block(Block::default().borders(Borders::ALL));
        frame.render_widget(title, chunks[0]);

        // API key input (masked)
        let masked_key = "*".repeat(self.api_key.len());
        let input_text = if self.api_key_cursor < masked_key.len() {
            format!(
                "{}█{}",
                &masked_key[..self.api_key_cursor],
                &masked_key[self.api_key_cursor..]
            )
        } else {
            format!("{}█", masked_key)
        };

        let input = Paragraph::new(input_text)
            .style(Style::default().fg(Color::Yellow))
            .block(
                Block::default()
                    .borders(Borders::ALL)
                    .title(format!("Enter API Key for {}", self.selected_provider())),
            );
        frame.render_widget(input, chunks[1]);

        // Error message or instructions
        let message_widget = if let Some(ref error) = self.error_message {
            Paragraph::new(error.as_str())
                .style(Style::default().fg(Color::Red))
                .alignment(Alignment::Center)
        } else if let Some(ref existing_key) = self.existing_api_key {
            // Show masked existing key
            let masked = mask_api_key(existing_key);
            Paragraph::new(format!(
                "Current API key: {}\n\
                Press Enter to keep existing key, or type a new key to replace it\n\
                Your API key will be securely stored in ~/.reflex/config.toml",
                masked
            ))
            .style(Style::default().fg(Color::Yellow))
            .alignment(Alignment::Center)
        } else {
            Paragraph::new("Your API key will be securely stored in ~/.reflex/config.toml")
                .style(Style::default().fg(Color::Green))
                .alignment(Alignment::Center)
        };
        frame.render_widget(message_widget, chunks[2]);

        // Help text
        let help = Paragraph::new("Enter to continue, Esc to go back, Ctrl+C to quit")
            .style(Style::default().fg(Color::DarkGray))
            .alignment(Alignment::Center);
        frame.render_widget(help, chunks[3]);
    }

    /// Render model selection screen
    fn render_model_selection(&mut self, frame: &mut Frame) {
        let is_openrouter = self.selected_provider() == "openrouter";
        let supports_filter = self.supports_filter();
        let filtered = self.filtered_model_ids();
        let model_count = filtered.len();

        let constraints = if is_openrouter {
            vec![
                Constraint::Length(3), // Title
                Constraint::Length(3), // Filter input
                Constraint::Min(0),    // Model list
                Constraint::Length(3), // Help text
            ]
        } else {
            vec![
                Constraint::Length(3), // Title
                Constraint::Min(0),    // Model list
                Constraint::Length(3), // Help text
            ]
        };

        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .margin(2)
            .constraints(constraints)
            .split(frame.area());

        // Title
        let title_text = if is_openrouter {
            format!(
                "Select Model for {} ({} models)",
                self.selected_provider(),
                model_count
            )
        } else {
            format!("Select Model for {}", self.selected_provider())
        };
        let title = Paragraph::new(title_text)
            .style(
                Style::default()
                    .fg(Color::Cyan)
                    .add_modifier(Modifier::BOLD),
            )
            .alignment(Alignment::Center)
            .block(Block::default().borders(Borders::ALL));
        frame.render_widget(title, chunks[0]);

        // Filter input (OpenRouter only)
        let (list_chunk, help_chunk) = if is_openrouter {
            let filter_text = format!("{}█", self.model_filter);
            let filter_input = Paragraph::new(filter_text)
                .style(Style::default().fg(Color::Yellow))
                .block(
                    Block::default()
                        .borders(Borders::ALL)
                        .title("Filter (type to search)"),
                );
            frame.render_widget(filter_input, chunks[1]);
            (chunks[2], chunks[3])
        } else {
            (chunks[1], chunks[2])
        };

        // Model list
        if model_count == 0 && supports_filter {
            let empty_msg = Paragraph::new("No models match filter")
                .style(Style::default().fg(Color::DarkGray))
                .alignment(Alignment::Center)
                .block(Block::default().borders(Borders::ALL).title("Models"));
            frame.render_widget(empty_msg, list_chunk);
        } else {
            let model_items: Vec<ListItem> = filtered
                .iter()
                .enumerate()
                .map(|(idx, model_id)| {
                    let model_display = if is_openrouter {
                        if let Some(m) = self.filtered_openrouter_model(idx) {
                            format!(
                                "{}  ${:.2} / ${:.2} per 1M tokens",
                                model_id, m.prompt_price, m.completion_price
                            )
                        } else {
                            model_id.to_string()
                        }
                    } else if idx == 0 {
                        format!("{} (recommended)", model_id)
                    } else {
                        model_id.to_string()
                    };

                    ListItem::new(model_display)
                })
                .collect();

            let list_title = if supports_filter {
                "Models (↑/↓ to navigate, type to filter, Enter to select, Esc to go back)"
            } else {
                "Select Model (↑/↓ to navigate, Enter to select, Esc to go back, Ctrl+C to quit)"
            };
            let list = List::new(model_items)
                .block(Block::default().borders(Borders::ALL).title(list_title))
                .highlight_style(
                    Style::default()
                        .fg(Color::Yellow)
                        .add_modifier(Modifier::BOLD),
                )
                .highlight_symbol("> ");

            let mut list_state = ListState::default().with_selected(Some(self.selected_model_idx));
            frame.render_stateful_widget(list, list_chunk, &mut list_state);
        }

        // Help text
        let help_text = if supports_filter {
            "Type to filter, ↑/↓ to navigate, Enter to select, Esc to go back, Ctrl+C to quit"
        } else {
            "Use arrow keys or j/k to navigate, Enter to select, Esc to go back, Ctrl+C to quit"
        };
        let help = Paragraph::new(help_text)
            .style(Style::default().fg(Color::DarkGray))
            .alignment(Alignment::Center);
        frame.render_widget(help, help_chunk);
    }

    /// Render base URL input screen (openai-compatible only)
    fn render_base_url_input(&mut self, frame: &mut Frame) {
        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .margin(2)
            .constraints([
                Constraint::Length(3),
                Constraint::Length(3),
                Constraint::Min(0),
                Constraint::Length(3),
            ])
            .split(frame.area());

        let title = Paragraph::new("Configure OpenAI-Compatible Endpoint")
            .style(
                Style::default()
                    .fg(Color::Cyan)
                    .add_modifier(Modifier::BOLD),
            )
            .alignment(Alignment::Center)
            .block(Block::default().borders(Borders::ALL));
        frame.render_widget(title, chunks[0]);

        let input_text = if self.base_url_cursor < self.base_url.len() {
            format!(
                "{}█{}",
                &self.base_url[..self.base_url_cursor],
                &self.base_url[self.base_url_cursor..]
            )
        } else {
            format!("{}█", self.base_url)
        };

        let input = Paragraph::new(input_text)
            .style(Style::default().fg(Color::Yellow))
            .block(
                Block::default()
                    .borders(Borders::ALL)
                    .title("Base URL (e.g. http://localhost:1234/v1)"),
            );
        frame.render_widget(input, chunks[1]);

        let message_widget = if let Some(ref error) = self.error_message {
            Paragraph::new(error.as_str())
                .style(Style::default().fg(Color::Red))
                .alignment(Alignment::Center)
        } else if let Some(ref existing) = self.existing_base_url {
            Paragraph::new(format!(
                "Current base URL: {}\n\
                Examples: LMStudio http://localhost:1234/v1 · Ollama http://localhost:11434/v1\n\
                Press Enter to continue.",
                existing
            ))
            .style(Style::default().fg(Color::Yellow))
            .alignment(Alignment::Center)
            .wrap(Wrap { trim: true })
        } else {
            Paragraph::new(
                "Enter the base URL of your OpenAI-compatible endpoint.\n\
                Examples: LMStudio http://localhost:1234/v1 · Ollama http://localhost:11434/v1\n\
                The /chat/completions path will be appended automatically.",
            )
            .style(Style::default().fg(Color::Green))
            .alignment(Alignment::Center)
            .wrap(Wrap { trim: true })
        };
        frame.render_widget(message_widget, chunks[2]);

        let help = Paragraph::new("Enter to continue, Esc to go back, Ctrl+C to quit")
            .style(Style::default().fg(Color::DarkGray))
            .alignment(Alignment::Center);
        frame.render_widget(help, chunks[3]);
    }

    /// Render free-text model input screen (openai-compatible only)
    fn render_model_text_input(&mut self, frame: &mut Frame) {
        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .margin(2)
            .constraints([
                Constraint::Length(3),
                Constraint::Length(3),
                Constraint::Min(0),
                Constraint::Length(3),
            ])
            .split(frame.area());

        let title = Paragraph::new("Specify Model Name")
            .style(
                Style::default()
                    .fg(Color::Cyan)
                    .add_modifier(Modifier::BOLD),
            )
            .alignment(Alignment::Center)
            .block(Block::default().borders(Borders::ALL));
        frame.render_widget(title, chunks[0]);

        let input_text = if self.model_text_cursor < self.model_text.len() {
            format!(
                "{}█{}",
                &self.model_text[..self.model_text_cursor],
                &self.model_text[self.model_text_cursor..]
            )
        } else {
            format!("{}█", self.model_text)
        };

        let input = Paragraph::new(input_text)
            .style(Style::default().fg(Color::Yellow))
            .block(
                Block::default()
                    .borders(Borders::ALL)
                    .title("Model name (as it appears on your endpoint)"),
            );
        frame.render_widget(input, chunks[1]);

        let message_widget = if let Some(ref error) = self.error_message {
            Paragraph::new(error.as_str())
                .style(Style::default().fg(Color::Red))
                .alignment(Alignment::Center)
        } else if let Some(ref existing) = self.existing_compatible_model {
            Paragraph::new(format!(
                "Current model: {}\n\
                Type the exact model identifier loaded on your server.",
                existing
            ))
            .style(Style::default().fg(Color::Yellow))
            .alignment(Alignment::Center)
            .wrap(Wrap { trim: true })
        } else {
            Paragraph::new(
                "Enter the model name your server hosts.\n\
                Examples: qwen2.5-coder-32b-instruct, llama-3.1-8b-instruct, mistral-7b",
            )
            .style(Style::default().fg(Color::Green))
            .alignment(Alignment::Center)
            .wrap(Wrap { trim: true })
        };
        frame.render_widget(message_widget, chunks[2]);

        let help = Paragraph::new("Enter to test connection, Esc to go back, Ctrl+C to quit")
            .style(Style::default().fg(Color::DarkGray))
            .alignment(Alignment::Center);
        frame.render_widget(help, chunks[3]);
    }

    /// Render fetching models loading screen
    fn render_fetching_models(&mut self, frame: &mut Frame) {
        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .margin(2)
            .constraints([Constraint::Length(3), Constraint::Min(0)])
            .split(frame.area());

        // Title
        let title = Paragraph::new("Fetching Available Models...")
            .style(
                Style::default()
                    .fg(Color::Cyan)
                    .add_modifier(Modifier::BOLD),
            )
            .alignment(Alignment::Center)
            .block(Block::default().borders(Borders::ALL));
        frame.render_widget(title, chunks[0]);

        // Loading message — name the actual provider being queried
        let provider_label = match self.selected_provider() {
            "openrouter" => "OpenRouter",
            "openai" => "OpenAI",
            "anthropic" => "Anthropic",
            other => other,
        };
        let body = format!(
            "Loading models from {}...\n\nPlease wait...",
            provider_label
        );
        let message = Paragraph::new(body)
            .style(Style::default().fg(Color::Yellow))
            .alignment(Alignment::Center)
            .wrap(Wrap { trim: true });
        frame.render_widget(message, chunks[1]);
    }

    /// Render sort strategy selection screen (OpenRouter only)
    fn render_sort_strategy_selection(&mut self, frame: &mut Frame) {
        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .margin(2)
            .constraints([
                Constraint::Length(3),
                Constraint::Min(0),
                Constraint::Length(3),
            ])
            .split(frame.area());

        // Title
        let title = Paragraph::new("Select Provider Sort Strategy (OpenRouter)")
            .style(
                Style::default()
                    .fg(Color::Cyan)
                    .add_modifier(Modifier::BOLD),
            )
            .alignment(Alignment::Center)
            .block(Block::default().borders(Borders::ALL));
        frame.render_widget(title, chunks[0]);

        // Sort strategy list
        let strategy_items: Vec<ListItem> = OPENROUTER_SORT_STRATEGIES
            .iter()
            .enumerate()
            .map(|(idx, (name, description))| {
                let display = if idx == 0 {
                    format!("{} - {} (recommended)", name, description)
                } else {
                    format!("{} - {}", name, description)
                };

                ListItem::new(display)
            })
            .collect();

        let list =
            List::new(strategy_items)
                .block(Block::default().borders(Borders::ALL).title(
                    "Select Sort Strategy (↑/↓ to navigate, Enter to select, Esc to go back)",
                ))
                .highlight_style(
                    Style::default()
                        .fg(Color::Yellow)
                        .add_modifier(Modifier::BOLD),
                )
                .highlight_symbol("> ");

        let mut list_state = ListState::default().with_selected(Some(self.selected_sort_idx));
        frame.render_stateful_widget(list, chunks[1], &mut list_state);

        // Help text
        let help = Paragraph::new(
            "Controls how OpenRouter selects the upstream provider for your chosen model",
        )
        .style(Style::default().fg(Color::DarkGray))
        .alignment(Alignment::Center);
        frame.render_widget(help, chunks[2]);
    }

    /// Render connectivity test screen
    fn render_connectivity_test(&mut self, frame: &mut Frame) {
        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .margin(2)
            .constraints([Constraint::Length(3), Constraint::Min(0)])
            .split(frame.area());

        // Title
        let title = Paragraph::new("Testing Connection...")
            .style(
                Style::default()
                    .fg(Color::Cyan)
                    .add_modifier(Modifier::BOLD),
            )
            .alignment(Alignment::Center)
            .block(Block::default().borders(Borders::ALL));
        frame.render_widget(title, chunks[0]);

        // Loading message
        let message = Paragraph::new(format!(
            "Testing connection to {}...\n\nPlease wait...",
            self.selected_provider()
        ))
        .style(Style::default().fg(Color::Yellow))
        .alignment(Alignment::Center)
        .wrap(Wrap { trim: true });
        frame.render_widget(message, chunks[1]);
    }

    /// Render result screen
    fn render_result(&mut self, frame: &mut Frame, success: bool, message: &str) {
        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .margin(2)
            .constraints([
                Constraint::Length(3),
                Constraint::Min(0),
                Constraint::Length(3),
            ])
            .split(frame.area());

        // Title
        let title = if success {
            Paragraph::new("Configuration Successful!").style(
                Style::default()
                    .fg(Color::Green)
                    .add_modifier(Modifier::BOLD),
            )
        } else {
            Paragraph::new("Configuration Failed")
                .style(Style::default().fg(Color::Red).add_modifier(Modifier::BOLD))
        };
        let title = title
            .alignment(Alignment::Center)
            .block(Block::default().borders(Borders::ALL));
        frame.render_widget(title, chunks[0]);

        // Message
        let message_widget = Paragraph::new(message)
            .style(if success {
                Style::default().fg(Color::Green)
            } else {
                Style::default().fg(Color::Red)
            })
            .alignment(Alignment::Center)
            .wrap(Wrap { trim: true });
        frame.render_widget(message_widget, chunks[1]);

        // Help text
        let help = Paragraph::new(if success {
            "Press Enter, q, or Ctrl+C to exit"
        } else {
            "Press Enter, q, or Ctrl+C to exit (configuration not saved)"
        })
        .style(Style::default().fg(Color::DarkGray))
        .alignment(Alignment::Center);
        frame.render_widget(help, chunks[2]);
    }
}

/// Setup terminal for TUI
fn setup_terminal() -> Result<Terminal<CrosstermBackend<Stdout>>> {
    enable_raw_mode().context("Failed to enable raw mode")?;
    let mut stdout = io::stdout();
    execute!(stdout, EnterAlternateScreen).context("Failed to enter alternate screen")?;
    let backend = CrosstermBackend::new(stdout);
    Terminal::new(backend).context("Failed to create terminal")
}

/// Restore terminal to normal mode
fn restore_terminal(terminal: &mut Terminal<CrosstermBackend<Stdout>>) -> Result<()> {
    disable_raw_mode().context("Failed to disable raw mode")?;
    execute!(terminal.backend_mut(), LeaveAlternateScreen)
        .context("Failed to leave alternate screen")?;
    terminal.show_cursor().context("Failed to show cursor")?;
    Ok(())
}

/// Run the configuration wizard
pub fn run_configure_wizard() -> Result<()> {
    use std::io::IsTerminal;
    if !std::io::stdin().is_terminal() {
        anyhow::bail!(
            "The configuration wizard requires an interactive terminal.\n\
             \n\
             Run `rfx llm config` in an interactive terminal session, or configure\n\
             via environment variables instead:\n\
             \n\
             For OpenAI:     export OPENAI_API_KEY=sk-...\n\
             For Anthropic:  export ANTHROPIC_API_KEY=sk-ant-...\n\
             For OpenRouter: export OPENROUTER_API_KEY=sk-or-..."
        );
    }
    let mut terminal = setup_terminal()?;
    let mut wizard = ConfigWizard::new();

    let result = run_wizard_loop(&mut terminal, &mut wizard);

    // Always restore terminal
    restore_terminal(&mut terminal)?;

    result
}

/// Main wizard event loop
fn run_wizard_loop(
    terminal: &mut Terminal<CrosstermBackend<Stdout>>,
    wizard: &mut ConfigWizard,
) -> Result<()> {
    loop {
        // Render current screen
        terminal.draw(|frame| wizard.render(frame))?;

        // Dispatch /v1/models fetch by provider. OpenRouter dead-ends on error
        // because pricing data is critical to its UX; OpenAI and Anthropic fall
        // back to a small offline list so the wizard remains usable.
        if wizard.screen == WizardScreen::FetchingModels {
            let provider = wizard.selected_provider().to_string();
            match provider.as_str() {
                "openrouter" => match fetch_openrouter_models(&wizard.api_key) {
                    Ok(models) => {
                        wizard.fetched_models = models;
                        wizard.selected_model_idx = 0;
                        wizard.model_filter.clear();
                        wizard.error_message = None;
                        wizard.screen = WizardScreen::ModelSelection;
                    }
                    Err(e) => {
                        wizard.screen = WizardScreen::Result {
                            success: false,
                            message: format!(
                                "Failed to fetch models from OpenRouter: {}\n\n\
                                Please check your API key and try again.",
                                e
                            ),
                        };
                    }
                },
                "openai" => match fetch_openai_models_blocking(&wizard.api_key) {
                    Ok(ids) => {
                        wizard.fetched_dynamic_models = ids;
                        wizard.selected_model_idx = 0;
                        wizard.model_filter.clear();
                        wizard.error_message = None;
                        wizard.screen = WizardScreen::ModelSelection;
                    }
                    Err(e) => {
                        log::warn!("OpenAI /v1/models fetch failed, using fallback list: {}", e);
                        wizard.fetched_dynamic_models = OPENAI_FALLBACK_MODELS
                            .iter()
                            .map(|s| s.to_string())
                            .collect();
                        wizard.selected_model_idx = 0;
                        wizard.model_filter.clear();
                        wizard.error_message = Some(
                            "Could not reach api.openai.com — showing recent models. \
                            Some newer models may be missing."
                                .to_string(),
                        );
                        wizard.screen = WizardScreen::ModelSelection;
                    }
                },
                "anthropic" => match fetch_anthropic_models_blocking(&wizard.api_key) {
                    Ok(ids) => {
                        wizard.fetched_dynamic_models = ids;
                        wizard.selected_model_idx = 0;
                        wizard.model_filter.clear();
                        wizard.error_message = None;
                        wizard.screen = WizardScreen::ModelSelection;
                    }
                    Err(e) => {
                        log::warn!(
                            "Anthropic /v1/models fetch failed, using fallback list: {}",
                            e
                        );
                        wizard.fetched_dynamic_models = ANTHROPIC_FALLBACK_MODELS
                            .iter()
                            .map(|s| s.to_string())
                            .collect();
                        wizard.selected_model_idx = 0;
                        wizard.model_filter.clear();
                        wizard.error_message = Some(
                            "Could not reach api.anthropic.com — showing recent models. \
                            Some newer models may be missing."
                                .to_string(),
                        );
                        wizard.screen = WizardScreen::ModelSelection;
                    }
                },
                _ => {
                    // Unexpected provider in FetchingModels; fall through.
                    wizard.screen = WizardScreen::ModelSelection;
                }
            }
            continue;
        }

        // Handle connectivity test asynchronously
        if wizard.screen == WizardScreen::ConnectivityTest {
            let provider = wizard.selected_provider().to_string();
            let is_compatible = provider == "openai-compatible";

            // openai-compatible uses free-text model input; others use list selection
            let selected_model = if is_compatible {
                wizard.model_text.clone()
            } else {
                wizard.selected_model()
            };

            // Build provider options. openai-compatible needs base_url here, or
            // the factory will bail and the connectivity test would never reach
            // the network. OpenRouter passes sort via save (not test).
            let options = if is_compatible {
                let mut opts = HashMap::new();
                opts.insert("base_url".to_string(), wizard.base_url.clone());
                Some(opts)
            } else {
                None
            };

            let result = test_connectivity(&provider, &wizard.api_key, &selected_model, options);
            match result {
                Ok(_) => {
                    // Save configuration
                    let sort = if provider == "openrouter" {
                        Some(wizard.selected_sort())
                    } else {
                        None
                    };
                    let base_url = if is_compatible {
                        Some(wizard.base_url.as_str())
                    } else {
                        None
                    };
                    if let Err(e) = save_user_config(
                        &provider,
                        &wizard.api_key,
                        &selected_model,
                        sort,
                        base_url,
                    ) {
                        wizard.screen = WizardScreen::Result {
                            success: false,
                            message: format!("Failed to save configuration: {}", e),
                        };
                    } else {
                        wizard.screen = WizardScreen::Result {
                            success: true,
                            message: format!(
                                "Configuration saved successfully!\n\n\
                                Provider: {}\n\
                                Config file: ~/.reflex/config.toml\n\n\
                                You can now use 'rfx ask' to query your codebase.",
                                provider
                            ),
                        };
                    }
                }
                Err(e) => {
                    wizard.screen = WizardScreen::Result {
                        success: false,
                        message: format!(
                            "Connectivity test failed: {}\n\n\
                            Please check your endpoint, model, and credentials and try again.",
                            e
                        ),
                    };
                }
            }
            continue;
        }

        // Handle keyboard input
        if event::poll(std::time::Duration::from_millis(100))? {
            if let Event::Key(key) = event::read()? {
                let should_exit = wizard.handle_key(key)?;
                if should_exit {
                    break;
                }
            }
        }
    }

    Ok(())
}

/// Test connectivity to the selected provider
fn test_connectivity(
    provider_name: &str,
    api_key: &str,
    model: &str,
    options: Option<HashMap<String, String>>,
) -> Result<()> {
    // Create a tokio runtime for async operations
    let runtime = tokio::runtime::Runtime::new().context("Failed to create async runtime")?;

    runtime.block_on(async {
        // openai-compatible needs the model passed through; other providers
        // can fall back to their built-in defaults if no model is given.
        let model_arg = if model.is_empty() {
            None
        } else {
            Some(model.to_string())
        };

        // Create provider instance
        let provider = crate::semantic::providers::create_provider(
            provider_name,
            api_key.to_string(),
            model_arg,
            options,
            crate::semantic::config::SemanticConfig::default().timeout_seconds,
        )?;

        // Try to make a simple API call to test connectivity
        // Note: Must contain "json" for OpenAI structured output requirement
        let test_prompt = "Please respond with valid JSON: {\"status\": \"ok\"}";

        // Call complete method (json_mode: true for test). Some local servers
        // do not honor response_format, but the call should still complete.
        provider.complete(test_prompt, true).await?;

        Ok::<(), anyhow::Error>(())
    })?;

    Ok(())
}

/// Fetch models from OpenRouter API (blocking wrapper)
fn fetch_openrouter_models(api_key: &str) -> Result<Vec<OpenRouterModel>> {
    let runtime = tokio::runtime::Runtime::new().context("Failed to create async runtime")?;
    runtime.block_on(async { crate::semantic::providers::openrouter::fetch_models(api_key).await })
}

/// Fetch chat models from OpenAI's /v1/models (blocking wrapper)
fn fetch_openai_models_blocking(api_key: &str) -> Result<Vec<String>> {
    let runtime = tokio::runtime::Runtime::new().context("Failed to create async runtime")?;
    runtime.block_on(async { crate::semantic::providers::openai::fetch_models(api_key).await })
}

/// Fetch chat models from Anthropic's /v1/models (blocking wrapper)
fn fetch_anthropic_models_blocking(api_key: &str) -> Result<Vec<String>> {
    let runtime = tokio::runtime::Runtime::new().context("Failed to create async runtime")?;
    runtime.block_on(async { crate::semantic::providers::anthropic::fetch_models(api_key).await })
}

/// Save user configuration to ~/.reflex/config.toml
fn save_user_config(
    provider: &str,
    api_key: &str,
    model: &str,
    sort: Option<&str>,
    base_url: Option<&str>,
) -> Result<()> {
    use serde::{Deserialize, Serialize};
    use std::fs;

    #[derive(Debug, Serialize, Deserialize)]
    struct UserConfig {
        #[serde(default)]
        semantic: SemanticSection,
        #[serde(default)]
        credentials: HashMap<String, String>,
    }

    #[derive(Debug, Serialize, Deserialize)]
    struct SemanticSection {
        provider: String,
    }

    impl Default for SemanticSection {
        fn default() -> Self {
            Self {
                provider: "openai".to_string(),
            }
        }
    }

    let home =
        dirs::home_dir().ok_or_else(|| anyhow::anyhow!("Could not determine home directory"))?;

    let config_dir = home.join(".reflex");
    fs::create_dir_all(&config_dir).context("Failed to create ~/.reflex directory")?;

    let config_path = config_dir.join("config.toml");

    // Load existing config if it exists
    let mut config = if config_path.exists() {
        let config_str =
            fs::read_to_string(&config_path).context("Failed to read existing config file")?;
        toml::from_str::<UserConfig>(&config_str).unwrap_or_else(|_| UserConfig {
            semantic: SemanticSection::default(),
            credentials: HashMap::new(),
        })
    } else {
        UserConfig {
            semantic: SemanticSection::default(),
            credentials: HashMap::new(),
        }
    };

    // The [semantic] provider value stays in its user-facing kebab-case form
    // (e.g. "openai-compatible"), but credential field names use underscores
    // to match the serde fields on the Credentials struct.
    config.semantic.provider = provider.to_string();
    let cred_prefix = provider.replace('-', "_");

    // Update the specific provider's key and model in credentials
    let key_name = format!("{}_api_key", cred_prefix);
    let model_name = format!("{}_model", cred_prefix);
    config.credentials.insert(key_name, api_key.to_string());
    config.credentials.insert(model_name, model.to_string());

    // Save sort strategy for OpenRouter
    if let Some(sort_value) = sort {
        config
            .credentials
            .insert("openrouter_sort".to_string(), sort_value.to_string());
    }

    // Save base URL for openai-compatible
    if let Some(url) = base_url {
        config
            .credentials
            .insert(format!("{}_base_url", cred_prefix), url.to_string());
    }

    // Serialize to TOML
    let toml_content =
        toml::to_string_pretty(&config).context("Failed to serialize config to TOML")?;

    // Prepend comment header
    let final_content = format!(
        "# Reflex User Configuration\n\
         # This file stores your AI provider API keys\n\
         # Location: ~/.reflex/config.toml\n\
         \n\
         {}",
        toml_content
    );

    fs::write(&config_path, final_content).context("Failed to write configuration file")?;

    log::info!("Configuration saved to {:?}", config_path);

    Ok(())
}