costHook.py 210 B

12345678
  1. from __future__ import annotations
  2. from .cost_tracker import CostTracker
  3. def apply_cost_hook(tracker: CostTracker, label: str, units: int) -> CostTracker:
  4. tracker.record(label, units)
  5. return tracker