@cocorograph/hub-agent 0.6.7 → 0.6.9

package/package.json

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

package/src/main.mjs

@@ -37,7 +37,7 @@ import {
   listWorktreeStubs,
   removeWorktree as removeWorktreeDir,
 } from "./tmux.mjs"
-import { getSessionUsages, getUsage } from "./usage.mjs"
+import { getSessionUsages, getUsage, recordChatRateLimit } from "./usage.mjs"
 
 const logger = pino({ name: "hub-agent" })
 
@@ -179,6 +179,15 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
   // claude.error を返す経路に分岐 (dispatch 側で判定)。
   if (claudeBridge) {
     claudeBridge.on("event", ({ stream_id, session_id, event }) => {
+      // SDK の rate_limit_event を捕捉して usage の 5h/7d 実値に反映する
+      // (チャットモードでは statusLine が更新されないため、これが live 値の唯一の源)。
+      if (event?.type === "rate_limit_event" && event.rate_limit_info) {
+        try {
+          recordChatRateLimit(event.rate_limit_info)
+        } catch {
+          /* ignore */
+        }
+      }
       client.send({ type: "claude.event", stream_id, session_id, event })
     })
     claudeBridge.on("permission", ({ stream_id, request_id, tool_name, input }) => {

package/src/usage.mjs

@@ -107,6 +107,42 @@ async function readOrNull(p) {
   }
 }
 
+// ---------------------------------------------------------------------------
+// チャット(SDK) の rate_limit_event から取得した最新 rate-limit (プロセス内共有)。
+// statusLine cache はターミナルでしか更新されないため、チャットモードでは
+// SDK stream の rate_limit_event を捕捉してここに溜め、getUsage で 5h/7d に反映する。
+// ---------------------------------------------------------------------------
+const chatRateLimits = {
+  /** @type {{percent:number, resetAtMs:number|null}|null} */ five_hour: null,
+  /** @type {{percent:number, resetAtMs:number|null}|null} */ seven_day: null,
+  updatedAtMs: 0,
+}
+
+/** utilization (0-1 割合 or 0-100 %) を % に正規化。 */
+function utilizationToPercent(u) {
+  if (typeof u !== "number" || !Number.isFinite(u)) return null
+  return u <= 1 ? Math.round(u * 1000) / 10 : Math.round(u * 10) / 10
+}
+
+/**
+ * SDK の SDKRateLimitInfo を取り込む (main.mjs の rate_limit_event ハンドラから呼ぶ)。
+ * rateLimitType: 'five_hour' / 'seven_day*' を 5h / 7d スロットに振り分ける。
+ */
+export function recordChatRateLimit(info) {
+  if (!info || typeof info !== "object") return
+  const type = info.rateLimitType
+  const slot = type === "five_hour" ? "five_hour" : typeof type === "string" && type.startsWith("seven_day") ? "seven_day" : null
+  if (!slot) return // overage 等は対象外
+  const percent = utilizationToPercent(info.utilization)
+  if (percent === null) return
+  let resetAtMs = null
+  if (typeof info.resetsAt === "number") {
+    resetAtMs = info.resetsAt < 1e12 ? info.resetsAt * 1000 : info.resetsAt
+  }
+  chatRateLimits[slot] = { percent, resetAtMs }
+  chatRateLimits.updatedAtMs = Date.now()
+}
+
 async function readOfficial(now) {
   const text = await readOrNull(statuslineCache())
   if (!text) return null
@@ -245,6 +281,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 +335,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 +358,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
@@ -318,10 +375,10 @@ export async function getUsage() {
   const now = Date.now()
   const official = await readOfficial(now)
   const result = official || (await readEstimate(now))
+  const cacheMtime = await statuslineCacheMtime()
 
   const est = await latestJsonlContext(now)
   if (est) {
-    const cacheMtime = await statuslineCacheMtime()
     // statusLine が jsonl と同程度以上に新しければ official.context が信頼できる
     // (ターミナルモード)。jsonl の方が新しい (チャットモード) なら推定で上書き。
     const officialFresh =
@@ -333,6 +390,33 @@ export async function getUsage() {
       result.context = est.percent
     }
   }
+
+  // チャット(SDK) の rate_limit_event が statusLine cache より新しければ、5h/7d を
+  // その実値 (utilization) で上書きする。ターミナルが statusLine を更新していない
+  // チャットモードでも 5h/7d がライブで動くようになる。
+  if (
+    chatRateLimits.updatedAtMs > 0 &&
+    (cacheMtime === null || chatRateLimits.updatedAtMs > cacheMtime)
+  ) {
+    if (chatRateLimits.five_hour) {
+      result.last5h = {
+        ...result.last5h,
+        percent: chatRateLimits.five_hour.percent,
+        resetAtMs: chatRateLimits.five_hour.resetAtMs ?? result.last5h?.resetAtMs ?? null,
+      }
+    }
+    if (chatRateLimits.seven_day) {
+      result.last7d = {
+        ...result.last7d,
+        percent: chatRateLimits.seven_day.percent,
+        resetAtMs: chatRateLimits.seven_day.resetAtMs ?? result.last7d?.resetAtMs ?? null,
+      }
+    }
+    if (chatRateLimits.five_hour || chatRateLimits.seven_day) {
+      // rate_limit_event は API 実値なので official 扱い。
+      result.source = "official"
+    }
+  }
   return result
 }