@cocorograph/hub-agent 0.6.22 → 0.6.24

package/package.json

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

package/src/main.mjs

@@ -37,7 +37,12 @@ import {
   listWorktreeStubs,
   removeWorktree as removeWorktreeDir,
 } from "./tmux.mjs"
-import { getSessionUsages, getUsage, recordChatRateLimit } from "./usage.mjs"
+import {
+  contextWindowSize,
+  getSessionUsages,
+  getUsage,
+  recordChatRateLimit,
+} from "./usage.mjs"
 import { getChatSignal, recordChatActivity } from "./chat-signals.mjs"
 
 const logger = pino({ name: "hub-agent" })
@@ -319,15 +324,38 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
 const SESSION_EVENTS_DIR =
   process.env.COCKPIT_SESSION_EVENTS_DIR || "/tmp/cockpit_session_events"
 
-// context 窓サイズ (トークン)。usage.mjs と同じ既定。1M ベータ等は env で上書き。
-const CONTEXT_WINDOW_TOKENS = Number(process.env.HUB_CONTEXT_WINDOW) || 200000
+// context 窓サイズ (トークン) の既定。env HUB_CONTEXT_WINDOW があれば最優先 (手動上書き)。
+// 無ければ起動後に statusLine cache の context_window_size (Opus 4.8 の 1M 等) を
+// refreshContextWindow() で解決して resolvedContextWindow を上書きする。
+const CONTEXT_WINDOW_FALLBACK = Number(process.env.HUB_CONTEXT_WINDOW) || 200000
+
+// 実際に context% 計算で使う窓サイズ。startStateLoop の tick で定期リフレッシュされる。
+// 200k 固定だと 1M ベータのセッションで分母が 5 倍小さく、ドーナツが常に振り切れる
+// (273k トークン → 137% を 100% にクランプ表示) 不具合があったため、実窓サイズを使う。
+let resolvedContextWindow = CONTEXT_WINDOW_FALLBACK
+
+/**
+ * 実コンテキスト窓サイズ (1M ベータ等) を statusLine cache から解決して
+ * resolvedContextWindow を更新する。env HUB_CONTEXT_WINDOW 指定時は手動上書きを
+ * 尊重して何もしない。ファイル読み失敗時は前回値を保持する。
+ */
+export async function refreshContextWindow() {
+  if (process.env.HUB_CONTEXT_WINDOW) return // 手動上書きが最優先
+  try {
+    const size = await contextWindowSize()
+    if (typeof size === "number" && size > 0) resolvedContextWindow = size
+  } catch {
+    /* 前回値を保持 */
+  }
+}
 
 /**
  * SDK assistant メッセージの usage から context 使用率 (%) を概算する。
  * context = input + cache_read + cache_creation + output(直近応答)。
  * プロンプトキャッシュ有効時は cache_* に大半が乗るため 4 種を合算する。
+ * 分母は resolvedContextWindow (実窓サイズ。1M ベータ等を自動反映)。
  */
-function contextPctFromUsage(u) {
+export function contextPctFromUsage(u) {
   if (!u) return null
   const tokens =
     (u.input_tokens || 0) +
@@ -335,7 +363,7 @@ function contextPctFromUsage(u) {
     (u.cache_creation_input_tokens || 0) +
     (u.output_tokens || 0)
   if (tokens <= 0) return null
-  return Math.min(100, Math.round((tokens / CONTEXT_WINDOW_TOKENS) * 1000) / 10)
+  return Math.min(100, Math.round((tokens / resolvedContextWindow) * 1000) / 10)
 }
 
 /**
@@ -483,6 +511,9 @@ function startStateLoop({ client, plugins, logger, intervalMs }) {
   const tick = async () => {
     if (stopped) return
     try {
+      // 実コンテキスト窓サイズ (1M ベータ等) を反映。contextPctFromUsage の分母を
+      // 5s ごとに最新化し、ドーナツが 200k 固定で振り切れるのを防ぐ。
+      await refreshContextWindow()
       const states = await listSessionStates({ plugins, logger })
       for (const s of states) {
         let status = s.status

package/src/usage.mjs

@@ -360,7 +360,7 @@ async function statuslineCacheMtime() {
  * あればそれを使う (1M ベータ等を自動反映)。無ければ HUB_CONTEXT_WINDOW / 200000。
  * ※ cache が stale でも窓サイズ自体は不変なので利用できる。
  */
-async function contextWindowSize() {
+export async function contextWindowSize() {
   const text = await readOrNull(statuslineCache())
   if (text) {
     try {