@ai-devkit/agent-manager 0.37.0 → 0.39.0

package/dist/capacity/{zai.js → providers/zai.js}

@@ -1,8 +1,4 @@
-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",
@@ -50,9 +46,6 @@ const UNIT_METADATA = new Map([
 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)) {
@@ -60,26 +53,23 @@ function optionalNumber(value, 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;
+function durationMinutes(unit, count, type) {
+    if (type === "TIME_LIMIT" && unit === 5 && count === 1) return 30 * 24 * 60;
     const metadata = UNIT_METADATA.get(unit);
-    return metadata === undefined || number <= 0 ? null : metadata.minutes * number;
+    return metadata === undefined || count <= 0 ? null : metadata.minutes * count;
 }
-function windowLabel(type, unit, number) {
-    if (type === "TIME_LIMIT" && unit === 5 && number === 1) return "MCP · monthly";
+function windowLabel(type, unit, count) {
+    if (type === "TIME_LIMIT" && unit === 5 && count === 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"}`;
+    return `${kind} · ${count} ${unitLabel}${count === 1 ? "" : "s"}`;
 }
 function resetTime(value) {
     const milliseconds = optionalNumber(value, "limit nextResetTime");
@@ -90,10 +80,7 @@ function resetTime(value) {
 }
 function availability(windows) {
     if (windows.length === 0) return "unknown";
-    if (windows.some((window)=>window.usedPercent !== null && window.usedPercent < 100)) {
-        return "yes";
-    }
-    return "no";
+    return windows.some((window)=>window.usedPercent !== null && window.usedPercent < 100) ? "yes" : "no";
 }
 function parseLimit(value, index) {
     const limit = record(value);
@@ -121,39 +108,14 @@ function parseLimit(value, index) {
         label: windowLabel(limit.type, unit, count),
         limitType: limit.type,
         durationMinutes: durationMinutes(unit, count, limit.type),
-        usedPercent: clampPercent(usedPercent),
+        usedPercent: Math.max(0, Math.min(100, 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) {
+export function parseZaiCapacity(raw) {
     const root = record(raw);
     const data = record(root?.data);
     if (root?.success !== true || root.code !== 200 || !data || !Array.isArray(data.limits)) {
@@ -161,19 +123,13 @@ export function parseZaiQuota(raw, checkedAt) {
     }
     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);
+export async function fetchZaiCapacity(key, options = {}) {
     let response;
     try {
         response = await (options.fetch ?? globalThis.fetch)(ZAI_QUOTA_URL, {
@@ -181,23 +137,22 @@ export async function probeZaiCapacity(options) {
             headers: {
                 Authorization: `Bearer ${key}`
             },
-            signal: controller.signal
+            signal: AbortSignal.timeout(options.timeoutMs ?? 5000)
         });
     } 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 parseZaiCapacity(await response.json());
+    } catch (error) {
+        if (error instanceof SyntaxError) {
+            throw new Error("z.ai quota response is not valid JSON");
+        }
+        throw error;
     }
-    return parseZaiQuota(raw, options.checkedAt);
 }
 
 //# sourceMappingURL=zai.js.map

package/dist/capacity/providers/zai.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/capacity/providers/zai.ts"],"sourcesContent":["import type {\n  Availability,\n  CapacityLimitType,\n  CapacityWindow,\n  ProviderCapacitySnapshot,\n} from \"../types.js\";\n\nconst ZAI_QUOTA_URL = \"https://api.z.ai/api/monitor/usage/quota/limit\";\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 ZaiRequestOptions = {\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 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 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(unit: number, count: number, type: CapacityLimitType): number | null {\n  if (type === \"TIME_LIMIT\" && unit === 5 && count === 1) return 30 * 24 * 60;\n  const metadata = UNIT_METADATA.get(unit);\n  return metadata === undefined || count <= 0 ? null : metadata.minutes * count;\n}\n\nfunction windowLabel(type: CapacityLimitType, unit: number, count: number): string {\n  if (type === \"TIME_LIMIT\" && unit === 5 && count === 1) 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} · ${count} ${unitLabel}${count === 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())) 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  return windows.some((window) => window.usedPercent !== null && window.usedPercent < 100)\n    ? \"yes\"\n    : \"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) 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 = total !== null && remaining !== null ? total - remaining : null;\n  if (current !== null) consumed = Math.max(consumed ?? current, current);\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  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: Math.max(0, Math.min(100, usedPercent)),\n    resetsAt: resetTime(limit.nextResetTime),\n    total,\n    current,\n    remaining,\n  };\n}\n\nexport function parseZaiCapacity(raw: unknown): ProviderCapacitySnapshot {\n  const root = record(raw);\n  const data = record(root?.data);\n  if (root?.success !== true || root.code !== 200 || !data || !Array.isArray(data.limits)) {\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    authenticated: true,\n    available: availability(windows),\n    windows,\n    creditsRemaining: null,\n  };\n}\n\nexport async function fetchZaiCapacity(\n  key: string,\n  options: ZaiRequestOptions = {},\n): Promise<ProviderCapacitySnapshot> {\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: AbortSignal.timeout(options.timeoutMs ?? 5000),\n    });\n  } catch {\n    throw new Error(\"z.ai quota request failed\");\n  }\n  if (response.status !== 200) {\n    throw new Error(`z.ai quota request failed: HTTP ${response.status}`);\n  }\n  try {\n    return parseZaiCapacity(await response.json());\n  } catch (error) {\n    if (error instanceof SyntaxError) {\n      throw new Error(\"z.ai quota response is not valid JSON\");\n    }\n    throw error;\n  }\n}\n"],"names":["ZAI_QUOTA_URL","LIMIT_METADATA","TOKENS_LIMIT","id","label","CREDIT_LIMIT","TIME_LIMIT","UNIT_METADATA","Map","minutes","record","value","Array","isArray","optionalNumber","field","undefined","Number","isFinite","Error","integer","isInteger","isCapacityLimitType","Object","hasOwn","durationMinutes","unit","count","type","metadata","get","windowLabel","kind","unitLabel","resetTime","milliseconds","date","Date","isNaN","getTime","toISOString","availability","windows","length","some","window","usedPercent","parseLimit","index","limit","number","serverPercent","percentage","total","usage","current","currentValue","remaining","consumed","Math","max","limitType","min","resetsAt","nextResetTime","parseZaiCapacity","raw","root","data","success","code","limits","map","filter","authenticated","available","creditsRemaining","fetchZaiCapacity","key","options","response","fetch","globalThis","method","headers","Authorization","signal","AbortSignal","timeout","timeoutMs","status","json","error","SyntaxError"],"mappings":"AAOA,MAAMA,gBAAgB;AACtB,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;AAQD,SAASM,OAAOC,KAAc;IAC5B,OAAOA,UAAU,QAAQ,OAAOA,UAAU,YAAY,CAACC,MAAMC,OAAO,CAACF,SAChEA,QACD;AACN;AAEA,SAASG,eAAeH,KAAc,EAAEI,KAAa;IACnD,IAAIJ,UAAUK,aAAaL,UAAU,MAAM,OAAO;IAClD,IAAI,OAAOA,UAAU,YAAY,CAACM,OAAOC,QAAQ,CAACP,QAAQ;QACxD,MAAM,IAAIQ,MAAM,CAAC,mBAAmB,EAAEJ,OAAO;IAC/C;IACA,OAAOJ;AACT;AAEA,SAASS,QAAQT,KAAc;IAC7B,OAAO,OAAOA,UAAU,YAAYM,OAAOI,SAAS,CAACV,SAASA,QAAQ;AACxE;AAEA,SAASW,oBAAoBX,KAAa;IACxC,OAAOY,OAAOC,MAAM,CAACvB,gBAAgBU;AACvC;AAEA,SAASc,gBAAgBC,IAAY,EAAEC,KAAa,EAAEC,IAAuB;IAC3E,IAAIA,SAAS,gBAAgBF,SAAS,KAAKC,UAAU,GAAG,OAAO,KAAK,KAAK;IACzE,MAAME,WAAWtB,cAAcuB,GAAG,CAACJ;IACnC,OAAOG,aAAab,aAAaW,SAAS,IAAI,OAAOE,SAASpB,OAAO,GAAGkB;AAC1E;AAEA,SAASI,YAAYH,IAAuB,EAAEF,IAAY,EAAEC,KAAa;IACvE,IAAIC,SAAS,gBAAgBF,SAAS,KAAKC,UAAU,GAAG,OAAO;IAC/D,MAAMK,OAAO/B,cAAc,CAAC2B,KAAK,CAACxB,KAAK;IACvC,MAAM6B,YAAY1B,cAAcuB,GAAG,CAACJ,OAAOtB;IAC3C,IAAI,CAAC6B,WAAW,OAAOD;IACvB,OAAO,GAAGA,KAAK,GAAG,EAAEL,MAAM,CAAC,EAAEM,YAAYN,UAAU,IAAI,KAAK,KAAK;AACnE;AAEA,SAASO,UAAUvB,KAAc;IAC/B,MAAMwB,eAAerB,eAAeH,OAAO;IAC3C,IAAIwB,iBAAiB,MAAM,OAAO;IAClC,MAAMC,OAAO,IAAIC,KAAKF;IACtB,IAAIlB,OAAOqB,KAAK,CAACF,KAAKG,OAAO,KAAK,MAAM,IAAIpB,MAAM;IAClD,OAAOiB,KAAKI,WAAW;AACzB;AAEA,SAASC,aAAaC,OAAyB;IAC7C,IAAIA,QAAQC,MAAM,KAAK,GAAG,OAAO;IACjC,OAAOD,QAAQE,IAAI,CAAC,CAACC,SAAWA,OAAOC,WAAW,KAAK,QAAQD,OAAOC,WAAW,GAAG,OAChF,QACA;AACN;AAEA,SAASC,WAAWpC,KAAc,EAAEqC,KAAa;IAC/C,MAAMC,QAAQvC,OAAOC;IACrB,IAAI,CAACsC,OAAO,MAAM,IAAI9B,MAAM;IAC5B,MAAMO,OAAON,QAAQ6B,MAAMvB,IAAI;IAC/B,MAAMC,QAAQP,QAAQ6B,MAAMC,MAAM;IAClC,IAAI,OAAOD,MAAMrB,IAAI,KAAK,YAAYF,SAAS,QAAQC,UAAU,MAAM;QACrE,MAAM,IAAIR,MAAM;IAClB;IACA,MAAMgC,gBAAgBrC,eAAemC,MAAMG,UAAU,EAAE;IACvD,IAAID,kBAAkB,MAAM,MAAM,IAAIhC,MAAM;IAC5C,IAAI,CAACG,oBAAoB2B,MAAMrB,IAAI,GAAG,OAAO;IAE7C,MAAMyB,QAAQvC,eAAemC,MAAMK,KAAK,EAAE;IAC1C,MAAMC,UAAUzC,eAAemC,MAAMO,YAAY,EAAE;IACnD,MAAMC,YAAY3C,eAAemC,MAAMQ,SAAS,EAAE;IAClD,IAAIC,WAAWL,UAAU,QAAQI,cAAc,OAAOJ,QAAQI,YAAY;IAC1E,IAAIF,YAAY,MAAMG,WAAWC,KAAKC,GAAG,CAACF,YAAYH,SAASA;IAC/D,IAAIT,cAAcK;IAClB,IAAIE,UAAU,QAAQA,QAAQ,KAAKK,aAAa,MAAM;QACpDZ,cAAc,AAACY,WAAWL,QAAS;IACrC;IACA,MAAMxB,WAAW5B,cAAc,CAACgD,MAAMrB,IAAI,CAAC;IAC3C,OAAO;QACLzB,IAAI,CAAC,IAAI,EAAE0B,SAAS1B,EAAE,CAAC,CAAC,EAAE6C,QAAQ,GAAG;QACrC5C,OAAO2B,YAAYkB,MAAMrB,IAAI,EAAEF,MAAMC;QACrCkC,WAAWZ,MAAMrB,IAAI;QACrBH,iBAAiBA,gBAAgBC,MAAMC,OAAOsB,MAAMrB,IAAI;QACxDkB,aAAaa,KAAKC,GAAG,CAAC,GAAGD,KAAKG,GAAG,CAAC,KAAKhB;QACvCiB,UAAU7B,UAAUe,MAAMe,aAAa;QACvCX;QACAE;QACAE;IACF;AACF;AAEA,OAAO,SAASQ,iBAAiBC,GAAY;IAC3C,MAAMC,OAAOzD,OAAOwD;IACpB,MAAME,OAAO1D,OAAOyD,MAAMC;IAC1B,IAAID,MAAME,YAAY,QAAQF,KAAKG,IAAI,KAAK,OAAO,CAACF,QAAQ,CAACxD,MAAMC,OAAO,CAACuD,KAAKG,MAAM,GAAG;QACvF,MAAM,IAAIpD,MAAM;IAClB;IACA,MAAMuB,UAAU0B,KAAKG,MAAM,CACxBC,GAAG,CAACzB,YACJ0B,MAAM,CAAC,CAAC5B,SAAqCA,WAAW;IAC3D,OAAO;QACL6B,eAAe;QACfC,WAAWlC,aAAaC;QACxBA;QACAkC,kBAAkB;IACpB;AACF;AAEA,OAAO,eAAeC,iBACpBC,GAAW,EACXC,UAA6B,CAAC,CAAC;IAE/B,IAAIC;IACJ,IAAI;QACFA,WAAW,MAAM,AAACD,CAAAA,QAAQE,KAAK,IAAIC,WAAWD,KAAK,AAAD,EAAGjF,eAAe;YAClEmF,QAAQ;YACRC,SAAS;gBAAEC,eAAe,CAAC,OAAO,EAAEP,KAAK;YAAC;YAC1CQ,QAAQC,YAAYC,OAAO,CAACT,QAAQU,SAAS,IAAI;QACnD;IACF,EAAE,OAAM;QACN,MAAM,IAAItE,MAAM;IAClB;IACA,IAAI6D,SAASU,MAAM,KAAK,KAAK;QAC3B,MAAM,IAAIvE,MAAM,CAAC,gCAAgC,EAAE6D,SAASU,MAAM,EAAE;IACtE;IACA,IAAI;QACF,OAAOzB,iBAAiB,MAAMe,SAASW,IAAI;IAC7C,EAAE,OAAOC,OAAO;QACd,IAAIA,iBAAiBC,aAAa;YAChC,MAAM,IAAI1E,MAAM;QAClB;QACA,MAAMyE;IACR;AACF"}

package/dist/capacity/sources/claude.d.ts

@@ -0,0 +1,9 @@
+import { type AnthropicCapacityRequestOptions } from "../providers/anthropic.js";
+import type { CapacityReport } from "../types.js";
+import { type ClaudeCredentialOptions } from "../../harnesses/claude/credentials.js";
+export type ClaudeCapacityOptions = ClaudeCredentialOptions & AnthropicCapacityRequestOptions;
+export type ClaudeProbeOptions = ClaudeCapacityOptions & {
+    checkedAt: string;
+};
+export declare function probeClaudeCapacity(options: ClaudeProbeOptions): Promise<CapacityReport>;
+//# sourceMappingURL=claude.d.ts.map

package/dist/capacity/sources/claude.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"claude.d.ts","sourceRoot":"","sources":["../../../src/capacity/sources/claude.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,+BAA+B,EACrC,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,cAAc,EAA4B,MAAM,aAAa,CAAC;AAC5E,OAAO,EAEL,KAAK,uBAAuB,EAC7B,MAAM,uCAAuC,CAAC;AAE/C,MAAM,MAAM,qBAAqB,GAAG,uBAAuB,GAAG,+BAA+B,CAAC;AAE9F,MAAM,MAAM,kBAAkB,GAAG,qBAAqB,GAAG;IACvD,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAWF,wBAAsB,mBAAmB,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,cAAc,CAAC,CAI9F"}

package/dist/capacity/sources/claude.js

@@ -0,0 +1,17 @@
+import { fetchAnthropicCapacity } from "../providers/anthropic.js";
+import { resolveClaudeAnthropicToken } from "../../harnesses/claude/credentials.js";
+function claudeReport(snapshot, checkedAt) {
+    return {
+        harness: "claude",
+        provider: "anthropic",
+        generatedAt: checkedAt,
+        ...snapshot
+    };
+}
+export async function probeClaudeCapacity(options) {
+    const token = await resolveClaudeAnthropicToken(options);
+    const snapshot = await fetchAnthropicCapacity(token, options);
+    return claudeReport(snapshot, options.checkedAt);
+}
+
+//# sourceMappingURL=claude.js.map

package/dist/capacity/sources/claude.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/capacity/sources/claude.ts"],"sourcesContent":["import {\n  fetchAnthropicCapacity,\n  type AnthropicCapacityRequestOptions,\n} from \"../providers/anthropic.js\";\nimport type { CapacityReport, ProviderCapacitySnapshot } from \"../types.js\";\nimport {\n  resolveClaudeAnthropicToken,\n  type ClaudeCredentialOptions,\n} from \"../../harnesses/claude/credentials.js\";\n\nexport type ClaudeCapacityOptions = ClaudeCredentialOptions & AnthropicCapacityRequestOptions;\n\nexport type ClaudeProbeOptions = ClaudeCapacityOptions & {\n  checkedAt: string;\n};\n\nfunction claudeReport(snapshot: ProviderCapacitySnapshot, checkedAt: string): CapacityReport {\n  return {\n    harness: \"claude\",\n    provider: \"anthropic\",\n    generatedAt: checkedAt,\n    ...snapshot,\n  };\n}\n\nexport async function probeClaudeCapacity(options: ClaudeProbeOptions): Promise<CapacityReport> {\n  const token = await resolveClaudeAnthropicToken(options);\n  const snapshot = await fetchAnthropicCapacity(token, options);\n  return claudeReport(snapshot, options.checkedAt);\n}\n"],"names":["fetchAnthropicCapacity","resolveClaudeAnthropicToken","claudeReport","snapshot","checkedAt","harness","provider","generatedAt","probeClaudeCapacity","options","token"],"mappings":"AAAA,SACEA,sBAAsB,QAEjB,4BAA4B;AAEnC,SACEC,2BAA2B,QAEtB,wCAAwC;AAQ/C,SAASC,aAAaC,QAAkC,EAAEC,SAAiB;IACzE,OAAO;QACLC,SAAS;QACTC,UAAU;QACVC,aAAaH;QACb,GAAGD,QAAQ;IACb;AACF;AAEA,OAAO,eAAeK,oBAAoBC,OAA2B;IACnE,MAAMC,QAAQ,MAAMT,4BAA4BQ;IAChD,MAAMN,WAAW,MAAMH,uBAAuBU,OAAOD;IACrD,OAAOP,aAAaC,UAAUM,QAAQL,SAAS;AACjD"}

package/dist/capacity/{codex.d.ts → sources/codex.d.ts}

@@ -1,5 +1,5 @@
-import type { CapacityReport, CapacityWindow } from "./types.js";
-export { toRateWindow } from "./wham.js";
+import type { CapacityReport, CapacityWindow } from "../types.js";
+import { type CodexCredentialOptions } from "../../harnesses/codex/credentials.js";
 type CodexUsageSource = "pat" | "oauth" | "cli";
 type UsageSnapshot = {
     windows: CapacityWindow[];
@@ -17,20 +17,16 @@ type CliResponses = {
     account: unknown;
 };
 type CodexRpc = (messages: RpcMessage[]) => Promise<CliResponses>;
-export declare const CODEX_APP_SERVER_ARGS: readonly ["-s", "read-only", "-a", "untrusted", "app-server"];
-type CodexProbeOptions = {
+export type CodexProbeOptions = CodexCredentialOptions & {
     installed: boolean;
     checkedAt: string;
-    readFile?: (path: string, encoding: BufferEncoding) => Promise<string>;
     fetch?: typeof globalThis.fetch;
     rpc?: CodexRpc;
     timeoutMs?: number;
-    env?: NodeJS.ProcessEnv;
-    now?: () => Date;
 };
-export declare function resolveCodexAuthPath(env?: NodeJS.ProcessEnv): string;
+export declare const CODEX_APP_SERVER_ARGS: readonly ["-s", "read-only", "-a", "untrusted", "app-server"];
 export declare function parseUsage(raw: unknown, source: "pat" | "oauth"): UsageSnapshot;
-export declare function parseCliUsage(raw: unknown): UsageSnapshot;
 export declare function codexUnavailableReport(checkedAt: string): CapacityReport;
 export declare function probeCodexCapacity(options: CodexProbeOptions): Promise<CapacityReport>;
+export {};
 //# sourceMappingURL=codex.d.ts.map

package/dist/capacity/sources/codex.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"codex.d.ts","sourceRoot":"","sources":["../../../src/capacity/sources/codex.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAClE,OAAO,EAEL,KAAK,sBAAsB,EAC5B,MAAM,sCAAsC,CAAC;AAE9C,KAAK,gBAAgB,GAAG,KAAK,GAAG,OAAO,GAAG,KAAK,CAAC;AAChD,KAAK,aAAa,GAAG;IACnB,OAAO,EAAE,cAAc,EAAE,CAAC;IAC1B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,MAAM,EAAE,gBAAgB,CAAC;CAC1B,CAAC;AAEF,KAAK,aAAa,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAC7C,KAAK,UAAU,GAAG;IAAE,EAAE,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,aAAa,CAAA;CAAE,CAAC;AAC1E,KAAK,YAAY,GAAG;IAAE,UAAU,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,OAAO,CAAA;CAAE,CAAC;AAC9D,KAAK,QAAQ,GAAG,CAAC,QAAQ,EAAE,UAAU,EAAE,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC;AAElE,MAAM,MAAM,iBAAiB,GAAG,sBAAsB,GAAG;IACvD,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAChC,GAAG,CAAC,EAAE,QAAQ,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,eAAO,MAAM,qBAAqB,+DAAgE,CAAC;AAgBnG,wBAAgB,UAAU,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,GAAG,OAAO,GAAG,aAAa,CAG/E;AA8GD,wBAAgB,sBAAsB,CAAC,SAAS,EAAE,MAAM,GAAG,cAAc,CAUxE;AAuDD,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,cAAc,CAAC,CAiC5F"}

package/dist/capacity/{codex.js → sources/codex.js}

@@ -1,8 +1,6 @@
 import { spawn } from "node:child_process";
-import { readFile } from "node:fs/promises";
-import { join } from "node:path";
-import { parseWhamUsage, resetTime, safeIdentifier } from "./wham.js";
-export { toRateWindow } from "./wham.js";
+import { fetchOpenAiPatAccountId, fetchOpenAiSubscriptionCapacity, parseWhamUsage, resetTime, safeIdentifier } from "../providers/openai-subscription.js";
+import { resolveCodexCredentials } from "../../harnesses/codex/credentials.js";
 export const CODEX_APP_SERVER_ARGS = [
     "-s",
     "read-only",
@@ -19,10 +17,6 @@ function finiteNumber(value) {
 function nonEmptyText(value) {
     return typeof value === "string" && value.length > 0 ? value : null;
 }
-export function resolveCodexAuthPath(env = process.env) {
-    const root = env.CODEX_HOME || join(env.HOME || "", ".codex");
-    return join(root, "auth.json");
-}
 export function parseUsage(raw, source) {
     const { windows, creditsRemaining } = parseWhamUsage(raw);
     return {
@@ -51,13 +45,15 @@ function cliSnapshotWindows(value, fallbackId) {
         cliWindow(snapshot.secondary, `${scope}:secondary`, `${scope} secondary`)
     ].filter((item)=>item !== null);
 }
-export function parseCliUsage(raw) {
+function parseCliUsage(raw) {
     const response = record(raw) ?? {};
     const primary = record(response.rateLimits);
     const windows = primary ? cliSnapshotWindows(primary, "codex") : [];
     const buckets = record(response.rateLimitsByLimitId);
     if (buckets) {
-        for (const [id, snapshot] of Object.entries(buckets))windows.push(...cliSnapshotWindows(snapshot, id));
+        for (const [id, snapshot] of Object.entries(buckets)){
+            windows.push(...cliSnapshotWindows(snapshot, id));
+        }
     }
     const unique = [
         ...new Map(windows.map((window)=>[
@@ -71,7 +67,7 @@ export function parseCliUsage(raw) {
         source: "cli"
     };
 }
-function capacityFromSnapshot(snapshot, context, raw) {
+function capacityFromSnapshot(snapshot, options, raw) {
     const hasUsage = snapshot.windows.some((window)=>window.usedPercent !== null);
     const rateLimits = record(record(raw)?.rateLimits);
     const reached = nonEmptyText(rateLimits?.rateLimitReachedType);
@@ -79,56 +75,13 @@ function capacityFromSnapshot(snapshot, context, raw) {
     return {
         harness: "codex",
         provider: "openai",
-        generatedAt: context.checkedAt,
+        generatedAt: options.checkedAt,
         authenticated: true,
         available: reached ? "no" : hasUsage ? "yes" : "unknown",
         windows: snapshot.windows,
         creditsRemaining: snapshot.creditsRemaining ?? finiteNumber(resetCredits?.availableCount)
     };
 }
-function jwtExpiry(token) {
-    const part = token.split(".")[1];
-    if (!part) return null;
-    try {
-        return finiteNumber(record(JSON.parse(Buffer.from(part, "base64url").toString("utf8")))?.exp);
-    } catch  {
-        return null;
-    }
-}
-function staleOAuth(tokens, token, now) {
-    const metadata = tokens.expires_at ?? tokens.expiresAt ?? tokens.expiry;
-    let expiry = finiteNumber(metadata);
-    if (typeof metadata === "string") {
-        const parsed = Date.parse(metadata);
-        expiry = Number.isNaN(parsed) ? null : parsed / 1000;
-    }
-    expiry ??= jwtExpiry(token);
-    return expiry !== null && expiry <= now.getTime() / 1000;
-}
-async function fetchJson(fetcher, url, init, timeoutMs) {
-    const controller = new AbortController();
-    const timer = setTimeout(()=>controller.abort(), timeoutMs);
-    try {
-        const response = await fetcher(url, {
-            ...init,
-            signal: controller.signal
-        });
-        if (!response.ok) throw new Error(response.status === 401 ? "unauthorized" : "request failed");
-        return await response.json();
-    } finally{
-        clearTimeout(timer);
-    }
-}
-async function apiSnapshot(token, accountId, source, options) {
-    const fetcher = options.fetch ?? globalThis.fetch;
-    const raw = await fetchJson(fetcher, "https://chatgpt.com/backend-api/wham/usage", {
-        headers: {
-            Authorization: `Bearer ${token}`,
-            "ChatGPT-Account-Id": accountId
-        }
-    }, options.timeoutMs ?? 5000);
-    return parseUsage(raw, source);
-}
 function appServerRpc(messages, timeoutMs = 5000) {
     return new Promise((resolve, reject)=>{
         const child = spawn("codex", CODEX_APP_SERVER_ARGS, {
@@ -195,11 +148,8 @@ export function codexUnavailableReport(checkedAt) {
         creditsRemaining: null
     };
 }
-function unavailable(options) {
-    return codexUnavailableReport(options.checkedAt);
-}
 async function cliFallback(options) {
-    if (!options.installed) return unavailable(options);
+    if (!options.installed) return codexUnavailableReport(options.checkedAt);
     const messages = [
         {
             id: 1,
@@ -236,40 +186,37 @@ async function cliFallback(options) {
         }
         return result;
     } catch  {
-        return unavailable(options);
+        return codexUnavailableReport(options.checkedAt);
     }
 }
+async function subscriptionSnapshot(access, accountId, source, options) {
+    const snapshot = await fetchOpenAiSubscriptionCapacity({
+        access,
+        accountId
+    }, options);
+    if (snapshot.authenticated === false) throw new Error("unauthorized");
+    return {
+        windows: snapshot.windows,
+        creditsRemaining: snapshot.creditsRemaining,
+        source
+    };
+}
 export async function probeCodexCapacity(options) {
-    let parsed = null;
-    try {
-        const contents = await (options.readFile ?? readFile)(resolveCodexAuthPath(options.env), "utf8");
-        parsed = record(JSON.parse(contents));
-    } catch  {
-        return cliFallback(options);
-    }
-    const auth = parsed ?? {};
-    const pat = nonEmptyText(auth.personal_access_token);
-    if (pat) {
+    const credentials = await resolveCodexCredentials(options);
+    if (!credentials) return cliFallback(options);
+    if (credentials.personalAccessToken) {
         try {
-            const fetcher = options.fetch ?? globalThis.fetch;
-            const whoami = record(await fetchJson(fetcher, "https://auth.openai.com/api/accounts/v1/user-auth-credential/whoami", {
-                headers: {
-                    Authorization: `Bearer ${pat}`
-                }
-            }, options.timeoutMs ?? 5000));
-            const accountId = nonEmptyText(whoami?.chatgpt_account_id);
-            if (!accountId) throw new Error("account unavailable");
-            return capacityFromSnapshot(await apiSnapshot(pat, accountId, "pat", options), options);
+            const accountId = await fetchOpenAiPatAccountId(credentials.personalAccessToken, options);
+            const snapshot = await subscriptionSnapshot(credentials.personalAccessToken, accountId, "pat", options);
+            return capacityFromSnapshot(snapshot, options);
         } catch  {
         // Continue to a separately available OAuth credential before using the CLI.
         }
     }
-    const tokens = record(auth.tokens);
-    const accessToken = nonEmptyText(tokens?.access_token);
-    const accountId = nonEmptyText(tokens?.account_id);
-    if (tokens && accessToken && accountId && !staleOAuth(tokens, accessToken, (options.now ?? (()=>new Date()))())) {
+    if (credentials.oauth) {
         try {
-            return capacityFromSnapshot(await apiSnapshot(accessToken, accountId, "oauth", options), options);
+            const snapshot = await subscriptionSnapshot(credentials.oauth.access, credentials.oauth.accountId, "oauth", options);
+            return capacityFromSnapshot(snapshot, options);
         } catch  {
             return cliFallback(options);
         }

package/dist/capacity/sources/codex.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/capacity/sources/codex.ts"],"sourcesContent":["import { spawn } from \"node:child_process\";\nimport {\n  fetchOpenAiPatAccountId,\n  fetchOpenAiSubscriptionCapacity,\n  parseWhamUsage,\n  resetTime,\n  safeIdentifier,\n} from \"../providers/openai-subscription.js\";\nimport type { CapacityReport, CapacityWindow } from \"../types.js\";\nimport {\n  resolveCodexCredentials,\n  type CodexCredentialOptions,\n} from \"../../harnesses/codex/credentials.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 type CodexProbeOptions = CodexCredentialOptions & {\n  installed: boolean;\n  checkedAt: string;\n  fetch?: typeof globalThis.fetch;\n  rpc?: CodexRpc;\n  timeoutMs?: number;\n};\n\nexport const CODEX_APP_SERVER_ARGS = [\"-s\", \"read-only\", \"-a\", \"untrusted\", \"app-server\"] as const;\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\nexport function parseUsage(raw: unknown, source: \"pat\" | \"oauth\"): UsageSnapshot {\n  const { windows, creditsRemaining } = parseWhamUsage(raw);\n  return { windows, creditsRemaining, source };\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\nfunction 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  }\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  options: 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    harness: \"codex\",\n    provider: \"openai\",\n    generatedAt: options.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 appServerRpc(messages: RpcMessage[], timeoutMs = 5000): Promise<CliResponses> {\n  return new Promise((resolve, reject) => {\n    const child = spawn(\"codex\", CODEX_APP_SERVER_ARGS, { stdio: [\"pipe\", \"pipe\", \"ignore\"] });\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\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\nasync function cliFallback(options: CodexProbeOptions): Promise<CapacityReport> {\n  if (!options.installed) return codexUnavailableReport(options.checkedAt);\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 codexUnavailableReport(options.checkedAt);\n  }\n}\n\nasync function subscriptionSnapshot(\n  access: string,\n  accountId: string,\n  source: \"pat\" | \"oauth\",\n  options: CodexProbeOptions,\n): Promise<UsageSnapshot> {\n  const snapshot = await fetchOpenAiSubscriptionCapacity({ access, accountId }, options);\n  if (snapshot.authenticated === false) throw new Error(\"unauthorized\");\n  return {\n    windows: snapshot.windows,\n    creditsRemaining: snapshot.creditsRemaining,\n    source,\n  };\n}\n\nexport async function probeCodexCapacity(options: CodexProbeOptions): Promise<CapacityReport> {\n  const credentials = await resolveCodexCredentials(options);\n  if (!credentials) return cliFallback(options);\n\n  if (credentials.personalAccessToken) {\n    try {\n      const accountId = await fetchOpenAiPatAccountId(credentials.personalAccessToken, options);\n      const snapshot = await subscriptionSnapshot(\n        credentials.personalAccessToken,\n        accountId,\n        \"pat\",\n        options,\n      );\n      return capacityFromSnapshot(snapshot, options);\n    } catch {\n      // Continue to a separately available OAuth credential before using the CLI.\n    }\n  }\n\n  if (credentials.oauth) {\n    try {\n      const snapshot = await subscriptionSnapshot(\n        credentials.oauth.access,\n        credentials.oauth.accountId,\n        \"oauth\",\n        options,\n      );\n      return capacityFromSnapshot(snapshot, options);\n    } catch {\n      return cliFallback(options);\n    }\n  }\n  return cliFallback(options);\n}\n"],"names":["spawn","fetchOpenAiPatAccountId","fetchOpenAiSubscriptionCapacity","parseWhamUsage","resetTime","safeIdentifier","resolveCodexCredentials","CODEX_APP_SERVER_ARGS","record","value","Array","isArray","finiteNumber","Number","isFinite","nonEmptyText","length","parseUsage","raw","source","windows","creditsRemaining","cliWindow","id","label","input","durationMinutes","windowDurationMins","usedPercent","resetsAt","cliSnapshotWindows","fallbackId","snapshot","scope","limitId","primary","secondary","filter","item","parseCliUsage","response","rateLimits","buckets","rateLimitsByLimitId","Object","entries","push","unique","Map","map","window","values","capacityFromSnapshot","options","hasUsage","some","reached","rateLimitReachedType","resetCredits","rateLimitResetCredits","usageLimitResetCredits","harness","provider","generatedAt","checkedAt","authenticated","available","availableCount","appServerRpc","messages","timeoutMs","Promise","resolve","reject","child","stdio","results","buffer","settled","finish","error","clearTimeout","timer","kill","setTimeout","Error","once","stdout","setEncoding","on","chunk","newline","indexOf","line","slice","trim","message","JSON","parse","request","stdin","write","stringify","result","account","codexUnavailableReport","cliFallback","installed","method","params","clientInfo","name","title","version","capabilities","rpc","requests","accountEnvelope","hasOwn","subscriptionSnapshot","access","accountId","probeCodexCapacity","credentials","personalAccessToken","oauth"],"mappings":"AAAA,SAASA,KAAK,QAAQ,qBAAqB;AAC3C,SACEC,uBAAuB,EACvBC,+BAA+B,EAC/BC,cAAc,EACdC,SAAS,EACTC,cAAc,QACT,sCAAsC;AAE7C,SACEC,uBAAuB,QAElB,uCAAuC;AAsB9C,OAAO,MAAMC,wBAAwB;IAAC;IAAM;IAAa;IAAM;IAAa;CAAa,CAAU;AAEnG,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,OAAO,SAASQ,WAAWC,GAAY,EAAEC,MAAuB;IAC9D,MAAM,EAAEC,OAAO,EAAEC,gBAAgB,EAAE,GAAGlB,eAAee;IACrD,OAAO;QAAEE;QAASC;QAAkBF;IAAO;AAC7C;AAEA,SAASG,UAAUb,KAAc,EAAEc,EAAU,EAAEC,KAAa;IAC1D,MAAMC,QAAQjB,OAAOC;IACrB,IAAI,CAACgB,OAAO,OAAO;IACnB,OAAO;QACLF;QACAC;QACAE,iBAAiBd,aAAaa,MAAME,kBAAkB;QACtDC,aAAahB,aAAaa,MAAMG,WAAW;QAC3CC,UAAUzB,UAAUqB,MAAMI,QAAQ;IACpC;AACF;AAEA,SAASC,mBAAmBrB,KAAc,EAAEsB,UAAkB;IAC5D,MAAMC,WAAWxB,OAAOC;IACxB,IAAI,CAACuB,UAAU,OAAO,EAAE;IACxB,MAAMC,QAAQ5B,eAAe2B,SAASE,OAAO,KAAK7B,eAAe0B,eAAe;IAChF,OAAO;QACLT,UAAUU,SAASG,OAAO,EAAE,GAAGF,MAAM,QAAQ,CAAC,EAAE,GAAGA,MAAM,QAAQ,CAAC;QAClEX,UAAUU,SAASI,SAAS,EAAE,GAAGH,MAAM,UAAU,CAAC,EAAE,GAAGA,MAAM,UAAU,CAAC;KACzE,CAACI,MAAM,CAAC,CAACC,OAAiCA,SAAS;AACtD;AAEA,SAASC,cAAcrB,GAAY;IACjC,MAAMsB,WAAWhC,OAAOU,QAAQ,CAAC;IACjC,MAAMiB,UAAU3B,OAAOgC,SAASC,UAAU;IAC1C,MAAMrB,UAAUe,UAAUL,mBAAmBK,SAAS,WAAW,EAAE;IACnE,MAAMO,UAAUlC,OAAOgC,SAASG,mBAAmB;IACnD,IAAID,SAAS;QACX,KAAK,MAAM,CAACnB,IAAIS,SAAS,IAAIY,OAAOC,OAAO,CAACH,SAAU;YACpDtB,QAAQ0B,IAAI,IAAIhB,mBAAmBE,UAAUT;QAC/C;IACF;IACA,MAAMwB,SAAS;WAAI,IAAIC,IAAI5B,QAAQ6B,GAAG,CAAC,CAACC,SAAW;gBAACA,OAAO3B,EAAE;gBAAE2B;aAAO,GAAGC,MAAM;KAAG;IAClF,OAAO;QAAE/B,SAAS2B;QAAQ1B,kBAAkB;QAAMF,QAAQ;IAAM;AAClE;AAEA,SAASiC,qBACPpB,QAAuB,EACvBqB,OAA0B,EAC1BnC,GAAa;IAEb,MAAMoC,WAAWtB,SAASZ,OAAO,CAACmC,IAAI,CAAC,CAACL,SAAWA,OAAOtB,WAAW,KAAK;IAC1E,MAAMa,aAAajC,OAAOA,OAAOU,MAAMuB;IACvC,MAAMe,UAAUzC,aAAa0B,YAAYgB;IACzC,MAAMC,eACJlD,OAAOA,OAAOU,MAAMyC,0BAA0BnD,OAAOA,OAAOU,MAAM0C;IACpE,OAAO;QACLC,SAAS;QACTC,UAAU;QACVC,aAAaV,QAAQW,SAAS;QAC9BC,eAAe;QACfC,WAAWV,UAAU,OAAOF,WAAW,QAAQ;QAC/ClC,SAASY,SAASZ,OAAO;QACzBC,kBAAkBW,SAASX,gBAAgB,IAAIT,aAAa8C,cAAcS;IAC5E;AACF;AAEA,SAASC,aAAaC,QAAsB,EAAEC,YAAY,IAAI;IAC5D,OAAO,IAAIC,QAAQ,CAACC,SAASC;QAC3B,MAAMC,QAAQ1E,MAAM,SAASO,uBAAuB;YAAEoE,OAAO;gBAAC;gBAAQ;gBAAQ;aAAS;QAAC;QACxF,MAAMC,UAAiC,CAAC;QACxC,IAAIC,SAAS;QACb,IAAIC,UAAU;QACd,MAAMC,SAAS,CAACC;YACd,IAAIF,SAAS;YACbA,UAAU;YACVG,aAAaC;YACbR,MAAMS,IAAI;YACV,IAAIH,OAAOP,OAAOO;iBACbR,QAAQI;QACf;QACA,MAAMM,QAAQE,WAAW,IAAML,OAAO,IAAIM,MAAM,2BAA2Bf;QAC3EI,MAAMY,IAAI,CAAC,SAAS,IAAMP,OAAO,IAAIM,MAAM;QAC3CX,MAAMY,IAAI,CAAC,QAAQ;YACjB,IAAI,CAACR,SAASC,OAAO,IAAIM,MAAM;QACjC;QACAX,MAAMa,MAAM,CAACC,WAAW,CAAC;QACzBd,MAAMa,MAAM,CAACE,EAAE,CAAC,QAAQ,CAACC;YACvBb,UAAUa;YACV,OAAS;gBACP,MAAMC,UAAUd,OAAOe,OAAO,CAAC;gBAC/B,IAAID,UAAU,GAAG;gBACjB,MAAME,OAAOhB,OAAOiB,KAAK,CAAC,GAAGH,SAASI,IAAI;gBAC1ClB,SAASA,OAAOiB,KAAK,CAACH,UAAU;gBAChC,IAAI,CAACE,MAAM;gBACX,IAAIG;gBACJ,IAAI;oBACFA,UAAUC,KAAKC,KAAK,CAACL;gBACvB,EAAE,OAAM;oBACN;gBACF;gBACA,IAAIG,QAAQzE,EAAE,KAAK,GAAG;oBACpB,KAAK,MAAM4E,WAAW9B,SAASyB,KAAK,CAAC,GACnCpB,MAAM0B,KAAK,CAACC,KAAK,CAAC,GAAGJ,KAAKK,SAAS,CAACH,SAAS,EAAE,CAAC;gBACpD,OAAO,IAAIH,QAAQzE,EAAE,KAAK,GAAG;oBAC3B,IAAIyE,QAAQhB,KAAK,EAAED,OAAO,IAAIM,MAAM;yBAC/BT,QAAQnC,UAAU,GAAGuD,QAAQO,MAAM;gBAC1C,OAAO,IAAIP,QAAQzE,EAAE,KAAK,GAAG;oBAC3B,IAAIyE,QAAQhB,KAAK,EAAED,OAAO,IAAIM,MAAM;yBAC/BT,QAAQ4B,OAAO,GAAGR,QAAQO,MAAM;gBACvC;gBACA,IAAI,gBAAgB3B,WAAW,aAAaA,SAASG;YACvD;QACF;QACAL,MAAM0B,KAAK,CAACC,KAAK,CAAC,GAAGJ,KAAKK,SAAS,CAACjC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC;IACtD;AACF;AAEA,OAAO,SAASoC,uBAAuBzC,SAAiB;IACtD,OAAO;QACLH,SAAS;QACTC,UAAU;QACVC,aAAaC;QACbC,eAAe;QACfC,WAAW;QACX9C,SAAS,EAAE;QACXC,kBAAkB;IACpB;AACF;AAEA,eAAeqF,YAAYrD,OAA0B;IACnD,IAAI,CAACA,QAAQsD,SAAS,EAAE,OAAOF,uBAAuBpD,QAAQW,SAAS;IACvE,MAAMK,WAAyB;QAC7B;YACE9C,IAAI;YACJqF,QAAQ;YACRC,QAAQ;gBACNC,YAAY;oBAAEC,MAAM;oBAAaC,OAAO;oBAAMC,SAAS;gBAAI;gBAC3DC,cAAc;YAChB;QACF;QACA;YAAEN,QAAQ;QAAc;QACxB;YAAErF,IAAI;YAAGqF,QAAQ;QAA0B;QAC3C;YAAErF,IAAI;YAAGqF,QAAQ;QAAe;KACjC;IACD,IAAI;QACF,MAAMO,MAAM9D,QAAQ8D,GAAG,IAAK,CAAA,CAACC,WAAahD,aAAagD,UAAU/D,QAAQiB,SAAS,CAAA;QAClF,MAAM9B,WAAW,MAAM2E,IAAI9C;QAC3B,MAAMkC,SAASnD,qBACbb,cAAcC,SAASC,UAAU,GACjCY,SACAb,SAASC,UAAU;QAErB,MAAM4E,kBAAkB7G,OAAOgC,SAASgE,OAAO;QAC/C,IACEa,mBACAzE,OAAO0E,MAAM,CAACD,iBAAiB,cAC/B,CAAC7G,OAAO6G,gBAAgBb,OAAO,GAC/B;YACAD,OAAOtC,aAAa,GAAG;YACvBsC,OAAOrC,SAAS,GAAG;QACrB;QACA,OAAOqC;IACT,EAAE,OAAM;QACN,OAAOE,uBAAuBpD,QAAQW,SAAS;IACjD;AACF;AAEA,eAAeuD,qBACbC,MAAc,EACdC,SAAiB,EACjBtG,MAAuB,EACvBkC,OAA0B;IAE1B,MAAMrB,WAAW,MAAM9B,gCAAgC;QAAEsH;QAAQC;IAAU,GAAGpE;IAC9E,IAAIrB,SAASiC,aAAa,KAAK,OAAO,MAAM,IAAIoB,MAAM;IACtD,OAAO;QACLjE,SAASY,SAASZ,OAAO;QACzBC,kBAAkBW,SAASX,gBAAgB;QAC3CF;IACF;AACF;AAEA,OAAO,eAAeuG,mBAAmBrE,OAA0B;IACjE,MAAMsE,cAAc,MAAMrH,wBAAwB+C;IAClD,IAAI,CAACsE,aAAa,OAAOjB,YAAYrD;IAErC,IAAIsE,YAAYC,mBAAmB,EAAE;QACnC,IAAI;YACF,MAAMH,YAAY,MAAMxH,wBAAwB0H,YAAYC,mBAAmB,EAAEvE;YACjF,MAAMrB,WAAW,MAAMuF,qBACrBI,YAAYC,mBAAmB,EAC/BH,WACA,OACApE;YAEF,OAAOD,qBAAqBpB,UAAUqB;QACxC,EAAE,OAAM;QACN,4EAA4E;QAC9E;IACF;IAEA,IAAIsE,YAAYE,KAAK,EAAE;QACrB,IAAI;YACF,MAAM7F,WAAW,MAAMuF,qBACrBI,YAAYE,KAAK,CAACL,MAAM,EACxBG,YAAYE,KAAK,CAACJ,SAAS,EAC3B,SACApE;YAEF,OAAOD,qBAAqBpB,UAAUqB;QACxC,EAAE,OAAM;YACN,OAAOqD,YAAYrD;QACrB;IACF;IACA,OAAOqD,YAAYrD;AACrB"}

package/dist/capacity/sources/pi.d.ts

@@ -0,0 +1,30 @@
+import type { CapacityReport } from "../types.js";
+import { type PiCredentialOptions } from "../../harnesses/pi/credentials.js";
+export type PiAnthropicCapacityOptions = PiCredentialOptions & {
+    now?: () => Date;
+    fetch?: typeof globalThis.fetch;
+    timeoutMs?: number;
+};
+export type PiAnthropicProbeOptions = PiAnthropicCapacityOptions & {
+    checkedAt: string;
+};
+export type PiOpenAiCapacityOptions = PiCredentialOptions & {
+    now?: () => Date;
+    fetch?: typeof globalThis.fetch;
+    timeoutMs?: number;
+};
+export type PiOpenAiProbeOptions = PiOpenAiCapacityOptions & {
+    checkedAt: string;
+};
+export type PiZaiCapacityOptions = PiCredentialOptions & {
+    now?: () => Date;
+    fetch?: typeof globalThis.fetch;
+    timeoutMs?: number;
+};
+export type PiZaiProbeOptions = PiZaiCapacityOptions & {
+    checkedAt: string;
+};
+export declare function probePiAnthropicCapacity(options: PiAnthropicProbeOptions): Promise<CapacityReport>;
+export declare function probeOpenAiCapacity(options: PiOpenAiProbeOptions): Promise<CapacityReport>;
+export declare function probeZaiCapacity(options: PiZaiProbeOptions): Promise<CapacityReport>;
+//# sourceMappingURL=pi.d.ts.map

package/dist/capacity/sources/pi.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pi.d.ts","sourceRoot":"","sources":["../../../src/capacity/sources/pi.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAA4B,MAAM,aAAa,CAAC;AAC5E,OAAO,EAIL,KAAK,mBAAmB,EAEzB,MAAM,mCAAmC,CAAC;AAE3C,MAAM,MAAM,0BAA0B,GAAG,mBAAmB,GAAG;IAC7D,GAAG,CAAC,EAAE,MAAM,IAAI,CAAC;IACjB,KAAK,CAAC,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAAG,0BAA0B,GAAG;IACjE,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAAG,mBAAmB,GAAG;IAC1D,GAAG,CAAC,EAAE,MAAM,IAAI,CAAC;IACjB,KAAK,CAAC,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG,uBAAuB,GAAG;IAC3D,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG,mBAAmB,GAAG;IACvD,GAAG,CAAC,EAAE,MAAM,IAAI,CAAC;IACjB,KAAK,CAAC,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG,oBAAoB,GAAG;IACrD,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,wBAAsB,wBAAwB,CAC5C,OAAO,EAAE,uBAAuB,GAC/B,OAAO,CAAC,cAAc,CAAC,CAqBzB;AAgCD,wBAAsB,mBAAmB,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC,CAqBhG;AAED,wBAAsB,gBAAgB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,cAAc,CAAC,CAS1F"}

package/dist/capacity/sources/pi.js

@@ -0,0 +1,81 @@
+import { fetchAnthropicCapacity } from "../providers/anthropic.js";
+import { fetchOpenAiPlatformCapacity } from "../providers/openai-platform.js";
+import { fetchOpenAiSubscriptionCapacity } from "../providers/openai-subscription.js";
+import { fetchZaiCapacity } from "../providers/zai.js";
+import { resolvePiAnthropicCredential, resolvePiOpenAiCredential, resolvePiZaiApiKey } from "../../harnesses/pi/credentials.js";
+export async function probePiAnthropicCapacity(options) {
+    const credential = await resolvePiAnthropicCredential(options);
+    const now = options.now?.() ?? new Date();
+    if (credential.expiresMs !== null && credential.expiresMs <= now.getTime()) {
+        return {
+            harness: "pi",
+            provider: "anthropic",
+            generatedAt: options.checkedAt,
+            authenticated: true,
+            available: "unknown",
+            windows: [],
+            creditsRemaining: null
+        };
+    }
+    const snapshot = await fetchAnthropicCapacity(credential.access, options);
+    return {
+        harness: "pi",
+        provider: "anthropic",
+        generatedAt: options.checkedAt,
+        ...snapshot
+    };
+}
+function piOpenAiReport(snapshot, checkedAt) {
+    return {
+        harness: "pi",
+        provider: "openai",
+        generatedAt: checkedAt,
+        ...snapshot
+    };
+}
+function jwtExpiryMs(token) {
+    const part = token.split(".")[1];
+    if (!part) return null;
+    try {
+        const payload = JSON.parse(Buffer.from(part, "base64url").toString("utf8"));
+        if (!payload || typeof payload !== "object" || Array.isArray(payload)) return null;
+        const exp = payload.exp;
+        return typeof exp === "number" && Number.isFinite(exp) ? exp * 1000 : null;
+    } catch  {
+        return null;
+    }
+}
+function oauthExpired(credential, nowMs) {
+    const expiryMs = credential.expiresMs ?? jwtExpiryMs(credential.access);
+    return expiryMs !== null && expiryMs <= nowMs;
+}
+export async function probeOpenAiCapacity(options) {
+    const credential = await resolvePiOpenAiCredential(options);
+    if (credential.kind === "platform") {
+        const snapshot = await fetchOpenAiPlatformCapacity(credential.key, options);
+        return piOpenAiReport(snapshot, options.checkedAt);
+    }
+    const nowMs = Date.parse(options.checkedAt);
+    if (Number.isNaN(nowMs) || oauthExpired(credential, nowMs)) {
+        return piOpenAiReport({
+            authenticated: true,
+            available: "unknown",
+            windows: [],
+            creditsRemaining: null
+        }, options.checkedAt);
+    }
+    const snapshot = await fetchOpenAiSubscriptionCapacity(credential, options);
+    return piOpenAiReport(snapshot, options.checkedAt);
+}
+export async function probeZaiCapacity(options) {
+    const key = await resolvePiZaiApiKey(options);
+    const snapshot = await fetchZaiCapacity(key, options);
+    return {
+        harness: "pi",
+        provider: "zai",
+        generatedAt: options.checkedAt,
+        ...snapshot
+    };
+}
+
+//# sourceMappingURL=pi.js.map

package/dist/capacity/sources/pi.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/capacity/sources/pi.ts"],"sourcesContent":["import { fetchAnthropicCapacity } from \"../providers/anthropic.js\";\nimport { fetchOpenAiPlatformCapacity } from \"../providers/openai-platform.js\";\nimport { fetchOpenAiSubscriptionCapacity } from \"../providers/openai-subscription.js\";\nimport { fetchZaiCapacity } from \"../providers/zai.js\";\nimport type { CapacityReport, ProviderCapacitySnapshot } from \"../types.js\";\nimport {\n  resolvePiAnthropicCredential,\n  resolvePiOpenAiCredential,\n  resolvePiZaiApiKey,\n  type PiCredentialOptions,\n  type PiOpenAiCredential,\n} from \"../../harnesses/pi/credentials.js\";\n\nexport type PiAnthropicCapacityOptions = PiCredentialOptions & {\n  now?: () => Date;\n  fetch?: typeof globalThis.fetch;\n  timeoutMs?: number;\n};\n\nexport type PiAnthropicProbeOptions = PiAnthropicCapacityOptions & {\n  checkedAt: string;\n};\n\nexport type PiOpenAiCapacityOptions = PiCredentialOptions & {\n  now?: () => Date;\n  fetch?: typeof globalThis.fetch;\n  timeoutMs?: number;\n};\n\nexport type PiOpenAiProbeOptions = PiOpenAiCapacityOptions & {\n  checkedAt: string;\n};\n\nexport type PiZaiCapacityOptions = PiCredentialOptions & {\n  now?: () => Date;\n  fetch?: typeof globalThis.fetch;\n  timeoutMs?: number;\n};\n\nexport type PiZaiProbeOptions = PiZaiCapacityOptions & {\n  checkedAt: string;\n};\n\nexport async function probePiAnthropicCapacity(\n  options: PiAnthropicProbeOptions,\n): Promise<CapacityReport> {\n  const credential = await resolvePiAnthropicCredential(options);\n  const now = options.now?.() ?? new Date();\n  if (credential.expiresMs !== null && credential.expiresMs <= now.getTime()) {\n    return {\n      harness: \"pi\",\n      provider: \"anthropic\",\n      generatedAt: options.checkedAt,\n      authenticated: true,\n      available: \"unknown\",\n      windows: [],\n      creditsRemaining: null,\n    };\n  }\n  const snapshot = await fetchAnthropicCapacity(credential.access, options);\n  return {\n    harness: \"pi\",\n    provider: \"anthropic\",\n    generatedAt: options.checkedAt,\n    ...snapshot,\n  };\n}\n\nfunction piOpenAiReport(snapshot: ProviderCapacitySnapshot, checkedAt: string): CapacityReport {\n  return {\n    harness: \"pi\",\n    provider: \"openai\",\n    generatedAt: checkedAt,\n    ...snapshot,\n  };\n}\n\nfunction jwtExpiryMs(token: string): number | null {\n  const part = token.split(\".\")[1];\n  if (!part) return null;\n  try {\n    const payload = JSON.parse(Buffer.from(part, \"base64url\").toString(\"utf8\")) as unknown;\n    if (!payload || typeof payload !== \"object\" || Array.isArray(payload)) return null;\n    const exp = (payload as Record<string, unknown>).exp;\n    return typeof exp === \"number\" && Number.isFinite(exp) ? exp * 1000 : null;\n  } catch {\n    return null;\n  }\n}\n\nfunction oauthExpired(\n  credential: Extract<PiOpenAiCredential, { kind: \"oauth\" }>,\n  nowMs: number,\n): boolean {\n  const expiryMs = credential.expiresMs ?? jwtExpiryMs(credential.access);\n  return expiryMs !== null && expiryMs <= nowMs;\n}\n\nexport async function probeOpenAiCapacity(options: PiOpenAiProbeOptions): Promise<CapacityReport> {\n  const credential = await resolvePiOpenAiCredential(options);\n  if (credential.kind === \"platform\") {\n    const snapshot = await fetchOpenAiPlatformCapacity(credential.key, options);\n    return piOpenAiReport(snapshot, options.checkedAt);\n  }\n\n  const nowMs = Date.parse(options.checkedAt);\n  if (Number.isNaN(nowMs) || oauthExpired(credential, nowMs)) {\n    return piOpenAiReport(\n      {\n        authenticated: true,\n        available: \"unknown\",\n        windows: [],\n        creditsRemaining: null,\n      },\n      options.checkedAt,\n    );\n  }\n  const snapshot = await fetchOpenAiSubscriptionCapacity(credential, options);\n  return piOpenAiReport(snapshot, options.checkedAt);\n}\n\nexport async function probeZaiCapacity(options: PiZaiProbeOptions): Promise<CapacityReport> {\n  const key = await resolvePiZaiApiKey(options);\n  const snapshot = await fetchZaiCapacity(key, options);\n  return {\n    harness: \"pi\",\n    provider: \"zai\",\n    generatedAt: options.checkedAt,\n    ...snapshot,\n  };\n}\n"],"names":["fetchAnthropicCapacity","fetchOpenAiPlatformCapacity","fetchOpenAiSubscriptionCapacity","fetchZaiCapacity","resolvePiAnthropicCredential","resolvePiOpenAiCredential","resolvePiZaiApiKey","probePiAnthropicCapacity","options","credential","now","Date","expiresMs","getTime","harness","provider","generatedAt","checkedAt","authenticated","available","windows","creditsRemaining","snapshot","access","piOpenAiReport","jwtExpiryMs","token","part","split","payload","JSON","parse","Buffer","from","toString","Array","isArray","exp","Number","isFinite","oauthExpired","nowMs","expiryMs","probeOpenAiCapacity","kind","key","isNaN","probeZaiCapacity"],"mappings":"AAAA,SAASA,sBAAsB,QAAQ,4BAA4B;AACnE,SAASC,2BAA2B,QAAQ,kCAAkC;AAC9E,SAASC,+BAA+B,QAAQ,sCAAsC;AACtF,SAASC,gBAAgB,QAAQ,sBAAsB;AAEvD,SACEC,4BAA4B,EAC5BC,yBAAyB,EACzBC,kBAAkB,QAGb,oCAAoC;AAgC3C,OAAO,eAAeC,yBACpBC,OAAgC;IAEhC,MAAMC,aAAa,MAAML,6BAA6BI;IACtD,MAAME,MAAMF,QAAQE,GAAG,QAAQ,IAAIC;IACnC,IAAIF,WAAWG,SAAS,KAAK,QAAQH,WAAWG,SAAS,IAAIF,IAAIG,OAAO,IAAI;QAC1E,OAAO;YACLC,SAAS;YACTC,UAAU;YACVC,aAAaR,QAAQS,SAAS;YAC9BC,eAAe;YACfC,WAAW;YACXC,SAAS,EAAE;YACXC,kBAAkB;QACpB;IACF;IACA,MAAMC,WAAW,MAAMtB,uBAAuBS,WAAWc,MAAM,EAAEf;IACjE,OAAO;QACLM,SAAS;QACTC,UAAU;QACVC,aAAaR,QAAQS,SAAS;QAC9B,GAAGK,QAAQ;IACb;AACF;AAEA,SAASE,eAAeF,QAAkC,EAAEL,SAAiB;IAC3E,OAAO;QACLH,SAAS;QACTC,UAAU;QACVC,aAAaC;QACb,GAAGK,QAAQ;IACb;AACF;AAEA,SAASG,YAAYC,KAAa;IAChC,MAAMC,OAAOD,MAAME,KAAK,CAAC,IAAI,CAAC,EAAE;IAChC,IAAI,CAACD,MAAM,OAAO;IAClB,IAAI;QACF,MAAME,UAAUC,KAAKC,KAAK,CAACC,OAAOC,IAAI,CAACN,MAAM,aAAaO,QAAQ,CAAC;QACnE,IAAI,CAACL,WAAW,OAAOA,YAAY,YAAYM,MAAMC,OAAO,CAACP,UAAU,OAAO;QAC9E,MAAMQ,MAAM,AAACR,QAAoCQ,GAAG;QACpD,OAAO,OAAOA,QAAQ,YAAYC,OAAOC,QAAQ,CAACF,OAAOA,MAAM,OAAO;IACxE,EAAE,OAAM;QACN,OAAO;IACT;AACF;AAEA,SAASG,aACP/B,UAA0D,EAC1DgC,KAAa;IAEb,MAAMC,WAAWjC,WAAWG,SAAS,IAAIa,YAAYhB,WAAWc,MAAM;IACtE,OAAOmB,aAAa,QAAQA,YAAYD;AAC1C;AAEA,OAAO,eAAeE,oBAAoBnC,OAA6B;IACrE,MAAMC,aAAa,MAAMJ,0BAA0BG;IACnD,IAAIC,WAAWmC,IAAI,KAAK,YAAY;QAClC,MAAMtB,WAAW,MAAMrB,4BAA4BQ,WAAWoC,GAAG,EAAErC;QACnE,OAAOgB,eAAeF,UAAUd,QAAQS,SAAS;IACnD;IAEA,MAAMwB,QAAQ9B,KAAKoB,KAAK,CAACvB,QAAQS,SAAS;IAC1C,IAAIqB,OAAOQ,KAAK,CAACL,UAAUD,aAAa/B,YAAYgC,QAAQ;QAC1D,OAAOjB,eACL;YACEN,eAAe;YACfC,WAAW;YACXC,SAAS,EAAE;YACXC,kBAAkB;QACpB,GACAb,QAAQS,SAAS;IAErB;IACA,MAAMK,WAAW,MAAMpB,gCAAgCO,YAAYD;IACnE,OAAOgB,eAAeF,UAAUd,QAAQS,SAAS;AACnD;AAEA,OAAO,eAAe8B,iBAAiBvC,OAA0B;IAC/D,MAAMqC,MAAM,MAAMvC,mBAAmBE;IACrC,MAAMc,WAAW,MAAMnB,iBAAiB0C,KAAKrC;IAC7C,OAAO;QACLM,SAAS;QACTC,UAAU;QACVC,aAAaR,QAAQS,SAAS;QAC9B,GAAGK,QAAQ;IACb;AACF"}

package/dist/capacity/types.d.ts

@@ -22,4 +22,6 @@ export interface CapacityReport {
     windows: CapacityWindow[];
     creditsRemaining: number | null;
 }
+/** Provider quota data before a harness/source identity is attached. */
+export type ProviderCapacitySnapshot = Omit<CapacityReport, "harness" | "provider" | "generatedAt">;
 //# sourceMappingURL=types.d.ts.map

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;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"}
+{"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;AAED,wEAAwE;AACxE,MAAM,MAAM,wBAAwB,GAAG,IAAI,CAAC,cAAc,EAAE,SAAS,GAAG,UAAU,GAAG,aAAa,CAAC,CAAC"}