Struct ModelRegistry 

pub struct ModelRegistry { /* private fields */ }

Thread-safe registry of loaded model metadata.

Models are keyed by name; each registration parses (or accepts) GGUF metadata without loading tensor weights.

Example

use oxirs_graphrag::model_loader::{ModelRegistry, GgufMetadata, GgufParser};
use std::path::PathBuf;

let registry = ModelRegistry::new();
// In real usage you would supply a path to a .gguf file:
// let handle = registry.register("llama-3-8b", PathBuf::from("model.gguf")).unwrap();
assert!(registry.is_empty());

Implementations

impl ModelRegistry

pub fn new() -> Self

Create an empty registry.

pub fn register( &self, name: &str, path: PathBuf, ) -> Result<ModelHandle, RegistryError>

Register a model by parsing its GGUF file.

Returns an error if name is already registered or the file cannot be parsed.

pub fn register_with_metadata( &self, name: &str, path: PathBuf, metadata: GgufMetadata, ) -> Result<ModelHandle, RegistryError>

Register a model with pre-parsed metadata.

Useful for tests that construct synthetic metadata without a real file.

pub fn get(&self, handle: &ModelHandle) -> Option<ModelInfo>

Look up a model by its handle.

pub fn get_by_name(&self, name: &str) -> Option<ModelInfo>

Look up a model by name.

pub fn list(&self) -> Vec<ModelHandle>

List all registered model handles.

pub fn remove(&self, handle: &ModelHandle) -> bool

Remove a model from the registry.

Returns true if the model existed and was removed.

pub fn len(&self) -> usize

Return the number of registered models.

pub fn is_empty(&self) -> bool

Return true if no models are registered.

Trait Implementations

impl Default for ModelRegistry

fn default() -> ModelRegistry

Returns the “default value” for a type.