Struct ModelRegistry 

pub struct ModelRegistry { /* private fields */ }

Model registry for managing model lifecycle

Thread Safety

ModelRegistry uses Arc internally for efficient cloning and sharing across threads. The internal HashMap is immutable after creation, making concurrent reads safe without locks.

Implementations

impl ModelRegistry

pub fn new() -> Self

Create a new registry with default cache directory

pub fn from_file(path: &str) -> Result<Self>

Create a registry from a JSON file

Arguments

• path - Path to registry.json file

Example

let registry = ModelRegistry::from_file("models/registry.json")?;

pub fn get_model_metadata( &self, task: ModelTask, variant: ModelVariant, ) -> Result<&ModelMetadata>

Get metadata for a specific model

Arguments

• task - The model task
• variant - The model variant

Returns

Reference to model metadata, or Error if not found

Example

let metadata = registry.get_model_metadata(
    ModelTask::PromptInjection,
    ModelVariant::FP16
)?;
println!("Model ID: {}", metadata.id);

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

List all available models in the registry

Returns

Vector of references to all model metadata

Example

let all_models = registry.list_models();
for model in all_models {
    println!("Model: {} ({:?}/{:?})", model.id, model.task, model.variant);
}

pub fn list_models_for_task(&self, task: ModelTask) -> Vec<&ModelMetadata>

List all models for a specific task

Arguments

• task - The model task to filter by

Returns

Vector of references to model metadata matching the task

Example

let models = registry.list_models_for_task(ModelTask::PromptInjection);
println!("Found {} PromptInjection models", models.len());

pub fn get_available_variants(&self, task: ModelTask) -> Vec<ModelVariant>

Get all available variants for a specific task

Arguments

• task - The model task

Returns

Vector of all available model variants for the task

Example

let variants = registry.get_available_variants(ModelTask::PromptInjection);
for variant in variants {
    println!("Available variant: {:?}", variant);
}

pub fn has_model(&self, task: ModelTask, variant: ModelVariant) -> bool

Check if a specific model is available in the registry

Arguments

• task - The model task
• variant - The model variant

Returns

true if the model is registered, false otherwise

Example

if registry.has_model(ModelTask::PromptInjection, ModelVariant::FP16) {
    println!("Model is available!");
} else {
    println!("Model not found");
}

pub fn model_count(&self) -> usize

Get the total number of registered models

Returns

Count of all models in the registry

Example

println!("Registry contains {} models", registry.model_count());

pub fn is_empty(&self) -> bool

Check if the registry is empty

Returns

true if no models are registered, false otherwise

Example

if registry.is_empty() {
    println!("No models registered");
}

pub async fn ensure_model_available( &self, task: ModelTask, variant: ModelVariant, ) -> Result<PathBuf>

Ensure a model is available locally (download if needed)

This method:

1. Checks if model is already cached
2. Verifies checksum if cached
3. Downloads if not cached or verification fails
4. Verifies checksum after download

Arguments

• task - The model task
• variant - The model variant

Returns

Path to the local model file

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.