Struct EmbeddingRegistry 

pub struct EmbeddingRegistry { /* private fields */ }

A runtime registry of named EmbeddingProvider implementations.

Providers are keyed by EmbeddingProviderInfo::id. An optional default can be set explicitly via EmbeddingRegistry::set_default; if none is set, EmbeddingRegistry::default_provider returns the first registered provider.

Implementations

impl EmbeddingRegistry

pub fn new() -> Self

Create an empty registry.

pub fn register(&mut self, provider: Arc<dyn EmbeddingProvider>)

Register a provider.

If a provider with the same id already exists it is replaced. The insertion order of new ids is preserved for use by EmbeddingRegistry::default_provider.

pub fn get(&self, id: &str) -> Option<Arc<dyn EmbeddingProvider>>

Look up a provider by id.

Returns None if no provider with that id has been registered.

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

Return metadata for all registered providers, in registration order.

pub fn default_provider(&self) -> Option<Arc<dyn EmbeddingProvider>>

Return the default provider.

• If a default was set via EmbeddingRegistry::set_default, that provider is returned.
• Otherwise the first registered provider is returned.
• Returns None if the registry is empty.

pub fn set_default(&mut self, id: &str) -> Result<()>

Change the default provider.

Returns EngramError::InvalidInput if id is not registered.

pub fn count(&self) -> usize

Return the number of registered providers.

Trait Implementations

impl Default for EmbeddingRegistry

fn default() -> Self

Returns the “default value” for a type.