Struct ModelRegistry 

pub struct ModelRegistry { /* private fields */ }

Centralized model catalog with pricing, capabilities, and provider metadata.

Lookup order for pricing/model queries:

1. Exact match on "provider::model"
2. Fuzzy match — any registered model whose name is a substring of the query (or vice-versa), scoped to the same provider
3. Provider-level default entry (cache multipliers only, via get_pricing)

Implementations

impl ModelRegistry

pub fn new() -> Self

Create an empty registry.

pub fn with_defaults() -> Self

Create a registry pre-loaded with the embedded defaults.

pub fn from_toml(toml_str: &str) -> Result<Self, String>

Parse a TOML string into a registry.

pub fn merge(&mut self, other: Self)

Merge another registry on top (overrides win).

pub fn register(&mut self, provider: &str, model_id: &str, info: ModelInfo)

Register a single model dynamically (e.g. from Ollama discovery).

pub fn get(&self, provider: &str, model: &str) -> Option<&ModelInfo>

Exact-match lookup.

pub fn get_fuzzy(&self, provider: &str, model: &str) -> Option<&ModelInfo>

Fuzzy lookup: tries exact match first, then substring matching against all models for the given provider.

pub fn get_pricing(&self, provider: &str, model: &str) -> Option<CostRates>

Get pricing for a model (convenience wrapper returning just CostRates). Falls back to provider-level cache multipliers for unknown models.

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

Get provider metadata.

pub fn provider_names(&self) -> Vec<&str>

List all known provider names, sorted alphabetically.

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

Get the default model for a provider.

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

Get the API key env var for a provider.

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

List all model IDs for a provider, sorted alphabetically.

pub fn models_by_provider(&self) -> HashMap<String, Vec<String>>

Get a map of provider → model list, suitable for the settings API.

pub fn len(&self) -> usize

Number of models in the registry.

pub fn is_empty(&self) -> bool

Whether the registry is empty.

Trait Implementations

impl Clone for ModelRegistry

fn clone(&self) -> ModelRegistry

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ModelRegistry

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ModelRegistry

fn default() -> Self

Returns the “default value” for a type.