Struct ModelRegistry 

pub struct ModelRegistry { /* private fields */ }

Model registry manager

Implementations

impl ModelRegistry

pub fn new(storage: Arc<ModelStorage>) -> Result<Self>

pub async fn refresh_openrouter_catalog_from_api( &mut self, api_key: &str, ) -> Result<()>

Refresh the OpenRouter model catalog from the live OpenRouter API. This replaces any existing OpenRouter entries with the up-to-date list of models visible to the provided API key.

pub async fn refresh_huggingface_catalog_from_api( &mut self, limit: usize, ) -> Result<()>

Refresh the Hugging Face GGUF/GGML model catalog from the HF Hub API. Fetches top models by downloads and extracts available quantized files.

pub fn update_compatibility_scores( &mut self, recommender: &ModelRecommender, hardware: &HardwareProfile, )

Recompute compatibility scores for all known local models based on the current hardware profile and user preferences. This is used by the model manager to support “Best Match” sorting in the UI.

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

Scan local storage for existing models and populate registry

pub async fn update_model_status_from_storage( &mut self, model_id: &str, ) -> Result<()>

Update model status based on file existence

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

Update all model statuses based on file existence in storage

pub fn get_installed_model_path(&self, model_id: &str) -> Option<PathBuf>

Get the path of an installed model by ID

pub async fn get_model_metadata(&self, model_id: &str) -> Option<ModelMetadata>

Get the complete model metadata including runtime binaries information

pub fn add_model(&mut self, model_info: ModelInfo)

Add a model to the registry

pub fn get_model(&self, model_id: &str) -> Option<&ModelInfo>

Get model information by ID

pub fn get_model_mut(&mut self, model_id: &str) -> Option<&mut ModelInfo>

Get mutable reference to model information

pub fn list_models(&self) -> Vec<&ModelInfo>

List all models in registry

pub fn list_models_by_status(&self, status: ModelStatus) -> Vec<&ModelInfo>

List models by status

pub fn search_models(&self, query: &str) -> Vec<&ModelInfo>

Search models by name or tags

pub fn get_recommended_models(&self, max_results: usize) -> Vec<&ModelInfo>

Get models sorted by compatibility score for current hardware

pub fn update_model_status(&mut self, model_id: &str, status: ModelStatus)

Update model status

pub fn remove_model(&mut self, model_id: &str) -> bool

Remove a model from registry

pub fn get_statistics(&self) -> RegistryStats

Get registry statistics

pub fn get_models_by_category(&self, category: &str) -> Vec<&ModelInfo>

Get models by category/tags

pub fn get_trending_models(&self, limit: usize) -> Vec<&ModelInfo>

Get trending models (recently added or popular tags)

pub fn get_models_by_task(&self, task: &str) -> Vec<&ModelInfo>

Get models by task type (e.g., “chat”, “coding”, “text-generation”)

pub async fn save_registry(&self) -> Result<()>

Save registry to persistent storage

pub fn populate_default_catalog(&mut self)

Populate the registry with well-known models from all sources. Only adds models that are not already in the registry. Also removes stale models that are no longer available.

pub async fn populate_default_openrouter_models(&mut self)

Populate registry with OpenRouter models fetched from the public API. This is called when no API key is available to show users what models are available.