Struct CostTracker 

pub struct CostTracker { /* private fields */ }

Thread-safe cumulative cost tracker.

Construct once per run (or share one per session), attach it to a crate::token_counter::TokenTrackingLLM with with_cost_tracker, optionally share the same Arc with an AgentExecutor::with_cost_tracker for hard budget enforcement.

Implementations

impl CostTracker

pub fn new(table: impl Into<Arc<PricingTable>>) -> Self

Tracker over the given pricing table.

pub fn with_builtin_prices() -> Self

Tracker over the built-in price snapshot.

pub fn with_scope(self, scope: impl Into<String>) -> Self

Labels this tracker with a run/session id (carried in reports and emitted cost events).

pub fn with_metrics_sink(self, sink: Arc<dyn MetricsSink>) -> Self

Attaches an observability sink: every record emits one ObsEvent::Cost. Sink failures are warned, never propagated.

pub fn pricing(&self) -> &PricingTable

The pricing table backing this tracker.

pub async fn record( &self, provider: Option<&str>, model: &str, prompt_tokens: usize, completion_tokens: usize, ) -> f64

Records one LLM call and returns its USD cost.

Models absent from the table price at 0.0 but still count toward token / call aggregates. The observability event (when attached) is exported after aggregation; export failure only logs a warning.

pub async fn record_usage( &self, provider: Option<&str>, model: &str, usage: &TokenUsage, ) -> f64

Records one call from a provider-annotated TokenUsage.

pub async fn total_cost_usd(&self) -> f64

Cumulative USD spend (the value the budget gate reads).

pub async fn report(&self) -> CostReport

Snapshot report of everything recorded so far.

pub async fn records(&self) -> Vec<CostRecord>

Every individual call recorded so far (oldest first).

pub async fn reset(&self)

Resets all accumulation (e.g. start a new run while keeping the same session-scoped tracker).