Struct ToolLoopCostTrackingContext 

pub struct ToolLoopCostTrackingContext {
    pub tracker: Option<Arc<CostTracker>>,
    pub model_provider_pricing: Arc<ModelProviderPricing>,
    pub turn_usage: Arc<Mutex<TurnUsage>>,
    pub agent_alias: Option<String>,
}

Context for cost tracking within the tool call loop. Scoped via tokio::task_local! at call sites (channels, gateway).

Fields

tracker: Option<Arc<CostTracker>>

Shared cost tracker. None for usage-only contexts that accumulate per-turn token totals without persisting cost records or enforcing budgets (see Self::usage_only).

model_provider_pricing: Arc<ModelProviderPricing>

turn_usage: Arc<Mutex<TurnUsage>>

Per-scope usage accumulator so wrapping code can read token/cost totals after the scoped future exits (without racing concurrent traffic sharing the same tracker).

agent_alias: Option<String>

Alias of the agent driving this turn. Stamped onto persisted CostRecords so /api/cost?agent=<alias> can attribute spend.

Implementations

impl ToolLoopCostTrackingContext

pub fn new( tracker: Arc<CostTracker>, model_provider_pricing: Arc<ModelProviderPricing>, ) -> Self

pub fn usage_only() -> Self

pub fn with_agent_alias(self, agent_alias: impl Into<String>) -> Self

Attach an agent alias to this context so subsequent record_tool_loop_cost_usage calls stamp records with it.

pub fn snapshot_turn_usage(&self) -> TurnUsage

Trait Implementations

impl Clone for ToolLoopCostTrackingContext

fn clone(&self) -> ToolLoopCostTrackingContext

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.