7. 실습 ⑤ — 노드별 성능 프로파일

import time
from collections import defaultdict
from graph import app
from state import init_state
def _new_stat():
    return {"count": 0, "total": 0.0}
def _sort_key(item):
    return -item[1]["total"]
def profile(questions):
    stats = defaultdict(_new_stat)
    for q in questions:
        s = init_state(q)
        s["deadline"] = time.time() + 60
        last = time.time()
        for step in app.stream(s, {"recursion_limit": 40}):
            now = time.time()
            for node in step.keys():
                stats[node]["count"] += 1
                stats[node]["total"] += now - last
            last = now
    print(f"{'노드':<12} {'호출':>5} {'총시간':>8} {'평균':>8}")
    print("-" * 40)
    rows = sorted(stats.items(), key=_sort_key)
    for node, s in rows:
        avg = s["total"] / s["count"]
        print(f"{node:<12} {s['count']:>5} "
              f"{s['total']:>7.1f}s {avg:>7.2f}s")
profile([t["q"] for t in TESTS])

실행 결과 예시
────────────────────────────────────────
노드            호출    총시간     평균
────────────────────────────────────────
generate         25     48.2s    1.93s   ← 최대 병목
verify           25     26.7s    1.07s
rewrite           5      6.1s    1.22s
classify         20      4.3s    0.22s
retrieve         25      3.8s    0.15s
fallback          3      0.0s    0.00s
bump              3      0.0s    0.00s