ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter10/generative-agents/monitor_campaign.py

@@ -0,0 +1,85 @@
+#!/usr/bin/env python3
+"""Report durable Experiment 10-5 progress without reading credentials."""
+
+from __future__ import annotations
+
+import argparse
+import gzip
+import json
+import os
+from pathlib import Path
+
+
+ARMS = ("baseline", "custom_goal", "no_reflection")
+
+
+def process_alive(pid: int | None) -> bool:
+    if not pid:
+        return False
+    try:
+        os.kill(pid, 0)
+    except OSError:
+        return False
+    return True
+
+
+def receipt_rows(path: Path) -> int:
+    opener = gzip.open if path.suffix == ".gz" else open
+    with opener(path, "rt", encoding="utf-8") as handle:
+        return sum(1 for _ in handle)
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser()
+    parser.add_argument("output", type=Path)
+    args = parser.parse_args()
+    output = args.output.resolve()
+    launch_path = output / "launch.json"
+    launch_by_arm = {}
+    if launch_path.exists():
+        launch = json.loads(launch_path.read_text(encoding="utf-8"))
+        launch_by_arm = {row["arm"]: row for row in launch.get("launches", [])}
+    supervisor_path = output / "supervisor_status.json"
+    if supervisor_path.exists():
+        supervisor = json.loads(supervisor_path.read_text(encoding="utf-8"))
+        for arm, pid in supervisor.get("pids", {}).items():
+            launch_by_arm.setdefault(arm, {})["pid"] = pid
+    result = {"seed": None, "arms": {}}
+    seed_path = output / "seed_status.json"
+    if seed_path.exists():
+        result["seed"] = json.loads(seed_path.read_text(encoding="utf-8"))
+    else:
+        live = output / "receipts" / "seed_history.jsonl"
+        result["seed"] = {
+            "complete": False,
+            "live_receipt_calls": receipt_rows(live) if live.exists() else 0,
+        }
+    for arm in ARMS:
+        status_path = output / "status" / f"{arm}.json"
+        status = json.loads(status_path.read_text(encoding="utf-8")) if status_path.exists() else {
+            "completed_steps": 0,
+            "target_steps": 17_280,
+            "complete": False,
+            "checkpoints": [],
+        }
+        launch = launch_by_arm.get(arm, {})
+        status["pid"] = launch.get("pid")
+        status["process_alive"] = process_alive(launch.get("pid"))
+        receipt_dir = output / "receipts" / arm
+        live = (
+            sorted(
+                path
+                for path in receipt_dir.glob("*.jsonl")
+                if ".failed-" not in path.name
+            )
+            if receipt_dir.exists()
+            else []
+        )
+        status["live_receipt_calls"] = sum(receipt_rows(path) for path in live)
+        result["arms"][arm] = status
+    print(json.dumps(result, indent=2, ensure_ascii=False))
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())