@cocorograph/hub-agent 0.8.80 → 0.8.82

package/src/usage.mjs

@@ -802,7 +802,7 @@ export function isFreshUnboundBind(boundVal) {
  * @param {string} cwd
  * @returns {Promise<{ percent: number | null, uuid: string | null, isReset: boolean }>}
  */
-export async function jsonlContextForCwd(cwd, sessionId = null) {
+export async function jsonlContextForCwd(cwd, sessionId = null, { contextWindow } = {}) {
   const empty = { percent: null, uuid: null, isReset: false }
   if (!cwd) return empty
   const dirName = encodeCwdToDirName(cwd)
@@ -817,12 +817,12 @@ export async function jsonlContextForCwd(cwd, sessionId = null) {
 
   // 末尾 assistant.usage を探す。`_jsonlCtxMemo` は fp+mtime+size をキーにしているので
   // 同 jsonl の連続呼び出しは I/O ゼロでヒットする。
+  const windowSize = typeof contextWindow === "number" && contextWindow > 0 ? contextWindow : await contextWindowSize()
   const memo = _jsonlCtxMemo.get(newest.fp)
   let percent = null
-  if (memo && memo.mtimeMs === newest.mtimeMs && memo.size === newest.size) {
+  if (memo && memo.mtimeMs === newest.mtimeMs && memo.size === newest.size && memo.windowSize === windowSize) {
     percent = memo.result ? memo.result.percent : null
   } else {
-    const windowSize = await contextWindowSize()
     const scan = (text) => {
       const lines = text.split("\n")
       for (let i = lines.length - 1; i >= 0; i--) {
@@ -857,6 +857,7 @@ export async function jsonlContextForCwd(cwd, sessionId = null) {
     _jsonlCtxMemo.set(newest.fp, {
       mtimeMs: newest.mtimeMs,
       size: newest.size,
+      windowSize,
       result,
     })
   }
@@ -2188,6 +2189,10 @@ export function codexContextPercentFromTokenUsage(tokenUsage) {
 /** producer(session.seq) の live context%(App Server 正本)。呼び出し側は cwd の権威 producer
  *  (liveSessionForCwd の seq) を渡す。live tokenUsage 未取得 / 算出不能なら null（呼び出し側が
  *  rollout へフォールバックする）。 */
+export function hasCodexLiveTokenUsage(producerSeq) {
+  return _codexLiveTokenUsageByProducer.has(producerSeq)
+}
+
 export function codexLiveContextForProducer(producerSeq) {
   if (typeof producerSeq !== "number") return null
   const e = _codexLiveTokenUsageByProducer.get(producerSeq)