Struct ComponentRegistry 

pub struct ComponentRegistry { /* private fields */ }

Global component registry for managing component factories

Implementations

impl ComponentRegistry

pub fn new() -> Self

Create a new empty registry

pub fn with_defaults() -> Self

Create a registry with all default factories pre-registered

pub fn register_defaults(&self)

Register all default component factories

pub fn register_tokenizer_factory( &self, name: impl Into<String>, factory: Arc<dyn ComponentFactory<Arc<dyn Tokenizer + Send + Sync>>>, )

Register a tokenizer factory

pub fn register_sampler_factory( &self, name: impl Into<String>, factory: Arc<dyn ComponentFactory<Arc<dyn Sampler + Send + Sync>>>, )

Register a sampler factory

pub fn register_scheduler_factory( &self, name: impl Into<String>, factory: Arc<dyn ComponentFactory<Arc<dyn Scheduler + Send + Sync>>>, )

Register a scheduler factory

pub fn register_kv_cache_factory( &self, name: impl Into<String>, factory: Arc<dyn ComponentFactory<Arc<dyn KvCacheManager + Send + Sync>>>, )

Register a KV cache factory

pub fn register_executor_factory( &self, name: impl Into<String>, factory: Arc<dyn ComponentFactory<Arc<dyn ModelExecutor + Send + Sync>>>, )

Register a model executor factory

pub fn get_tokenizer_factory( &self, name: &str, ) -> Option<Arc<dyn ComponentFactory<Arc<dyn Tokenizer + Send + Sync>>>>

Get a tokenizer factory by name

pub fn get_sampler_factory( &self, name: &str, ) -> Option<Arc<dyn ComponentFactory<Arc<dyn Sampler + Send + Sync>>>>

Get a sampler factory by name

pub fn get_scheduler_factory( &self, name: &str, ) -> Option<Arc<dyn ComponentFactory<Arc<dyn Scheduler + Send + Sync>>>>

Get a scheduler factory by name

pub fn get_kv_cache_factory( &self, name: &str, ) -> Option<Arc<dyn ComponentFactory<Arc<dyn KvCacheManager + Send + Sync>>>>

Get a KV cache factory by name

pub fn get_executor_factory( &self, name: &str, ) -> Option<Arc<dyn ComponentFactory<Arc<dyn ModelExecutor + Send + Sync>>>>

Get a model executor factory by name

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

List all registered tokenizer names

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

List all registered sampler names

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

List all registered scheduler names

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

List all registered KV cache names

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

List all registered executor names

pub async fn create_tokenizer( &self, name: &str, config: &ComponentConfig, ) -> Result<Arc<dyn Tokenizer + Send + Sync>>

Create a tokenizer by name

pub async fn create_sampler( &self, name: &str, config: &ComponentConfig, ) -> Result<Arc<dyn Sampler + Send + Sync>>

Create a sampler by name

pub async fn create_scheduler( &self, name: &str, config: &ComponentConfig, ) -> Result<Arc<dyn Scheduler + Send + Sync>>

Create a scheduler by name

pub async fn create_kv_cache( &self, name: &str, config: &ComponentConfig, ) -> Result<Arc<dyn KvCacheManager + Send + Sync>>

Create a KV cache manager by name

pub async fn create_executor( &self, name: &str, config: &ComponentConfig, ) -> Result<Arc<dyn ModelExecutor + Send + Sync>>

Create a model executor by name

Trait Implementations

impl Debug for ComponentRegistry

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ComponentRegistry

fn default() -> Self

Returns the “default value” for a type.