Struct ModelRegistry 

pub struct ModelRegistry { /* private fields */ }

The multi-model registry.

Manages a collection of ModelEndpoint instances and supports alias resolution so that clients can refer to models by friendly names.

Implementations

impl ModelRegistry

pub fn new() -> Self

Create an empty registry.

pub fn register(&mut self, endpoint: ModelEndpoint)

Register a model endpoint.

If the endpoint has is_default set, it becomes the default model. Replaces any existing endpoint with the same ID.

pub fn unregister(&mut self, id: &ModelId) -> Option<ModelEndpoint>

Remove an endpoint from the registry.

Also clears the default-model pointer if it pointed to the removed endpoint, and removes any aliases that targeted this ID.

pub fn add_alias(&mut self, alias: impl Into<String>, target: ModelId)

Add an alias: e.g. "gpt-4" maps to ModelId("bonsai-8b").

pub fn resolve(&self, id_or_alias: &str) -> Option<&ModelEndpoint>

Resolve a model identifier (checks ID first, then aliases).

Returns None if neither a direct ID nor an alias matches.

pub fn default_endpoint(&self) -> Option<&ModelEndpoint>

Get the default model endpoint.

pub fn available_endpoints(&self) -> Vec<&ModelEndpoint>

List all available (Ready) endpoints.

pub fn all_endpoints(&self) -> Vec<&ModelEndpoint>

List all registered endpoints (including non-ready ones).

pub fn set_status(&mut self, id: &ModelId, status: EndpointStatus) -> bool

Update an endpoint’s status.

Returns true if the endpoint was found and updated.

pub fn len(&self) -> usize

Number of registered endpoints.

pub fn is_empty(&self) -> bool

Is the registry empty?

Trait Implementations

impl Default for ModelRegistry

fn default() -> Self

Returns the “default value” for a type.