aa-gateway 0.0.1-beta.1

Control plane — policy enforcement engine and agent registry for Agent Assembly
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

//! Budget tracking engine for `aa-gateway`.
//!
//! Entry point: [`tracker::BudgetTracker::record_usage`].

pub mod types;
pub use types::{BudgetAlert, BudgetState, BudgetStatus, BudgetWindow, Model, Provider};

pub mod pricing;
pub use pricing::{PricingEntry, PricingLoadError, PricingTable};

pub mod persistence;
pub use persistence::{
    default_budget_path, load_from_disk, save_to_disk_atomic, start_background_writer, start_window_flush_task,
    PersistedAgentEntry, PersistedBudget, PersistenceError,
};

pub mod tracker;
pub use tracker::BudgetTracker;

pub mod rollup;
pub use rollup::{compute_budget_rollup, BudgetRollup, BudgetRow};