plugins/agent/providers/src/facts.rs

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

use maka_agent_ports::provider::Error;
type Result<T> = std::result::Result<T, Error>;
pub(crate) mod adapter;
pub(crate) mod discovery;
use adapter::{ModelRuntimeOverride, RuntimeAdapter};
use discovery::ModelDiscovery;
mod metadata;
use maka_agent_facts::configuration::ApiProtocol;
pub use metadata::{ModelMetadata, ThinkingOffBehavior};
use serde::Deserialize;
use std::{collections::BTreeMap, sync::OnceLock};

#[derive(Debug, Clone, Copy, PartialEq, Eq, Deserialize)]
#[serde(rename_all = "snake_case")]
pub enum AuthKind {
    ApiKey,
    OauthToken,
    None,
    OptionalApiKey,
}

#[derive(Debug, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct ProviderFacts {
    pub label: String,
    pub base_url: String,
    pub auth_kind: AuthKind,
    pub runtime_adapter: RuntimeAdapter,
    pub protocol_adapters: BTreeMap<ApiProtocol, RuntimeAdapter>,
    pub retired: bool,
    pub model_discovery: ModelDiscovery,
    pub fallback_models: Vec<String>,
    pub models: BTreeMap<String, ModelFacts>,
}

#[derive(Debug, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct ModelFacts {
    pub metadata: ModelMetadata,
    pub runtime_override: Option<ModelRuntimeOverride>,
    pub entry: CatalogDefaults,
    /// Native Anthropic SDK capability, generated with the bundled SDK version.
    pub anthropic_adaptive_thinking: Option<bool>,
}

pub(crate) fn all() -> Result<&'static BTreeMap<String, ProviderFacts>> {
    static FACTS: OnceLock<std::result::Result<BTreeMap<String, ProviderFacts>, String>> =
        OnceLock::new();
    FACTS
        .get_or_init(|| {
            serde_json::from_str(include_str!("../data/catalog-facts.json"))
                .map_err(|error| error.to_string())
        })
        .as_ref()
        .map_err(|error| Error::Invalid(error.clone()))
}

pub fn provider_facts(provider: &str) -> Result<&'static ProviderFacts> {
    all()?
        .get(provider)
        .ok_or_else(|| Error::Invalid(format!("unknown provider: {provider}")))
}

#[derive(Debug, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct CatalogDefaults {
    pub can_use_as_chat_default: bool,
    pub supports_vision: bool,
    pub thinking_levels: Vec<maka_agent_facts::execution::ThinkingLevel>,
}