Struct ProviderRegistry

pub struct ProviderRegistry { /* private fields */ }

Registry for managing LLM providers

Implementations

impl ProviderRegistry

pub fn new(storage: Arc<dyn AuthStorage>) -> Self

Create a new provider registry with authentication storage

pub fn register_provider(&mut self, provider: Arc<dyn Provider>) -> Result<()>

Register a provider

pub fn get_provider(&self, provider_id: &str) -> Result<Arc<dyn Provider>>

Get a provider by ID

pub fn list_providers(&self) -> Vec<String>

List all registered providers

pub async fn get_model( &mut self, provider_id: &str, model_id: &str, ) -> Result<Arc<dyn Model>>

Get a model by provider and model ID

pub fn parse_model_string(&self, model_string: &str) -> Result<(String, String)>

Parse a model string (format: “provider/model” or “provider:model”)

pub fn set_default_provider(&mut self, provider_id: &str) -> Result<()>

Set default provider

pub fn get_default_provider(&self) -> Option<&str>

Get default provider

pub async fn list_all_models(&self) -> Result<Vec<ModelInfo>>

List all available models

pub async fn get_all_provider_health(&self) -> HashMap<String, ProviderHealth>

Get provider health for all providers

pub fn clear_model_cache(&mut self)

Clear cached models

pub fn remove_provider(&mut self, provider_id: &str) -> Result<()>

Remove a provider

pub async fn discover_from_env(&mut self) -> Result<()>

Discover providers from environment variables

pub async fn discover_from_storage(&mut self) -> Result<()>

Discover providers from storage

pub async fn initialize_all(&mut self) -> Result<()>

Initialize all registered providers

pub async fn load_models_dev(&mut self) -> Result<()>

Load models from models.dev API

pub async fn load_configs(&mut self, path: &str) -> Result<()>

Load configurations from a file

pub async fn get(&self, provider_id: &str) -> Option<Arc<dyn Provider>>

Get a provider by ID (async version)

pub fn parse_model(model_str: &str) -> (String, String)

Parse a model string (format: “provider/model” or just “model”)

pub async fn get_default_model( &self, provider_id: &str, ) -> Result<Arc<dyn Model>>

Get the default model for a provider

pub async fn available(&self) -> Vec<String>

Get list of available providers (those that can authenticate)

pub async fn list(&self) -> Vec<String>

List all registered provider IDs

pub async fn register(&mut self, provider: Arc<dyn Provider>)

Register a provider (async version)