@ai-devkit/agent-manager 0.33.1 → 0.35.0

package/dist/capacity/codex.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/capacity/codex.ts"],"sourcesContent":["import { spawn } from \"node:child_process\";\nimport { readFile } from \"node:fs/promises\";\nimport { join } from \"node:path\";\nimport type { CapacityReport, CapacityWindow } from \"./types.js\";\n\ntype CodexUsageSource = \"pat\" | \"oauth\" | \"cli\";\ntype UsageSnapshot = {\n  windows: CapacityWindow[];\n  creditsRemaining: number | null;\n  source: CodexUsageSource;\n};\n\ntype UnknownRecord = Record<string, unknown>;\ntype RpcMessage = { id?: number; method: string; params?: UnknownRecord };\ntype CliResponses = { rateLimits: unknown; account: unknown };\ntype CodexRpc = (messages: RpcMessage[]) => Promise<CliResponses>;\n\nexport const CODEX_APP_SERVER_ARGS = [\"-s\", \"read-only\", \"-a\", \"untrusted\", \"app-server\"] as const;\n\ntype CodexProbeOptions = {\n  installed: boolean;\n  checkedAt: string;\n  readFile?: (path: string, encoding: BufferEncoding) => Promise<string>;\n  fetch?: typeof globalThis.fetch;\n  rpc?: CodexRpc;\n  timeoutMs?: number;\n  env?: NodeJS.ProcessEnv;\n  now?: () => Date;\n};\n\nfunction record(value: unknown): UnknownRecord | null {\n  return value !== null && typeof value === \"object\" && !Array.isArray(value)\n    ? (value as UnknownRecord)\n    : null;\n}\n\nfunction finiteNumber(value: unknown): number | null {\n  return typeof value === \"number\" && Number.isFinite(value) ? value : null;\n}\n\nfunction nonEmptyText(value: unknown): string | null {\n  return typeof value === \"string\" && value.length > 0 ? value : null;\n}\n\nfunction resetTime(value: unknown): string | null {\n  const seconds = finiteNumber(value);\n  if (seconds !== null) return new Date(seconds * 1000).toISOString();\n  if (typeof value === \"string\" && !Number.isNaN(Date.parse(value)))\n    return new Date(value).toISOString();\n  return null;\n}\n\nfunction safeIdentifier(value: unknown): string | null {\n  const candidate = nonEmptyText(value);\n  if (!candidate || !/^[a-z][a-z0-9_-]{0,63}$/i.test(candidate)) return null;\n  if (/(?:account|token|secret|key)[_-]?\\d{6,}/i.test(candidate)) return null;\n  return candidate;\n}\n\nexport function resolveCodexAuthPath(env: NodeJS.ProcessEnv = process.env): string {\n  const root = env.CODEX_HOME || join(env.HOME || \"\", \".codex\");\n  return join(root, \"auth.json\");\n}\n\nexport function toRateWindow(value: unknown, id: string, label: string): CapacityWindow | null {\n  const input = record(value);\n  if (!input) return null;\n  const used = finiteNumber(input.used_percent);\n  const seconds = finiteNumber(input.limit_window_seconds);\n  return {\n    id,\n    label,\n    durationMinutes: seconds === null ? null : seconds / 60,\n    usedPercent: used,\n    resetsAt: resetTime(input.reset_at),\n  };\n}\n\nfunction extraWindows(value: unknown): CapacityWindow[] {\n  if (!Array.isArray(value)) return [];\n  return value.flatMap((entry, index) => {\n    const limit = record(entry);\n    if (!limit) return [];\n    const scope = safeIdentifier(limit.limit_name) ?? `extra-${index + 1}`;\n    const windows = record(limit.rate_limit) ?? limit;\n    return [\n      toRateWindow(windows.primary_window, `${scope}:primary`, `${scope} primary`),\n      toRateWindow(windows.secondary_window, `${scope}:secondary`, `${scope} secondary`),\n    ].filter((window): window is CapacityWindow => window !== null);\n  });\n}\n\nexport function parseUsage(raw: unknown, source: \"pat\" | \"oauth\"): UsageSnapshot {\n  const response = record(raw) ?? {};\n  const limits = record(response.rate_limit) ?? {};\n  const credits = record(response.credits) ?? {};\n  return {\n    windows: [\n      toRateWindow(limits.primary_window, \"session\", \"Session\"),\n      toRateWindow(limits.secondary_window, \"weekly\", \"Weekly\"),\n      ...extraWindows(response.additional_rate_limits),\n    ].filter((window): window is CapacityWindow => window !== null),\n    creditsRemaining: finiteNumber(credits.balance),\n    source,\n  };\n}\n\nfunction cliWindow(value: unknown, id: string, label: string): CapacityWindow | null {\n  const input = record(value);\n  if (!input) return null;\n  return {\n    id,\n    label,\n    durationMinutes: finiteNumber(input.windowDurationMins),\n    usedPercent: finiteNumber(input.usedPercent),\n    resetsAt: resetTime(input.resetsAt),\n  };\n}\n\nfunction cliSnapshotWindows(value: unknown, fallbackId: string): CapacityWindow[] {\n  const snapshot = record(value);\n  if (!snapshot) return [];\n  const scope = safeIdentifier(snapshot.limitId) ?? safeIdentifier(fallbackId) ?? \"codex\";\n  return [\n    cliWindow(snapshot.primary, `${scope}:primary`, `${scope} primary`),\n    cliWindow(snapshot.secondary, `${scope}:secondary`, `${scope} secondary`),\n  ].filter((item): item is CapacityWindow => item !== null);\n}\n\nexport function parseCliUsage(raw: unknown): UsageSnapshot {\n  const response = record(raw) ?? {};\n  const primary = record(response.rateLimits);\n  const windows = primary ? cliSnapshotWindows(primary, \"codex\") : [];\n  const buckets = record(response.rateLimitsByLimitId);\n  if (buckets) {\n    for (const [id, snapshot] of Object.entries(buckets))\n      windows.push(...cliSnapshotWindows(snapshot, id));\n  }\n  const unique = [...new Map(windows.map((window) => [window.id, window])).values()];\n  return { windows: unique, creditsRemaining: null, source: \"cli\" };\n}\n\nfunction capacityFromSnapshot(\n  snapshot: UsageSnapshot,\n  context: CodexProbeOptions,\n  raw?: unknown,\n): CapacityReport {\n  const hasUsage = snapshot.windows.some((window) => window.usedPercent !== null);\n  const rateLimits = record(record(raw)?.rateLimits);\n  const reached = nonEmptyText(rateLimits?.rateLimitReachedType);\n  const resetCredits =\n    record(record(raw)?.rateLimitResetCredits) ?? record(record(raw)?.usageLimitResetCredits);\n  return {\n    provider: \"codex\",\n    generatedAt: context.checkedAt,\n    authenticated: true,\n    available: reached ? \"no\" : hasUsage ? \"yes\" : \"unknown\",\n    windows: snapshot.windows,\n    creditsRemaining: snapshot.creditsRemaining ?? finiteNumber(resetCredits?.availableCount),\n  };\n}\n\nfunction jwtExpiry(token: string): number | null {\n  const part = token.split(\".\")[1];\n  if (!part) return null;\n  try {\n    return finiteNumber(record(JSON.parse(Buffer.from(part, \"base64url\").toString(\"utf8\")))?.exp);\n  } catch {\n    return null;\n  }\n}\n\nfunction staleOAuth(tokens: UnknownRecord, token: string, now: Date): boolean {\n  const metadata = tokens.expires_at ?? tokens.expiresAt ?? tokens.expiry;\n  let expiry: number | null = finiteNumber(metadata);\n  if (typeof metadata === \"string\") {\n    const parsed = Date.parse(metadata);\n    expiry = Number.isNaN(parsed) ? null : parsed / 1000;\n  }\n  expiry ??= jwtExpiry(token);\n  return expiry !== null && expiry <= now.getTime() / 1000;\n}\n\nasync function fetchJson(\n  fetcher: typeof globalThis.fetch,\n  url: string,\n  init: RequestInit,\n  timeoutMs: number,\n): Promise<unknown> {\n  const controller = new AbortController();\n  const timer = setTimeout(() => controller.abort(), timeoutMs);\n  try {\n    const response = await fetcher(url, { ...init, signal: controller.signal });\n    if (!response.ok) throw new Error(response.status === 401 ? \"unauthorized\" : \"request failed\");\n    return await response.json();\n  } finally {\n    clearTimeout(timer);\n  }\n}\n\nasync function apiSnapshot(\n  token: string,\n  accountId: string,\n  source: \"pat\" | \"oauth\",\n  options: CodexProbeOptions,\n): Promise<UsageSnapshot> {\n  const fetcher = options.fetch ?? globalThis.fetch;\n  const raw = await fetchJson(\n    fetcher,\n    \"https://chatgpt.com/backend-api/wham/usage\",\n    {\n      headers: { Authorization: `Bearer ${token}`, \"ChatGPT-Account-Id\": accountId },\n    },\n    options.timeoutMs ?? 5000,\n  );\n  return parseUsage(raw, source);\n}\n\nfunction appServerRpc(messages: RpcMessage[], timeoutMs = 5000): Promise<CliResponses> {\n  return new Promise((resolve, reject) => {\n    const child = spawn(\"codex\", CODEX_APP_SERVER_ARGS, {\n      stdio: [\"pipe\", \"pipe\", \"ignore\"],\n    });\n    const results: Partial<CliResponses> = {};\n    let buffer = \"\";\n    let settled = false;\n    const finish = (error?: Error) => {\n      if (settled) return;\n      settled = true;\n      clearTimeout(timer);\n      child.kill();\n      if (error) reject(error);\n      else resolve(results as CliResponses);\n    };\n    const timer = setTimeout(() => finish(new Error(\"codex probe timed out\")), timeoutMs);\n    child.once(\"error\", () => finish(new Error(\"codex app-server unavailable\")));\n    child.once(\"exit\", () => {\n      if (!settled) finish(new Error(\"codex app-server exited\"));\n    });\n    child.stdout.setEncoding(\"utf8\");\n    child.stdout.on(\"data\", (chunk: string) => {\n      buffer += chunk;\n      for (;;) {\n        const newline = buffer.indexOf(\"\\n\");\n        if (newline < 0) break;\n        const line = buffer.slice(0, newline).trim();\n        buffer = buffer.slice(newline + 1);\n        if (!line) continue;\n        let message: UnknownRecord;\n        try {\n          message = JSON.parse(line) as UnknownRecord;\n        } catch {\n          continue;\n        }\n        if (message.id === 1) {\n          for (const request of messages.slice(1))\n            child.stdin.write(`${JSON.stringify(request)}\\n`);\n        } else if (message.id === 2) {\n          if (message.error) finish(new Error(\"codex rate-limit method failed\"));\n          else results.rateLimits = message.result;\n        } else if (message.id === 3) {\n          if (message.error) finish(new Error(\"codex account method failed\"));\n          else results.account = message.result;\n        }\n        if (\"rateLimits\" in results && \"account\" in results) finish();\n      }\n    });\n    child.stdin.write(`${JSON.stringify(messages[0])}\\n`);\n  });\n}\n\nfunction unavailable(options: CodexProbeOptions): CapacityReport {\n  return {\n    provider: \"codex\",\n    generatedAt: options.checkedAt,\n    authenticated: null,\n    available: \"unknown\",\n    windows: [],\n    creditsRemaining: null,\n  };\n}\n\nasync function cliFallback(options: CodexProbeOptions): Promise<CapacityReport> {\n  if (!options.installed) return unavailable(options);\n  const messages: RpcMessage[] = [\n    {\n      id: 1,\n      method: \"initialize\",\n      params: {\n        clientInfo: { name: \"ai-devkit\", title: null, version: \"1\" },\n        capabilities: null,\n      },\n    },\n    { method: \"initialized\" },\n    { id: 2, method: \"account/rateLimits/read\" },\n    { id: 3, method: \"account/read\" },\n  ];\n  try {\n    const rpc = options.rpc ?? ((requests) => appServerRpc(requests, options.timeoutMs));\n    const response = await rpc(messages);\n    const result = capacityFromSnapshot(\n      parseCliUsage(response.rateLimits),\n      options,\n      response.rateLimits,\n    );\n    const accountEnvelope = record(response.account);\n    if (\n      accountEnvelope &&\n      Object.hasOwn(accountEnvelope, \"account\") &&\n      !record(accountEnvelope.account)\n    ) {\n      result.authenticated = false;\n      result.available = \"unknown\";\n    }\n    return result;\n  } catch {\n    return unavailable(options);\n  }\n}\n\nexport async function probeCodexCapacity(options: CodexProbeOptions): Promise<CapacityReport> {\n  let parsed: UnknownRecord | null = null;\n  try {\n    const contents = await (options.readFile ?? readFile)(\n      resolveCodexAuthPath(options.env),\n      \"utf8\",\n    );\n    parsed = record(JSON.parse(contents));\n  } catch {\n    return cliFallback(options);\n  }\n\n  const auth = parsed ?? {};\n  const pat = nonEmptyText(auth.personal_access_token);\n  if (pat) {\n    try {\n      const fetcher = options.fetch ?? globalThis.fetch;\n      const whoami = record(\n        await fetchJson(\n          fetcher,\n          \"https://auth.openai.com/api/accounts/v1/user-auth-credential/whoami\",\n          { headers: { Authorization: `Bearer ${pat}` } },\n          options.timeoutMs ?? 5000,\n        ),\n      );\n      const accountId = nonEmptyText(whoami?.chatgpt_account_id);\n      if (!accountId) throw new Error(\"account unavailable\");\n      return capacityFromSnapshot(await apiSnapshot(pat, accountId, \"pat\", options), options);\n    } catch {\n      // Continue to a separately available OAuth credential before using the CLI.\n    }\n  }\n\n  const tokens = record(auth.tokens);\n  const accessToken = nonEmptyText(tokens?.access_token);\n  const accountId = nonEmptyText(tokens?.account_id);\n  if (\n    tokens &&\n    accessToken &&\n    accountId &&\n    !staleOAuth(tokens, accessToken, (options.now ?? (() => new Date()))())\n  ) {\n    try {\n      return capacityFromSnapshot(\n        await apiSnapshot(accessToken, accountId, \"oauth\", options),\n        options,\n      );\n    } catch {\n      return cliFallback(options);\n    }\n  }\n  return cliFallback(options);\n}\n"],"names":["spawn","readFile","join","CODEX_APP_SERVER_ARGS","record","value","Array","isArray","finiteNumber","Number","isFinite","nonEmptyText","length","resetTime","seconds","Date","toISOString","isNaN","parse","safeIdentifier","candidate","test","resolveCodexAuthPath","env","process","root","CODEX_HOME","HOME","toRateWindow","id","label","input","used","used_percent","limit_window_seconds","durationMinutes","usedPercent","resetsAt","reset_at","extraWindows","flatMap","entry","index","limit","scope","limit_name","windows","rate_limit","primary_window","secondary_window","filter","window","parseUsage","raw","source","response","limits","credits","additional_rate_limits","creditsRemaining","balance","cliWindow","windowDurationMins","cliSnapshotWindows","fallbackId","snapshot","limitId","primary","secondary","item","parseCliUsage","rateLimits","buckets","rateLimitsByLimitId","Object","entries","push","unique","Map","map","values","capacityFromSnapshot","context","hasUsage","some","reached","rateLimitReachedType","resetCredits","rateLimitResetCredits","usageLimitResetCredits","provider","generatedAt","checkedAt","authenticated","available","availableCount","jwtExpiry","token","part","split","JSON","Buffer","from","toString","exp","staleOAuth","tokens","now","metadata","expires_at","expiresAt","expiry","parsed","getTime","fetchJson","fetcher","url","init","timeoutMs","controller","AbortController","timer","setTimeout","abort","signal","ok","Error","status","json","clearTimeout","apiSnapshot","accountId","options","fetch","globalThis","headers","Authorization","appServerRpc","messages","Promise","resolve","reject","child","stdio","results","buffer","settled","finish","error","kill","once","stdout","setEncoding","on","chunk","newline","indexOf","line","slice","trim","message","request","stdin","write","stringify","result","account","unavailable","cliFallback","installed","method","params","clientInfo","name","title","version","capabilities","rpc","requests","accountEnvelope","hasOwn","probeCodexCapacity","contents","auth","pat","personal_access_token","whoami","chatgpt_account_id","accessToken","access_token","account_id"],"mappings":"AAAA,SAASA,KAAK,QAAQ,qBAAqB;AAC3C,SAASC,QAAQ,QAAQ,mBAAmB;AAC5C,SAASC,IAAI,QAAQ,YAAY;AAejC,OAAO,MAAMC,wBAAwB;IAAC;IAAM;IAAa;IAAM;IAAa;CAAa,CAAU;AAanG,SAASC,OAAOC,KAAc;IAC5B,OAAOA,UAAU,QAAQ,OAAOA,UAAU,YAAY,CAACC,MAAMC,OAAO,CAACF,SAChEA,QACD;AACN;AAEA,SAASG,aAAaH,KAAc;IAClC,OAAO,OAAOA,UAAU,YAAYI,OAAOC,QAAQ,CAACL,SAASA,QAAQ;AACvE;AAEA,SAASM,aAAaN,KAAc;IAClC,OAAO,OAAOA,UAAU,YAAYA,MAAMO,MAAM,GAAG,IAAIP,QAAQ;AACjE;AAEA,SAASQ,UAAUR,KAAc;IAC/B,MAAMS,UAAUN,aAAaH;IAC7B,IAAIS,YAAY,MAAM,OAAO,IAAIC,KAAKD,UAAU,MAAME,WAAW;IACjE,IAAI,OAAOX,UAAU,YAAY,CAACI,OAAOQ,KAAK,CAACF,KAAKG,KAAK,CAACb,SACxD,OAAO,IAAIU,KAAKV,OAAOW,WAAW;IACpC,OAAO;AACT;AAEA,SAASG,eAAed,KAAc;IACpC,MAAMe,YAAYT,aAAaN;IAC/B,IAAI,CAACe,aAAa,CAAC,2BAA2BC,IAAI,CAACD,YAAY,OAAO;IACtE,IAAI,2CAA2CC,IAAI,CAACD,YAAY,OAAO;IACvE,OAAOA;AACT;AAEA,OAAO,SAASE,qBAAqBC,MAAyBC,QAAQD,GAAG;IACvE,MAAME,OAAOF,IAAIG,UAAU,IAAIxB,KAAKqB,IAAII,IAAI,IAAI,IAAI;IACpD,OAAOzB,KAAKuB,MAAM;AACpB;AAEA,OAAO,SAASG,aAAavB,KAAc,EAAEwB,EAAU,EAAEC,KAAa;IACpE,MAAMC,QAAQ3B,OAAOC;IACrB,IAAI,CAAC0B,OAAO,OAAO;IACnB,MAAMC,OAAOxB,aAAauB,MAAME,YAAY;IAC5C,MAAMnB,UAAUN,aAAauB,MAAMG,oBAAoB;IACvD,OAAO;QACLL;QACAC;QACAK,iBAAiBrB,YAAY,OAAO,OAAOA,UAAU;QACrDsB,aAAaJ;QACbK,UAAUxB,UAAUkB,MAAMO,QAAQ;IACpC;AACF;AAEA,SAASC,aAAalC,KAAc;IAClC,IAAI,CAACC,MAAMC,OAAO,CAACF,QAAQ,OAAO,EAAE;IACpC,OAAOA,MAAMmC,OAAO,CAAC,CAACC,OAAOC;QAC3B,MAAMC,QAAQvC,OAAOqC;QACrB,IAAI,CAACE,OAAO,OAAO,EAAE;QACrB,MAAMC,QAAQzB,eAAewB,MAAME,UAAU,KAAK,CAAC,MAAM,EAAEH,QAAQ,GAAG;QACtE,MAAMI,UAAU1C,OAAOuC,MAAMI,UAAU,KAAKJ;QAC5C,OAAO;YACLf,aAAakB,QAAQE,cAAc,EAAE,GAAGJ,MAAM,QAAQ,CAAC,EAAE,GAAGA,MAAM,QAAQ,CAAC;YAC3EhB,aAAakB,QAAQG,gBAAgB,EAAE,GAAGL,MAAM,UAAU,CAAC,EAAE,GAAGA,MAAM,UAAU,CAAC;SAClF,CAACM,MAAM,CAAC,CAACC,SAAqCA,WAAW;IAC5D;AACF;AAEA,OAAO,SAASC,WAAWC,GAAY,EAAEC,MAAuB;IAC9D,MAAMC,WAAWnD,OAAOiD,QAAQ,CAAC;IACjC,MAAMG,SAASpD,OAAOmD,SAASR,UAAU,KAAK,CAAC;IAC/C,MAAMU,UAAUrD,OAAOmD,SAASE,OAAO,KAAK,CAAC;IAC7C,OAAO;QACLX,SAAS;YACPlB,aAAa4B,OAAOR,cAAc,EAAE,WAAW;YAC/CpB,aAAa4B,OAAOP,gBAAgB,EAAE,UAAU;eAC7CV,aAAagB,SAASG,sBAAsB;SAChD,CAACR,MAAM,CAAC,CAACC,SAAqCA,WAAW;QAC1DQ,kBAAkBnD,aAAaiD,QAAQG,OAAO;QAC9CN;IACF;AACF;AAEA,SAASO,UAAUxD,KAAc,EAAEwB,EAAU,EAAEC,KAAa;IAC1D,MAAMC,QAAQ3B,OAAOC;IACrB,IAAI,CAAC0B,OAAO,OAAO;IACnB,OAAO;QACLF;QACAC;QACAK,iBAAiB3B,aAAauB,MAAM+B,kBAAkB;QACtD1B,aAAa5B,aAAauB,MAAMK,WAAW;QAC3CC,UAAUxB,UAAUkB,MAAMM,QAAQ;IACpC;AACF;AAEA,SAAS0B,mBAAmB1D,KAAc,EAAE2D,UAAkB;IAC5D,MAAMC,WAAW7D,OAAOC;IACxB,IAAI,CAAC4D,UAAU,OAAO,EAAE;IACxB,MAAMrB,QAAQzB,eAAe8C,SAASC,OAAO,KAAK/C,eAAe6C,eAAe;IAChF,OAAO;QACLH,UAAUI,SAASE,OAAO,EAAE,GAAGvB,MAAM,QAAQ,CAAC,EAAE,GAAGA,MAAM,QAAQ,CAAC;QAClEiB,UAAUI,SAASG,SAAS,EAAE,GAAGxB,MAAM,UAAU,CAAC,EAAE,GAAGA,MAAM,UAAU,CAAC;KACzE,CAACM,MAAM,CAAC,CAACmB,OAAiCA,SAAS;AACtD;AAEA,OAAO,SAASC,cAAcjB,GAAY;IACxC,MAAME,WAAWnD,OAAOiD,QAAQ,CAAC;IACjC,MAAMc,UAAU/D,OAAOmD,SAASgB,UAAU;IAC1C,MAAMzB,UAAUqB,UAAUJ,mBAAmBI,SAAS,WAAW,EAAE;IACnE,MAAMK,UAAUpE,OAAOmD,SAASkB,mBAAmB;IACnD,IAAID,SAAS;QACX,KAAK,MAAM,CAAC3C,IAAIoC,SAAS,IAAIS,OAAOC,OAAO,CAACH,SAC1C1B,QAAQ8B,IAAI,IAAIb,mBAAmBE,UAAUpC;IACjD;IACA,MAAMgD,SAAS;WAAI,IAAIC,IAAIhC,QAAQiC,GAAG,CAAC,CAAC5B,SAAW;gBAACA,OAAOtB,EAAE;gBAAEsB;aAAO,GAAG6B,MAAM;KAAG;IAClF,OAAO;QAAElC,SAAS+B;QAAQlB,kBAAkB;QAAML,QAAQ;IAAM;AAClE;AAEA,SAAS2B,qBACPhB,QAAuB,EACvBiB,OAA0B,EAC1B7B,GAAa;IAEb,MAAM8B,WAAWlB,SAASnB,OAAO,CAACsC,IAAI,CAAC,CAACjC,SAAWA,OAAOf,WAAW,KAAK;IAC1E,MAAMmC,aAAanE,OAAOA,OAAOiD,MAAMkB;IACvC,MAAMc,UAAU1E,aAAa4D,YAAYe;IACzC,MAAMC,eACJnF,OAAOA,OAAOiD,MAAMmC,0BAA0BpF,OAAOA,OAAOiD,MAAMoC;IACpE,OAAO;QACLC,UAAU;QACVC,aAAaT,QAAQU,SAAS;QAC9BC,eAAe;QACfC,WAAWT,UAAU,OAAOF,WAAW,QAAQ;QAC/CrC,SAASmB,SAASnB,OAAO;QACzBa,kBAAkBM,SAASN,gBAAgB,IAAInD,aAAa+E,cAAcQ;IAC5E;AACF;AAEA,SAASC,UAAUC,KAAa;IAC9B,MAAMC,OAAOD,MAAME,KAAK,CAAC,IAAI,CAAC,EAAE;IAChC,IAAI,CAACD,MAAM,OAAO;IAClB,IAAI;QACF,OAAO1F,aAAaJ,OAAOgG,KAAKlF,KAAK,CAACmF,OAAOC,IAAI,CAACJ,MAAM,aAAaK,QAAQ,CAAC,WAAWC;IAC3F,EAAE,OAAM;QACN,OAAO;IACT;AACF;AAEA,SAASC,WAAWC,MAAqB,EAAET,KAAa,EAAEU,GAAS;IACjE,MAAMC,WAAWF,OAAOG,UAAU,IAAIH,OAAOI,SAAS,IAAIJ,OAAOK,MAAM;IACvE,IAAIA,SAAwBvG,aAAaoG;IACzC,IAAI,OAAOA,aAAa,UAAU;QAChC,MAAMI,SAASjG,KAAKG,KAAK,CAAC0F;QAC1BG,SAAStG,OAAOQ,KAAK,CAAC+F,UAAU,OAAOA,SAAS;IAClD;IACAD,WAAWf,UAAUC;IACrB,OAAOc,WAAW,QAAQA,UAAUJ,IAAIM,OAAO,KAAK;AACtD;AAEA,eAAeC,UACbC,OAAgC,EAChCC,GAAW,EACXC,IAAiB,EACjBC,SAAiB;IAEjB,MAAMC,aAAa,IAAIC;IACvB,MAAMC,QAAQC,WAAW,IAAMH,WAAWI,KAAK,IAAIL;IACnD,IAAI;QACF,MAAM/D,WAAW,MAAM4D,QAAQC,KAAK;YAAE,GAAGC,IAAI;YAAEO,QAAQL,WAAWK,MAAM;QAAC;QACzE,IAAI,CAACrE,SAASsE,EAAE,EAAE,MAAM,IAAIC,MAAMvE,SAASwE,MAAM,KAAK,MAAM,iBAAiB;QAC7E,OAAO,MAAMxE,SAASyE,IAAI;IAC5B,SAAU;QACRC,aAAaR;IACf;AACF;AAEA,eAAeS,YACbjC,KAAa,EACbkC,SAAiB,EACjB7E,MAAuB,EACvB8E,OAA0B;IAE1B,MAAMjB,UAAUiB,QAAQC,KAAK,IAAIC,WAAWD,KAAK;IACjD,MAAMhF,MAAM,MAAM6D,UAChBC,SACA,8CACA;QACEoB,SAAS;YAAEC,eAAe,CAAC,OAAO,EAAEvC,OAAO;YAAE,sBAAsBkC;QAAU;IAC/E,GACAC,QAAQd,SAAS,IAAI;IAEvB,OAAOlE,WAAWC,KAAKC;AACzB;AAEA,SAASmF,aAAaC,QAAsB,EAAEpB,YAAY,IAAI;IAC5D,OAAO,IAAIqB,QAAQ,CAACC,SAASC;QAC3B,MAAMC,QAAQ9I,MAAM,SAASG,uBAAuB;YAClD4I,OAAO;gBAAC;gBAAQ;gBAAQ;aAAS;QACnC;QACA,MAAMC,UAAiC,CAAC;QACxC,IAAIC,SAAS;QACb,IAAIC,UAAU;QACd,MAAMC,SAAS,CAACC;YACd,IAAIF,SAAS;YACbA,UAAU;YACVjB,aAAaR;YACbqB,MAAMO,IAAI;YACV,IAAID,OAAOP,OAAOO;iBACbR,QAAQI;QACf;QACA,MAAMvB,QAAQC,WAAW,IAAMyB,OAAO,IAAIrB,MAAM,2BAA2BR;QAC3EwB,MAAMQ,IAAI,CAAC,SAAS,IAAMH,OAAO,IAAIrB,MAAM;QAC3CgB,MAAMQ,IAAI,CAAC,QAAQ;YACjB,IAAI,CAACJ,SAASC,OAAO,IAAIrB,MAAM;QACjC;QACAgB,MAAMS,MAAM,CAACC,WAAW,CAAC;QACzBV,MAAMS,MAAM,CAACE,EAAE,CAAC,QAAQ,CAACC;YACvBT,UAAUS;YACV,OAAS;gBACP,MAAMC,UAAUV,OAAOW,OAAO,CAAC;gBAC/B,IAAID,UAAU,GAAG;gBACjB,MAAME,OAAOZ,OAAOa,KAAK,CAAC,GAAGH,SAASI,IAAI;gBAC1Cd,SAASA,OAAOa,KAAK,CAACH,UAAU;gBAChC,IAAI,CAACE,MAAM;gBACX,IAAIG;gBACJ,IAAI;oBACFA,UAAU5D,KAAKlF,KAAK,CAAC2I;gBACvB,EAAE,OAAM;oBACN;gBACF;gBACA,IAAIG,QAAQnI,EAAE,KAAK,GAAG;oBACpB,KAAK,MAAMoI,WAAWvB,SAASoB,KAAK,CAAC,GACnChB,MAAMoB,KAAK,CAACC,KAAK,CAAC,GAAG/D,KAAKgE,SAAS,CAACH,SAAS,EAAE,CAAC;gBACpD,OAAO,IAAID,QAAQnI,EAAE,KAAK,GAAG;oBAC3B,IAAImI,QAAQZ,KAAK,EAAED,OAAO,IAAIrB,MAAM;yBAC/BkB,QAAQzE,UAAU,GAAGyF,QAAQK,MAAM;gBAC1C,OAAO,IAAIL,QAAQnI,EAAE,KAAK,GAAG;oBAC3B,IAAImI,QAAQZ,KAAK,EAAED,OAAO,IAAIrB,MAAM;yBAC/BkB,QAAQsB,OAAO,GAAGN,QAAQK,MAAM;gBACvC;gBACA,IAAI,gBAAgBrB,WAAW,aAAaA,SAASG;YACvD;QACF;QACAL,MAAMoB,KAAK,CAACC,KAAK,CAAC,GAAG/D,KAAKgE,SAAS,CAAC1B,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC;IACtD;AACF;AAEA,SAAS6B,YAAYnC,OAA0B;IAC7C,OAAO;QACL1C,UAAU;QACVC,aAAayC,QAAQxC,SAAS;QAC9BC,eAAe;QACfC,WAAW;QACXhD,SAAS,EAAE;QACXa,kBAAkB;IACpB;AACF;AAEA,eAAe6G,YAAYpC,OAA0B;IACnD,IAAI,CAACA,QAAQqC,SAAS,EAAE,OAAOF,YAAYnC;IAC3C,MAAMM,WAAyB;QAC7B;YACE7G,IAAI;YACJ6I,QAAQ;YACRC,QAAQ;gBACNC,YAAY;oBAAEC,MAAM;oBAAaC,OAAO;oBAAMC,SAAS;gBAAI;gBAC3DC,cAAc;YAChB;QACF;QACA;YAAEN,QAAQ;QAAc;QACxB;YAAE7I,IAAI;YAAG6I,QAAQ;QAA0B;QAC3C;YAAE7I,IAAI;YAAG6I,QAAQ;QAAe;KACjC;IACD,IAAI;QACF,MAAMO,MAAM7C,QAAQ6C,GAAG,IAAK,CAAA,CAACC,WAAazC,aAAayC,UAAU9C,QAAQd,SAAS,CAAA;QAClF,MAAM/D,WAAW,MAAM0H,IAAIvC;QAC3B,MAAM2B,SAASpF,qBACbX,cAAcf,SAASgB,UAAU,GACjC6D,SACA7E,SAASgB,UAAU;QAErB,MAAM4G,kBAAkB/K,OAAOmD,SAAS+G,OAAO;QAC/C,IACEa,mBACAzG,OAAO0G,MAAM,CAACD,iBAAiB,cAC/B,CAAC/K,OAAO+K,gBAAgBb,OAAO,GAC/B;YACAD,OAAOxE,aAAa,GAAG;YACvBwE,OAAOvE,SAAS,GAAG;QACrB;QACA,OAAOuE;IACT,EAAE,OAAM;QACN,OAAOE,YAAYnC;IACrB;AACF;AAEA,OAAO,eAAeiD,mBAAmBjD,OAA0B;IACjE,IAAIpB,SAA+B;IACnC,IAAI;QACF,MAAMsE,WAAW,MAAM,AAAClD,CAAAA,QAAQnI,QAAQ,IAAIA,QAAO,EACjDqB,qBAAqB8G,QAAQ7G,GAAG,GAChC;QAEFyF,SAAS5G,OAAOgG,KAAKlF,KAAK,CAACoK;IAC7B,EAAE,OAAM;QACN,OAAOd,YAAYpC;IACrB;IAEA,MAAMmD,OAAOvE,UAAU,CAAC;IACxB,MAAMwE,MAAM7K,aAAa4K,KAAKE,qBAAqB;IACnD,IAAID,KAAK;QACP,IAAI;YACF,MAAMrE,UAAUiB,QAAQC,KAAK,IAAIC,WAAWD,KAAK;YACjD,MAAMqD,SAAStL,OACb,MAAM8G,UACJC,SACA,uEACA;gBAAEoB,SAAS;oBAAEC,eAAe,CAAC,OAAO,EAAEgD,KAAK;gBAAC;YAAE,GAC9CpD,QAAQd,SAAS,IAAI;YAGzB,MAAMa,YAAYxH,aAAa+K,QAAQC;YACvC,IAAI,CAACxD,WAAW,MAAM,IAAIL,MAAM;YAChC,OAAO7C,qBAAqB,MAAMiD,YAAYsD,KAAKrD,WAAW,OAAOC,UAAUA;QACjF,EAAE,OAAM;QACN,4EAA4E;QAC9E;IACF;IAEA,MAAM1B,SAAStG,OAAOmL,KAAK7E,MAAM;IACjC,MAAMkF,cAAcjL,aAAa+F,QAAQmF;IACzC,MAAM1D,YAAYxH,aAAa+F,QAAQoF;IACvC,IACEpF,UACAkF,eACAzD,aACA,CAAC1B,WAAWC,QAAQkF,aAAa,AAACxD,CAAAA,QAAQzB,GAAG,IAAK,CAAA,IAAM,IAAI5F,MAAK,CAAC,MAClE;QACA,IAAI;YACF,OAAOkE,qBACL,MAAMiD,YAAY0D,aAAazD,WAAW,SAASC,UACnDA;QAEJ,EAAE,OAAM;YACN,OAAOoC,YAAYpC;QACrB;IACF;IACA,OAAOoC,YAAYpC;AACrB"}
+{"version":3,"sources":["../../src/capacity/codex.ts"],"sourcesContent":["import { spawn } from \"node:child_process\";\nimport { readFile } from \"node:fs/promises\";\nimport { join } from \"node:path\";\nimport type { CapacityReport, CapacityWindow } from \"./types.js\";\n\ntype CodexUsageSource = \"pat\" | \"oauth\" | \"cli\";\ntype UsageSnapshot = {\n  windows: CapacityWindow[];\n  creditsRemaining: number | null;\n  source: CodexUsageSource;\n};\n\ntype UnknownRecord = Record<string, unknown>;\ntype RpcMessage = { id?: number; method: string; params?: UnknownRecord };\ntype CliResponses = { rateLimits: unknown; account: unknown };\ntype CodexRpc = (messages: RpcMessage[]) => Promise<CliResponses>;\n\nexport const CODEX_APP_SERVER_ARGS = [\n  \"-s\",\n  \"read-only\",\n  \"-a\",\n  \"untrusted\",\n  \"app-server\",\n] as const;\n\ntype CodexProbeOptions = {\n  installed: boolean;\n  checkedAt: string;\n  readFile?: (path: string, encoding: BufferEncoding) => Promise<string>;\n  fetch?: typeof globalThis.fetch;\n  rpc?: CodexRpc;\n  timeoutMs?: number;\n  env?: NodeJS.ProcessEnv;\n  now?: () => Date;\n};\n\nfunction record(value: unknown): UnknownRecord | null {\n  return value !== null && typeof value === \"object\" && !Array.isArray(value)\n    ? (value as UnknownRecord)\n    : null;\n}\n\nfunction finiteNumber(value: unknown): number | null {\n  return typeof value === \"number\" && Number.isFinite(value) ? value : null;\n}\n\nfunction nonEmptyText(value: unknown): string | null {\n  return typeof value === \"string\" && value.length > 0 ? value : null;\n}\n\nfunction resetTime(value: unknown): string | null {\n  const seconds = finiteNumber(value);\n  if (seconds !== null) return new Date(seconds * 1000).toISOString();\n  if (typeof value === \"string\" && !Number.isNaN(Date.parse(value)))\n    return new Date(value).toISOString();\n  return null;\n}\n\nfunction safeIdentifier(value: unknown): string | null {\n  const candidate = nonEmptyText(value);\n  if (!candidate || !/^[a-z][a-z0-9_-]{0,63}$/i.test(candidate)) return null;\n  if (/(?:account|token|secret|key)[_-]?\\d{6,}/i.test(candidate)) return null;\n  return candidate;\n}\n\nexport function resolveCodexAuthPath(\n  env: NodeJS.ProcessEnv = process.env,\n): string {\n  const root = env.CODEX_HOME || join(env.HOME || \"\", \".codex\");\n  return join(root, \"auth.json\");\n}\n\nexport function toRateWindow(\n  value: unknown,\n  id: string,\n  label: string,\n): CapacityWindow | null {\n  const input = record(value);\n  if (!input) return null;\n  const used = finiteNumber(input.used_percent);\n  const seconds = finiteNumber(input.limit_window_seconds);\n  return {\n    id,\n    label,\n    durationMinutes: seconds === null ? null : seconds / 60,\n    usedPercent: used,\n    resetsAt: resetTime(input.reset_at),\n  };\n}\n\nfunction extraWindows(value: unknown): CapacityWindow[] {\n  if (!Array.isArray(value)) return [];\n  return value.flatMap((entry, index) => {\n    const limit = record(entry);\n    if (!limit) return [];\n    const scope = safeIdentifier(limit.limit_name) ?? `extra-${index + 1}`;\n    const windows = record(limit.rate_limit) ?? limit;\n    return [\n      toRateWindow(\n        windows.primary_window,\n        `${scope}:primary`,\n        `${scope} primary`,\n      ),\n      toRateWindow(\n        windows.secondary_window,\n        `${scope}:secondary`,\n        `${scope} secondary`,\n      ),\n    ].filter((window): window is CapacityWindow => window !== null);\n  });\n}\n\nexport function parseUsage(\n  raw: unknown,\n  source: \"pat\" | \"oauth\",\n): UsageSnapshot {\n  const response = record(raw) ?? {};\n  const limits = record(response.rate_limit) ?? {};\n  const credits = record(response.credits) ?? {};\n  return {\n    windows: [\n      toRateWindow(limits.primary_window, \"session\", \"Session\"),\n      toRateWindow(limits.secondary_window, \"weekly\", \"Weekly\"),\n      ...extraWindows(response.additional_rate_limits),\n    ].filter((window): window is CapacityWindow => window !== null),\n    creditsRemaining: finiteNumber(credits.balance),\n    source,\n  };\n}\n\nfunction cliWindow(\n  value: unknown,\n  id: string,\n  label: string,\n): CapacityWindow | null {\n  const input = record(value);\n  if (!input) return null;\n  return {\n    id,\n    label,\n    durationMinutes: finiteNumber(input.windowDurationMins),\n    usedPercent: finiteNumber(input.usedPercent),\n    resetsAt: resetTime(input.resetsAt),\n  };\n}\n\nfunction cliSnapshotWindows(\n  value: unknown,\n  fallbackId: string,\n): CapacityWindow[] {\n  const snapshot = record(value);\n  if (!snapshot) return [];\n  const scope =\n    safeIdentifier(snapshot.limitId) ?? safeIdentifier(fallbackId) ?? \"codex\";\n  return [\n    cliWindow(snapshot.primary, `${scope}:primary`, `${scope} primary`),\n    cliWindow(snapshot.secondary, `${scope}:secondary`, `${scope} secondary`),\n  ].filter((item): item is CapacityWindow => item !== null);\n}\n\nexport function parseCliUsage(raw: unknown): UsageSnapshot {\n  const response = record(raw) ?? {};\n  const primary = record(response.rateLimits);\n  const windows = primary ? cliSnapshotWindows(primary, \"codex\") : [];\n  const buckets = record(response.rateLimitsByLimitId);\n  if (buckets) {\n    for (const [id, snapshot] of Object.entries(buckets))\n      windows.push(...cliSnapshotWindows(snapshot, id));\n  }\n  const unique = [\n    ...new Map(windows.map((window) => [window.id, window])).values(),\n  ];\n  return { windows: unique, creditsRemaining: null, source: \"cli\" };\n}\n\nfunction capacityFromSnapshot(\n  snapshot: UsageSnapshot,\n  context: CodexProbeOptions,\n  raw?: unknown,\n): CapacityReport {\n  const hasUsage = snapshot.windows.some(\n    (window) => window.usedPercent !== null,\n  );\n  const rateLimits = record(record(raw)?.rateLimits);\n  const reached = nonEmptyText(rateLimits?.rateLimitReachedType);\n  const resetCredits =\n    record(record(raw)?.rateLimitResetCredits) ??\n    record(record(raw)?.usageLimitResetCredits);\n  return {\n    harness: \"codex\",\n    provider: \"openai\",\n    generatedAt: context.checkedAt,\n    authenticated: true,\n    available: reached ? \"no\" : hasUsage ? \"yes\" : \"unknown\",\n    windows: snapshot.windows,\n    creditsRemaining:\n      snapshot.creditsRemaining ?? finiteNumber(resetCredits?.availableCount),\n  };\n}\n\nfunction jwtExpiry(token: string): number | null {\n  const part = token.split(\".\")[1];\n  if (!part) return null;\n  try {\n    return finiteNumber(\n      record(JSON.parse(Buffer.from(part, \"base64url\").toString(\"utf8\")))?.exp,\n    );\n  } catch {\n    return null;\n  }\n}\n\nfunction staleOAuth(tokens: UnknownRecord, token: string, now: Date): boolean {\n  const metadata = tokens.expires_at ?? tokens.expiresAt ?? tokens.expiry;\n  let expiry: number | null = finiteNumber(metadata);\n  if (typeof metadata === \"string\") {\n    const parsed = Date.parse(metadata);\n    expiry = Number.isNaN(parsed) ? null : parsed / 1000;\n  }\n  expiry ??= jwtExpiry(token);\n  return expiry !== null && expiry <= now.getTime() / 1000;\n}\n\nasync function fetchJson(\n  fetcher: typeof globalThis.fetch,\n  url: string,\n  init: RequestInit,\n  timeoutMs: number,\n): Promise<unknown> {\n  const controller = new AbortController();\n  const timer = setTimeout(() => controller.abort(), timeoutMs);\n  try {\n    const response = await fetcher(url, { ...init, signal: controller.signal });\n    if (!response.ok)\n      throw new Error(\n        response.status === 401 ? \"unauthorized\" : \"request failed\",\n      );\n    return await response.json();\n  } finally {\n    clearTimeout(timer);\n  }\n}\n\nasync function apiSnapshot(\n  token: string,\n  accountId: string,\n  source: \"pat\" | \"oauth\",\n  options: CodexProbeOptions,\n): Promise<UsageSnapshot> {\n  const fetcher = options.fetch ?? globalThis.fetch;\n  const raw = await fetchJson(\n    fetcher,\n    \"https://chatgpt.com/backend-api/wham/usage\",\n    {\n      headers: {\n        Authorization: `Bearer ${token}`,\n        \"ChatGPT-Account-Id\": accountId,\n      },\n    },\n    options.timeoutMs ?? 5000,\n  );\n  return parseUsage(raw, source);\n}\n\nfunction appServerRpc(\n  messages: RpcMessage[],\n  timeoutMs = 5000,\n): Promise<CliResponses> {\n  return new Promise((resolve, reject) => {\n    const child = spawn(\"codex\", CODEX_APP_SERVER_ARGS, {\n      stdio: [\"pipe\", \"pipe\", \"ignore\"],\n    });\n    const results: Partial<CliResponses> = {};\n    let buffer = \"\";\n    let settled = false;\n    const finish = (error?: Error) => {\n      if (settled) return;\n      settled = true;\n      clearTimeout(timer);\n      child.kill();\n      if (error) reject(error);\n      else resolve(results as CliResponses);\n    };\n    const timer = setTimeout(\n      () => finish(new Error(\"codex probe timed out\")),\n      timeoutMs,\n    );\n    child.once(\"error\", () =>\n      finish(new Error(\"codex app-server unavailable\")),\n    );\n    child.once(\"exit\", () => {\n      if (!settled) finish(new Error(\"codex app-server exited\"));\n    });\n    child.stdout.setEncoding(\"utf8\");\n    child.stdout.on(\"data\", (chunk: string) => {\n      buffer += chunk;\n      for (;;) {\n        const newline = buffer.indexOf(\"\\n\");\n        if (newline < 0) break;\n        const line = buffer.slice(0, newline).trim();\n        buffer = buffer.slice(newline + 1);\n        if (!line) continue;\n        let message: UnknownRecord;\n        try {\n          message = JSON.parse(line) as UnknownRecord;\n        } catch {\n          continue;\n        }\n        if (message.id === 1) {\n          for (const request of messages.slice(1))\n            child.stdin.write(`${JSON.stringify(request)}\\n`);\n        } else if (message.id === 2) {\n          if (message.error)\n            finish(new Error(\"codex rate-limit method failed\"));\n          else results.rateLimits = message.result;\n        } else if (message.id === 3) {\n          if (message.error) finish(new Error(\"codex account method failed\"));\n          else results.account = message.result;\n        }\n        if (\"rateLimits\" in results && \"account\" in results) finish();\n      }\n    });\n    child.stdin.write(`${JSON.stringify(messages[0])}\\n`);\n  });\n}\n\nexport function codexUnavailableReport(checkedAt: string): CapacityReport {\n  return {\n    harness: \"codex\",\n    provider: \"openai\",\n    generatedAt: checkedAt,\n    authenticated: null,\n    available: \"unknown\",\n    windows: [],\n    creditsRemaining: null,\n  };\n}\n\nfunction unavailable(options: CodexProbeOptions): CapacityReport {\n  return codexUnavailableReport(options.checkedAt);\n}\n\nasync function cliFallback(\n  options: CodexProbeOptions,\n): Promise<CapacityReport> {\n  if (!options.installed) return unavailable(options);\n  const messages: RpcMessage[] = [\n    {\n      id: 1,\n      method: \"initialize\",\n      params: {\n        clientInfo: { name: \"ai-devkit\", title: null, version: \"1\" },\n        capabilities: null,\n      },\n    },\n    { method: \"initialized\" },\n    { id: 2, method: \"account/rateLimits/read\" },\n    { id: 3, method: \"account/read\" },\n  ];\n  try {\n    const rpc =\n      options.rpc ?? ((requests) => appServerRpc(requests, options.timeoutMs));\n    const response = await rpc(messages);\n    const result = capacityFromSnapshot(\n      parseCliUsage(response.rateLimits),\n      options,\n      response.rateLimits,\n    );\n    const accountEnvelope = record(response.account);\n    if (\n      accountEnvelope &&\n      Object.hasOwn(accountEnvelope, \"account\") &&\n      !record(accountEnvelope.account)\n    ) {\n      result.authenticated = false;\n      result.available = \"unknown\";\n    }\n    return result;\n  } catch {\n    return unavailable(options);\n  }\n}\n\nexport async function probeCodexCapacity(\n  options: CodexProbeOptions,\n): Promise<CapacityReport> {\n  let parsed: UnknownRecord | null = null;\n  try {\n    const contents = await (options.readFile ?? readFile)(\n      resolveCodexAuthPath(options.env),\n      \"utf8\",\n    );\n    parsed = record(JSON.parse(contents));\n  } catch {\n    return cliFallback(options);\n  }\n\n  const auth = parsed ?? {};\n  const pat = nonEmptyText(auth.personal_access_token);\n  if (pat) {\n    try {\n      const fetcher = options.fetch ?? globalThis.fetch;\n      const whoami = record(\n        await fetchJson(\n          fetcher,\n          \"https://auth.openai.com/api/accounts/v1/user-auth-credential/whoami\",\n          { headers: { Authorization: `Bearer ${pat}` } },\n          options.timeoutMs ?? 5000,\n        ),\n      );\n      const accountId = nonEmptyText(whoami?.chatgpt_account_id);\n      if (!accountId) throw new Error(\"account unavailable\");\n      return capacityFromSnapshot(\n        await apiSnapshot(pat, accountId, \"pat\", options),\n        options,\n      );\n    } catch {\n      // Continue to a separately available OAuth credential before using the CLI.\n    }\n  }\n\n  const tokens = record(auth.tokens);\n  const accessToken = nonEmptyText(tokens?.access_token);\n  const accountId = nonEmptyText(tokens?.account_id);\n  if (\n    tokens &&\n    accessToken &&\n    accountId &&\n    !staleOAuth(tokens, accessToken, (options.now ?? (() => new Date()))())\n  ) {\n    try {\n      return capacityFromSnapshot(\n        await apiSnapshot(accessToken, accountId, \"oauth\", options),\n        options,\n      );\n    } catch {\n      return cliFallback(options);\n    }\n  }\n  return cliFallback(options);\n}\n"],"names":["spawn","readFile","join","CODEX_APP_SERVER_ARGS","record","value","Array","isArray","finiteNumber","Number","isFinite","nonEmptyText","length","resetTime","seconds","Date","toISOString","isNaN","parse","safeIdentifier","candidate","test","resolveCodexAuthPath","env","process","root","CODEX_HOME","HOME","toRateWindow","id","label","input","used","used_percent","limit_window_seconds","durationMinutes","usedPercent","resetsAt","reset_at","extraWindows","flatMap","entry","index","limit","scope","limit_name","windows","rate_limit","primary_window","secondary_window","filter","window","parseUsage","raw","source","response","limits","credits","additional_rate_limits","creditsRemaining","balance","cliWindow","windowDurationMins","cliSnapshotWindows","fallbackId","snapshot","limitId","primary","secondary","item","parseCliUsage","rateLimits","buckets","rateLimitsByLimitId","Object","entries","push","unique","Map","map","values","capacityFromSnapshot","context","hasUsage","some","reached","rateLimitReachedType","resetCredits","rateLimitResetCredits","usageLimitResetCredits","harness","provider","generatedAt","checkedAt","authenticated","available","availableCount","jwtExpiry","token","part","split","JSON","Buffer","from","toString","exp","staleOAuth","tokens","now","metadata","expires_at","expiresAt","expiry","parsed","getTime","fetchJson","fetcher","url","init","timeoutMs","controller","AbortController","timer","setTimeout","abort","signal","ok","Error","status","json","clearTimeout","apiSnapshot","accountId","options","fetch","globalThis","headers","Authorization","appServerRpc","messages","Promise","resolve","reject","child","stdio","results","buffer","settled","finish","error","kill","once","stdout","setEncoding","on","chunk","newline","indexOf","line","slice","trim","message","request","stdin","write","stringify","result","account","codexUnavailableReport","unavailable","cliFallback","installed","method","params","clientInfo","name","title","version","capabilities","rpc","requests","accountEnvelope","hasOwn","probeCodexCapacity","contents","auth","pat","personal_access_token","whoami","chatgpt_account_id","accessToken","access_token","account_id"],"mappings":"AAAA,SAASA,KAAK,QAAQ,qBAAqB;AAC3C,SAASC,QAAQ,QAAQ,mBAAmB;AAC5C,SAASC,IAAI,QAAQ,YAAY;AAejC,OAAO,MAAMC,wBAAwB;IACnC;IACA;IACA;IACA;IACA;CACD,CAAU;AAaX,SAASC,OAAOC,KAAc;IAC5B,OAAOA,UAAU,QAAQ,OAAOA,UAAU,YAAY,CAACC,MAAMC,OAAO,CAACF,SAChEA,QACD;AACN;AAEA,SAASG,aAAaH,KAAc;IAClC,OAAO,OAAOA,UAAU,YAAYI,OAAOC,QAAQ,CAACL,SAASA,QAAQ;AACvE;AAEA,SAASM,aAAaN,KAAc;IAClC,OAAO,OAAOA,UAAU,YAAYA,MAAMO,MAAM,GAAG,IAAIP,QAAQ;AACjE;AAEA,SAASQ,UAAUR,KAAc;IAC/B,MAAMS,UAAUN,aAAaH;IAC7B,IAAIS,YAAY,MAAM,OAAO,IAAIC,KAAKD,UAAU,MAAME,WAAW;IACjE,IAAI,OAAOX,UAAU,YAAY,CAACI,OAAOQ,KAAK,CAACF,KAAKG,KAAK,CAACb,SACxD,OAAO,IAAIU,KAAKV,OAAOW,WAAW;IACpC,OAAO;AACT;AAEA,SAASG,eAAed,KAAc;IACpC,MAAMe,YAAYT,aAAaN;IAC/B,IAAI,CAACe,aAAa,CAAC,2BAA2BC,IAAI,CAACD,YAAY,OAAO;IACtE,IAAI,2CAA2CC,IAAI,CAACD,YAAY,OAAO;IACvE,OAAOA;AACT;AAEA,OAAO,SAASE,qBACdC,MAAyBC,QAAQD,GAAG;IAEpC,MAAME,OAAOF,IAAIG,UAAU,IAAIxB,KAAKqB,IAAII,IAAI,IAAI,IAAI;IACpD,OAAOzB,KAAKuB,MAAM;AACpB;AAEA,OAAO,SAASG,aACdvB,KAAc,EACdwB,EAAU,EACVC,KAAa;IAEb,MAAMC,QAAQ3B,OAAOC;IACrB,IAAI,CAAC0B,OAAO,OAAO;IACnB,MAAMC,OAAOxB,aAAauB,MAAME,YAAY;IAC5C,MAAMnB,UAAUN,aAAauB,MAAMG,oBAAoB;IACvD,OAAO;QACLL;QACAC;QACAK,iBAAiBrB,YAAY,OAAO,OAAOA,UAAU;QACrDsB,aAAaJ;QACbK,UAAUxB,UAAUkB,MAAMO,QAAQ;IACpC;AACF;AAEA,SAASC,aAAalC,KAAc;IAClC,IAAI,CAACC,MAAMC,OAAO,CAACF,QAAQ,OAAO,EAAE;IACpC,OAAOA,MAAMmC,OAAO,CAAC,CAACC,OAAOC;QAC3B,MAAMC,QAAQvC,OAAOqC;QACrB,IAAI,CAACE,OAAO,OAAO,EAAE;QACrB,MAAMC,QAAQzB,eAAewB,MAAME,UAAU,KAAK,CAAC,MAAM,EAAEH,QAAQ,GAAG;QACtE,MAAMI,UAAU1C,OAAOuC,MAAMI,UAAU,KAAKJ;QAC5C,OAAO;YACLf,aACEkB,QAAQE,cAAc,EACtB,GAAGJ,MAAM,QAAQ,CAAC,EAClB,GAAGA,MAAM,QAAQ,CAAC;YAEpBhB,aACEkB,QAAQG,gBAAgB,EACxB,GAAGL,MAAM,UAAU,CAAC,EACpB,GAAGA,MAAM,UAAU,CAAC;SAEvB,CAACM,MAAM,CAAC,CAACC,SAAqCA,WAAW;IAC5D;AACF;AAEA,OAAO,SAASC,WACdC,GAAY,EACZC,MAAuB;IAEvB,MAAMC,WAAWnD,OAAOiD,QAAQ,CAAC;IACjC,MAAMG,SAASpD,OAAOmD,SAASR,UAAU,KAAK,CAAC;IAC/C,MAAMU,UAAUrD,OAAOmD,SAASE,OAAO,KAAK,CAAC;IAC7C,OAAO;QACLX,SAAS;YACPlB,aAAa4B,OAAOR,cAAc,EAAE,WAAW;YAC/CpB,aAAa4B,OAAOP,gBAAgB,EAAE,UAAU;eAC7CV,aAAagB,SAASG,sBAAsB;SAChD,CAACR,MAAM,CAAC,CAACC,SAAqCA,WAAW;QAC1DQ,kBAAkBnD,aAAaiD,QAAQG,OAAO;QAC9CN;IACF;AACF;AAEA,SAASO,UACPxD,KAAc,EACdwB,EAAU,EACVC,KAAa;IAEb,MAAMC,QAAQ3B,OAAOC;IACrB,IAAI,CAAC0B,OAAO,OAAO;IACnB,OAAO;QACLF;QACAC;QACAK,iBAAiB3B,aAAauB,MAAM+B,kBAAkB;QACtD1B,aAAa5B,aAAauB,MAAMK,WAAW;QAC3CC,UAAUxB,UAAUkB,MAAMM,QAAQ;IACpC;AACF;AAEA,SAAS0B,mBACP1D,KAAc,EACd2D,UAAkB;IAElB,MAAMC,WAAW7D,OAAOC;IACxB,IAAI,CAAC4D,UAAU,OAAO,EAAE;IACxB,MAAMrB,QACJzB,eAAe8C,SAASC,OAAO,KAAK/C,eAAe6C,eAAe;IACpE,OAAO;QACLH,UAAUI,SAASE,OAAO,EAAE,GAAGvB,MAAM,QAAQ,CAAC,EAAE,GAAGA,MAAM,QAAQ,CAAC;QAClEiB,UAAUI,SAASG,SAAS,EAAE,GAAGxB,MAAM,UAAU,CAAC,EAAE,GAAGA,MAAM,UAAU,CAAC;KACzE,CAACM,MAAM,CAAC,CAACmB,OAAiCA,SAAS;AACtD;AAEA,OAAO,SAASC,cAAcjB,GAAY;IACxC,MAAME,WAAWnD,OAAOiD,QAAQ,CAAC;IACjC,MAAMc,UAAU/D,OAAOmD,SAASgB,UAAU;IAC1C,MAAMzB,UAAUqB,UAAUJ,mBAAmBI,SAAS,WAAW,EAAE;IACnE,MAAMK,UAAUpE,OAAOmD,SAASkB,mBAAmB;IACnD,IAAID,SAAS;QACX,KAAK,MAAM,CAAC3C,IAAIoC,SAAS,IAAIS,OAAOC,OAAO,CAACH,SAC1C1B,QAAQ8B,IAAI,IAAIb,mBAAmBE,UAAUpC;IACjD;IACA,MAAMgD,SAAS;WACV,IAAIC,IAAIhC,QAAQiC,GAAG,CAAC,CAAC5B,SAAW;gBAACA,OAAOtB,EAAE;gBAAEsB;aAAO,GAAG6B,MAAM;KAChE;IACD,OAAO;QAAElC,SAAS+B;QAAQlB,kBAAkB;QAAML,QAAQ;IAAM;AAClE;AAEA,SAAS2B,qBACPhB,QAAuB,EACvBiB,OAA0B,EAC1B7B,GAAa;IAEb,MAAM8B,WAAWlB,SAASnB,OAAO,CAACsC,IAAI,CACpC,CAACjC,SAAWA,OAAOf,WAAW,KAAK;IAErC,MAAMmC,aAAanE,OAAOA,OAAOiD,MAAMkB;IACvC,MAAMc,UAAU1E,aAAa4D,YAAYe;IACzC,MAAMC,eACJnF,OAAOA,OAAOiD,MAAMmC,0BACpBpF,OAAOA,OAAOiD,MAAMoC;IACtB,OAAO;QACLC,SAAS;QACTC,UAAU;QACVC,aAAaV,QAAQW,SAAS;QAC9BC,eAAe;QACfC,WAAWV,UAAU,OAAOF,WAAW,QAAQ;QAC/CrC,SAASmB,SAASnB,OAAO;QACzBa,kBACEM,SAASN,gBAAgB,IAAInD,aAAa+E,cAAcS;IAC5D;AACF;AAEA,SAASC,UAAUC,KAAa;IAC9B,MAAMC,OAAOD,MAAME,KAAK,CAAC,IAAI,CAAC,EAAE;IAChC,IAAI,CAACD,MAAM,OAAO;IAClB,IAAI;QACF,OAAO3F,aACLJ,OAAOiG,KAAKnF,KAAK,CAACoF,OAAOC,IAAI,CAACJ,MAAM,aAAaK,QAAQ,CAAC,WAAWC;IAEzE,EAAE,OAAM;QACN,OAAO;IACT;AACF;AAEA,SAASC,WAAWC,MAAqB,EAAET,KAAa,EAAEU,GAAS;IACjE,MAAMC,WAAWF,OAAOG,UAAU,IAAIH,OAAOI,SAAS,IAAIJ,OAAOK,MAAM;IACvE,IAAIA,SAAwBxG,aAAaqG;IACzC,IAAI,OAAOA,aAAa,UAAU;QAChC,MAAMI,SAASlG,KAAKG,KAAK,CAAC2F;QAC1BG,SAASvG,OAAOQ,KAAK,CAACgG,UAAU,OAAOA,SAAS;IAClD;IACAD,WAAWf,UAAUC;IACrB,OAAOc,WAAW,QAAQA,UAAUJ,IAAIM,OAAO,KAAK;AACtD;AAEA,eAAeC,UACbC,OAAgC,EAChCC,GAAW,EACXC,IAAiB,EACjBC,SAAiB;IAEjB,MAAMC,aAAa,IAAIC;IACvB,MAAMC,QAAQC,WAAW,IAAMH,WAAWI,KAAK,IAAIL;IACnD,IAAI;QACF,MAAMhE,WAAW,MAAM6D,QAAQC,KAAK;YAAE,GAAGC,IAAI;YAAEO,QAAQL,WAAWK,MAAM;QAAC;QACzE,IAAI,CAACtE,SAASuE,EAAE,EACd,MAAM,IAAIC,MACRxE,SAASyE,MAAM,KAAK,MAAM,iBAAiB;QAE/C,OAAO,MAAMzE,SAAS0E,IAAI;IAC5B,SAAU;QACRC,aAAaR;IACf;AACF;AAEA,eAAeS,YACbjC,KAAa,EACbkC,SAAiB,EACjB9E,MAAuB,EACvB+E,OAA0B;IAE1B,MAAMjB,UAAUiB,QAAQC,KAAK,IAAIC,WAAWD,KAAK;IACjD,MAAMjF,MAAM,MAAM8D,UAChBC,SACA,8CACA;QACEoB,SAAS;YACPC,eAAe,CAAC,OAAO,EAAEvC,OAAO;YAChC,sBAAsBkC;QACxB;IACF,GACAC,QAAQd,SAAS,IAAI;IAEvB,OAAOnE,WAAWC,KAAKC;AACzB;AAEA,SAASoF,aACPC,QAAsB,EACtBpB,YAAY,IAAI;IAEhB,OAAO,IAAIqB,QAAQ,CAACC,SAASC;QAC3B,MAAMC,QAAQ/I,MAAM,SAASG,uBAAuB;YAClD6I,OAAO;gBAAC;gBAAQ;gBAAQ;aAAS;QACnC;QACA,MAAMC,UAAiC,CAAC;QACxC,IAAIC,SAAS;QACb,IAAIC,UAAU;QACd,MAAMC,SAAS,CAACC;YACd,IAAIF,SAAS;YACbA,UAAU;YACVjB,aAAaR;YACbqB,MAAMO,IAAI;YACV,IAAID,OAAOP,OAAOO;iBACbR,QAAQI;QACf;QACA,MAAMvB,QAAQC,WACZ,IAAMyB,OAAO,IAAIrB,MAAM,2BACvBR;QAEFwB,MAAMQ,IAAI,CAAC,SAAS,IAClBH,OAAO,IAAIrB,MAAM;QAEnBgB,MAAMQ,IAAI,CAAC,QAAQ;YACjB,IAAI,CAACJ,SAASC,OAAO,IAAIrB,MAAM;QACjC;QACAgB,MAAMS,MAAM,CAACC,WAAW,CAAC;QACzBV,MAAMS,MAAM,CAACE,EAAE,CAAC,QAAQ,CAACC;YACvBT,UAAUS;YACV,OAAS;gBACP,MAAMC,UAAUV,OAAOW,OAAO,CAAC;gBAC/B,IAAID,UAAU,GAAG;gBACjB,MAAME,OAAOZ,OAAOa,KAAK,CAAC,GAAGH,SAASI,IAAI;gBAC1Cd,SAASA,OAAOa,KAAK,CAACH,UAAU;gBAChC,IAAI,CAACE,MAAM;gBACX,IAAIG;gBACJ,IAAI;oBACFA,UAAU5D,KAAKnF,KAAK,CAAC4I;gBACvB,EAAE,OAAM;oBACN;gBACF;gBACA,IAAIG,QAAQpI,EAAE,KAAK,GAAG;oBACpB,KAAK,MAAMqI,WAAWvB,SAASoB,KAAK,CAAC,GACnChB,MAAMoB,KAAK,CAACC,KAAK,CAAC,GAAG/D,KAAKgE,SAAS,CAACH,SAAS,EAAE,CAAC;gBACpD,OAAO,IAAID,QAAQpI,EAAE,KAAK,GAAG;oBAC3B,IAAIoI,QAAQZ,KAAK,EACfD,OAAO,IAAIrB,MAAM;yBACdkB,QAAQ1E,UAAU,GAAG0F,QAAQK,MAAM;gBAC1C,OAAO,IAAIL,QAAQpI,EAAE,KAAK,GAAG;oBAC3B,IAAIoI,QAAQZ,KAAK,EAAED,OAAO,IAAIrB,MAAM;yBAC/BkB,QAAQsB,OAAO,GAAGN,QAAQK,MAAM;gBACvC;gBACA,IAAI,gBAAgBrB,WAAW,aAAaA,SAASG;YACvD;QACF;QACAL,MAAMoB,KAAK,CAACC,KAAK,CAAC,GAAG/D,KAAKgE,SAAS,CAAC1B,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC;IACtD;AACF;AAEA,OAAO,SAAS6B,uBAAuB3E,SAAiB;IACtD,OAAO;QACLH,SAAS;QACTC,UAAU;QACVC,aAAaC;QACbC,eAAe;QACfC,WAAW;QACXjD,SAAS,EAAE;QACXa,kBAAkB;IACpB;AACF;AAEA,SAAS8G,YAAYpC,OAA0B;IAC7C,OAAOmC,uBAAuBnC,QAAQxC,SAAS;AACjD;AAEA,eAAe6E,YACbrC,OAA0B;IAE1B,IAAI,CAACA,QAAQsC,SAAS,EAAE,OAAOF,YAAYpC;IAC3C,MAAMM,WAAyB;QAC7B;YACE9G,IAAI;YACJ+I,QAAQ;YACRC,QAAQ;gBACNC,YAAY;oBAAEC,MAAM;oBAAaC,OAAO;oBAAMC,SAAS;gBAAI;gBAC3DC,cAAc;YAChB;QACF;QACA;YAAEN,QAAQ;QAAc;QACxB;YAAE/I,IAAI;YAAG+I,QAAQ;QAA0B;QAC3C;YAAE/I,IAAI;YAAG+I,QAAQ;QAAe;KACjC;IACD,IAAI;QACF,MAAMO,MACJ9C,QAAQ8C,GAAG,IAAK,CAAA,CAACC,WAAa1C,aAAa0C,UAAU/C,QAAQd,SAAS,CAAA;QACxE,MAAMhE,WAAW,MAAM4H,IAAIxC;QAC3B,MAAM2B,SAASrF,qBACbX,cAAcf,SAASgB,UAAU,GACjC8D,SACA9E,SAASgB,UAAU;QAErB,MAAM8G,kBAAkBjL,OAAOmD,SAASgH,OAAO;QAC/C,IACEc,mBACA3G,OAAO4G,MAAM,CAACD,iBAAiB,cAC/B,CAACjL,OAAOiL,gBAAgBd,OAAO,GAC/B;YACAD,OAAOxE,aAAa,GAAG;YACvBwE,OAAOvE,SAAS,GAAG;QACrB;QACA,OAAOuE;IACT,EAAE,OAAM;QACN,OAAOG,YAAYpC;IACrB;AACF;AAEA,OAAO,eAAekD,mBACpBlD,OAA0B;IAE1B,IAAIpB,SAA+B;IACnC,IAAI;QACF,MAAMuE,WAAW,MAAM,AAACnD,CAAAA,QAAQpI,QAAQ,IAAIA,QAAO,EACjDqB,qBAAqB+G,QAAQ9G,GAAG,GAChC;QAEF0F,SAAS7G,OAAOiG,KAAKnF,KAAK,CAACsK;IAC7B,EAAE,OAAM;QACN,OAAOd,YAAYrC;IACrB;IAEA,MAAMoD,OAAOxE,UAAU,CAAC;IACxB,MAAMyE,MAAM/K,aAAa8K,KAAKE,qBAAqB;IACnD,IAAID,KAAK;QACP,IAAI;YACF,MAAMtE,UAAUiB,QAAQC,KAAK,IAAIC,WAAWD,KAAK;YACjD,MAAMsD,SAASxL,OACb,MAAM+G,UACJC,SACA,uEACA;gBAAEoB,SAAS;oBAAEC,eAAe,CAAC,OAAO,EAAEiD,KAAK;gBAAC;YAAE,GAC9CrD,QAAQd,SAAS,IAAI;YAGzB,MAAMa,YAAYzH,aAAaiL,QAAQC;YACvC,IAAI,CAACzD,WAAW,MAAM,IAAIL,MAAM;YAChC,OAAO9C,qBACL,MAAMkD,YAAYuD,KAAKtD,WAAW,OAAOC,UACzCA;QAEJ,EAAE,OAAM;QACN,4EAA4E;QAC9E;IACF;IAEA,MAAM1B,SAASvG,OAAOqL,KAAK9E,MAAM;IACjC,MAAMmF,cAAcnL,aAAagG,QAAQoF;IACzC,MAAM3D,YAAYzH,aAAagG,QAAQqF;IACvC,IACErF,UACAmF,eACA1D,aACA,CAAC1B,WAAWC,QAAQmF,aAAa,AAACzD,CAAAA,QAAQzB,GAAG,IAAK,CAAA,IAAM,IAAI7F,MAAK,CAAC,MAClE;QACA,IAAI;YACF,OAAOkE,qBACL,MAAMkD,YAAY2D,aAAa1D,WAAW,SAASC,UACnDA;QAEJ,EAAE,OAAM;YACN,OAAOqC,YAAYrC;QACrB;IACF;IACA,OAAOqC,YAAYrC;AACrB"}

package/dist/capacity/index.d.ts

@@ -7,5 +7,13 @@ export type CapacityProbeOptions = {
     access?: (target: string) => Promise<void>;
     probe?: typeof probeCodexCapacity;
 };
+export type ZaiCapacityOptions = {
+    now?: () => Date;
+    env?: NodeJS.ProcessEnv;
+    readFile?: (path: string, encoding: BufferEncoding) => Promise<string>;
+    fetch?: typeof globalThis.fetch;
+    timeoutMs?: number;
+};
 export declare function getCodexCapacityReport(options?: CapacityProbeOptions): Promise<CapacityReport>;
+export declare function getZaiCapacityReport(options?: ZaiCapacityOptions): Promise<CapacityReport>;
 //# sourceMappingURL=index.d.ts.map

package/dist/capacity/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/capacity/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAChD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD,YAAY,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjE,MAAM,MAAM,oBAAoB,GAAG;IACjC,GAAG,CAAC,EAAE,MAAM,IAAI,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3C,KAAK,CAAC,EAAE,OAAO,kBAAkB,CAAC;CACnC,CAAC;AA+BF,wBAAsB,sBAAsB,CAC1C,OAAO,GAAE,oBAAyB,GACjC,OAAO,CAAC,cAAc,CAAC,CAezB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/capacity/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAA0B,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAExE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD,YAAY,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjE,MAAM,MAAM,oBAAoB,GAAG;IACjC,GAAG,CAAC,EAAE,MAAM,IAAI,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3C,KAAK,CAAC,EAAE,OAAO,kBAAkB,CAAC;CACnC,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,GAAG,CAAC,EAAE,MAAM,IAAI,CAAC;IACjB,GAAG,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC;IACxB,QAAQ,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IACvE,KAAK,CAAC,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AA+BF,wBAAsB,sBAAsB,CAC1C,OAAO,GAAE,oBAAyB,GACjC,OAAO,CAAC,cAAc,CAAC,CAczB;AAED,wBAAsB,oBAAoB,CACxC,OAAO,GAAE,kBAAuB,GAC/B,OAAO,CAAC,cAAc,CAAC,CAMzB"}

package/dist/capacity/index.js

@@ -1,7 +1,8 @@
 import { constants } from "node:fs";
 import { access as fsAccess } from "node:fs/promises";
 import path from "node:path";
-import { probeCodexCapacity } from "./codex.js";
+import { codexUnavailableReport, probeCodexCapacity } from "./codex.js";
+import { probeZaiCapacity } from "./zai.js";
 async function canAccess(target, mode) {
     try {
         await fsAccess(target, mode);
@@ -35,15 +36,15 @@ export async function getCodexCapacityReport(options = {}) {
             checkedAt: generatedAt
         });
     } catch  {
-        return {
-            provider: "codex",
-            generatedAt,
-            authenticated: null,
-            available: "unknown",
-            windows: [],
-            creditsRemaining: null
-        };
+        return codexUnavailableReport(generatedAt);
     }
 }
+export async function getZaiCapacityReport(options = {}) {
+    const { now, ...probeOptions } = options;
+    return probeZaiCapacity({
+        ...probeOptions,
+        checkedAt: (now?.() ?? new Date()).toISOString()
+    });
+}
 
 //# sourceMappingURL=index.js.map

package/dist/capacity/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/capacity/index.ts"],"sourcesContent":["import { constants } from \"node:fs\";\nimport { access as fsAccess } from \"node:fs/promises\";\nimport path from \"node:path\";\nimport { probeCodexCapacity } from \"./codex.js\";\nimport type { CapacityReport } from \"./types.js\";\n\nexport type { CapacityReport, CapacityWindow } from \"./types.js\";\n\nexport type CapacityProbeOptions = {\n  now?: () => Date;\n  path?: string;\n  access?: (target: string) => Promise<void>;\n  probe?: typeof probeCodexCapacity;\n};\n\nasync function canAccess(target: string, mode: number): Promise<boolean> {\n  try {\n    await fsAccess(target, mode);\n    return true;\n  } catch {\n    return false;\n  }\n}\n\nasync function isCodexInstalled(\n  pathValue: string,\n  checkAccess?: (target: string) => Promise<void>,\n): Promise<boolean> {\n  const directories = pathValue.split(path.delimiter).filter(Boolean);\n  for (const directory of directories) {\n    const executable = path.join(directory, \"codex\");\n    if (checkAccess) {\n      try {\n        await checkAccess(executable);\n        return true;\n      } catch {\n        continue;\n      }\n    }\n    if (await canAccess(executable, constants.X_OK)) return true;\n  }\n  return false;\n}\n\nexport async function getCodexCapacityReport(\n  options: CapacityProbeOptions = {},\n): Promise<CapacityReport> {\n  const generatedAt = (options.now?.() ?? new Date()).toISOString();\n  const installed = await isCodexInstalled(options.path ?? process.env.PATH ?? \"\", options.access);\n  try {\n    return await (options.probe ?? probeCodexCapacity)({ installed, checkedAt: generatedAt });\n  } catch {\n    return {\n      provider: \"codex\",\n      generatedAt,\n      authenticated: null,\n      available: \"unknown\",\n      windows: [],\n      creditsRemaining: null,\n    };\n  }\n}\n"],"names":["constants","access","fsAccess","path","probeCodexCapacity","canAccess","target","mode","isCodexInstalled","pathValue","checkAccess","directories","split","delimiter","filter","Boolean","directory","executable","join","X_OK","getCodexCapacityReport","options","generatedAt","now","Date","toISOString","installed","process","env","PATH","probe","checkedAt","provider","authenticated","available","windows","creditsRemaining"],"mappings":"AAAA,SAASA,SAAS,QAAQ,UAAU;AACpC,SAASC,UAAUC,QAAQ,QAAQ,mBAAmB;AACtD,OAAOC,UAAU,YAAY;AAC7B,SAASC,kBAAkB,QAAQ,aAAa;AAYhD,eAAeC,UAAUC,MAAc,EAAEC,IAAY;IACnD,IAAI;QACF,MAAML,SAASI,QAAQC;QACvB,OAAO;IACT,EAAE,OAAM;QACN,OAAO;IACT;AACF;AAEA,eAAeC,iBACbC,SAAiB,EACjBC,WAA+C;IAE/C,MAAMC,cAAcF,UAAUG,KAAK,CAACT,KAAKU,SAAS,EAAEC,MAAM,CAACC;IAC3D,KAAK,MAAMC,aAAaL,YAAa;QACnC,MAAMM,aAAad,KAAKe,IAAI,CAACF,WAAW;QACxC,IAAIN,aAAa;YACf,IAAI;gBACF,MAAMA,YAAYO;gBAClB,OAAO;YACT,EAAE,OAAM;gBACN;YACF;QACF;QACA,IAAI,MAAMZ,UAAUY,YAAYjB,UAAUmB,IAAI,GAAG,OAAO;IAC1D;IACA,OAAO;AACT;AAEA,OAAO,eAAeC,uBACpBC,UAAgC,CAAC,CAAC;IAElC,MAAMC,cAAc,AAACD,CAAAA,QAAQE,GAAG,QAAQ,IAAIC,MAAK,EAAGC,WAAW;IAC/D,MAAMC,YAAY,MAAMlB,iBAAiBa,QAAQlB,IAAI,IAAIwB,QAAQC,GAAG,CAACC,IAAI,IAAI,IAAIR,QAAQpB,MAAM;IAC/F,IAAI;QACF,OAAO,MAAM,AAACoB,CAAAA,QAAQS,KAAK,IAAI1B,kBAAiB,EAAG;YAAEsB;YAAWK,WAAWT;QAAY;IACzF,EAAE,OAAM;QACN,OAAO;YACLU,UAAU;YACVV;YACAW,eAAe;YACfC,WAAW;YACXC,SAAS,EAAE;YACXC,kBAAkB;QACpB;IACF;AACF"}
+{"version":3,"sources":["../../src/capacity/index.ts"],"sourcesContent":["import { constants } from \"node:fs\";\nimport { access as fsAccess } from \"node:fs/promises\";\nimport path from \"node:path\";\nimport { codexUnavailableReport, probeCodexCapacity } from \"./codex.js\";\nimport { probeZaiCapacity } from \"./zai.js\";\nimport type { CapacityReport } from \"./types.js\";\n\nexport type { CapacityReport, CapacityWindow } from \"./types.js\";\n\nexport type CapacityProbeOptions = {\n  now?: () => Date;\n  path?: string;\n  access?: (target: string) => Promise<void>;\n  probe?: typeof probeCodexCapacity;\n};\n\nexport type ZaiCapacityOptions = {\n  now?: () => Date;\n  env?: NodeJS.ProcessEnv;\n  readFile?: (path: string, encoding: BufferEncoding) => Promise<string>;\n  fetch?: typeof globalThis.fetch;\n  timeoutMs?: number;\n};\n\nasync function canAccess(target: string, mode: number): Promise<boolean> {\n  try {\n    await fsAccess(target, mode);\n    return true;\n  } catch {\n    return false;\n  }\n}\n\nasync function isCodexInstalled(\n  pathValue: string,\n  checkAccess?: (target: string) => Promise<void>,\n): Promise<boolean> {\n  const directories = pathValue.split(path.delimiter).filter(Boolean);\n  for (const directory of directories) {\n    const executable = path.join(directory, \"codex\");\n    if (checkAccess) {\n      try {\n        await checkAccess(executable);\n        return true;\n      } catch {\n        continue;\n      }\n    }\n    if (await canAccess(executable, constants.X_OK)) return true;\n  }\n  return false;\n}\n\nexport async function getCodexCapacityReport(\n  options: CapacityProbeOptions = {},\n): Promise<CapacityReport> {\n  const generatedAt = (options.now?.() ?? new Date()).toISOString();\n  const installed = await isCodexInstalled(\n    options.path ?? process.env.PATH ?? \"\",\n    options.access,\n  );\n  try {\n    return await (options.probe ?? probeCodexCapacity)({\n      installed,\n      checkedAt: generatedAt,\n    });\n  } catch {\n    return codexUnavailableReport(generatedAt);\n  }\n}\n\nexport async function getZaiCapacityReport(\n  options: ZaiCapacityOptions = {},\n): Promise<CapacityReport> {\n  const { now, ...probeOptions } = options;\n  return probeZaiCapacity({\n    ...probeOptions,\n    checkedAt: (now?.() ?? new Date()).toISOString(),\n  });\n}\n"],"names":["constants","access","fsAccess","path","codexUnavailableReport","probeCodexCapacity","probeZaiCapacity","canAccess","target","mode","isCodexInstalled","pathValue","checkAccess","directories","split","delimiter","filter","Boolean","directory","executable","join","X_OK","getCodexCapacityReport","options","generatedAt","now","Date","toISOString","installed","process","env","PATH","probe","checkedAt","getZaiCapacityReport","probeOptions"],"mappings":"AAAA,SAASA,SAAS,QAAQ,UAAU;AACpC,SAASC,UAAUC,QAAQ,QAAQ,mBAAmB;AACtD,OAAOC,UAAU,YAAY;AAC7B,SAASC,sBAAsB,EAAEC,kBAAkB,QAAQ,aAAa;AACxE,SAASC,gBAAgB,QAAQ,WAAW;AAoB5C,eAAeC,UAAUC,MAAc,EAAEC,IAAY;IACnD,IAAI;QACF,MAAMP,SAASM,QAAQC;QACvB,OAAO;IACT,EAAE,OAAM;QACN,OAAO;IACT;AACF;AAEA,eAAeC,iBACbC,SAAiB,EACjBC,WAA+C;IAE/C,MAAMC,cAAcF,UAAUG,KAAK,CAACX,KAAKY,SAAS,EAAEC,MAAM,CAACC;IAC3D,KAAK,MAAMC,aAAaL,YAAa;QACnC,MAAMM,aAAahB,KAAKiB,IAAI,CAACF,WAAW;QACxC,IAAIN,aAAa;YACf,IAAI;gBACF,MAAMA,YAAYO;gBAClB,OAAO;YACT,EAAE,OAAM;gBACN;YACF;QACF;QACA,IAAI,MAAMZ,UAAUY,YAAYnB,UAAUqB,IAAI,GAAG,OAAO;IAC1D;IACA,OAAO;AACT;AAEA,OAAO,eAAeC,uBACpBC,UAAgC,CAAC,CAAC;IAElC,MAAMC,cAAc,AAACD,CAAAA,QAAQE,GAAG,QAAQ,IAAIC,MAAK,EAAGC,WAAW;IAC/D,MAAMC,YAAY,MAAMlB,iBACtBa,QAAQpB,IAAI,IAAI0B,QAAQC,GAAG,CAACC,IAAI,IAAI,IACpCR,QAAQtB,MAAM;IAEhB,IAAI;QACF,OAAO,MAAM,AAACsB,CAAAA,QAAQS,KAAK,IAAI3B,kBAAiB,EAAG;YACjDuB;YACAK,WAAWT;QACb;IACF,EAAE,OAAM;QACN,OAAOpB,uBAAuBoB;IAChC;AACF;AAEA,OAAO,eAAeU,qBACpBX,UAA8B,CAAC,CAAC;IAEhC,MAAM,EAAEE,GAAG,EAAE,GAAGU,cAAc,GAAGZ;IACjC,OAAOjB,iBAAiB;QACtB,GAAG6B,YAAY;QACfF,WAAW,AAACR,CAAAA,WAAW,IAAIC,MAAK,EAAGC,WAAW;IAChD;AACF"}

package/dist/capacity/types.d.ts

@@ -1,12 +1,20 @@
 export type Availability = "yes" | "no" | "unknown";
+export type CapacityLimitType = "TOKENS_LIMIT" | "CREDIT_LIMIT" | "TIME_LIMIT";
 export interface CapacityWindow {
     id: string;
     label: string;
     durationMinutes: number | null;
     usedPercent: number | null;
     resetsAt: string | null;
+    limitType?: CapacityLimitType;
+    total?: number | null;
+    current?: number | null;
+    remaining?: number | null;
 }
 export interface CapacityReport {
+    /** Harness whose quota pool is measured, e.g. "pi" or "codex". */
+    harness: string;
+    /** Model provider backing the harness, e.g. "zai" or "openai". */
     provider: string;
     generatedAt: string;
     authenticated: boolean | null;

package/dist/capacity/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/capacity/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,YAAY,GAAG,KAAK,GAAG,IAAI,GAAG,SAAS,CAAC;AAEpD,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;CACzB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,SAAS,EAAE,YAAY,CAAC;IACxB,OAAO,EAAE,cAAc,EAAE,CAAC;IAC1B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/capacity/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,YAAY,GAAG,KAAK,GAAG,IAAI,GAAG,SAAS,CAAC;AACpD,MAAM,MAAM,iBAAiB,GAAG,cAAc,GAAG,cAAc,GAAG,YAAY,CAAC;AAE/E,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,SAAS,CAAC,EAAE,iBAAiB,CAAC;IAC9B,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC3B;AAED,MAAM,WAAW,cAAc;IAC7B,kEAAkE;IAClE,OAAO,EAAE,MAAM,CAAC;IAChB,kEAAkE;IAClE,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,SAAS,EAAE,YAAY,CAAC;IACxB,OAAO,EAAE,cAAc,EAAE,CAAC;IAC1B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC"}

package/dist/capacity/types.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/capacity/types.ts"],"sourcesContent":["export type Availability = \"yes\" | \"no\" | \"unknown\";\n\nexport interface CapacityWindow {\n  id: string;\n  label: string;\n  durationMinutes: number | null;\n  usedPercent: number | null;\n  resetsAt: string | null;\n}\n\nexport interface CapacityReport {\n  provider: string;\n  generatedAt: string;\n  authenticated: boolean | null;\n  available: Availability;\n  windows: CapacityWindow[];\n  creditsRemaining: number | null;\n}\n"],"names":[],"mappings":"AAUA,WAOC"}
+{"version":3,"sources":["../../src/capacity/types.ts"],"sourcesContent":["export type Availability = \"yes\" | \"no\" | \"unknown\";\nexport type CapacityLimitType = \"TOKENS_LIMIT\" | \"CREDIT_LIMIT\" | \"TIME_LIMIT\";\n\nexport interface CapacityWindow {\n  id: string;\n  label: string;\n  durationMinutes: number | null;\n  usedPercent: number | null;\n  resetsAt: string | null;\n  limitType?: CapacityLimitType;\n  total?: number | null;\n  current?: number | null;\n  remaining?: number | null;\n}\n\nexport interface CapacityReport {\n  /** Harness whose quota pool is measured, e.g. \"pi\" or \"codex\". */\n  harness: string;\n  /** Model provider backing the harness, e.g. \"zai\" or \"openai\". */\n  provider: string;\n  generatedAt: string;\n  authenticated: boolean | null;\n  available: Availability;\n  windows: CapacityWindow[];\n  creditsRemaining: number | null;\n}\n"],"names":[],"mappings":"AAeA,WAUC"}

package/dist/capacity/zai.d.ts

@@ -0,0 +1,14 @@
+import type { CapacityReport } from "./types.js";
+export type ZaiCredentialOptions = {
+    env?: NodeJS.ProcessEnv;
+    readFile?: (path: string, encoding: BufferEncoding) => Promise<string>;
+};
+export type ZaiProbeOptions = ZaiCredentialOptions & {
+    checkedAt: string;
+    fetch?: typeof globalThis.fetch;
+    timeoutMs?: number;
+};
+export declare function resolveZaiApiKey(options?: ZaiCredentialOptions): Promise<string>;
+export declare function parseZaiQuota(raw: unknown, checkedAt: string): CapacityReport;
+export declare function probeZaiCapacity(options: ZaiProbeOptions): Promise<CapacityReport>;
+//# sourceMappingURL=zai.d.ts.map

package/dist/capacity/zai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"zai.d.ts","sourceRoot":"","sources":["../../src/capacity/zai.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAGV,cAAc,EAEf,MAAM,YAAY,CAAC;AAkBpB,MAAM,MAAM,oBAAoB,GAAG;IACjC,GAAG,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC;IACxB,QAAQ,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;CACxE,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG,oBAAoB,GAAG;IACnD,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAqHF,wBAAsB,gBAAgB,CACpC,OAAO,GAAE,oBAAyB,GACjC,OAAO,CAAC,MAAM,CAAC,CA2BjB;AAED,wBAAgB,aAAa,CAAC,GAAG,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,GAAG,cAAc,CAuB7E;AAED,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,cAAc,CAAC,CA0BzB"}

package/dist/capacity/zai.js

@@ -0,0 +1,203 @@
+import { readFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { join } from "node:path";
+const ZAI_QUOTA_URL = "https://api.z.ai/api/monitor/usage/quota/limit";
+const MISSING_API_KEY_MESSAGE = "z.ai API key not found; set Z_AI_API_KEY or configure Pi provider zai";
+const LIMIT_METADATA = {
+    TOKENS_LIMIT: {
+        id: "tokens",
+        label: "Tokens"
+    },
+    CREDIT_LIMIT: {
+        id: "credit",
+        label: "Credit"
+    },
+    TIME_LIMIT: {
+        id: "time",
+        label: "Time"
+    }
+};
+const UNIT_METADATA = new Map([
+    [
+        1,
+        {
+            minutes: 24 * 60,
+            label: "day"
+        }
+    ],
+    [
+        3,
+        {
+            minutes: 60,
+            label: "hour"
+        }
+    ],
+    [
+        5,
+        {
+            minutes: 1,
+            label: "minute"
+        }
+    ],
+    [
+        6,
+        {
+            minutes: 7 * 24 * 60,
+            label: "week"
+        }
+    ]
+]);
+function record(value) {
+    return value !== null && typeof value === "object" && !Array.isArray(value) ? value : null;
+}
+function nonEmptyText(value) {
+    return typeof value === "string" && value.trim() ? value.trim() : null;
+}
+function optionalNumber(value, field) {
+    if (value === undefined || value === null) return null;
+    if (typeof value !== "number" || !Number.isFinite(value)) {
+        throw new Error(`Invalid z.ai quota ${field}`);
+    }
+    return value;
+}
+function clampPercent(value) {
+    return Math.max(0, Math.min(100, value));
+}
+function integer(value) {
+    return typeof value === "number" && Number.isInteger(value) ? value : null;
+}
+function isCapacityLimitType(value) {
+    return Object.hasOwn(LIMIT_METADATA, value);
+}
+function durationMinutes(unit, number, type) {
+    if (type === "TIME_LIMIT" && unit === 5 && number === 1) return 30 * 24 * 60;
+    const metadata = UNIT_METADATA.get(unit);
+    return metadata === undefined || number <= 0 ? null : metadata.minutes * number;
+}
+function windowLabel(type, unit, number) {
+    if (type === "TIME_LIMIT" && unit === 5 && number === 1) return "MCP · monthly";
+    const kind = LIMIT_METADATA[type].label;
+    const unitLabel = UNIT_METADATA.get(unit)?.label;
+    if (!unitLabel) return kind;
+    return `${kind} · ${number} ${unitLabel}${number === 1 ? "" : "s"}`;
+}
+function resetTime(value) {
+    const milliseconds = optionalNumber(value, "limit nextResetTime");
+    if (milliseconds === null) return null;
+    const date = new Date(milliseconds);
+    if (Number.isNaN(date.getTime())) throw new Error("Invalid z.ai quota limit nextResetTime");
+    return date.toISOString();
+}
+function availability(windows) {
+    if (windows.length === 0) return "unknown";
+    if (windows.some((window)=>window.usedPercent !== null && window.usedPercent < 100)) {
+        return "yes";
+    }
+    return "no";
+}
+function parseLimit(value, index) {
+    const limit = record(value);
+    if (!limit) throw new Error("Invalid z.ai quota limit entry");
+    const unit = integer(limit.unit);
+    const count = integer(limit.number);
+    if (typeof limit.type !== "string" || unit === null || count === null) {
+        throw new Error("Invalid z.ai quota limit entry");
+    }
+    const serverPercent = optionalNumber(limit.percentage, "limit percentage");
+    if (serverPercent === null) throw new Error("Invalid z.ai quota limit percentage");
+    if (!isCapacityLimitType(limit.type)) return null;
+    const total = optionalNumber(limit.usage, "limit usage");
+    const current = optionalNumber(limit.currentValue, "limit currentValue");
+    const remaining = optionalNumber(limit.remaining, "limit remaining");
+    let consumed = total !== null && remaining !== null ? total - remaining : null;
+    if (current !== null) consumed = Math.max(consumed ?? current, current);
+    let usedPercent = serverPercent;
+    if (total !== null && total > 0 && consumed !== null) {
+        usedPercent = consumed / total * 100;
+    }
+    const metadata = LIMIT_METADATA[limit.type];
+    return {
+        id: `zai:${metadata.id}:${index + 1}`,
+        label: windowLabel(limit.type, unit, count),
+        limitType: limit.type,
+        durationMinutes: durationMinutes(unit, count, limit.type),
+        usedPercent: clampPercent(usedPercent),
+        resetsAt: resetTime(limit.nextResetTime),
+        total,
+        current,
+        remaining
+    };
+}
+export async function resolveZaiApiKey(options = {}) {
+    const env = options.env ?? process.env;
+    const environmentKey = nonEmptyText(env.Z_AI_API_KEY);
+    if (environmentKey) return environmentKey;
+    const authPath = join(env.HOME || homedir(), ".pi", "agent", "auth.json");
+    let parsed;
+    try {
+        const contents = await (options.readFile ?? readFile)(authPath, "utf8");
+        const root = record(JSON.parse(contents));
+        if (!root) throw new Error("invalid");
+        parsed = root;
+    } catch (error) {
+        if (record(error)?.code === "ENOENT") {
+            throw new Error(MISSING_API_KEY_MESSAGE);
+        }
+        throw new Error("z.ai Pi auth file is malformed");
+    }
+    const credential = record(parsed.zai);
+    if (!credential) {
+        throw new Error(MISSING_API_KEY_MESSAGE);
+    }
+    const key = credential.type === "api_key" ? nonEmptyText(credential.key) : null;
+    if (!key) throw new Error("z.ai Pi auth file has a malformed zai credential");
+    return key;
+}
+export function parseZaiQuota(raw, checkedAt) {
+    const root = record(raw);
+    const data = record(root?.data);
+    if (root?.success !== true || root.code !== 200 || !data || !Array.isArray(data.limits)) {
+        throw new Error("Invalid z.ai quota response");
+    }
+    const windows = data.limits.map(parseLimit).filter((window)=>window !== null);
+    return {
+        harness: "pi",
+        provider: "zai",
+        generatedAt: checkedAt,
+        authenticated: true,
+        available: availability(windows),
+        windows,
+        creditsRemaining: null
+    };
+}
+export async function probeZaiCapacity(options) {
+    const key = await resolveZaiApiKey(options);
+    const controller = new AbortController();
+    const timer = setTimeout(()=>controller.abort(), options.timeoutMs ?? 5000);
+    let response;
+    try {
+        response = await (options.fetch ?? globalThis.fetch)(ZAI_QUOTA_URL, {
+            method: "GET",
+            headers: {
+                Authorization: `Bearer ${key}`
+            },
+            signal: controller.signal
+        });
+    } catch  {
+        throw new Error("z.ai quota request failed");
+    } finally{
+        clearTimeout(timer);
+    }
+    if (response.status !== 200) {
+        throw new Error(`z.ai quota request failed: HTTP ${response.status}`);
+    }
+    let raw;
+    try {
+        raw = await response.json();
+    } catch  {
+        throw new Error("z.ai quota response is not valid JSON");
+    }
+    return parseZaiQuota(raw, options.checkedAt);
+}
+
+//# sourceMappingURL=zai.js.map

package/dist/capacity/zai.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/capacity/zai.ts"],"sourcesContent":["import { readFile } from \"node:fs/promises\";\nimport { homedir } from \"node:os\";\nimport { join } from \"node:path\";\nimport type {\n  Availability,\n  CapacityLimitType,\n  CapacityReport,\n  CapacityWindow,\n} from \"./types.js\";\n\nconst ZAI_QUOTA_URL = \"https://api.z.ai/api/monitor/usage/quota/limit\";\nconst MISSING_API_KEY_MESSAGE =\n  \"z.ai API key not found; set Z_AI_API_KEY or configure Pi provider zai\";\nconst LIMIT_METADATA = {\n  TOKENS_LIMIT: { id: \"tokens\", label: \"Tokens\" },\n  CREDIT_LIMIT: { id: \"credit\", label: \"Credit\" },\n  TIME_LIMIT: { id: \"time\", label: \"Time\" },\n} as const satisfies Record<CapacityLimitType, { id: string; label: string }>;\nconst UNIT_METADATA = new Map<number, { minutes: number; label: string }>([\n  [1, { minutes: 24 * 60, label: \"day\" }],\n  [3, { minutes: 60, label: \"hour\" }],\n  [5, { minutes: 1, label: \"minute\" }],\n  [6, { minutes: 7 * 24 * 60, label: \"week\" }],\n]);\ntype UnknownRecord = Record<string, unknown>;\n\nexport type ZaiCredentialOptions = {\n  env?: NodeJS.ProcessEnv;\n  readFile?: (path: string, encoding: BufferEncoding) => Promise<string>;\n};\n\nexport type ZaiProbeOptions = ZaiCredentialOptions & {\n  checkedAt: string;\n  fetch?: typeof globalThis.fetch;\n  timeoutMs?: number;\n};\n\nfunction record(value: unknown): UnknownRecord | null {\n  return value !== null && typeof value === \"object\" && !Array.isArray(value)\n    ? (value as UnknownRecord)\n    : null;\n}\n\nfunction nonEmptyText(value: unknown): string | null {\n  return typeof value === \"string\" && value.trim() ? value.trim() : null;\n}\n\nfunction optionalNumber(value: unknown, field: string): number | null {\n  if (value === undefined || value === null) return null;\n  if (typeof value !== \"number\" || !Number.isFinite(value)) {\n    throw new Error(`Invalid z.ai quota ${field}`);\n  }\n  return value;\n}\n\nfunction clampPercent(value: number): number {\n  return Math.max(0, Math.min(100, value));\n}\n\nfunction integer(value: unknown): number | null {\n  return typeof value === \"number\" && Number.isInteger(value) ? value : null;\n}\n\nfunction isCapacityLimitType(value: string): value is CapacityLimitType {\n  return Object.hasOwn(LIMIT_METADATA, value);\n}\n\nfunction durationMinutes(\n  unit: number,\n  number: number,\n  type: CapacityLimitType,\n): number | null {\n  if (type === \"TIME_LIMIT\" && unit === 5 && number === 1) return 30 * 24 * 60;\n  const metadata = UNIT_METADATA.get(unit);\n  return metadata === undefined || number <= 0\n    ? null\n    : metadata.minutes * number;\n}\n\nfunction windowLabel(\n  type: CapacityLimitType,\n  unit: number,\n  number: number,\n): string {\n  if (type === \"TIME_LIMIT\" && unit === 5 && number === 1)\n    return \"MCP · monthly\";\n  const kind = LIMIT_METADATA[type].label;\n  const unitLabel = UNIT_METADATA.get(unit)?.label;\n  if (!unitLabel) return kind;\n  return `${kind} · ${number} ${unitLabel}${number === 1 ? \"\" : \"s\"}`;\n}\n\nfunction resetTime(value: unknown): string | null {\n  const milliseconds = optionalNumber(value, \"limit nextResetTime\");\n  if (milliseconds === null) return null;\n  const date = new Date(milliseconds);\n  if (Number.isNaN(date.getTime()))\n    throw new Error(\"Invalid z.ai quota limit nextResetTime\");\n  return date.toISOString();\n}\n\nfunction availability(windows: CapacityWindow[]): Availability {\n  if (windows.length === 0) return \"unknown\";\n  if (\n    windows.some(\n      (window) => window.usedPercent !== null && window.usedPercent < 100,\n    )\n  ) {\n    return \"yes\";\n  }\n  return \"no\";\n}\n\nfunction parseLimit(value: unknown, index: number): CapacityWindow | null {\n  const limit = record(value);\n  if (!limit) throw new Error(\"Invalid z.ai quota limit entry\");\n  const unit = integer(limit.unit);\n  const count = integer(limit.number);\n  if (typeof limit.type !== \"string\" || unit === null || count === null) {\n    throw new Error(\"Invalid z.ai quota limit entry\");\n  }\n  const serverPercent = optionalNumber(limit.percentage, \"limit percentage\");\n  if (serverPercent === null)\n    throw new Error(\"Invalid z.ai quota limit percentage\");\n  if (!isCapacityLimitType(limit.type)) return null;\n\n  const total = optionalNumber(limit.usage, \"limit usage\");\n  const current = optionalNumber(limit.currentValue, \"limit currentValue\");\n  const remaining = optionalNumber(limit.remaining, \"limit remaining\");\n  let consumed =\n    total !== null && remaining !== null ? total - remaining : null;\n  if (current !== null) consumed = Math.max(consumed ?? current, current);\n\n  let usedPercent = serverPercent;\n  if (total !== null && total > 0 && consumed !== null) {\n    usedPercent = (consumed / total) * 100;\n  }\n  const metadata = LIMIT_METADATA[limit.type];\n\n  return {\n    id: `zai:${metadata.id}:${index + 1}`,\n    label: windowLabel(limit.type, unit, count),\n    limitType: limit.type,\n    durationMinutes: durationMinutes(unit, count, limit.type),\n    usedPercent: clampPercent(usedPercent),\n    resetsAt: resetTime(limit.nextResetTime),\n    total,\n    current,\n    remaining,\n  };\n}\n\nexport async function resolveZaiApiKey(\n  options: ZaiCredentialOptions = {},\n): Promise<string> {\n  const env = options.env ?? process.env;\n  const environmentKey = nonEmptyText(env.Z_AI_API_KEY);\n  if (environmentKey) return environmentKey;\n\n  const authPath = join(env.HOME || homedir(), \".pi\", \"agent\", \"auth.json\");\n  let parsed: UnknownRecord;\n  try {\n    const contents = await (options.readFile ?? readFile)(authPath, \"utf8\");\n    const root = record(JSON.parse(contents));\n    if (!root) throw new Error(\"invalid\");\n    parsed = root;\n  } catch (error) {\n    if (record(error)?.code === \"ENOENT\") {\n      throw new Error(MISSING_API_KEY_MESSAGE);\n    }\n    throw new Error(\"z.ai Pi auth file is malformed\");\n  }\n\n  const credential = record(parsed.zai);\n  if (!credential) {\n    throw new Error(MISSING_API_KEY_MESSAGE);\n  }\n  const key =\n    credential.type === \"api_key\" ? nonEmptyText(credential.key) : null;\n  if (!key) throw new Error(\"z.ai Pi auth file has a malformed zai credential\");\n  return key;\n}\n\nexport function parseZaiQuota(raw: unknown, checkedAt: string): CapacityReport {\n  const root = record(raw);\n  const data = record(root?.data);\n  if (\n    root?.success !== true ||\n    root.code !== 200 ||\n    !data ||\n    !Array.isArray(data.limits)\n  ) {\n    throw new Error(\"Invalid z.ai quota response\");\n  }\n  const windows = data.limits\n    .map(parseLimit)\n    .filter((window): window is CapacityWindow => window !== null);\n  return {\n    harness: \"pi\",\n    provider: \"zai\",\n    generatedAt: checkedAt,\n    authenticated: true,\n    available: availability(windows),\n    windows,\n    creditsRemaining: null,\n  };\n}\n\nexport async function probeZaiCapacity(\n  options: ZaiProbeOptions,\n): Promise<CapacityReport> {\n  const key = await resolveZaiApiKey(options);\n  const controller = new AbortController();\n  const timer = setTimeout(() => controller.abort(), options.timeoutMs ?? 5000);\n  let response: Response;\n  try {\n    response = await (options.fetch ?? globalThis.fetch)(ZAI_QUOTA_URL, {\n      method: \"GET\",\n      headers: { Authorization: `Bearer ${key}` },\n      signal: controller.signal,\n    });\n  } catch {\n    throw new Error(\"z.ai quota request failed\");\n  } finally {\n    clearTimeout(timer);\n  }\n  if (response.status !== 200) {\n    throw new Error(`z.ai quota request failed: HTTP ${response.status}`);\n  }\n  let raw: unknown;\n  try {\n    raw = await response.json();\n  } catch {\n    throw new Error(\"z.ai quota response is not valid JSON\");\n  }\n  return parseZaiQuota(raw, options.checkedAt);\n}\n"],"names":["readFile","homedir","join","ZAI_QUOTA_URL","MISSING_API_KEY_MESSAGE","LIMIT_METADATA","TOKENS_LIMIT","id","label","CREDIT_LIMIT","TIME_LIMIT","UNIT_METADATA","Map","minutes","record","value","Array","isArray","nonEmptyText","trim","optionalNumber","field","undefined","Number","isFinite","Error","clampPercent","Math","max","min","integer","isInteger","isCapacityLimitType","Object","hasOwn","durationMinutes","unit","number","type","metadata","get","windowLabel","kind","unitLabel","resetTime","milliseconds","date","Date","isNaN","getTime","toISOString","availability","windows","length","some","window","usedPercent","parseLimit","index","limit","count","serverPercent","percentage","total","usage","current","currentValue","remaining","consumed","limitType","resetsAt","nextResetTime","resolveZaiApiKey","options","env","process","environmentKey","Z_AI_API_KEY","authPath","HOME","parsed","contents","root","JSON","parse","error","code","credential","zai","key","parseZaiQuota","raw","checkedAt","data","success","limits","map","filter","harness","provider","generatedAt","authenticated","available","creditsRemaining","probeZaiCapacity","controller","AbortController","timer","setTimeout","abort","timeoutMs","response","fetch","globalThis","method","headers","Authorization","signal","clearTimeout","status","json"],"mappings":"AAAA,SAASA,QAAQ,QAAQ,mBAAmB;AAC5C,SAASC,OAAO,QAAQ,UAAU;AAClC,SAASC,IAAI,QAAQ,YAAY;AAQjC,MAAMC,gBAAgB;AACtB,MAAMC,0BACJ;AACF,MAAMC,iBAAiB;IACrBC,cAAc;QAAEC,IAAI;QAAUC,OAAO;IAAS;IAC9CC,cAAc;QAAEF,IAAI;QAAUC,OAAO;IAAS;IAC9CE,YAAY;QAAEH,IAAI;QAAQC,OAAO;IAAO;AAC1C;AACA,MAAMG,gBAAgB,IAAIC,IAAgD;IACxE;QAAC;QAAG;YAAEC,SAAS,KAAK;YAAIL,OAAO;QAAM;KAAE;IACvC;QAAC;QAAG;YAAEK,SAAS;YAAIL,OAAO;QAAO;KAAE;IACnC;QAAC;QAAG;YAAEK,SAAS;YAAGL,OAAO;QAAS;KAAE;IACpC;QAAC;QAAG;YAAEK,SAAS,IAAI,KAAK;YAAIL,OAAO;QAAO;KAAE;CAC7C;AAcD,SAASM,OAAOC,KAAc;IAC5B,OAAOA,UAAU,QAAQ,OAAOA,UAAU,YAAY,CAACC,MAAMC,OAAO,CAACF,SAChEA,QACD;AACN;AAEA,SAASG,aAAaH,KAAc;IAClC,OAAO,OAAOA,UAAU,YAAYA,MAAMI,IAAI,KAAKJ,MAAMI,IAAI,KAAK;AACpE;AAEA,SAASC,eAAeL,KAAc,EAAEM,KAAa;IACnD,IAAIN,UAAUO,aAAaP,UAAU,MAAM,OAAO;IAClD,IAAI,OAAOA,UAAU,YAAY,CAACQ,OAAOC,QAAQ,CAACT,QAAQ;QACxD,MAAM,IAAIU,MAAM,CAAC,mBAAmB,EAAEJ,OAAO;IAC/C;IACA,OAAON;AACT;AAEA,SAASW,aAAaX,KAAa;IACjC,OAAOY,KAAKC,GAAG,CAAC,GAAGD,KAAKE,GAAG,CAAC,KAAKd;AACnC;AAEA,SAASe,QAAQf,KAAc;IAC7B,OAAO,OAAOA,UAAU,YAAYQ,OAAOQ,SAAS,CAAChB,SAASA,QAAQ;AACxE;AAEA,SAASiB,oBAAoBjB,KAAa;IACxC,OAAOkB,OAAOC,MAAM,CAAC7B,gBAAgBU;AACvC;AAEA,SAASoB,gBACPC,IAAY,EACZC,MAAc,EACdC,IAAuB;IAEvB,IAAIA,SAAS,gBAAgBF,SAAS,KAAKC,WAAW,GAAG,OAAO,KAAK,KAAK;IAC1E,MAAME,WAAW5B,cAAc6B,GAAG,CAACJ;IACnC,OAAOG,aAAajB,aAAae,UAAU,IACvC,OACAE,SAAS1B,OAAO,GAAGwB;AACzB;AAEA,SAASI,YACPH,IAAuB,EACvBF,IAAY,EACZC,MAAc;IAEd,IAAIC,SAAS,gBAAgBF,SAAS,KAAKC,WAAW,GACpD,OAAO;IACT,MAAMK,OAAOrC,cAAc,CAACiC,KAAK,CAAC9B,KAAK;IACvC,MAAMmC,YAAYhC,cAAc6B,GAAG,CAACJ,OAAO5B;IAC3C,IAAI,CAACmC,WAAW,OAAOD;IACvB,OAAO,GAAGA,KAAK,GAAG,EAAEL,OAAO,CAAC,EAAEM,YAAYN,WAAW,IAAI,KAAK,KAAK;AACrE;AAEA,SAASO,UAAU7B,KAAc;IAC/B,MAAM8B,eAAezB,eAAeL,OAAO;IAC3C,IAAI8B,iBAAiB,MAAM,OAAO;IAClC,MAAMC,OAAO,IAAIC,KAAKF;IACtB,IAAItB,OAAOyB,KAAK,CAACF,KAAKG,OAAO,KAC3B,MAAM,IAAIxB,MAAM;IAClB,OAAOqB,KAAKI,WAAW;AACzB;AAEA,SAASC,aAAaC,OAAyB;IAC7C,IAAIA,QAAQC,MAAM,KAAK,GAAG,OAAO;IACjC,IACED,QAAQE,IAAI,CACV,CAACC,SAAWA,OAAOC,WAAW,KAAK,QAAQD,OAAOC,WAAW,GAAG,MAElE;QACA,OAAO;IACT;IACA,OAAO;AACT;AAEA,SAASC,WAAW1C,KAAc,EAAE2C,KAAa;IAC/C,MAAMC,QAAQ7C,OAAOC;IACrB,IAAI,CAAC4C,OAAO,MAAM,IAAIlC,MAAM;IAC5B,MAAMW,OAAON,QAAQ6B,MAAMvB,IAAI;IAC/B,MAAMwB,QAAQ9B,QAAQ6B,MAAMtB,MAAM;IAClC,IAAI,OAAOsB,MAAMrB,IAAI,KAAK,YAAYF,SAAS,QAAQwB,UAAU,MAAM;QACrE,MAAM,IAAInC,MAAM;IAClB;IACA,MAAMoC,gBAAgBzC,eAAeuC,MAAMG,UAAU,EAAE;IACvD,IAAID,kBAAkB,MACpB,MAAM,IAAIpC,MAAM;IAClB,IAAI,CAACO,oBAAoB2B,MAAMrB,IAAI,GAAG,OAAO;IAE7C,MAAMyB,QAAQ3C,eAAeuC,MAAMK,KAAK,EAAE;IAC1C,MAAMC,UAAU7C,eAAeuC,MAAMO,YAAY,EAAE;IACnD,MAAMC,YAAY/C,eAAeuC,MAAMQ,SAAS,EAAE;IAClD,IAAIC,WACFL,UAAU,QAAQI,cAAc,OAAOJ,QAAQI,YAAY;IAC7D,IAAIF,YAAY,MAAMG,WAAWzC,KAAKC,GAAG,CAACwC,YAAYH,SAASA;IAE/D,IAAIT,cAAcK;IAClB,IAAIE,UAAU,QAAQA,QAAQ,KAAKK,aAAa,MAAM;QACpDZ,cAAc,AAACY,WAAWL,QAAS;IACrC;IACA,MAAMxB,WAAWlC,cAAc,CAACsD,MAAMrB,IAAI,CAAC;IAE3C,OAAO;QACL/B,IAAI,CAAC,IAAI,EAAEgC,SAAShC,EAAE,CAAC,CAAC,EAAEmD,QAAQ,GAAG;QACrClD,OAAOiC,YAAYkB,MAAMrB,IAAI,EAAEF,MAAMwB;QACrCS,WAAWV,MAAMrB,IAAI;QACrBH,iBAAiBA,gBAAgBC,MAAMwB,OAAOD,MAAMrB,IAAI;QACxDkB,aAAa9B,aAAa8B;QAC1Bc,UAAU1B,UAAUe,MAAMY,aAAa;QACvCR;QACAE;QACAE;IACF;AACF;AAEA,OAAO,eAAeK,iBACpBC,UAAgC,CAAC,CAAC;IAElC,MAAMC,MAAMD,QAAQC,GAAG,IAAIC,QAAQD,GAAG;IACtC,MAAME,iBAAiB1D,aAAawD,IAAIG,YAAY;IACpD,IAAID,gBAAgB,OAAOA;IAE3B,MAAME,WAAW5E,KAAKwE,IAAIK,IAAI,IAAI9E,WAAW,OAAO,SAAS;IAC7D,IAAI+E;IACJ,IAAI;QACF,MAAMC,WAAW,MAAM,AAACR,CAAAA,QAAQzE,QAAQ,IAAIA,QAAO,EAAG8E,UAAU;QAChE,MAAMI,OAAOpE,OAAOqE,KAAKC,KAAK,CAACH;QAC/B,IAAI,CAACC,MAAM,MAAM,IAAIzD,MAAM;QAC3BuD,SAASE;IACX,EAAE,OAAOG,OAAO;QACd,IAAIvE,OAAOuE,QAAQC,SAAS,UAAU;YACpC,MAAM,IAAI7D,MAAMrB;QAClB;QACA,MAAM,IAAIqB,MAAM;IAClB;IAEA,MAAM8D,aAAazE,OAAOkE,OAAOQ,GAAG;IACpC,IAAI,CAACD,YAAY;QACf,MAAM,IAAI9D,MAAMrB;IAClB;IACA,MAAMqF,MACJF,WAAWjD,IAAI,KAAK,YAAYpB,aAAaqE,WAAWE,GAAG,IAAI;IACjE,IAAI,CAACA,KAAK,MAAM,IAAIhE,MAAM;IAC1B,OAAOgE;AACT;AAEA,OAAO,SAASC,cAAcC,GAAY,EAAEC,SAAiB;IAC3D,MAAMV,OAAOpE,OAAO6E;IACpB,MAAME,OAAO/E,OAAOoE,MAAMW;IAC1B,IACEX,MAAMY,YAAY,QAClBZ,KAAKI,IAAI,KAAK,OACd,CAACO,QACD,CAAC7E,MAAMC,OAAO,CAAC4E,KAAKE,MAAM,GAC1B;QACA,MAAM,IAAItE,MAAM;IAClB;IACA,MAAM2B,UAAUyC,KAAKE,MAAM,CACxBC,GAAG,CAACvC,YACJwC,MAAM,CAAC,CAAC1C,SAAqCA,WAAW;IAC3D,OAAO;QACL2C,SAAS;QACTC,UAAU;QACVC,aAAaR;QACbS,eAAe;QACfC,WAAWnD,aAAaC;QACxBA;QACAmD,kBAAkB;IACpB;AACF;AAEA,OAAO,eAAeC,iBACpB/B,OAAwB;IAExB,MAAMgB,MAAM,MAAMjB,iBAAiBC;IACnC,MAAMgC,aAAa,IAAIC;IACvB,MAAMC,QAAQC,WAAW,IAAMH,WAAWI,KAAK,IAAIpC,QAAQqC,SAAS,IAAI;IACxE,IAAIC;IACJ,IAAI;QACFA,WAAW,MAAM,AAACtC,CAAAA,QAAQuC,KAAK,IAAIC,WAAWD,KAAK,AAAD,EAAG7G,eAAe;YAClE+G,QAAQ;YACRC,SAAS;gBAAEC,eAAe,CAAC,OAAO,EAAE3B,KAAK;YAAC;YAC1C4B,QAAQZ,WAAWY,MAAM;QAC3B;IACF,EAAE,OAAM;QACN,MAAM,IAAI5F,MAAM;IAClB,SAAU;QACR6F,aAAaX;IACf;IACA,IAAII,SAASQ,MAAM,KAAK,KAAK;QAC3B,MAAM,IAAI9F,MAAM,CAAC,gCAAgC,EAAEsF,SAASQ,MAAM,EAAE;IACtE;IACA,IAAI5B;IACJ,IAAI;QACFA,MAAM,MAAMoB,SAASS,IAAI;IAC3B,EAAE,OAAM;QACN,MAAM,IAAI/F,MAAM;IAClB;IACA,OAAOiE,cAAcC,KAAKlB,QAAQmB,SAAS;AAC7C"}

package/dist/index.d.ts

@@ -1,14 +1,14 @@
 export { AgentManager, AgentNotRunningError } from "./AgentManager.js";
-export { getCodexCapacityReport } from "./capacity/index.js";
-export type { CapacityProbeOptions, CapacityReport, CapacityWindow } from "./capacity/index.js";
+export { getCodexCapacityReport, getZaiCapacityReport } from "./capacity/index.js";
+export type { CapacityProbeOptions, CapacityReport, CapacityWindow, ZaiCapacityOptions, } from "./capacity/index.js";
 export { getAgentReadinessReport, getAgentReadinessReports, worstReadinessStatus, } from "./readiness/AgentReadiness.js";
 export type { AgentReadinessOptions, AgentReadinessReport, AuthReadinessCheck, BuiltInSkillsReadinessCheck, DirectoryReadinessCheck, ExecutableReadinessCheck, IntegrationReadinessCheck, ReadinessAgentType, ReadinessAuthState, ReadinessCheck, ReadinessStatus, } from "./readiness/AgentReadiness.js";
 export { ClaudeCodeAdapter } from "./providers/claude/ClaudeCodeAdapter.js";
 export { CodexAdapter } from "./providers/codex/CodexAdapter.js";
 export { CopilotAdapter } from "./providers/copilot/CopilotAdapter.js";
 export { GeminiCliAdapter } from "./providers/gemini/GeminiCliAdapter.js";
-export { GrokCliAdapter } from "./adapters/GrokCliAdapter.js";
-export { OpenCodeAdapter } from "./adapters/OpenCodeAdapter.js";
+export { GrokCliAdapter } from "./providers/grok/GrokCliAdapter.js";
+export { OpenCodeAdapter } from "./providers/opencode/OpenCodeAdapter.js";
 export { PiAdapter } from "./providers/pi/PiAdapter.js";
 export { AgentStatus } from "./adapters/AgentAdapter.js";
 export type { AgentAdapter, AgentType, AgentInfo, ProcessInfo, ConversationMessage, SessionSummary, ListSessionsOptions, AgentDetectionContext, } from "./adapters/AgentAdapter.js";

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AACvE,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,YAAY,EAAE,oBAAoB,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAChG,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,oBAAoB,GACrB,MAAM,+BAA+B,CAAC;AACvC,YAAY,EACV,qBAAqB,EACrB,oBAAoB,EACpB,kBAAkB,EAClB,2BAA2B,EAC3B,uBAAuB,EACvB,wBAAwB,EACxB,yBAAyB,EACzB,kBAAkB,EAClB,kBAAkB,EAClB,cAAc,EACd,eAAe,GAChB,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,uCAAuC,CAAC;AACvE,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAChE,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AACxD,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,YAAY,EACV,YAAY,EACZ,SAAS,EACT,SAAS,EACT,WAAW,EACX,mBAAmB,EACnB,cAAc,EACd,mBAAmB,EACnB,qBAAqB,GACtB,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,oBAAoB,EAAE,YAAY,EAAE,MAAM,oCAAoC,CAAC;AACxF,YAAY,EAAE,gBAAgB,EAAE,MAAM,oCAAoC,CAAC;AAC3E,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAEpD,YAAY,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAChF,OAAO,EACL,sBAAsB,EACtB,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,GACtB,MAAM,wCAAwC,CAAC;AAChD,YAAY,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAE7E,OAAO,EACL,aAAa,EACb,mBAAmB,EACnB,mBAAmB,EACnB,uBAAuB,EACvB,mBAAmB,GACpB,MAAM,0BAA0B,CAAC;AAClC,YAAY,EACV,oBAAoB,EACpB,aAAa,EACb,oBAAoB,EACpB,cAAc,GACf,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,YAAY,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC5F,OAAO,EACL,UAAU,EACV,SAAS,EACT,UAAU,EACV,eAAe,EACf,oBAAoB,EACpB,mBAAmB,EACnB,wBAAwB,EACxB,4BAA4B,EAC5B,0BAA0B,EAC1B,4BAA4B,EAC5B,2BAA2B,GAC5B,MAAM,kCAAkC,CAAC;AAC1C,YAAY,EACV,cAAc,EACd,cAAc,EACd,iBAAiB,EACjB,aAAa,EACb,eAAe,EACf,gBAAgB,EAChB,mBAAmB,GACpB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,YAAY,EAAE,WAAW,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAEzE,YAAY,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,2BAA2B,CAAC;AAEnC,OAAO,EACL,WAAW,EACX,iBAAiB,EACjB,qBAAqB,EACrB,yBAAyB,EACzB,2BAA2B,EAC3B,6BAA6B,EAC7B,gBAAgB,EAChB,eAAe,EACf,YAAY,GACb,MAAM,2BAA2B,CAAC;AACnC,YAAY,EACV,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,EAClB,iBAAiB,EACjB,cAAc,EACd,eAAe,EACf,mBAAmB,EACnB,iBAAiB,EACjB,oBAAoB,EACpB,gBAAgB,EAChB,gBAAgB,EAChB,iBAAiB,EACjB,gBAAgB,EAChB,eAAe,GAChB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,sBAAsB,EAAE,MAAM,qCAAqC,CAAC;AAC7E,YAAY,EACV,uBAAuB,EACvB,6BAA6B,EAC7B,gBAAgB,EAChB,oBAAoB,GACrB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,uBAAuB,EAAE,MAAM,uDAAuD,CAAC;AAChG,YAAY,EACV,8BAA8B,EAC9B,qBAAqB,GACtB,MAAM,uDAAuD,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,4CAA4C,CAAC;AAC3E,YAAY,EAAE,oBAAoB,EAAE,MAAM,4CAA4C,CAAC;AACvF,OAAO,EAAE,gBAAgB,EAAE,MAAM,+CAA+C,CAAC;AACjF,YAAY,EACV,uBAAuB,EACvB,oBAAoB,EACpB,mBAAmB,GACpB,MAAM,+CAA+C,CAAC;AACvD,OAAO,EAAE,sBAAsB,EAAE,MAAM,qDAAqD,CAAC;AAC7F,YAAY,EACV,6BAA6B,EAC7B,oBAAoB,GACrB,MAAM,qDAAqD,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,sCAAsC,CAAC;AAClE,YAAY,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAC9E,OAAO,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAC;AACxE,YAAY,EACV,oBAAoB,EACpB,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,mBAAmB,EAAE,MAAM,+CAA+C,CAAC;AACpF,YAAY,EACV,0BAA0B,EAC1B,iBAAiB,GAClB,MAAM,+CAA+C,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AACvE,OAAO,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AACnF,YAAY,EACV,oBAAoB,EACpB,cAAc,EACd,cAAc,EACd,kBAAkB,GACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,oBAAoB,GACrB,MAAM,+BAA+B,CAAC;AACvC,YAAY,EACV,qBAAqB,EACrB,oBAAoB,EACpB,kBAAkB,EAClB,2BAA2B,EAC3B,uBAAuB,EACvB,wBAAwB,EACxB,yBAAyB,EACzB,kBAAkB,EAClB,kBAAkB,EAClB,cAAc,EACd,eAAe,GAChB,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,uCAAuC,CAAC;AACvE,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AACxD,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,YAAY,EACV,YAAY,EACZ,SAAS,EACT,SAAS,EACT,WAAW,EACX,mBAAmB,EACnB,cAAc,EACd,mBAAmB,EACnB,qBAAqB,GACtB,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,oBAAoB,EAAE,YAAY,EAAE,MAAM,oCAAoC,CAAC;AACxF,YAAY,EAAE,gBAAgB,EAAE,MAAM,oCAAoC,CAAC;AAC3E,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAEpD,YAAY,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAChF,OAAO,EACL,sBAAsB,EACtB,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,GACtB,MAAM,wCAAwC,CAAC;AAChD,YAAY,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAE7E,OAAO,EACL,aAAa,EACb,mBAAmB,EACnB,mBAAmB,EACnB,uBAAuB,EACvB,mBAAmB,GACpB,MAAM,0BAA0B,CAAC;AAClC,YAAY,EACV,oBAAoB,EACpB,aAAa,EACb,oBAAoB,EACpB,cAAc,GACf,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,YAAY,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC5F,OAAO,EACL,UAAU,EACV,SAAS,EACT,UAAU,EACV,eAAe,EACf,oBAAoB,EACpB,mBAAmB,EACnB,wBAAwB,EACxB,4BAA4B,EAC5B,0BAA0B,EAC1B,4BAA4B,EAC5B,2BAA2B,GAC5B,MAAM,kCAAkC,CAAC;AAC1C,YAAY,EACV,cAAc,EACd,cAAc,EACd,iBAAiB,EACjB,aAAa,EACb,eAAe,EACf,gBAAgB,EAChB,mBAAmB,GACpB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,YAAY,EAAE,WAAW,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAEzE,YAAY,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,2BAA2B,CAAC;AAEnC,OAAO,EACL,WAAW,EACX,iBAAiB,EACjB,qBAAqB,EACrB,yBAAyB,EACzB,2BAA2B,EAC3B,6BAA6B,EAC7B,gBAAgB,EAChB,eAAe,EACf,YAAY,GACb,MAAM,2BAA2B,CAAC;AACnC,YAAY,EACV,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,EAClB,iBAAiB,EACjB,cAAc,EACd,eAAe,EACf,mBAAmB,EACnB,iBAAiB,EACjB,oBAAoB,EACpB,gBAAgB,EAChB,gBAAgB,EAChB,iBAAiB,EACjB,gBAAgB,EAChB,eAAe,GAChB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,sBAAsB,EAAE,MAAM,qCAAqC,CAAC;AAC7E,YAAY,EACV,uBAAuB,EACvB,6BAA6B,EAC7B,gBAAgB,EAChB,oBAAoB,GACrB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,uBAAuB,EAAE,MAAM,uDAAuD,CAAC;AAChG,YAAY,EACV,8BAA8B,EAC9B,qBAAqB,GACtB,MAAM,uDAAuD,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,4CAA4C,CAAC;AAC3E,YAAY,EAAE,oBAAoB,EAAE,MAAM,4CAA4C,CAAC;AACvF,OAAO,EAAE,gBAAgB,EAAE,MAAM,+CAA+C,CAAC;AACjF,YAAY,EACV,uBAAuB,EACvB,oBAAoB,EACpB,mBAAmB,GACpB,MAAM,+CAA+C,CAAC;AACvD,OAAO,EAAE,sBAAsB,EAAE,MAAM,qDAAqD,CAAC;AAC7F,YAAY,EACV,6BAA6B,EAC7B,oBAAoB,GACrB,MAAM,qDAAqD,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,sCAAsC,CAAC;AAClE,YAAY,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAC9E,OAAO,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAC;AACxE,YAAY,EACV,oBAAoB,EACpB,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,mBAAmB,EAAE,MAAM,+CAA+C,CAAC;AACpF,YAAY,EACV,0BAA0B,EAC1B,iBAAiB,GAClB,MAAM,+CAA+C,CAAC"}

package/dist/index.js

@@ -1,12 +1,12 @@
 export { AgentManager, AgentNotRunningError } from "./AgentManager.js";
-export { getCodexCapacityReport } from "./capacity/index.js";
+export { getCodexCapacityReport, getZaiCapacityReport } from "./capacity/index.js";
 export { getAgentReadinessReport, getAgentReadinessReports, worstReadinessStatus } from "./readiness/AgentReadiness.js";
 export { ClaudeCodeAdapter } from "./providers/claude/ClaudeCodeAdapter.js";
 export { CodexAdapter } from "./providers/codex/CodexAdapter.js";
 export { CopilotAdapter } from "./providers/copilot/CopilotAdapter.js";
 export { GeminiCliAdapter } from "./providers/gemini/GeminiCliAdapter.js";
-export { GrokCliAdapter } from "./adapters/GrokCliAdapter.js";
-export { OpenCodeAdapter } from "./adapters/OpenCodeAdapter.js";
+export { GrokCliAdapter } from "./providers/grok/GrokCliAdapter.js";
+export { OpenCodeAdapter } from "./providers/opencode/OpenCodeAdapter.js";
 export { PiAdapter } from "./providers/pi/PiAdapter.js";
 export { AgentStatus } from "./adapters/AgentAdapter.js";
 export { TerminalFocusManager, TerminalType } from "./terminal/TerminalFocusManager.js";

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts"],"sourcesContent":["export { AgentManager, AgentNotRunningError } from \"./AgentManager.js\";\nexport { getCodexCapacityReport } from \"./capacity/index.js\";\nexport type { CapacityProbeOptions, CapacityReport, CapacityWindow } from \"./capacity/index.js\";\nexport {\n  getAgentReadinessReport,\n  getAgentReadinessReports,\n  worstReadinessStatus,\n} from \"./readiness/AgentReadiness.js\";\nexport type {\n  AgentReadinessOptions,\n  AgentReadinessReport,\n  AuthReadinessCheck,\n  BuiltInSkillsReadinessCheck,\n  DirectoryReadinessCheck,\n  ExecutableReadinessCheck,\n  IntegrationReadinessCheck,\n  ReadinessAgentType,\n  ReadinessAuthState,\n  ReadinessCheck,\n  ReadinessStatus,\n} from \"./readiness/AgentReadiness.js\";\n\nexport { ClaudeCodeAdapter } from \"./providers/claude/ClaudeCodeAdapter.js\";\nexport { CodexAdapter } from \"./providers/codex/CodexAdapter.js\";\nexport { CopilotAdapter } from \"./providers/copilot/CopilotAdapter.js\";\nexport { GeminiCliAdapter } from \"./providers/gemini/GeminiCliAdapter.js\";\nexport { GrokCliAdapter } from \"./adapters/GrokCliAdapter.js\";\nexport { OpenCodeAdapter } from \"./adapters/OpenCodeAdapter.js\";\nexport { PiAdapter } from \"./providers/pi/PiAdapter.js\";\nexport { AgentStatus } from \"./adapters/AgentAdapter.js\";\nexport type {\n  AgentAdapter,\n  AgentType,\n  AgentInfo,\n  ProcessInfo,\n  ConversationMessage,\n  SessionSummary,\n  ListSessionsOptions,\n  AgentDetectionContext,\n} from \"./adapters/AgentAdapter.js\";\n\nexport { TerminalFocusManager, TerminalType } from \"./terminal/TerminalFocusManager.js\";\nexport type { TerminalLocation } from \"./terminal/TerminalFocusManager.js\";\nexport { TtyWriter } from \"./terminal/TtyWriter.js\";\n\nexport type { AgentManagerOptions, ListAgentsOptions } from \"./AgentManager.js\";\nexport {\n  extractHerdrAgentPanes,\n  fetchHerdrAgentPanes,\n  findMatchingHerdrPane,\n  herdrPaneToRuntimeRef,\n} from \"./runtime/herdr/HerdrAgentDiscovery.js\";\nexport type { HerdrAgentPane } from \"./runtime/herdr/HerdrAgentDiscovery.js\";\n\nexport {\n  AgentRegistry,\n  RenameNotFoundError,\n  RenameConflictError,\n  AGENT_RUNTIME_PROVIDERS,\n  parseTmuxRuntimeRef,\n} from \"./utils/AgentRegistry.js\";\nexport type {\n  AgentRegistryOptions,\n  RegistryEntry,\n  AgentRuntimeProvider,\n  TmuxRuntimeRef,\n} from \"./utils/AgentRegistry.js\";\nexport { TmuxManager } from \"./terminal/TmuxManager.js\";\nexport type { AgentRuntimeAvailability, InteractiveAgentRuntime } from \"./runtime/types.js\";\nexport {\n  startAgent,\n  stopAgent,\n  focusAgent,\n  sendAgentPrompt,\n  TmuxUnavailableError,\n  AgentNameInUseError,\n  AgentPidPollTimeoutError,\n  AgentRuntimeUnavailableError,\n  AgentTerminalNotFoundError,\n  DEFAULT_PID_POLL_INTERVAL_MS,\n  DEFAULT_PID_POLL_TIMEOUT_MS,\n} from \"./runtime/ManagedAgentRuntime.js\";\nexport type {\n  FocusAgentDeps,\n  StartAgentDeps,\n  StartAgentOptions,\n  StopAgentDeps,\n  StopAgentResult,\n  FocusAgentResult,\n  SendAgentPromptDeps,\n} from \"./runtime/ManagedAgentRuntime.js\";\nexport { AGENTS } from \"./utils/agents.js\";\nexport type { AgentConfig, StartableAgentType } from \"./utils/agents.js\";\n\nexport type { AgentRequest } from \"./utils/agent-requests.js\";\nexport {\n  getAgentRequestPath,\n  readLatestAgentRequest,\n  writeAgentRequest,\n} from \"./utils/agent-requests.js\";\n\nexport {\n  AGENT_MODES,\n  DurableAgentError,\n  DurableAgentBusyError,\n  DurableAgentNotFoundError,\n  DurableAgentRepositoryError,\n  DurableAgentNameConflictError,\n  ClaudePrintError,\n  CodexPrintError,\n  PiPrintError,\n} from \"./durable/DurableAgent.js\";\nexport type {\n  DurableAgent,\n  DurableAgentBase,\n  ClaudeDurableAgent,\n  CodexDurableAgent,\n  PiDurableAgent,\n  DurableProvider,\n  CodexPrintErrorCode,\n  DurableAgentState,\n  DurableSessionHealth,\n  DurableRunStatus,\n  DurableActiveRun,\n  DurableLastResult,\n  PiPrintErrorCode,\n  ProcessIdentity,\n} from \"./durable/DurableAgent.js\";\nexport { DurableAgentRepository } from \"./durable/DurableAgentRepository.js\";\nexport type {\n  CreateDurableAgentInput,\n  DurableAgentRepositoryOptions,\n  ProcessInspector,\n  DurableRunCompletion,\n} from \"./durable/DurableAgentRepository.js\";\nexport { ClaudePrintAgentService } from \"./providers/claude/durable/ClaudePrintAgentService.js\";\nexport type {\n  ClaudePrintAgentServiceOptions,\n  ClaudePrintSendResult,\n} from \"./providers/claude/durable/ClaudePrintAgentService.js\";\nexport { CodexCliProbe } from \"./providers/codex/durable/CodexCliProbe.js\";\nexport type { CodexCliProbeOptions } from \"./providers/codex/durable/CodexCliProbe.js\";\nexport { CodexPrintRunner } from \"./providers/codex/durable/CodexPrintRunner.js\";\nexport type {\n  CodexPrintRunnerOptions,\n  CodexPrintRunRequest,\n  CodexPrintRunResult,\n} from \"./providers/codex/durable/CodexPrintRunner.js\";\nexport { CodexPrintAgentService } from \"./providers/codex/durable/CodexPrintAgentService.js\";\nexport type {\n  CodexPrintAgentServiceOptions,\n  CodexPrintSendResult,\n} from \"./providers/codex/durable/CodexPrintAgentService.js\";\nexport { PiCliProbe } from \"./providers/pi/durable/PiCliProbe.js\";\nexport type { PiCliProbeOptions } from \"./providers/pi/durable/PiCliProbe.js\";\nexport { PiPrintRunner } from \"./providers/pi/durable/PiPrintRunner.js\";\nexport type {\n  PiPrintRunnerOptions,\n  PiPrintRunRequest,\n  PiPrintRunResult,\n} from \"./providers/pi/durable/PiPrintRunner.js\";\nexport { PiPrintAgentService } from \"./providers/pi/durable/PiPrintAgentService.js\";\nexport type {\n  PiPrintAgentServiceOptions,\n  PiPrintSendResult,\n} from \"./providers/pi/durable/PiPrintAgentService.js\";\n"],"names":["AgentManager","AgentNotRunningError","getCodexCapacityReport","getAgentReadinessReport","getAgentReadinessReports","worstReadinessStatus","ClaudeCodeAdapter","CodexAdapter","CopilotAdapter","GeminiCliAdapter","GrokCliAdapter","OpenCodeAdapter","PiAdapter","AgentStatus","TerminalFocusManager","TerminalType","TtyWriter","extractHerdrAgentPanes","fetchHerdrAgentPanes","findMatchingHerdrPane","herdrPaneToRuntimeRef","AgentRegistry","RenameNotFoundError","RenameConflictError","AGENT_RUNTIME_PROVIDERS","parseTmuxRuntimeRef","TmuxManager","startAgent","stopAgent","focusAgent","sendAgentPrompt","TmuxUnavailableError","AgentNameInUseError","AgentPidPollTimeoutError","AgentRuntimeUnavailableError","AgentTerminalNotFoundError","DEFAULT_PID_POLL_INTERVAL_MS","DEFAULT_PID_POLL_TIMEOUT_MS","AGENTS","getAgentRequestPath","readLatestAgentRequest","writeAgentRequest","AGENT_MODES","DurableAgentError","DurableAgentBusyError","DurableAgentNotFoundError","DurableAgentRepositoryError","DurableAgentNameConflictError","ClaudePrintError","CodexPrintError","PiPrintError","DurableAgentRepository","ClaudePrintAgentService","CodexCliProbe","CodexPrintRunner","CodexPrintAgentService","PiCliProbe","PiPrintRunner","PiPrintAgentService"],"mappings":"AAAA,SAASA,YAAY,EAAEC,oBAAoB,QAAQ,oBAAoB;AACvE,SAASC,sBAAsB,QAAQ,sBAAsB;AAE7D,SACEC,uBAAuB,EACvBC,wBAAwB,EACxBC,oBAAoB,QACf,gCAAgC;AAevC,SAASC,iBAAiB,QAAQ,0CAA0C;AAC5E,SAASC,YAAY,QAAQ,oCAAoC;AACjE,SAASC,cAAc,QAAQ,wCAAwC;AACvE,SAASC,gBAAgB,QAAQ,yCAAyC;AAC1E,SAASC,cAAc,QAAQ,+BAA+B;AAC9D,SAASC,eAAe,QAAQ,gCAAgC;AAChE,SAASC,SAAS,QAAQ,8BAA8B;AACxD,SAASC,WAAW,QAAQ,6BAA6B;AAYzD,SAASC,oBAAoB,EAAEC,YAAY,QAAQ,qCAAqC;AAExF,SAASC,SAAS,QAAQ,0BAA0B;AAGpD,SACEC,sBAAsB,EACtBC,oBAAoB,EACpBC,qBAAqB,EACrBC,qBAAqB,QAChB,yCAAyC;AAGhD,SACEC,aAAa,EACbC,mBAAmB,EACnBC,mBAAmB,EACnBC,uBAAuB,EACvBC,mBAAmB,QACd,2BAA2B;AAOlC,SAASC,WAAW,QAAQ,4BAA4B;AAExD,SACEC,UAAU,EACVC,SAAS,EACTC,UAAU,EACVC,eAAe,EACfC,oBAAoB,EACpBC,mBAAmB,EACnBC,wBAAwB,EACxBC,4BAA4B,EAC5BC,0BAA0B,EAC1BC,4BAA4B,EAC5BC,2BAA2B,QACtB,mCAAmC;AAU1C,SAASC,MAAM,QAAQ,oBAAoB;AAI3C,SACEC,mBAAmB,EACnBC,sBAAsB,EACtBC,iBAAiB,QACZ,4BAA4B;AAEnC,SACEC,WAAW,EACXC,iBAAiB,EACjBC,qBAAqB,EACrBC,yBAAyB,EACzBC,2BAA2B,EAC3BC,6BAA6B,EAC7BC,gBAAgB,EAChBC,eAAe,EACfC,YAAY,QACP,4BAA4B;AAiBnC,SAASC,sBAAsB,QAAQ,sCAAsC;AAO7E,SAASC,uBAAuB,QAAQ,wDAAwD;AAKhG,SAASC,aAAa,QAAQ,6CAA6C;AAE3E,SAASC,gBAAgB,QAAQ,gDAAgD;AAMjF,SAASC,sBAAsB,QAAQ,sDAAsD;AAK7F,SAASC,UAAU,QAAQ,uCAAuC;AAElE,SAASC,aAAa,QAAQ,0CAA0C;AAMxE,SAASC,mBAAmB,QAAQ,gDAAgD"}
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["export { AgentManager, AgentNotRunningError } from \"./AgentManager.js\";\nexport { getCodexCapacityReport, getZaiCapacityReport } from \"./capacity/index.js\";\nexport type {\n  CapacityProbeOptions,\n  CapacityReport,\n  CapacityWindow,\n  ZaiCapacityOptions,\n} from \"./capacity/index.js\";\nexport {\n  getAgentReadinessReport,\n  getAgentReadinessReports,\n  worstReadinessStatus,\n} from \"./readiness/AgentReadiness.js\";\nexport type {\n  AgentReadinessOptions,\n  AgentReadinessReport,\n  AuthReadinessCheck,\n  BuiltInSkillsReadinessCheck,\n  DirectoryReadinessCheck,\n  ExecutableReadinessCheck,\n  IntegrationReadinessCheck,\n  ReadinessAgentType,\n  ReadinessAuthState,\n  ReadinessCheck,\n  ReadinessStatus,\n} from \"./readiness/AgentReadiness.js\";\n\nexport { ClaudeCodeAdapter } from \"./providers/claude/ClaudeCodeAdapter.js\";\nexport { CodexAdapter } from \"./providers/codex/CodexAdapter.js\";\nexport { CopilotAdapter } from \"./providers/copilot/CopilotAdapter.js\";\nexport { GeminiCliAdapter } from \"./providers/gemini/GeminiCliAdapter.js\";\nexport { GrokCliAdapter } from \"./providers/grok/GrokCliAdapter.js\";\nexport { OpenCodeAdapter } from \"./providers/opencode/OpenCodeAdapter.js\";\nexport { PiAdapter } from \"./providers/pi/PiAdapter.js\";\nexport { AgentStatus } from \"./adapters/AgentAdapter.js\";\nexport type {\n  AgentAdapter,\n  AgentType,\n  AgentInfo,\n  ProcessInfo,\n  ConversationMessage,\n  SessionSummary,\n  ListSessionsOptions,\n  AgentDetectionContext,\n} from \"./adapters/AgentAdapter.js\";\n\nexport { TerminalFocusManager, TerminalType } from \"./terminal/TerminalFocusManager.js\";\nexport type { TerminalLocation } from \"./terminal/TerminalFocusManager.js\";\nexport { TtyWriter } from \"./terminal/TtyWriter.js\";\n\nexport type { AgentManagerOptions, ListAgentsOptions } from \"./AgentManager.js\";\nexport {\n  extractHerdrAgentPanes,\n  fetchHerdrAgentPanes,\n  findMatchingHerdrPane,\n  herdrPaneToRuntimeRef,\n} from \"./runtime/herdr/HerdrAgentDiscovery.js\";\nexport type { HerdrAgentPane } from \"./runtime/herdr/HerdrAgentDiscovery.js\";\n\nexport {\n  AgentRegistry,\n  RenameNotFoundError,\n  RenameConflictError,\n  AGENT_RUNTIME_PROVIDERS,\n  parseTmuxRuntimeRef,\n} from \"./utils/AgentRegistry.js\";\nexport type {\n  AgentRegistryOptions,\n  RegistryEntry,\n  AgentRuntimeProvider,\n  TmuxRuntimeRef,\n} from \"./utils/AgentRegistry.js\";\nexport { TmuxManager } from \"./terminal/TmuxManager.js\";\nexport type { AgentRuntimeAvailability, InteractiveAgentRuntime } from \"./runtime/types.js\";\nexport {\n  startAgent,\n  stopAgent,\n  focusAgent,\n  sendAgentPrompt,\n  TmuxUnavailableError,\n  AgentNameInUseError,\n  AgentPidPollTimeoutError,\n  AgentRuntimeUnavailableError,\n  AgentTerminalNotFoundError,\n  DEFAULT_PID_POLL_INTERVAL_MS,\n  DEFAULT_PID_POLL_TIMEOUT_MS,\n} from \"./runtime/ManagedAgentRuntime.js\";\nexport type {\n  FocusAgentDeps,\n  StartAgentDeps,\n  StartAgentOptions,\n  StopAgentDeps,\n  StopAgentResult,\n  FocusAgentResult,\n  SendAgentPromptDeps,\n} from \"./runtime/ManagedAgentRuntime.js\";\nexport { AGENTS } from \"./utils/agents.js\";\nexport type { AgentConfig, StartableAgentType } from \"./utils/agents.js\";\n\nexport type { AgentRequest } from \"./utils/agent-requests.js\";\nexport {\n  getAgentRequestPath,\n  readLatestAgentRequest,\n  writeAgentRequest,\n} from \"./utils/agent-requests.js\";\n\nexport {\n  AGENT_MODES,\n  DurableAgentError,\n  DurableAgentBusyError,\n  DurableAgentNotFoundError,\n  DurableAgentRepositoryError,\n  DurableAgentNameConflictError,\n  ClaudePrintError,\n  CodexPrintError,\n  PiPrintError,\n} from \"./durable/DurableAgent.js\";\nexport type {\n  DurableAgent,\n  DurableAgentBase,\n  ClaudeDurableAgent,\n  CodexDurableAgent,\n  PiDurableAgent,\n  DurableProvider,\n  CodexPrintErrorCode,\n  DurableAgentState,\n  DurableSessionHealth,\n  DurableRunStatus,\n  DurableActiveRun,\n  DurableLastResult,\n  PiPrintErrorCode,\n  ProcessIdentity,\n} from \"./durable/DurableAgent.js\";\nexport { DurableAgentRepository } from \"./durable/DurableAgentRepository.js\";\nexport type {\n  CreateDurableAgentInput,\n  DurableAgentRepositoryOptions,\n  ProcessInspector,\n  DurableRunCompletion,\n} from \"./durable/DurableAgentRepository.js\";\nexport { ClaudePrintAgentService } from \"./providers/claude/durable/ClaudePrintAgentService.js\";\nexport type {\n  ClaudePrintAgentServiceOptions,\n  ClaudePrintSendResult,\n} from \"./providers/claude/durable/ClaudePrintAgentService.js\";\nexport { CodexCliProbe } from \"./providers/codex/durable/CodexCliProbe.js\";\nexport type { CodexCliProbeOptions } from \"./providers/codex/durable/CodexCliProbe.js\";\nexport { CodexPrintRunner } from \"./providers/codex/durable/CodexPrintRunner.js\";\nexport type {\n  CodexPrintRunnerOptions,\n  CodexPrintRunRequest,\n  CodexPrintRunResult,\n} from \"./providers/codex/durable/CodexPrintRunner.js\";\nexport { CodexPrintAgentService } from \"./providers/codex/durable/CodexPrintAgentService.js\";\nexport type {\n  CodexPrintAgentServiceOptions,\n  CodexPrintSendResult,\n} from \"./providers/codex/durable/CodexPrintAgentService.js\";\nexport { PiCliProbe } from \"./providers/pi/durable/PiCliProbe.js\";\nexport type { PiCliProbeOptions } from \"./providers/pi/durable/PiCliProbe.js\";\nexport { PiPrintRunner } from \"./providers/pi/durable/PiPrintRunner.js\";\nexport type {\n  PiPrintRunnerOptions,\n  PiPrintRunRequest,\n  PiPrintRunResult,\n} from \"./providers/pi/durable/PiPrintRunner.js\";\nexport { PiPrintAgentService } from \"./providers/pi/durable/PiPrintAgentService.js\";\nexport type {\n  PiPrintAgentServiceOptions,\n  PiPrintSendResult,\n} from \"./providers/pi/durable/PiPrintAgentService.js\";\n"],"names":["AgentManager","AgentNotRunningError","getCodexCapacityReport","getZaiCapacityReport","getAgentReadinessReport","getAgentReadinessReports","worstReadinessStatus","ClaudeCodeAdapter","CodexAdapter","CopilotAdapter","GeminiCliAdapter","GrokCliAdapter","OpenCodeAdapter","PiAdapter","AgentStatus","TerminalFocusManager","TerminalType","TtyWriter","extractHerdrAgentPanes","fetchHerdrAgentPanes","findMatchingHerdrPane","herdrPaneToRuntimeRef","AgentRegistry","RenameNotFoundError","RenameConflictError","AGENT_RUNTIME_PROVIDERS","parseTmuxRuntimeRef","TmuxManager","startAgent","stopAgent","focusAgent","sendAgentPrompt","TmuxUnavailableError","AgentNameInUseError","AgentPidPollTimeoutError","AgentRuntimeUnavailableError","AgentTerminalNotFoundError","DEFAULT_PID_POLL_INTERVAL_MS","DEFAULT_PID_POLL_TIMEOUT_MS","AGENTS","getAgentRequestPath","readLatestAgentRequest","writeAgentRequest","AGENT_MODES","DurableAgentError","DurableAgentBusyError","DurableAgentNotFoundError","DurableAgentRepositoryError","DurableAgentNameConflictError","ClaudePrintError","CodexPrintError","PiPrintError","DurableAgentRepository","ClaudePrintAgentService","CodexCliProbe","CodexPrintRunner","CodexPrintAgentService","PiCliProbe","PiPrintRunner","PiPrintAgentService"],"mappings":"AAAA,SAASA,YAAY,EAAEC,oBAAoB,QAAQ,oBAAoB;AACvE,SAASC,sBAAsB,EAAEC,oBAAoB,QAAQ,sBAAsB;AAOnF,SACEC,uBAAuB,EACvBC,wBAAwB,EACxBC,oBAAoB,QACf,gCAAgC;AAevC,SAASC,iBAAiB,QAAQ,0CAA0C;AAC5E,SAASC,YAAY,QAAQ,oCAAoC;AACjE,SAASC,cAAc,QAAQ,wCAAwC;AACvE,SAASC,gBAAgB,QAAQ,yCAAyC;AAC1E,SAASC,cAAc,QAAQ,qCAAqC;AACpE,SAASC,eAAe,QAAQ,0CAA0C;AAC1E,SAASC,SAAS,QAAQ,8BAA8B;AACxD,SAASC,WAAW,QAAQ,6BAA6B;AAYzD,SAASC,oBAAoB,EAAEC,YAAY,QAAQ,qCAAqC;AAExF,SAASC,SAAS,QAAQ,0BAA0B;AAGpD,SACEC,sBAAsB,EACtBC,oBAAoB,EACpBC,qBAAqB,EACrBC,qBAAqB,QAChB,yCAAyC;AAGhD,SACEC,aAAa,EACbC,mBAAmB,EACnBC,mBAAmB,EACnBC,uBAAuB,EACvBC,mBAAmB,QACd,2BAA2B;AAOlC,SAASC,WAAW,QAAQ,4BAA4B;AAExD,SACEC,UAAU,EACVC,SAAS,EACTC,UAAU,EACVC,eAAe,EACfC,oBAAoB,EACpBC,mBAAmB,EACnBC,wBAAwB,EACxBC,4BAA4B,EAC5BC,0BAA0B,EAC1BC,4BAA4B,EAC5BC,2BAA2B,QACtB,mCAAmC;AAU1C,SAASC,MAAM,QAAQ,oBAAoB;AAI3C,SACEC,mBAAmB,EACnBC,sBAAsB,EACtBC,iBAAiB,QACZ,4BAA4B;AAEnC,SACEC,WAAW,EACXC,iBAAiB,EACjBC,qBAAqB,EACrBC,yBAAyB,EACzBC,2BAA2B,EAC3BC,6BAA6B,EAC7BC,gBAAgB,EAChBC,eAAe,EACfC,YAAY,QACP,4BAA4B;AAiBnC,SAASC,sBAAsB,QAAQ,sCAAsC;AAO7E,SAASC,uBAAuB,QAAQ,wDAAwD;AAKhG,SAASC,aAAa,QAAQ,6CAA6C;AAE3E,SAASC,gBAAgB,QAAQ,gDAAgD;AAMjF,SAASC,sBAAsB,QAAQ,sDAAsD;AAK7F,SAASC,UAAU,QAAQ,uCAAuC;AAElE,SAASC,aAAa,QAAQ,0CAA0C;AAMxE,SAASC,mBAAmB,QAAQ,gDAAgD"}

package/dist/providers/claude/ClaudeCodeAdapter.d.ts

@@ -25,5 +25,7 @@ export declare class ClaudeCodeAdapter implements AgentAdapter {
         verbose?: boolean;
     }): ConversationMessage[];
     listSessions(opts?: ListSessionsOptions): Promise<SessionSummary[]>;
+    findSessionsById(sessionId: string): Promise<SessionSummary[]>;
+    private toSessionSummary;
 }
 //# sourceMappingURL=ClaudeCodeAdapter.d.ts.map