Function test_model_api_key_override

core/src/config/tests.rs:302–362

()

#[test]
fn test_model_api_key_override() {
    let provider = ProviderConfig {
        name: "openai".to_string(),
        api_key: Some("provider-key".to_string()),
        base_url: Some("https://api.openai.com".to_string()),
        headers: HashMap::new(),
        session_id_header: None,
        models: vec![
            ModelConfig {
                id: "gpt-4".to_string(),
                name: "GPT-4".to_string(),
                family: "gpt".to_string(),
                api_key: None, // Uses provider key
                base_url: None,
                headers: HashMap::new(),
                session_id_header: None,
                attachment: false,
                reasoning: false,
                tool_call: true,
                temperature: true,
                release_date: None,
                modalities: ModelModalities::default(),
                cost: ModelCost::default(),
                limit: ModelLimit::default(),
            },
            ModelConfig {
                id: "custom-model".to_string(),
                name: "Custom Model".to_string(),
                family: "custom".to_string(),
                api_key: Some("model-specific-key".to_string()), // Override
                base_url: Some("https://custom.api.com".to_string()), // Override
                headers: HashMap::new(),
                session_id_header: None,
                attachment: false,
                reasoning: false,
                tool_call: true,
                temperature: true,
                release_date: None,
                modalities: ModelModalities::default(),
                cost: ModelCost::default(),
                limit: ModelLimit::default(),
            },
        ],
    };

    // Model without override uses provider key
    let model1 = provider.find_model("gpt-4").unwrap();
    assert_eq!(provider.get_api_key(model1), Some("provider-key"));
    assert_eq!(
        provider.get_base_url(model1),
        Some("https://api.openai.com")
    );

    // Model with override uses its own key
    let model2 = provider.find_model("custom-model").unwrap();
    assert_eq!(provider.get_api_key(model2), Some("model-specific-key"));
    assert_eq!(
        provider.get_base_url(model2),