cost_tracker.py 322 B

12345678910111213
  1. from __future__ import annotations
  2. from dataclasses import dataclass, field
  3. @dataclass
  4. class CostTracker:
  5. total_units: int = 0
  6. events: list[str] = field(default_factory=list)
  7. def record(self, label: str, units: int) -> None:
  8. self.total_units += units
  9. self.events.append(f'{label}:{units}')