fabro/lib/crates/fabro-agent/tests/it/guardrails.rs

use fabro_agent::{AgentProfile, AnthropicProfile, GeminiProfile, OpenAiProfile};
use fabro_model::{Catalog, Provider};

#[test]
fn profile_context_window_matches_catalog_for_default_models() {
    for &provider in Provider::ALL {
        let catalog_info = Catalog::builtin()
            .default_for_provider(provider)
            .cloned()
            .unwrap_or_else(|| panic!("no default model for {provider:?} in catalog"));
        let model = &catalog_info.id;
        let context_window = usize::try_from(catalog_info.context_window())
            .expect("catalog context window should be non-negative and fit in usize");

        let profile: Box<dyn AgentProfile> = match provider {
            Provider::OpenAi => Box::new(OpenAiProfile::new(model)),
            Provider::Kimi
            | Provider::Zai
            | Provider::Minimax
            | Provider::Inception
            | Provider::OpenAiCompatible => {
                Box::new(OpenAiProfile::new(model).with_provider(provider))
            }
            Provider::Gemini => Box::new(GeminiProfile::new(model)),
            Provider::Anthropic => Box::new(AnthropicProfile::new(model)),
        };

        assert_eq!(
            profile.context_window_size(),
            context_window,
            "context_window_size mismatch for {:?} model '{}': profile={} catalog={}",
            provider,
            model,
            profile.context_window_size(),
            context_window
        );
    }
}