@cocorograph/hub-agent 0.6.7 → 0.6.8

package/package.json

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

package/src/usage.mjs

@@ -245,6 +245,26 @@ async function statuslineCacheMtime() {
   }
 }
 
+/**
+ * context 窓サイズ (トークン) を解決する。statusLine cache に context_window_size が
+ * あればそれを使う (1M ベータ等を自動反映)。無ければ HUB_CONTEXT_WINDOW / 200000。
+ * ※ cache が stale でも窓サイズ自体は不変なので利用できる。
+ */
+async function contextWindowSize() {
+  const text = await readOrNull(statuslineCache())
+  if (text) {
+    try {
+      const j = JSON.parse(text)
+      const cw = j.context_window ?? j.contextWindow
+      const size = cw?.context_window_size ?? cw?.contextWindowSize
+      if (typeof size === "number" && size > 0) return size
+    } catch {
+      /* ignore */
+    }
+  }
+  return CONTEXT_WINDOW
+}
+
 /**
  * 直近にアクティブな session jsonl の最後の assistant.usage から context% を
  * 推定する。チャットモード(SDK headless)は statusLine を書き出さないため、
@@ -279,6 +299,7 @@ async function latestJsonlContext(now) {
     }),
   )
   if (!best) return null
+  const windowSize = await contextWindowSize()
   const text = await readOrNull(best.fp)
   if (!text) return null
   const lines = text.split("\n")
@@ -301,7 +322,7 @@ async function latestJsonlContext(now) {
       (u.cache_creation_input_tokens || 0) +
       (u.output_tokens || 0)
     if (tokens <= 0) continue
-    const percent = Math.min(100, Math.round((tokens / CONTEXT_WINDOW) * 1000) / 10)
+    const percent = Math.min(100, Math.round((tokens / windowSize) * 1000) / 10)
     return { percent, mtimeMs: best.mtimeMs }
   }
   return null