claude-code-conductor 2.21.0__tar.gz → 2.22.0__tar.gz

{claude_code_conductor-2.21.0 → claude_code_conductor-2.22.0}/.claude/hooks/select_tier.py

@@ -323,6 +323,7 @@ def write_tier_selection(
     escalation_reason: str | None = None,
     prompt_prefix: str | None = None,
     prompt_hash: str | None = None,
+    session_id: str | None = None,
 ) -> None:
     """直近の選択結果を ``tier_selection.json`` に書く。
 
@@ -335,6 +336,9 @@ def write_tier_selection(
 
     ``escalated`` / ``escalation_reason`` を任意で含める。
     failure rate に基づく昇格が起きた場合のみ True / 文字列が入る。
+
+    ``session_id`` を任意で含める。UserPromptSubmit payload の session UUID。
+    None のときは tier_selection.json のキー自体を省略する（後方互換）。
     """
     os.makedirs(os.path.dirname(TIER_SELECTION_PATH), exist_ok=True)
     payload: dict[str, object] = {
@@ -354,6 +358,8 @@ def write_tier_selection(
         payload["prompt_prefix"] = prompt_prefix
     if prompt_hash is not None:
         payload["prompt_hash"] = prompt_hash
+    if session_id is not None:
+        payload["session_id"] = session_id
     try:
         with open(TIER_SELECTION_PATH, "w", encoding="utf-8") as f:
             json.dump(payload, f, ensure_ascii=False)
@@ -415,6 +421,8 @@ def main() -> int:
     if not isinstance(prompt, str) or not prompt.strip():
         return 0
 
+    session_id = payload.get("session_id")
+
     # Phase 2-C: 類似度推定で complexity を補強
     heuristic_complexity = estimate_complexity(prompt)
     strong_complexity, weak_matches = similarity_boost(prompt)
@@ -448,6 +456,7 @@ def main() -> int:
         escalated=escalated, escalation_reason=escalation_reason,
         prompt_prefix=prompt_prefix,
         prompt_hash=prompt_hash,
+        session_id=session_id,
     )
 
     context_text = build_additional_context(

{claude_code_conductor-2.21.0 → claude_code_conductor-2.22.0}/.claude/skills/dev-workflow/scripts/record_tier_outcome.py

@@ -210,15 +210,19 @@ def main(argv: list[str] | None = None) -> int:
     # Phase 2-B: tier_recent_outcomes にも 1 件記録（escalation 判定の母数）。
     # tier_bandit 更新が失敗してもこちらは試みる（DB が一時的に詰まる程度なら
     # 片方だけ通る可能性もある）。
+    # v2.22.0: session_id を渡して cost 紐づけを可能にする。
+    # session_id キーが無い古い tier_selection.json の場合は None になる（後方互換）。
+    session_id = selection.get("session_id")
     try:
         c3_db.record_tier_recent_outcome(
             complexity=selection["complexity"],
             tier=selection["tier"],
             success=success,
+            session_id=session_id,
         )
     except Exception as exc:  # noqa: BLE001
         print(
-            f"[record_tier_outcome] tier_recent_outcomes record skipped: {exc}",
+            f"[record_tier_outcome] tier_recent_outcomes record skipped: {type(exc).__name__}",
             file=sys.stderr,
         )
 

{claude_code_conductor-2.21.0 → claude_code_conductor-2.22.0}/CHANGELOG.md

@@ -1,5 +1,31 @@
 # Changelog
 
+## [2.22.0] - 2026-05-25
+
+**tier-routing cost 紐づけデータ蓄積**: tier_recent_outcomes に session_id 列を追加し、agent_cost_runs と JOIN できるデータ基盤を整備する。`c3 tier stats` に complexity×tier 別の平均コストセクションを追加する。cost-aware routing 本体は v2.23.0。
+
+**スコープ注記**: 単価テーブル自動更新は公式 price API が非提供のため pricing.py の手動メンテ継続。tier_bandit の total_cost_usd / cost_samples 列は v2.23.0 用に確保のみ（書き込み・読み出しなし）。
+
+### 機能追加
+
+- **`src/c3/migrations/003_tier_cost.sql`（新規）**: `tier_recent_outcomes` に `session_id TEXT` 列と `idx_tier_recent_session` インデックスを追加。`tier_bandit` に `total_cost_usd REAL DEFAULT 0.0` / `cost_samples INTEGER DEFAULT 0` 列を追加（v2.23.0 用確保のみ）。既存データは ADD COLUMN DEFAULT で保持（破壊的変更なし）。
+
+- **`src/c3/db.py`: `record_tier_recent_outcome` に `session_id` 追加**: kw-only 引数 `session_id: str | None = None` を追加。既存呼び出しは後方互換（省略時 NULL 保存）。
+
+- **`src/c3/db.py`: `read_tier_cost_summary` 追加**: tier_recent_outcomes × agent_cost_runs を session_id で JOIN し、complexity×tier 別の sessions / total_cost_usd / avg_cost_usd を返す。2 段 CTE で session コストの 1 session 内重複計上を防ぐ（mainline 除外・session 単位 SUM → DISTINCT JOIN）。テーブル不在・データ不在・session_id 全 NULL で `[]`。
+
+- **`.claude/hooks/select_tier.py`: session_id 記録**: UserPromptSubmit payload から session_id を取得し tier_selection.json に追記。session_id が None のときはキーを省略（既存テスト互換）。
+
+- **`.claude/skills/dev-workflow/scripts/record_tier_outcome.py`: session_id 受け渡し**: tier_selection.json から session_id を読み `record_tier_recent_outcome` に渡す。session_id キーが無い古い json でも動作（None として扱う）。
+
+- **`c3 tier stats` に Tier 別平均コストセクション追加**: `_collect_snapshot()` に `read_tier_cost_summary()` の結果を `tier_cost` キーで追加。human 表示に「Tier 別平均コスト（粗い概算 / 精度向上は v2.23.0）」セクションを追加。tier_cost が空のときは「（cost 紐づけデータ未収集）」と表示。`--json` 出力の `tier_cost` キーにも自動反映。
+
+### 変更
+
+- **`src/c3/__init__.py`**: `__version__` を `"2.21.0"` から `"2.22.0"` に更新。
+
+- **注記更新**: `c3 tier stats` の「cost-aware routing は v2.22.0 予定」を「データ紐づけ蓄積。cost-aware routing 本体は v2.23.0 予定」へ更新。
+
 ## [2.21.0] - 2026-05-25
 
 **tier-routing コスト統合（データ収集基盤）**: Claude Code セッションログ（`~/.claude/projects/<slug>/<session>.jsonl` + subagent jsonl）を読み込み、モデル単価で USD 換算して c3.db に蓄積するデータ収集基盤を整備する。将来の cost-aware routing（v2.22.0）の土台。

{claude_code_conductor-2.21.0 → claude_code_conductor-2.22.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: claude-code-conductor
-Version: 2.21.0
+Version: 2.22.0
 Summary: Multi-agent orchestration framework for Claude Code with Codex/Cursor adapters (C3)
 Project-URL: Homepage, https://github.com/satoh-y-0323/claude-code-conductor
 Project-URL: Repository, https://github.com/satoh-y-0323/claude-code-conductor

{claude_code_conductor-2.21.0 → claude_code_conductor-2.22.0}/src/c3/__init__.py

@@ -1,3 +1,3 @@
 """Claude Code Conductor (C3) - multi-agent orchestration framework for Claude Code."""
 
-__version__ = "2.21.0"
+__version__ = "2.22.0"

{claude_code_conductor-2.21.0 → claude_code_conductor-2.22.0}/src/c3/cli_tier.py

@@ -89,7 +89,7 @@ def handle_stats(args: argparse.Namespace) -> int:
 
 
 def _collect_snapshot(db_path, recent_limit: int) -> dict[str, Any]:
-    """DB から tier_bandit / tier_recent_outcomes / agent_cost を読み snapshot dict を返す。"""
+    """DB から tier_bandit / tier_recent_outcomes / agent_cost / tier_cost を読み snapshot dict を返す。"""
     bandit_rows: list[dict[str, Any]] = []
     total_trials = 0
 
@@ -116,6 +116,8 @@ def _collect_snapshot(db_path, recent_limit: int) -> dict[str, Any]:
 
     agent_cost: list[dict[str, Any]] = c3_db.read_agent_cost_summary(db_path=db_path)
 
+    tier_cost: list[dict[str, Any]] = c3_db.read_tier_cost_summary(db_path=db_path)
+
     if total_trials < _LEARNING_THRESHOLD:
         mode = "uniform"
     else:
@@ -130,6 +132,7 @@ def _collect_snapshot(db_path, recent_limit: int) -> dict[str, Any]:
         "tier_bandit": bandit_rows,
         "recent_outcomes": recent_outcomes,
         "agent_cost": agent_cost,
+        "tier_cost": tier_cost,
     }
 
 
@@ -190,4 +193,19 @@ def _render_human(snapshot: dict[str, Any]) -> None:
                 f"{note}"
             )
     print()
-    print("（注: 本リリースはデータ収集基盤のみ。cost-aware routing は v2.22.0 予定）")
+
+    print("== Tier 別平均コスト（粗い概算 / 精度向上は v2.23.0） ==")
+    tier_cost = snapshot.get("tier_cost", [])
+    if not tier_cost:
+        print("（cost 紐づけデータ未収集）")
+    else:
+        print(f"{'complexity':<12} {'tier':<8} {'sessions':>8}  {'avg_usd':>10}  {'total_usd':>10}")
+        for row in tier_cost:
+            print(
+                f"{row['complexity']:<12} {row['tier']:<8} "
+                f"{row['sessions']:>8}  "
+                f"${row['avg_cost_usd']:>9.4f}  "
+                f"${row['total_cost_usd']:>9.4f}"
+            )
+    print()
+    print("（注: データ紐づけ蓄積。cost-aware routing 本体は v2.23.0 予定）")

{claude_code_conductor-2.21.0 → claude_code_conductor-2.22.0}/src/c3/db.py

@@ -358,12 +358,21 @@ def record_tier_recent_outcome(
     tier: str,
     success: bool,
     db_path: Path | None = None,
+    session_id: str | None = None,
 ) -> bool:
     """``tier_recent_outcomes`` に 1 件 INSERT する。
 
     Phase 2-B のエスカレーション判定用。tier_bandit の累積 α/β とは別に、
     直近 N 件の event を時系列で保持する。
 
+    Args:
+        complexity: 'simple' | 'medium' | 'complex'。
+        tier: 'haiku' | 'sonnet' | 'opus'。
+        success: True なら成功、False なら失敗。
+        db_path: c3.db のパス。省略時は :func:`locate_c3_db` で探索。
+        session_id: セッション UUID（v2.22.0+）。省略時は NULL で保存。
+            cost 紐づけに使用。None の場合は既存行との後方互換を維持する。
+
     Returns:
         INSERT 成功時 True、DB 不在 / エラー時 False。
     """
@@ -382,15 +391,15 @@ def record_tier_recent_outcome(
             _apply_busy_timeout(conn)
             conn.execute(
                 "INSERT INTO tier_recent_outcomes "
-                "(task_complexity, tier, success, ts) VALUES (?, ?, ?, ?)",
-                (complexity, tier, 1 if success else 0, now_iso),
+                "(task_complexity, tier, success, ts, session_id) VALUES (?, ?, ?, ?, ?)",
+                (complexity, tier, 1 if success else 0, now_iso, session_id),
             )
             conn.commit()
         finally:
             conn.close()
         return True
     except Exception as exc:  # noqa: BLE001
-        logger.warning("failed to record tier_recent_outcome: %s", exc)
+        logger.warning("failed to record tier_recent_outcome: %s", type(exc).__name__)
         return False
 
 
@@ -657,6 +666,99 @@ def read_agent_cost_summary(
     ]
 
 
+def read_tier_cost_summary(
+    *,
+    db_path: Path | None = None,
+) -> list[dict]:
+    """tier_recent_outcomes × agent_cost_runs を session_id で JOIN し、
+    complexity×tier 別コストを返す（粗い概算 / 精度向上は v2.23.0）。
+
+    2 段 CTE で重複計上を防ぐ:
+      - ``session_cost`` CTE: agent_cost_runs を mainline 除外しつつ
+        session_id 単位で SUM（1 session = 1 行）→ cost の重複計上を構造的に防ぐ
+      - ``outcome_sessions`` CTE: tier_recent_outcomes を
+        (session_id, task_complexity, tier) で DISTINCT 化。
+        session_id が NULL（v2.22.0 移行前行）は除外
+      - JOIN して complexity×tier 別に sessions / total_cost_usd / avg_cost_usd を算出
+
+    既知の不正確性（v2.22.0 許容・精度向上は v2.23.0）:
+      - session が複数の (complexity, tier) outcome を持つ場合、同一 session の cost が
+        複数の (complexity,tier) 行に重複加算されうる（1 session 内での cost 重複は防ぐが、
+        複数 outcome を持つ session の cross-outcome 重複は未解決）
+      - cost は session 全体の non-mainline 合計であり、特定 tier の model 行に限定しない
+      - agent_id 単位の紐づけはしていない
+
+    Args:
+        db_path: c3.db のパス。省略時は :func:`locate_c3_db` で探索。
+
+    Returns:
+        各行を dict にしたリスト。キー:
+        ``complexity``(str) / ``tier``(str) / ``sessions``(int) /
+        ``total_cost_usd``(float) / ``avg_cost_usd``(float)。
+        ORDER BY total_cost_usd DESC。
+        テーブル不在 / データ不在 / session_id 全 NULL / JOIN 0 行 /
+        DB 不在 / エラー時は空リストを返す。
+    """
+    if db_path is None:
+        db_path = locate_c3_db()
+        if db_path is None:
+            return []
+
+    try:
+        conn = sqlite3.connect(str(db_path))
+        try:
+            # WAL は書き込みヘルパー呼び出し時または migrate 時に設定済みの前提
+            # （read_agent_cost_summary 等の read ヘルパーと同方針 / CR-M-001）
+            _apply_busy_timeout(conn)
+            rows = conn.execute(
+                "WITH session_cost AS ("
+                "    SELECT session_id,"
+                "           SUM(total_cost_usd) AS session_cost_usd"
+                "    FROM agent_cost_runs"
+                "    WHERE agent_type <> 'mainline'"
+                "    GROUP BY session_id"
+                "),"
+                "outcome_sessions AS ("
+                "    SELECT DISTINCT session_id, task_complexity, tier"
+                "    FROM tier_recent_outcomes"
+                "    WHERE session_id IS NOT NULL"
+                ") "
+                "SELECT o.task_complexity            AS complexity,"
+                "       o.tier                       AS tier,"
+                "       COUNT(DISTINCT o.session_id) AS sessions,"
+                "       SUM(sc.session_cost_usd)     AS total_cost_usd,"
+                "       SUM(sc.session_cost_usd) / COUNT(DISTINCT o.session_id)"
+                "           AS avg_cost_usd "
+                "FROM outcome_sessions o "
+                "JOIN session_cost sc ON sc.session_id = o.session_id "
+                "GROUP BY o.task_complexity, o.tier "
+                "ORDER BY total_cost_usd DESC"
+            ).fetchall()
+        finally:
+            conn.close()
+    except sqlite3.OperationalError as exc:
+        # テーブル不在（no such table）は [] を返す（DB 未初期化でも止めない）
+        logger.debug(
+            "read_tier_cost_summary: table not found or inaccessible: %s",
+            type(exc).__name__,
+        )
+        return []
+    except Exception as exc:  # noqa: BLE001
+        logger.warning("read_tier_cost_summary: unexpected error: %s", type(exc).__name__)
+        return []
+
+    return [
+        {
+            "complexity": row[0],
+            "tier": row[1],
+            "sessions": int(row[2]),
+            "total_cost_usd": float(row[3]) if row[3] is not None else 0.0,
+            "avg_cost_usd": float(row[4]) if row[4] is not None else 0.0,
+        }
+        for row in rows
+    ]
+
+
 def get_ingest_offset(
     file_key: str,
     *,

claude_code_conductor-2.22.0/src/c3/migrations/003_tier_cost.sql

@@ -0,0 +1,25 @@
+-- C3 SQLite migration 003: tier_recent_outcomes に session_id 列・INDEX を追加し、
+--                           tier_bandit にコスト集計用列（v2.23.0 用）を確保する。
+--
+-- 変更内容:
+--   - tier_recent_outcomes.session_id TEXT（既存行は NULL）
+--   - idx_tier_recent_session ON tier_recent_outcomes(session_id)
+--   - tier_bandit.total_cost_usd REAL NOT NULL DEFAULT 0.0  （v2.23.0 用確保のみ）
+--   - tier_bandit.cost_samples   INTEGER NOT NULL DEFAULT 0  （v2.23.0 用確保のみ）
+--
+-- 後方互換: ADD COLUMN DEFAULT により既存行は NULL / 0.0 / 0 で埋まる。
+
+BEGIN;
+
+ALTER TABLE tier_recent_outcomes ADD COLUMN session_id TEXT;
+
+CREATE INDEX IF NOT EXISTS idx_tier_recent_session
+    ON tier_recent_outcomes(session_id);
+
+ALTER TABLE tier_bandit ADD COLUMN total_cost_usd REAL NOT NULL DEFAULT 0.0;
+
+ALTER TABLE tier_bandit ADD COLUMN cost_samples INTEGER NOT NULL DEFAULT 0;
+
+INSERT OR IGNORE INTO schema_migrations (version) VALUES ('003');
+
+COMMIT;

{claude_code_conductor-2.21.0 → claude_code_conductor-2.22.0}/tests/hooks/test_record_tier_outcome.py

@@ -499,3 +499,118 @@ class TestClaudeDirGuard:
 
         with pytest.raises(RuntimeError, match=r"_CLAUDE_DIR resolution broke"):
             self._load_from_path(wrong_hook)
+
+
+# ---------------------------------------------------------------------------
+# T4 (v2.22.0): session_id 受け渡し検証
+# ---------------------------------------------------------------------------
+
+
+class TestSessionIdPropagation:
+    """v2.22.0 T4: record_tier_outcome が tier_selection.json の session_id を
+    record_tier_recent_outcome に渡すことを検証する。"""
+
+    def _write_selection_with_session(
+        self,
+        path: Path,
+        *,
+        complexity: str,
+        tier: str,
+        session_id: str,
+    ) -> None:
+        path.parent.mkdir(parents=True, exist_ok=True)
+        path.write_text(
+            json.dumps({
+                "complexity": complexity,
+                "tier": tier,
+                "mode": "thompson",
+                "session_id": session_id,
+            }),
+            encoding="utf-8",
+        )
+
+    def test_session_id_passed_to_record_tier_recent_outcome(
+        self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
+    ) -> None:
+        """tier_selection.json に session_id があるとき、
+        record_tier_recent_outcome に session_id が渡されること。
+        monkeypatch で呼び出し引数をキャプチャして検証する。
+        """
+        db_path = tmp_path / "c3.db"
+        _create_c3_db(db_path)
+        sel_path = tmp_path / "state" / "tier_selection.json"
+        self._write_selection_with_session(
+            sel_path,
+            complexity="simple",
+            tier="haiku",
+            session_id="test-session-abc123",
+        )
+
+        mod = _load_hook_module("record_tier_outcome_t4_a")
+        monkeypatch.setattr(mod, "TIER_SELECTION_PATH", str(sel_path))
+
+        from c3 import db as c3_db
+        monkeypatch.setattr(c3_db, "locate_c3_db", lambda start=None: db_path)
+
+        # record_tier_recent_outcome の呼び出し引数をキャプチャ
+        captured: list[dict] = []
+        original_fn = c3_db.record_tier_recent_outcome
+
+        def _capture(**kwargs):  # type: ignore[no-untyped-def]
+            captured.append(kwargs)
+            return original_fn(**kwargs)
+
+        monkeypatch.setattr(c3_db, "record_tier_recent_outcome", _capture)
+
+        rc = mod.main(["--outcome", "success"])
+        assert rc == 0
+
+        assert len(captured) == 1
+        assert captured[0]["session_id"] == "test-session-abc123"
+        assert captured[0]["complexity"] == "simple"
+        assert captured[0]["tier"] == "haiku"
+        assert captured[0]["success"] is True
+
+    def test_no_session_id_in_json_passes_none(
+        self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
+    ) -> None:
+        """session_id キーが無い古い tier_selection.json（旧フォーマット）でも
+        crash せず、record_tier_recent_outcome に session_id=None が渡ること。
+        既存の tier_bandit 更新・recent_outcomes 記録が従来通り行われること。
+        """
+        db_path = tmp_path / "c3.db"
+        _create_c3_db(db_path)
+        sel_path = tmp_path / "state" / "tier_selection.json"
+        # session_id キーを含まない旧フォーマット
+        _write_selection(sel_path, "medium", "sonnet")
+
+        mod = _load_hook_module("record_tier_outcome_t4_b")
+        monkeypatch.setattr(mod, "TIER_SELECTION_PATH", str(sel_path))
+
+        from c3 import db as c3_db
+        monkeypatch.setattr(c3_db, "locate_c3_db", lambda start=None: db_path)
+
+        # record_tier_recent_outcome の呼び出し引数をキャプチャ
+        captured: list[dict] = []
+        original_fn = c3_db.record_tier_recent_outcome
+
+        def _capture(**kwargs):  # type: ignore[no-untyped-def]
+            captured.append(kwargs)
+            return original_fn(**kwargs)
+
+        monkeypatch.setattr(c3_db, "record_tier_recent_outcome", _capture)
+
+        rc = mod.main(["--outcome", "success"])
+        assert rc == 0
+
+        # crash していない & session_id=None が渡されている
+        assert len(captured) == 1
+        assert captured[0]["session_id"] is None
+        assert captured[0]["complexity"] == "medium"
+        assert captured[0]["tier"] == "sonnet"
+
+        # tier_bandit が従来通り更新されている
+        params = c3_db.read_tier_params("medium", db_path=db_path)
+        assert params["sonnet"] == (2.0, 1.0, 1)
+        # tier_selection.json も削除されている
+        assert not sel_path.exists()

{claude_code_conductor-2.21.0 → claude_code_conductor-2.22.0}/tests/hooks/test_select_tier.py

@@ -411,3 +411,85 @@ class TestMaskSecrets:
         data = json.loads((tmp_path / "tier_selection.json").read_text(encoding="utf-8"))
         assert "ghp_12345678abcdef" not in data.get("prompt_prefix", "")
         assert "token=***" in data.get("prompt_prefix", "")
+
+
+# ---------------------------------------------------------------------------
+# T3: session_id 記録（AC-3 / AC-9）
+# ---------------------------------------------------------------------------
+
+
+class TestSessionIdRecording:
+    """write_tier_selection / main の session_id 記録を検証する。"""
+
+    def test_write_tier_selection_with_session_id(
+        self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
+    ) -> None:
+        """session_id が渡されると tier_selection.json に session_id キーが入る。"""
+        mod = _load_hook_module()
+        target = tmp_path / "tier_selection.json"
+        monkeypatch.setattr(mod, "TIER_SELECTION_PATH", str(target))
+
+        mod.write_tier_selection("medium", "sonnet", "thompson", session_id="sess-abc123")
+        data = json.loads(target.read_text(encoding="utf-8"))
+        assert "session_id" in data
+        assert data["session_id"] == "sess-abc123"
+
+    def test_write_tier_selection_without_session_id_no_key(
+        self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
+    ) -> None:
+        """session_id が渡されない（None）ときは tier_selection.json に session_id キーが入らない。
+
+        既存の dict 完全一致テスト（test_write_tier_selection）と同形の確認。
+        """
+        mod = _load_hook_module()
+        target = tmp_path / "tier_selection.json"
+        monkeypatch.setattr(mod, "TIER_SELECTION_PATH", str(target))
+
+        mod.write_tier_selection("complex", "opus", "thompson")
+        data = json.loads(target.read_text(encoding="utf-8"))
+        # session_id キーが存在しないことを確認
+        assert "session_id" not in data
+        # 既存テストと同一の期待 dict と一致する
+        assert data == {
+            "complexity": "complex",
+            "tier": "opus",
+            "mode": "thompson",
+            "suggested_model": "opus",
+        }
+
+    def test_main_records_session_id_when_present(
+        self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture
+    ) -> None:
+        """payload に session_id があるとき tier_selection.json に session_id が入る。"""
+        mod = _load_hook_module()
+        monkeypatch.setattr(
+            mod, "TIER_SELECTION_PATH",
+            str(tmp_path / "tier_selection.json"),
+        )
+        payload = {"prompt": "新しい機能を追加してください", "session_id": "sess-xyz-999"}
+        monkeypatch.setattr(sys, "stdin", io.StringIO(json.dumps(payload)))
+
+        rc = mod.main()
+        assert rc == 0
+
+        data = json.loads((tmp_path / "tier_selection.json").read_text(encoding="utf-8"))
+        assert "session_id" in data
+        assert data["session_id"] == "sess-xyz-999"
+
+    def test_main_no_session_id_in_payload_no_key_in_json(
+        self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture
+    ) -> None:
+        """payload に session_id が無いとき tier_selection.json に session_id キーが入らず crash しない。"""
+        mod = _load_hook_module()
+        monkeypatch.setattr(
+            mod, "TIER_SELECTION_PATH",
+            str(tmp_path / "tier_selection.json"),
+        )
+        payload = {"prompt": "テストを書いてください"}
+        monkeypatch.setattr(sys, "stdin", io.StringIO(json.dumps(payload)))
+
+        rc = mod.main()
+        assert rc == 0
+
+        data = json.loads((tmp_path / "tier_selection.json").read_text(encoding="utf-8"))
+        assert "session_id" not in data