Struct TokenUsageTracker 

pub struct TokenUsageTracker { /* private fields */ }

Thread-safe token usage tracker. Tracks usage by session, model, and total.

Implementations

impl TokenUsageTracker

pub fn new() -> Self

Create a new token usage tracker.

pub async fn increment( &self, session_id: i64, model: &str, input_tokens: i64, output_tokens: i64, )

Increment token usage for a session and model.

pub async fn get_session_usage(&self, session_id: i64) -> Option<TokenMeter>

Get token usage for a specific session. Returns None if the session has no recorded usage.

pub async fn get_model_usage(&self, model: &str) -> Option<TokenMeter>

Get token usage for a specific model. Returns None if the model has no recorded usage.

pub async fn get_total_usage(&self) -> TokenMeter

Get total token usage across all sessions and models.

pub async fn get_all_session_usage(&self) -> HashMap<i64, TokenMeter>

Get all session usage as a map.

pub async fn get_all_model_usage(&self) -> HashMap<String, TokenMeter>

Get all model usage as a map.

pub async fn remove_session(&self, session_id: i64)

Remove usage tracking for a session (e.g., when session is deleted).

pub async fn session_count(&self) -> usize

Get the number of tracked sessions.

pub async fn model_count(&self) -> usize

Get the number of tracked models.

Trait Implementations

impl Default for TokenUsageTracker

fn default() -> Self

Returns the “default value” for a type.