Struct UsageMetadataCallbackGuard 

pub struct UsageMetadataCallbackGuard { /* private fields */ }

Guard type for the get_usage_metadata_callback function.

This guard provides access to the underlying UsageMetadataCallbackHandler and can be used with a callback manager to track usage metadata.

When the guard is dropped, any cleanup is performed automatically.

Example

use agent_chain_core::callbacks::get_usage_metadata_callback;

let callback_guard = get_usage_metadata_callback();
// Use callback_guard.handler() with your callback manager
// The handler can be cloned and added to managers

let usage = callback_guard.usage_metadata();
for (model, metadata) in usage.iter() {
    println!("{}: {} tokens", model, metadata.total_tokens);
}

Implementations

impl UsageMetadataCallbackGuard

pub fn handler(&self) -> &UsageMetadataCallbackHandler

Get a reference to the underlying handler.

pub fn handler_mut(&mut self) -> &mut UsageMetadataCallbackHandler

Get a mutable reference to the underlying handler.

pub fn usage_metadata(&self) -> HashMap<String, UsageMetadata>

Get the collected usage metadata.

pub fn as_arc_handler(&self) -> Arc<dyn BaseCallbackHandler>

Get an Arc-wrapped handler suitable for use with callback managers.

Methods from Deref<Target = UsageMetadataCallbackHandler>

pub fn usage_metadata(&self) -> HashMap<String, UsageMetadata>

Get the collected usage metadata.

Returns a clone of the current usage metadata map, keyed by model name.

Trait Implementations

impl Deref for UsageMetadataCallbackGuard

type Target = UsageMetadataCallbackHandler

The resulting type after dereferencing.

fn deref(&self) -> &Self::Target

Dereferences the value.

impl DerefMut for UsageMetadataCallbackGuard

fn deref_mut(&mut self) -> &mut Self::Target

Mutably dereferences the value.