aidaemon 0.9.33

use super::*;

impl Agent {
    /// Get the current model name.
    pub async fn current_model(&self) -> String {
        match tokio::time::timeout(Duration::from_secs(2), self.model.read()).await {
            Ok(guard) => guard.clone(),
            Err(_) => {
                warn!("Timed out acquiring model lock for current_model()");
                self.llm_runtime.snapshot().primary_model()
            }
        }
    }

    /// Switch the active model at runtime. Keeps the old model as fallback.
    /// Also disables auto-routing until `clear_model_override()` is called.
    pub async fn set_model(&self, model: String) {
        let mut m = match tokio::time::timeout(Duration::from_secs(2), self.model.write()).await {
            Ok(guard) => guard,
            Err(_) => {
                warn!("Timed out acquiring model write lock for set_model()");
                return;
            }
        };
        let mut fb =
            match tokio::time::timeout(Duration::from_secs(2), self.fallback_model.write()).await {
                Ok(guard) => guard,
                Err(_) => {
                    warn!("Timed out acquiring fallback_model write lock for set_model()");
                    return;
                }
            };
        info!(old = %*m, new = %model, "Model switched");
        *fb = m.clone();
        *m = model;
        match tokio::time::timeout(Duration::from_secs(2), self.model_override.write()).await {
            Ok(mut guard) => {
                *guard = true;
            }
            Err(_) => {
                warn!("Timed out acquiring model_override write lock for set_model()");
            }
        }
    }

    /// Re-enable auto-routing after a manual model override.
    pub async fn clear_model_override(&self) {
        match tokio::time::timeout(Duration::from_secs(2), self.model_override.write()).await {
            Ok(mut guard) => {
                *guard = false;
            }
            Err(_) => {
                warn!("Timed out acquiring model_override write lock for clear_model_override()");
                return;
            }
        }
        info!("Model override cleared, auto-routing re-enabled");
    }

    /// Rebuild and hot-swap the provider backend + router without a daemon restart.
    pub async fn reload_provider(
        &self,
        config: &crate::config::AppConfig,
    ) -> anyhow::Result<String> {
        let bundle = crate::startup::provider_router::build_provider_router(config)?;
        let new_primary = bundle.primary_model.clone();
        let new_fallback = config
            .provider
            .models
            .fallback_models
            .iter()
            .find(|m| m.as_str() != new_primary)
            .cloned()
            .unwrap_or_else(|| new_primary.clone());
        let old_model = match tokio::time::timeout(Duration::from_secs(2), self.model.read()).await
        {
            Ok(guard) => guard.clone(),
            Err(_) => {
                warn!("Timed out acquiring model lock during provider reload");
                self.llm_runtime.snapshot().primary_model()
            }
        };

        let old_runtime = self.llm_runtime.swap(
            bundle.provider,
            bundle.router,
            bundle.provider_kind,
            new_primary.clone(),
            bundle.failover_targets,
        );

        {
            let mut model = tokio::time::timeout(Duration::from_secs(2), self.model.write())
                .await
                .map_err(|_| anyhow::anyhow!("timed out acquiring model write lock"))?;
            let mut fallback =
                tokio::time::timeout(Duration::from_secs(2), self.fallback_model.write())
                    .await
                    .map_err(|_| anyhow::anyhow!("timed out acquiring fallback write lock"))?;
            *model = new_primary.clone();
            *fallback = new_fallback.clone();
        }
        {
            let mut override_guard =
                tokio::time::timeout(Duration::from_secs(2), self.model_override.write())
                    .await
                    .map_err(|_| anyhow::anyhow!("timed out acquiring model_override lock"))?;
            *override_guard = false;
        }

        info!(
            old_provider = ?old_runtime.provider_kind(),
            new_provider = ?config.provider.kind,
            old_model = %old_model,
            new_model = %new_primary,
            new_fallback = %new_fallback,
            "Provider runtime reloaded"
        );

        Ok(format!(
            "Provider: {:?} -> {:?}. Model: {} -> {}. Auto-routing re-enabled.",
            old_runtime.provider_kind(),
            config.provider.kind,
            old_model,
            new_primary
        ))
    }

    /// Clear conversation history for a session, preserving facts.
    pub async fn clear_session(&self, session_id: &str) -> anyhow::Result<()> {
        self.state.clear_session(session_id).await
    }

    /// List available models from the provider.
    pub async fn list_models(&self) -> anyhow::Result<Vec<String>> {
        self.llm_runtime.provider().list_models().await
    }

    /// Stamp the current config as "last known good" — called after a
    /// successful LLM response proves the config actually works.
    pub(super) async fn stamp_lastgood(&self) {
        let lastgood = self.config_path.with_extension("toml.lastgood");
        if let Err(e) = tokio::fs::copy(&self.config_path, &lastgood).await {
            warn!(error = %e, "Failed to stamp lastgood config");
        }
    }
}