init: company-haness 설계

.claude/hooks/bench_cascade/budget.py

@@ -0,0 +1,49 @@
+"""전 유료 모델 호출(arm-run·calibrate·judge·retry·LLM sanitize)에 대한 run-level 예산 receipt.
+receipt 없이 실행 거부 — 우발적 대량 API 소비 방지(Blocker 4)."""
+import json
+import os
+
+
+def approve(plan_id, max_tokens, max_cost, out_path):
+    rec = {"plan-id": plan_id, "max-tokens": int(max_tokens), "max-cost": float(max_cost),
+           "spent-tokens": 0, "spent-cost": 0.0}
+    os.makedirs(os.path.dirname(out_path), exist_ok=True)
+    with open(out_path, "w", encoding="utf-8") as f:
+        json.dump(rec, f)
+    return rec
+
+
+def load(path):
+    if not os.path.exists(path):
+        return None
+    with open(path, encoding="utf-8") as f:
+        return json.load(f)
+
+
+def remaining(path):
+    r = load(path)
+    if r is None:
+        return {"tokens": 0, "cost": 0.0}
+    return {"tokens": r["max-tokens"] - r["spent-tokens"], "cost": r["max-cost"] - r["spent-cost"]}
+
+
+def charge(path, tokens, cost):
+    r = load(path)
+    if r is None:
+        raise RuntimeError("예산 receipt 없음 — approve-budget 먼저")
+    if r["spent-tokens"] + tokens > r["max-tokens"] or r["spent-cost"] + cost > r["max-cost"]:
+        raise ValueError(f"예산 초과: 요구 {tokens}tok/{cost}$ > 잔여 {remaining(path)}")
+    r["spent-tokens"] += int(tokens)
+    r["spent-cost"] += float(cost)
+    with open(path, "w", encoding="utf-8") as f:
+        json.dump(r, f)
+    return r
+
+
+def require(path):
+    r = load(path)
+    if r is None:
+        raise RuntimeError("예산 receipt 없음 — 유료 실행 거부(approve-budget 필요)")
+    if r["max-tokens"] - r["spent-tokens"] <= 0 or r["max-cost"] - r["spent-cost"] <= 0:
+        raise RuntimeError("예산 소진 — 유료 실행 거부")
+    return r