Struct ModelRegistry 

pub struct ModelRegistry { /* private fields */ }

Model registry for looking up specifications.

Implementations

impl ModelRegistry

pub fn load() -> Result<Self>

Loads the model registry from embedded YAML.

pub fn get_model_spec(&self, api_identifier: &str) -> Option<&ModelSpec>

Returns the model specification for the given API identifier.

pub fn get_max_output_tokens(&self, api_identifier: &str) -> usize

Returns the max output tokens for a model, with fallback to provider defaults.

pub fn get_input_context(&self, api_identifier: &str) -> usize

Returns the input context limit for a model, with fallback to provider defaults.

pub fn is_legacy_model(&self, api_identifier: &str) -> bool

Checks if a model is legacy.

pub fn get_all_models(&self) -> &[ModelSpec]

Returns all available models.

pub fn get_models_by_provider(&self, provider: &str) -> Vec<&ModelSpec>

Returns models filtered by provider.

pub fn get_models_by_provider_and_tier( &self, provider: &str, tier: &str, ) -> Vec<&ModelSpec>

Returns models filtered by provider and tier.

pub fn get_default_model(&self, provider: &str) -> Option<&str>

Returns the default model identifier for a provider, as defined in models.yaml.

pub fn get_provider_config(&self, provider: &str) -> Option<&ProviderConfig>

Returns the provider configuration.

pub fn get_tier_info(&self, provider: &str, tier: &str) -> Option<&TierInfo>

Returns tier information for a provider.

pub fn get_beta_headers(&self, api_identifier: &str) -> &[BetaHeader]

Returns the beta headers for a model.

pub fn get_max_output_tokens_with_beta( &self, api_identifier: &str, beta_value: &str, ) -> usize

Returns the max output tokens for a model with a specific beta header active.

pub fn get_input_context_with_beta( &self, api_identifier: &str, beta_value: &str, ) -> usize

Returns the input context for a model with a specific beta header active.