Skip to main content

ModelRegistry

Trait ModelRegistry 

Source
pub trait ModelRegistry: Send + Sync {
    // Required methods
    fn register_model(
        &mut self,
        model_id: &ModelId,
        executor: Box<dyn ModelExecutor>,
    ) -> Result<()>;
    fn get_model(&self, model_id: &ModelId) -> Option<&dyn ModelExecutor>;
    fn remove_model(
        &mut self,
        model_id: &ModelId,
    ) -> Option<Box<dyn ModelExecutor>>;
    fn list_models(&self) -> Vec<ModelId>;
    fn get_model_info(&self, model_id: &ModelId) -> Option<&ModelInfo>;
    fn contains_model(&self, model_id: &ModelId) -> bool;
}
Expand description

Model registry for managing built models

Required Methods§

Source

fn register_model( &mut self, model_id: &ModelId, executor: Box<dyn ModelExecutor>, ) -> Result<()>

Register model executor

Source

fn get_model(&self, model_id: &ModelId) -> Option<&dyn ModelExecutor>

Get model executor

Source

fn remove_model(&mut self, model_id: &ModelId) -> Option<Box<dyn ModelExecutor>>

Remove model executor

Source

fn list_models(&self) -> Vec<ModelId>

List registered models

Source

fn get_model_info(&self, model_id: &ModelId) -> Option<&ModelInfo>

Get model information

Source

fn contains_model(&self, model_id: &ModelId) -> bool

Check if model exists

Implementors§