@cocorograph/hub-agent 0.6.16 → 0.6.17

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.6.16",
+  "version": "0.6.17",
   "description": "Hub Hosted Cockpit のローカル常駐 agent。Hub と outbound WSS で接続し、ローカルの tmux/pty を中継する。",
   "type": "module",
   "license": "UNLICENSED",

package/src/claude-stream-bridge.mjs

@@ -268,14 +268,89 @@ class ClaudeStreamSession {
   /** 再アタッチ: 走行中(または生存中)セッションに新しい stream_id を紐付け直し、
    *  idle 撤去タイマーを止める。以降のターンイベントはこの stream_id 経由で新しい
    *  browser 接続へライブに流れる (= 通常の生成中表示と同じ)。再アタッチ前の確定分は
-   *  browser 側の jsonl hydrate (history.request) で復元するため、ここでは replay しない。 */
-  reattach(stream_id) {
+   *  browser 側の jsonl hydrate (history.request) で復元するため、ここでは replay しない。
+   *
+   *  改修5 (2026-05-29): モデル/権限/拡張思考のターン切替。再アタッチ時に opts で
+   *  新しい値が渡され、現在値と異なれば applyRuntimeOptions で反映する。これにより
+   *  入力欄下バッジの変更が「同一セッション(常駐 query)を維持したまま次ターンから」
+   *  効くようになる。従来は reattach が stream_id だけ差し替えていたため、起動済み
+   *  query の model/permission/thinking は初期値のまま変わらなかった (バッジが効かない
+   *  不具合の原因)。 */
+  reattach(stream_id, opts = undefined) {
     this.stream_id = stream_id
     this._detached = false
     if (this._idleTimer) {
       clearTimeout(this._idleTimer)
       this._idleTimer = null
     }
+    if (opts) this.applyRuntimeOptions(opts)
+  }
+
+  /** 改修5: モデル/権限/拡張思考をランタイムに切り替える。
+   *
+   *  - 保持フィールド (this.model/permissionMode/maxThinkingTokens) を更新する。これは
+   *    常駐 query が異常終了して resume 再起動する際 (_runResidentQuery) に最新値で
+   *    再 spawn させるため、および per-message セッションが次ターンの options に反映する
+   *    ため。
+   *  - 起動済みの常駐 query があれば SDK の制御メソッド (setModel / setPermissionMode /
+   *    setMaxThinkingTokens) を呼び、プロセス再起動なしで次ターンから即反映する
+   *    (公式 streaming input mode のランタイム制御。stdin に control_request を流す)。
+   *
+   *  値が undefined のキーは「変更なし」として無視する (バッジ未送出時に既存値を消さない)。
+   *  model に空文字/null が来たら setModel(undefined) でデフォルトへ戻す。
+   *  maxThinkingTokens に 0/null が来たら setMaxThinkingTokens(null) でオフにする。 */
+  applyRuntimeOptions({ model, permissionMode, maxThinkingTokens } = {}) {
+    const q = this._residentQuery
+    // モデル
+    if (model !== undefined) {
+      const next = model || null
+      if (next !== this.model) {
+        this.model = next
+        if (q && typeof q.setModel === "function") {
+          q.setModel(next || undefined).catch((err) =>
+            this.logger?.warn(
+              { err: err?.message, stream_id: this.stream_id },
+              "setModel failed",
+            ),
+          )
+        }
+      }
+    }
+    // 権限モード
+    if (permissionMode !== undefined) {
+      const next = permissionMode || null
+      if (next !== this.permissionMode) {
+        this.permissionMode = next
+        // setPermissionMode は有効な mode を要求する。null/空 (=未指定へ戻す) は
+        // SDK 側に「解除」API が無いため、保持値の更新のみ (次回 spawn で既定に従う)。
+        if (next && q && typeof q.setPermissionMode === "function") {
+          q.setPermissionMode(next).catch((err) =>
+            this.logger?.warn(
+              { err: err?.message, stream_id: this.stream_id },
+              "setPermissionMode failed",
+            ),
+          )
+        }
+      }
+    }
+    // 拡張思考予算
+    if (maxThinkingTokens !== undefined) {
+      const next =
+        typeof maxThinkingTokens === "number" && maxThinkingTokens > 0
+          ? maxThinkingTokens
+          : null
+      if (next !== this.maxThinkingTokens) {
+        this.maxThinkingTokens = next
+        if (q && typeof q.setMaxThinkingTokens === "function") {
+          q.setMaxThinkingTokens(next).catch((err) =>
+            this.logger?.warn(
+              { err: err?.message, stream_id: this.stream_id },
+              "setMaxThinkingTokens failed",
+            ),
+          )
+        }
+      }
+    }
   }
 
   /** soft detach: browser 切断時にターンを中断せずセッションを生かしたまま detached に
@@ -740,10 +815,25 @@ export class ClaudeStreamBridge extends EventEmitter {
       const live = this._liveBySession.get(resumeSessionId)
       if (live && !live._closed) {
         this.sessions.delete(live.stream_id)
-        live.reattach(stream_id)
+        // 改修5: 再アタッチ時に model/permission/maxThinkingTokens を引き継ぎ反映する。
+        // browser はバッジ変更時に新しい値を載せた claude.attach を同一 resume で送る
+        // ため、ここで適用すると常駐 query を維持したまま次ターンから切り替わる。
+        live.reattach(stream_id, {
+          model,
+          permissionMode,
+          maxThinkingTokens:
+            typeof maxThinkingTokens === "number" ? maxThinkingTokens : null,
+        })
         this.sessions.set(stream_id, live)
         this.logger?.info(
-          { stream_id, resume: resumeSessionId, busy: live._busy },
+          {
+            stream_id,
+            resume: resumeSessionId,
+            busy: live._busy,
+            model: live.model,
+            permissionMode: live.permissionMode,
+            maxThinkingTokens: live.maxThinkingTokens,
+          },
           "claude stream reattached to live session",
         )
         return { stream_id, resuming: true, reattached: true }