Struct ProviderRegistry 

pub struct ProviderRegistry { /* private fields */ }

Registry of all available stream providers, keyed by API protocol.

Implementations

impl ProviderRegistry

pub fn new() -> Self

Create an empty registry.

pub fn register( &mut self, protocol: ApiProtocol, provider: impl StreamProvider + 'static, )

Register a provider for a given protocol.

pub fn resolve(&self, protocol: &ApiProtocol) -> Option<Arc<dyn StreamProvider>>

Resolve the provider for a protocol as a shared handle.

Unlike get, this hands out an owned Arc that can be stored (e.g. by Agent::from_config), so callers don’t have to borrow the registry for the provider’s lifetime.

pub fn get(&self, protocol: &ApiProtocol) -> Option<&dyn StreamProvider>

Borrow the provider for a protocol for a transient call. Prefer resolve when you need to keep the handle.

pub fn has(&self, protocol: &ApiProtocol) -> bool

Check if a protocol is registered.

pub fn protocols(&self) -> Vec<ApiProtocol>

List all registered protocols.

pub async fn stream( &self, model: &ModelConfig, config: StreamConfig, tx: UnboundedSender<StreamEvent>, cancel: CancellationToken, ) -> Result<Message, ProviderError>

Stream using the appropriate provider for the model’s API protocol.

Trait Implementations

impl Default for ProviderRegistry

fn default() -> Self

Create a registry with all built-in providers registered.