diff --git a/src/energiamesh/__init__.py b/src/energiamesh/__init__.py
index f815d94..0bac338 100644
--- a/src/energiamesh/__init__.py
+++ b/src/energiamesh/__init__.py
@@ -12,6 +12,7 @@ and a registry as the project evolves.
 """
 
 from .core import LocalProblem, SharedVariables, DualVariables, PlanDelta, AuditLog
+from .core import SafetyBudget, PrivacyBudget
 from .transport import TLSChannel
 from .registry import GraphOfContracts
 
@@ -21,6 +22,8 @@ __all__ = [
     "DualVariables",
     "PlanDelta",
     "AuditLog",
+    "SafetyBudget",
+    "PrivacyBudget",
     "GraphOfContracts",
     "TLSChannel",
 ]