Struct AgentTokenBudget 

pub struct AgentTokenBudget { /* private fields */ }

Agent token budget

AI agents get token budgets instead of simple rate limits. This allows for burst operations while enforcing daily/hourly limits.

Implementations

impl AgentTokenBudget

pub fn daily(agent_id: impl Into<String>, tokens: u64) -> Self

Create a new daily token budget

pub fn hourly(agent_id: impl Into<String>, tokens: u64) -> Self

Create a new hourly token budget

pub fn new(agent_id: impl Into<String>, tokens: u64, period: Duration) -> Self

Create a new token budget with custom period

pub fn with_warning_threshold(self, threshold: f64) -> Self

Set warning threshold (0.0 - 1.0)

pub fn with_hard_limit(self, hard: bool) -> Self

Set hard limit behavior

pub fn with_operation_costs(self, costs: HashMap<String, u64>) -> Self

Set operation costs

pub fn add_operation_cost(&mut self, operation: impl Into<String>, cost: u64)

Add operation cost

pub fn consume( &self, operation: &str, estimated_tokens: u64, ) -> Result<(), BudgetExceeded>

Consume tokens for an operation

pub fn check( &self, operation: &str, estimated_tokens: u64, ) -> Result<(), BudgetExceeded>

Check if budget is available (without consuming)

pub fn remaining(&self) -> u64

Get remaining tokens

pub fn used(&self) -> u64

Get used tokens

pub fn usage_percentage(&self) -> f64

Get usage percentage (0.0 - 1.0)

pub fn is_warning(&self) -> bool

Check if over warning threshold

pub fn time_until_reset(&self) -> Duration

Get time until reset

pub fn reset(&self)

Force reset

Trait Implementations

impl Clone for AgentTokenBudget

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for AgentTokenBudget

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.