Struct TokenBudgetEnforcer 

pub struct TokenBudgetEnforcer { /* private fields */ }

Token budget enforcer

Implements greedy token allocation by priority with optional truncation support.

Implementations

impl TokenBudgetEnforcer

pub fn new(config: TokenBudgetConfig) -> Self

Create a new budget enforcer

pub fn with_budget(budget: usize) -> Self

Create with simple budget (for backwards compatibility)

pub fn with_estimator(budget: usize, estimator: TokenEstimator) -> Self

Create with custom estimator

pub fn with_strategy(self, strategy: AllocationStrategy) -> Self

Set allocation strategy

pub fn reserve(&mut self, tokens: usize)

Reserve tokens for overhead (headers, separators, etc.)

pub fn available(&self) -> usize

Get available budget (total - reserved - allocated)

pub fn total_budget(&self) -> usize

Get total budget

pub fn allocated(&self) -> usize

Get allocated tokens

pub fn try_allocate(&self, tokens: usize) -> bool

Try to allocate tokens (returns true if successful)

pub fn allocate_sections(&self, sections: &[BudgetSection]) -> BudgetAllocation

Allocate sections by priority (dispatches to strategy-specific method)

pub fn reset(&self)

Reset allocation

pub fn estimator(&self) -> &TokenEstimator

Get the estimator