Struct BudgetManager 

pub struct BudgetManager { /* private fields */ }

Manages budgets for all agents with sliding window reset semantics.

Implementations

impl BudgetManager

pub fn new() -> Self

Creates a new empty budget manager.

pub fn set_budget(&self, limit: BudgetLimit)

Sets or updates the budget for an agent.

pub fn remove_budget(&self, agent_id: &AgentId)

Removes the budget for an agent.

pub fn reserve( &self, agent_id: &AgentId, tokens: u64, ) -> Result<(), BudgetExceeded>

Attempts to reserve tokens for an agent.

Returns Ok(()) if the tokens can be reserved. Returns Err(BudgetExceeded) if the agent has exceeded its token budget.

The usage window is automatically reset if it has expired.

pub fn release(&self, agent_id: &AgentId, tokens_used: u64)

Releases tokens back (e.g., on retry or error).

Tokens are subtracted from usage. Does not allow negative usage.

pub fn track_call(&self, agent_id: &AgentId) -> Result<(), BudgetExceeded>

Tracks a call for an agent.

Returns Err(BudgetExceeded) if the call limit has been exceeded.

pub fn remaining(&self, agent_id: &AgentId) -> BudgetInfo

Returns current budget information for an agent.

pub fn can_schedule(&self, agent_id: &AgentId) -> bool

Returns true if an agent can be scheduled (has budget remaining).

pub fn reset_window(&self, agent_id: &AgentId)

Manually resets the usage window for an agent.

pub fn full_info(&self, agent_id: &AgentId) -> Option<FullBudgetInfo>

Returns full budget information including limits and usage for an agent.

Returns None if no budget is configured for the agent.

pub fn all_full_info(&self) -> Vec<FullBudgetInfo>

Returns full budget info for all agents with configured budgets.

pub fn persist(&self, path: &Path) -> Result<()>

Persist budgets and usage to a JSON file at the given path.

pub fn restore(&self, path: &Path) -> Result<()>

Restore budgets and usage from a JSON file at the given path.

Returns Ok(()) if the file doesn’t exist (empty state). Returns an error if the file exists but cannot be parsed.

Trait Implementations

impl Default for BudgetManager

fn default() -> Self

Returns the “default value” for a type.