Struct TokenBudgetMiddleware 

pub struct TokenBudgetMiddleware { /* private fields */ }

Token budget middleware — tracks and enforces token budgets.

Implementations

impl TokenBudgetMiddleware

pub fn new(max_tokens: usize) -> Self

Create a token budget that terminates the agent once cumulative usage exceeds max_tokens.

pub fn with_cost_tracker( max_tokens: usize, tracker: Arc<CostTracker>, budget: f64, ) -> Self

Create with cost tracker integration for cost-based budget enforcement.

Trait Implementations

impl Middleware for TokenBudgetMiddleware

fn name(&self) -> &str

Human-readable name of this middleware, used for logging and lookup.

fn phases(&self) -> Vec<MiddlewarePhase>

Phases at which this middleware wishes to be invoked.

fn handle<'a>( &'a self, ctx: &'a MiddlewareContext, ) -> Pin<Box<dyn Future<Output = MiddlewareResult> + Send + 'a>>

Inspect the context for the current phase and return a result.