Struct ProviderBudget 

pub struct ProviderBudget { /* private fields */ }

Per-provider budget manager.

One ProviderBudget per kernel. The map is keyed by provider id (the same string the engine uses, e.g. "zai").

Implementations

impl ProviderBudget

pub fn new() -> Self

Create an empty budget. Use Self::from_config to populate from the user’s [token-maxing] config.

pub fn from_config(config: &TokenMaxingConfig) -> Self

Build from the user’s config. Providers not eligible under the metered-never rule are silently skipped — the caller should have validated the config first.

pub fn reload(&self, config: &TokenMaxingConfig)

Reload the configured providers while preserving existing usage counters. New providers start fresh at 0; removed providers are dropped; existing providers keep their tokens_used and window.

pub fn reserve(&self, provider: &str, tokens: u64) -> Result<(), ReserveError>

Reserve tokens against provider’s budget.

Ok(()) if the reservation fits; Err otherwise. The reserve here is the second line of defense — the primary gate is the availability verdict before the agent is dispatched.

pub fn release(&self, provider: &str, tokens: u64)

Release tokens back. Used on error/retry to free headroom.

pub fn snapshot(&self, provider: &str) -> Option<ProviderSnapshot>

Read a snapshot of a provider’s state, applying the reset-if-expired rule for callers that need an up-to-the-moment view.

pub fn snapshots(&self) -> Vec<(String, ProviderSnapshot)>

All provider states, snapshotted. Used by the QuotaTracker.

pub fn recalibrate( &self, provider: &str, used_after_calibration: u64, resets_at: Option<DateTime<Utc>>, ) -> bool

Snap a state’s tokens_used to a real value, then close the window. Called by QuotaTracker when a fetcher returns remaining_percent / resets_at.

used_after_calibration is what the counter should show right after the snap (i.e. token_limit - remaining).

Trait Implementations

impl Default for ProviderBudget

fn default() -> Self

Returns the “default value” for a type.