@ai-devkit/agent-manager 0.36.0 → 0.38.0

package/dist/capacity/providers/openai-subscription.js

@@ -0,0 +1,129 @@
+const WHAM_USAGE_URL = "https://chatgpt.com/backend-api/wham/usage";
+const OPENAI_WHOAMI_URL = "https://auth.openai.com/api/accounts/v1/user-auth-credential/whoami";
+function record(value) {
+    return value !== null && typeof value === "object" && !Array.isArray(value) ? value : null;
+}
+function finiteNumber(value) {
+    return typeof value === "number" && Number.isFinite(value) ? value : null;
+}
+function nonEmptyText(value) {
+    return typeof value === "string" && value.length > 0 ? value : null;
+}
+export function resetTime(value) {
+    const seconds = finiteNumber(value);
+    if (seconds !== null) return new Date(seconds * 1000).toISOString();
+    if (typeof value === "string" && !Number.isNaN(Date.parse(value))) {
+        return new Date(value).toISOString();
+    }
+    return null;
+}
+export function safeIdentifier(value) {
+    const candidate = nonEmptyText(value);
+    if (!candidate || !/^[a-z][a-z0-9_-]{0,63}$/i.test(candidate)) return null;
+    if (/(?:account|token|secret|key)[_-]?\d{6,}/i.test(candidate)) return null;
+    return candidate;
+}
+export function toRateWindow(value, id, label) {
+    const input = record(value);
+    if (!input) return null;
+    const used = finiteNumber(input.used_percent);
+    const seconds = finiteNumber(input.limit_window_seconds);
+    return {
+        id,
+        label,
+        durationMinutes: seconds === null ? null : seconds / 60,
+        usedPercent: used,
+        resetsAt: resetTime(input.reset_at)
+    };
+}
+function extraWindows(value) {
+    if (!Array.isArray(value)) return [];
+    return value.flatMap((entry, index)=>{
+        const limit = record(entry);
+        if (!limit) return [];
+        const scope = safeIdentifier(limit.limit_name) ?? `extra-${index + 1}`;
+        const windows = record(limit.rate_limit) ?? limit;
+        return [
+            toRateWindow(windows.primary_window, `${scope}:primary`, `${scope} primary`),
+            toRateWindow(windows.secondary_window, `${scope}:secondary`, `${scope} secondary`)
+        ].filter((window)=>window !== null);
+    });
+}
+export function parseWhamUsage(raw) {
+    const response = record(raw) ?? {};
+    const limits = record(response.rate_limit) ?? {};
+    const credits = record(response.credits) ?? {};
+    return {
+        windows: [
+            toRateWindow(limits.primary_window, "session", "Session"),
+            toRateWindow(limits.secondary_window, "weekly", "Weekly"),
+            ...extraWindows(response.additional_rate_limits)
+        ].filter((window)=>window !== null),
+        creditsRemaining: finiteNumber(credits.balance)
+    };
+}
+export async function fetchOpenAiSubscriptionCapacity(credential, options = {}) {
+    let response;
+    try {
+        response = await (options.fetch ?? globalThis.fetch)(WHAM_USAGE_URL, {
+            method: "GET",
+            headers: {
+                Authorization: `Bearer ${credential.access}`,
+                "ChatGPT-Account-Id": credential.accountId
+            },
+            signal: AbortSignal.timeout(options.timeoutMs ?? 5000)
+        });
+    } catch  {
+        throw new Error("OpenAI usage request failed");
+    }
+    if (response.status === 401 || response.status === 403) {
+        return {
+            authenticated: false,
+            available: "unknown",
+            windows: [],
+            creditsRemaining: null
+        };
+    }
+    if (response.status !== 200) {
+        throw new Error(`OpenAI usage request failed: HTTP ${response.status}`);
+    }
+    let raw;
+    try {
+        raw = await response.json();
+    } catch  {
+        throw new Error("OpenAI usage response is not valid JSON");
+    }
+    const snapshot = parseWhamUsage(raw);
+    const hasUsage = snapshot.windows.some((window)=>window.usedPercent !== null);
+    return {
+        authenticated: true,
+        available: hasUsage ? "yes" : "unknown",
+        windows: snapshot.windows,
+        creditsRemaining: snapshot.creditsRemaining
+    };
+}
+export async function fetchOpenAiPatAccountId(token, options = {}) {
+    let response;
+    try {
+        response = await (options.fetch ?? globalThis.fetch)(OPENAI_WHOAMI_URL, {
+            headers: {
+                Authorization: `Bearer ${token}`
+            },
+            signal: AbortSignal.timeout(options.timeoutMs ?? 5000)
+        });
+    } catch  {
+        throw new Error("OpenAI account request failed");
+    }
+    if (!response.ok) throw new Error("OpenAI account request failed");
+    let raw;
+    try {
+        raw = await response.json();
+    } catch  {
+        throw new Error("OpenAI account response is not valid JSON");
+    }
+    const accountId = nonEmptyText(record(raw)?.chatgpt_account_id);
+    if (!accountId) throw new Error("OpenAI account unavailable");
+    return accountId;
+}
+
+//# sourceMappingURL=openai-subscription.js.map

package/dist/capacity/providers/openai-subscription.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/capacity/providers/openai-subscription.ts"],"sourcesContent":["import type { CapacityWindow, ProviderCapacitySnapshot } from \"../types.js\";\n\nconst WHAM_USAGE_URL = \"https://chatgpt.com/backend-api/wham/usage\";\nconst OPENAI_WHOAMI_URL = \"https://auth.openai.com/api/accounts/v1/user-auth-credential/whoami\";\n\nexport type OpenAiSubscriptionCredential = {\n  access: string;\n  accountId: string;\n};\n\nexport type OpenAiSubscriptionCapacityOptions = {\n  fetch?: typeof globalThis.fetch;\n  timeoutMs?: number;\n};\n\nexport type WhamUsageSnapshot = {\n  windows: CapacityWindow[];\n  creditsRemaining: number | null;\n};\n\ntype UnknownRecord = Record<string, unknown>;\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 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  }\n  return null;\n}\n\nexport function 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 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 parseWhamUsage(raw: unknown): WhamUsageSnapshot {\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  };\n}\n\nexport async function fetchOpenAiSubscriptionCapacity(\n  credential: OpenAiSubscriptionCredential,\n  options: OpenAiSubscriptionCapacityOptions = {},\n): Promise<ProviderCapacitySnapshot> {\n  let response: Response;\n  try {\n    response = await (options.fetch ?? globalThis.fetch)(WHAM_USAGE_URL, {\n      method: \"GET\",\n      headers: {\n        Authorization: `Bearer ${credential.access}`,\n        \"ChatGPT-Account-Id\": credential.accountId,\n      },\n      signal: AbortSignal.timeout(options.timeoutMs ?? 5000),\n    });\n  } catch {\n    throw new Error(\"OpenAI usage request failed\");\n  }\n  if (response.status === 401 || response.status === 403) {\n    return {\n      authenticated: false,\n      available: \"unknown\",\n      windows: [],\n      creditsRemaining: null,\n    };\n  }\n  if (response.status !== 200) {\n    throw new Error(`OpenAI usage request failed: HTTP ${response.status}`);\n  }\n  let raw: unknown;\n  try {\n    raw = await response.json();\n  } catch {\n    throw new Error(\"OpenAI usage response is not valid JSON\");\n  }\n  const snapshot = parseWhamUsage(raw);\n  const hasUsage = snapshot.windows.some((window) => window.usedPercent !== null);\n  return {\n    authenticated: true,\n    available: hasUsage ? \"yes\" : \"unknown\",\n    windows: snapshot.windows,\n    creditsRemaining: snapshot.creditsRemaining,\n  };\n}\n\nexport async function fetchOpenAiPatAccountId(\n  token: string,\n  options: OpenAiSubscriptionCapacityOptions = {},\n): Promise<string> {\n  let response: Response;\n  try {\n    response = await (options.fetch ?? globalThis.fetch)(OPENAI_WHOAMI_URL, {\n      headers: { Authorization: `Bearer ${token}` },\n      signal: AbortSignal.timeout(options.timeoutMs ?? 5000),\n    });\n  } catch {\n    throw new Error(\"OpenAI account request failed\");\n  }\n  if (!response.ok) throw new Error(\"OpenAI account request failed\");\n  let raw: unknown;\n  try {\n    raw = await response.json();\n  } catch {\n    throw new Error(\"OpenAI account response is not valid JSON\");\n  }\n  const accountId = nonEmptyText(record(raw)?.chatgpt_account_id);\n  if (!accountId) throw new Error(\"OpenAI account unavailable\");\n  return accountId;\n}\n"],"names":["WHAM_USAGE_URL","OPENAI_WHOAMI_URL","record","value","Array","isArray","finiteNumber","Number","isFinite","nonEmptyText","length","resetTime","seconds","Date","toISOString","isNaN","parse","safeIdentifier","candidate","test","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","parseWhamUsage","raw","response","limits","credits","additional_rate_limits","creditsRemaining","balance","fetchOpenAiSubscriptionCapacity","credential","options","fetch","globalThis","method","headers","Authorization","access","accountId","signal","AbortSignal","timeout","timeoutMs","Error","status","authenticated","available","json","snapshot","hasUsage","some","fetchOpenAiPatAccountId","token","ok","chatgpt_account_id"],"mappings":"AAEA,MAAMA,iBAAiB;AACvB,MAAMC,oBAAoB;AAmB1B,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,UAAUR,KAAc;IACtC,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,SAAS;QACjE,OAAO,IAAIU,KAAKV,OAAOW,WAAW;IACpC;IACA,OAAO;AACT;AAEA,OAAO,SAASG,eAAed,KAAc;IAC3C,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,aAAajB,KAAc,EAAEkB,EAAU,EAAEC,KAAa;IACpE,MAAMC,QAAQrB,OAAOC;IACrB,IAAI,CAACoB,OAAO,OAAO;IACnB,MAAMC,OAAOlB,aAAaiB,MAAME,YAAY;IAC5C,MAAMb,UAAUN,aAAaiB,MAAMG,oBAAoB;IACvD,OAAO;QACLL;QACAC;QACAK,iBAAiBf,YAAY,OAAO,OAAOA,UAAU;QACrDgB,aAAaJ;QACbK,UAAUlB,UAAUY,MAAMO,QAAQ;IACpC;AACF;AAEA,SAASC,aAAa5B,KAAc;IAClC,IAAI,CAACC,MAAMC,OAAO,CAACF,QAAQ,OAAO,EAAE;IACpC,OAAOA,MAAM6B,OAAO,CAAC,CAACC,OAAOC;QAC3B,MAAMC,QAAQjC,OAAO+B;QACrB,IAAI,CAACE,OAAO,OAAO,EAAE;QACrB,MAAMC,QAAQnB,eAAekB,MAAME,UAAU,KAAK,CAAC,MAAM,EAAEH,QAAQ,GAAG;QACtE,MAAMI,UAAUpC,OAAOiC,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,eAAeC,GAAY;IACzC,MAAMC,WAAW5C,OAAO2C,QAAQ,CAAC;IACjC,MAAME,SAAS7C,OAAO4C,SAASP,UAAU,KAAK,CAAC;IAC/C,MAAMS,UAAU9C,OAAO4C,SAASE,OAAO,KAAK,CAAC;IAC7C,OAAO;QACLV,SAAS;YACPlB,aAAa2B,OAAOP,cAAc,EAAE,WAAW;YAC/CpB,aAAa2B,OAAON,gBAAgB,EAAE,UAAU;eAC7CV,aAAae,SAASG,sBAAsB;SAChD,CAACP,MAAM,CAAC,CAACC,SAAqCA,WAAW;QAC1DO,kBAAkB5C,aAAa0C,QAAQG,OAAO;IAChD;AACF;AAEA,OAAO,eAAeC,gCACpBC,UAAwC,EACxCC,UAA6C,CAAC,CAAC;IAE/C,IAAIR;IACJ,IAAI;QACFA,WAAW,MAAM,AAACQ,CAAAA,QAAQC,KAAK,IAAIC,WAAWD,KAAK,AAAD,EAAGvD,gBAAgB;YACnEyD,QAAQ;YACRC,SAAS;gBACPC,eAAe,CAAC,OAAO,EAAEN,WAAWO,MAAM,EAAE;gBAC5C,sBAAsBP,WAAWQ,SAAS;YAC5C;YACAC,QAAQC,YAAYC,OAAO,CAACV,QAAQW,SAAS,IAAI;QACnD;IACF,EAAE,OAAM;QACN,MAAM,IAAIC,MAAM;IAClB;IACA,IAAIpB,SAASqB,MAAM,KAAK,OAAOrB,SAASqB,MAAM,KAAK,KAAK;QACtD,OAAO;YACLC,eAAe;YACfC,WAAW;YACX/B,SAAS,EAAE;YACXY,kBAAkB;QACpB;IACF;IACA,IAAIJ,SAASqB,MAAM,KAAK,KAAK;QAC3B,MAAM,IAAID,MAAM,CAAC,kCAAkC,EAAEpB,SAASqB,MAAM,EAAE;IACxE;IACA,IAAItB;IACJ,IAAI;QACFA,MAAM,MAAMC,SAASwB,IAAI;IAC3B,EAAE,OAAM;QACN,MAAM,IAAIJ,MAAM;IAClB;IACA,MAAMK,WAAW3B,eAAeC;IAChC,MAAM2B,WAAWD,SAASjC,OAAO,CAACmC,IAAI,CAAC,CAAC9B,SAAWA,OAAOf,WAAW,KAAK;IAC1E,OAAO;QACLwC,eAAe;QACfC,WAAWG,WAAW,QAAQ;QAC9BlC,SAASiC,SAASjC,OAAO;QACzBY,kBAAkBqB,SAASrB,gBAAgB;IAC7C;AACF;AAEA,OAAO,eAAewB,wBACpBC,KAAa,EACbrB,UAA6C,CAAC,CAAC;IAE/C,IAAIR;IACJ,IAAI;QACFA,WAAW,MAAM,AAACQ,CAAAA,QAAQC,KAAK,IAAIC,WAAWD,KAAK,AAAD,EAAGtD,mBAAmB;YACtEyD,SAAS;gBAAEC,eAAe,CAAC,OAAO,EAAEgB,OAAO;YAAC;YAC5Cb,QAAQC,YAAYC,OAAO,CAACV,QAAQW,SAAS,IAAI;QACnD;IACF,EAAE,OAAM;QACN,MAAM,IAAIC,MAAM;IAClB;IACA,IAAI,CAACpB,SAAS8B,EAAE,EAAE,MAAM,IAAIV,MAAM;IAClC,IAAIrB;IACJ,IAAI;QACFA,MAAM,MAAMC,SAASwB,IAAI;IAC3B,EAAE,OAAM;QACN,MAAM,IAAIJ,MAAM;IAClB;IACA,MAAML,YAAYpD,aAAaP,OAAO2C,MAAMgC;IAC5C,IAAI,CAAChB,WAAW,MAAM,IAAIK,MAAM;IAChC,OAAOL;AACT"}

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

@@ -0,0 +1,8 @@
+import type { ProviderCapacitySnapshot } from "../types.js";
+export type ZaiRequestOptions = {
+    fetch?: typeof globalThis.fetch;
+    timeoutMs?: number;
+};
+export declare function parseZaiCapacity(raw: unknown): ProviderCapacitySnapshot;
+export declare function fetchZaiCapacity(key: string, options?: ZaiRequestOptions): Promise<ProviderCapacitySnapshot>;
+//# sourceMappingURL=zai.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"zai.d.ts","sourceRoot":"","sources":["../../../src/capacity/providers/zai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAIV,wBAAwB,EACzB,MAAM,aAAa,CAAC;AAgBrB,MAAM,MAAM,iBAAiB,GAAG;IAC9B,KAAK,CAAC,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAwFF,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,OAAO,GAAG,wBAAwB,CAevE;AAED,wBAAsB,gBAAgB,CACpC,GAAG,EAAE,MAAM,EACX,OAAO,GAAE,iBAAsB,GAC9B,OAAO,CAAC,wBAAwB,CAAC,CAsBnC"}

package/dist/capacity/providers/zai.js

@@ -0,0 +1,158 @@
+const ZAI_QUOTA_URL = "https://api.z.ai/api/monitor/usage/quota/limit";
+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 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 integer(value) {
+    return typeof value === "number" && Number.isInteger(value) ? value : null;
+}
+function isCapacityLimitType(value) {
+    return Object.hasOwn(LIMIT_METADATA, value);
+}
+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 || count <= 0 ? null : metadata.minutes * count;
+}
+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} · ${count} ${unitLabel}${count === 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";
+    return windows.some((window)=>window.usedPercent !== null && window.usedPercent < 100) ? "yes" : "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: Math.max(0, Math.min(100, usedPercent)),
+        resetsAt: resetTime(limit.nextResetTime),
+        total,
+        current,
+        remaining
+    };
+}
+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)) {
+        throw new Error("Invalid z.ai quota response");
+    }
+    const windows = data.limits.map(parseLimit).filter((window)=>window !== null);
+    return {
+        authenticated: true,
+        available: availability(windows),
+        windows,
+        creditsRemaining: null
+    };
+}
+export async function fetchZaiCapacity(key, options = {}) {
+    let response;
+    try {
+        response = await (options.fetch ?? globalThis.fetch)(ZAI_QUOTA_URL, {
+            method: "GET",
+            headers: {
+                Authorization: `Bearer ${key}`
+            },
+            signal: AbortSignal.timeout(options.timeoutMs ?? 5000)
+        });
+    } catch  {
+        throw new Error("z.ai quota request failed");
+    }
+    if (response.status !== 200) {
+        throw new Error(`z.ai quota request failed: HTTP ${response.status}`);
+    }
+    try {
+        return parseZaiCapacity(await response.json());
+    } catch (error) {
+        if (error instanceof SyntaxError) {
+            throw new Error("z.ai quota response is not valid JSON");
+        }
+        throw error;
+    }
+}
+
+//# 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/sources/codex.d.ts

@@ -0,0 +1,32 @@
+import type { CapacityReport, CapacityWindow } from "../types.js";
+import { type CodexCredentialOptions } from "../../harnesses/codex/credentials.js";
+type CodexUsageSource = "pat" | "oauth" | "cli";
+type UsageSnapshot = {
+    windows: CapacityWindow[];
+    creditsRemaining: number | null;
+    source: CodexUsageSource;
+};
+type UnknownRecord = Record<string, unknown>;
+type RpcMessage = {
+    id?: number;
+    method: string;
+    params?: UnknownRecord;
+};
+type CliResponses = {
+    rateLimits: unknown;
+    account: unknown;
+};
+type CodexRpc = (messages: RpcMessage[]) => Promise<CliResponses>;
+export type CodexProbeOptions = CodexCredentialOptions & {
+    installed: boolean;
+    checkedAt: string;
+    fetch?: typeof globalThis.fetch;
+    rpc?: CodexRpc;
+    timeoutMs?: number;
+};
+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 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/sources/codex.js

@@ -0,0 +1,227 @@
+import { spawn } from "node:child_process";
+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",
+    "-a",
+    "untrusted",
+    "app-server"
+];
+function record(value) {
+    return value !== null && typeof value === "object" && !Array.isArray(value) ? value : null;
+}
+function finiteNumber(value) {
+    return typeof value === "number" && Number.isFinite(value) ? value : null;
+}
+function nonEmptyText(value) {
+    return typeof value === "string" && value.length > 0 ? value : null;
+}
+export function parseUsage(raw, source) {
+    const { windows, creditsRemaining } = parseWhamUsage(raw);
+    return {
+        windows,
+        creditsRemaining,
+        source
+    };
+}
+function cliWindow(value, id, label) {
+    const input = record(value);
+    if (!input) return null;
+    return {
+        id,
+        label,
+        durationMinutes: finiteNumber(input.windowDurationMins),
+        usedPercent: finiteNumber(input.usedPercent),
+        resetsAt: resetTime(input.resetsAt)
+    };
+}
+function cliSnapshotWindows(value, fallbackId) {
+    const snapshot = record(value);
+    if (!snapshot) return [];
+    const scope = safeIdentifier(snapshot.limitId) ?? safeIdentifier(fallbackId) ?? "codex";
+    return [
+        cliWindow(snapshot.primary, `${scope}:primary`, `${scope} primary`),
+        cliWindow(snapshot.secondary, `${scope}:secondary`, `${scope} secondary`)
+    ].filter((item)=>item !== null);
+}
+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));
+        }
+    }
+    const unique = [
+        ...new Map(windows.map((window)=>[
+                window.id,
+                window
+            ])).values()
+    ];
+    return {
+        windows: unique,
+        creditsRemaining: null,
+        source: "cli"
+    };
+}
+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);
+    const resetCredits = record(record(raw)?.rateLimitResetCredits) ?? record(record(raw)?.usageLimitResetCredits);
+    return {
+        harness: "codex",
+        provider: "openai",
+        generatedAt: options.checkedAt,
+        authenticated: true,
+        available: reached ? "no" : hasUsage ? "yes" : "unknown",
+        windows: snapshot.windows,
+        creditsRemaining: snapshot.creditsRemaining ?? finiteNumber(resetCredits?.availableCount)
+    };
+}
+function appServerRpc(messages, timeoutMs = 5000) {
+    return new Promise((resolve, reject)=>{
+        const child = spawn("codex", CODEX_APP_SERVER_ARGS, {
+            stdio: [
+                "pipe",
+                "pipe",
+                "ignore"
+            ]
+        });
+        const results = {};
+        let buffer = "";
+        let settled = false;
+        const finish = (error)=>{
+            if (settled) return;
+            settled = true;
+            clearTimeout(timer);
+            child.kill();
+            if (error) reject(error);
+            else resolve(results);
+        };
+        const timer = setTimeout(()=>finish(new Error("codex probe timed out")), timeoutMs);
+        child.once("error", ()=>finish(new Error("codex app-server unavailable")));
+        child.once("exit", ()=>{
+            if (!settled) finish(new Error("codex app-server exited"));
+        });
+        child.stdout.setEncoding("utf8");
+        child.stdout.on("data", (chunk)=>{
+            buffer += chunk;
+            for(;;){
+                const newline = buffer.indexOf("\n");
+                if (newline < 0) break;
+                const line = buffer.slice(0, newline).trim();
+                buffer = buffer.slice(newline + 1);
+                if (!line) continue;
+                let message;
+                try {
+                    message = JSON.parse(line);
+                } catch  {
+                    continue;
+                }
+                if (message.id === 1) {
+                    for (const request of messages.slice(1))child.stdin.write(`${JSON.stringify(request)}\n`);
+                } else if (message.id === 2) {
+                    if (message.error) finish(new Error("codex rate-limit method failed"));
+                    else results.rateLimits = message.result;
+                } else if (message.id === 3) {
+                    if (message.error) finish(new Error("codex account method failed"));
+                    else results.account = message.result;
+                }
+                if ("rateLimits" in results && "account" in results) finish();
+            }
+        });
+        child.stdin.write(`${JSON.stringify(messages[0])}\n`);
+    });
+}
+export function codexUnavailableReport(checkedAt) {
+    return {
+        harness: "codex",
+        provider: "openai",
+        generatedAt: checkedAt,
+        authenticated: null,
+        available: "unknown",
+        windows: [],
+        creditsRemaining: null
+    };
+}
+async function cliFallback(options) {
+    if (!options.installed) return codexUnavailableReport(options.checkedAt);
+    const messages = [
+        {
+            id: 1,
+            method: "initialize",
+            params: {
+                clientInfo: {
+                    name: "ai-devkit",
+                    title: null,
+                    version: "1"
+                },
+                capabilities: null
+            }
+        },
+        {
+            method: "initialized"
+        },
+        {
+            id: 2,
+            method: "account/rateLimits/read"
+        },
+        {
+            id: 3,
+            method: "account/read"
+        }
+    ];
+    try {
+        const rpc = options.rpc ?? ((requests)=>appServerRpc(requests, options.timeoutMs));
+        const response = await rpc(messages);
+        const result = capacityFromSnapshot(parseCliUsage(response.rateLimits), options, response.rateLimits);
+        const accountEnvelope = record(response.account);
+        if (accountEnvelope && Object.hasOwn(accountEnvelope, "account") && !record(accountEnvelope.account)) {
+            result.authenticated = false;
+            result.available = "unknown";
+        }
+        return result;
+    } catch  {
+        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) {
+    const credentials = await resolveCodexCredentials(options);
+    if (!credentials) return cliFallback(options);
+    if (credentials.personalAccessToken) {
+        try {
+            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.
+        }
+    }
+    if (credentials.oauth) {
+        try {
+            const snapshot = await subscriptionSnapshot(credentials.oauth.access, credentials.oauth.accountId, "oauth", options);
+            return capacityFromSnapshot(snapshot, options);
+        } catch  {
+            return cliFallback(options);
+        }
+    }
+    return cliFallback(options);
+}
+
+//# sourceMappingURL=codex.js.map