@pentoshi/clai 4.18.0 → 5.0.0

package/dist/llm/freebuff-fingerprint.js

@@ -0,0 +1,137 @@
+import { createHash, randomBytes } from "node:crypto";
+import { readFileSync } from "node:fs";
+import { arch as osArch, cpus as osCpus, hostname as osHostname, networkInterfaces as osNetworkInterfaces, platform as osPlatform, } from "node:os";
+import { join } from "node:path";
+import { normalizeFreebuffProcessName } from "./freebuff-env-profile.js";
+const DEFAULT_MACHINE_ID_PATHS = ["/etc/machine-id", "/var/lib/dbus/machine-id"];
+const DEFAULT_DMI_DIR = "/sys/class/dmi/id";
+const DEFAULT_CPUINFO_PATH = "/proc/cpuinfo";
+const DEFAULT_OS_RELEASE_PATH = "/etc/os-release";
+const ZERO_MAC = "00:00:00:00:00:00";
+export function createLegacyFreebuffFingerprintId() {
+    return `codebuff-cli-${randomBytes(6).toString("base64url").substring(0, 8)}`;
+}
+function readTextDefault(path) {
+    try {
+        return readFileSync(path, "utf8");
+    }
+    catch {
+        return null;
+    }
+}
+function machineIdOf(deps) {
+    const read = deps.readText ?? readTextDefault;
+    for (const path of deps.machineIdPaths ?? DEFAULT_MACHINE_ID_PATHS) {
+        const value = read(path)?.trim();
+        if (value && value !== "unknown" && value.length >= 8)
+            return value;
+    }
+    return null;
+}
+function dmiValue(deps, field) {
+    const read = deps.readText ?? readTextDefault;
+    return read(join(deps.dmiDir ?? DEFAULT_DMI_DIR, field))?.trim() ?? "";
+}
+function parseCpuInfo(text) {
+    let manufacturer = "";
+    let brand = "";
+    let physicalCores = 0;
+    for (const line of text.split("\n")) {
+        const separator = line.indexOf(":");
+        if (separator < 0)
+            continue;
+        const name = line.slice(0, separator).trim();
+        const value = line.slice(separator + 1).trim();
+        if (!value)
+            continue;
+        if (!manufacturer && (name === "vendor_id" || name === "CPU implementer"))
+            manufacturer = value;
+        if (!brand && (name === "model name" || name === "Processor" || name === "Hardware"))
+            brand = value;
+        if (!physicalCores && name === "cpu cores") {
+            const cores = Number(value);
+            if (Number.isSafeInteger(cores) && cores > 0)
+                physicalCores = cores;
+        }
+    }
+    return { manufacturer, brand, physicalCores };
+}
+function distroOf(deps) {
+    const read = deps.readText ?? readTextDefault;
+    const release = read(deps.osReleasePath ?? DEFAULT_OS_RELEASE_PATH);
+    if (!release)
+        return "";
+    const match = /^PRETTY_NAME=(.*)$/m.exec(release);
+    return match?.[1]?.trim().replace(/^"|"$/g, "") ?? "";
+}
+function networkSnapshot(deps) {
+    const interfaces = deps.networkInterfaces
+        ? deps.networkInterfaces()
+        : osNetworkInterfaces();
+    const macAddresses = Object.values(interfaces)
+        .flatMap((entries) => (entries ?? []).filter((entry) => entry && !entry.internal && entry.mac && entry.mac !== ZERO_MAC))
+        .map((entry) => entry.mac)
+        .sort();
+    return { macAddresses, interfaceCount: Object.keys(interfaces).length };
+}
+async function enhancedFreebuffFingerprintId(deps) {
+    try {
+        const machineId = machineIdOf(deps);
+        if (!machineId)
+            return null;
+        const env = deps.env ?? process.env;
+        const read = deps.readText ?? readTextDefault;
+        const cpu = parseCpuInfo(read(deps.cpuInfoPath ?? DEFAULT_CPUINFO_PATH) ?? "");
+        const platform = deps.platform ?? osPlatform();
+        const network = networkSnapshot(deps);
+        const profile = {
+            system: {
+                manufacturer: dmiValue(deps, "sys_vendor"),
+                model: dmiValue(deps, "product_name"),
+                serial: dmiValue(deps, "product_serial"),
+                uuid: dmiValue(deps, "product_uuid"),
+            },
+            cpu: {
+                manufacturer: cpu.manufacturer,
+                brand: cpu.brand,
+                cores: (deps.cpuCount ?? (() => osCpus().length))(),
+                physicalCores: cpu.physicalCores,
+            },
+            os: {
+                platform,
+                distro: distroOf(deps),
+                arch: deps.arch ?? osArch(),
+                hostname: (deps.hostname ?? osHostname)(),
+            },
+            runtime: {
+                nodeVersion: deps.nodeVersion ?? process.version,
+                platform,
+                arch: deps.arch ?? osArch(),
+                shell: normalizeFreebuffProcessName(env.SHELL),
+                cpuCount: (deps.cpuCount ?? (() => osCpus().length))(),
+            },
+            network,
+            machineId,
+            fingerprintVersion: "2.0",
+        };
+        const digest = createHash("sha256").update(JSON.stringify(profile)).digest("base64url");
+        return `enhanced-${digest}`;
+    }
+    catch {
+        return null;
+    }
+}
+let cachedFingerprint;
+async function computeFreebuffFingerprintId(deps) {
+    return (await enhancedFreebuffFingerprintId(deps)) ?? createLegacyFreebuffFingerprintId();
+}
+export function freebuffFingerprintId(deps = {}) {
+    if (!cachedFingerprint) {
+        cachedFingerprint = computeFreebuffFingerprintId(deps).catch(() => createLegacyFreebuffFingerprintId());
+    }
+    return cachedFingerprint;
+}
+export function resetFreebuffFingerprint() {
+    cachedFingerprint = undefined;
+}
+//# sourceMappingURL=freebuff-fingerprint.js.map

package/dist/llm/freebuff-fingerprint.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"freebuff-fingerprint.js","sourceRoot":"","sources":["../../src/llm/freebuff-fingerprint.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EACL,IAAI,IAAI,MAAM,EACd,IAAI,IAAI,MAAM,EACd,QAAQ,IAAI,UAAU,EACtB,iBAAiB,IAAI,mBAAmB,EACxC,QAAQ,IAAI,UAAU,GACvB,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,4BAA4B,EAAE,MAAM,2BAA2B,CAAC;AAsBzE,MAAM,wBAAwB,GAAG,CAAC,iBAAiB,EAAE,0BAA0B,CAAC,CAAC;AACjF,MAAM,eAAe,GAAG,mBAAmB,CAAC;AAC5C,MAAM,oBAAoB,GAAG,eAAe,CAAC;AAC7C,MAAM,uBAAuB,GAAG,iBAAiB,CAAC;AAClD,MAAM,QAAQ,GAAG,mBAAmB,CAAC;AAErC,MAAM,UAAU,iCAAiC;IAC/C,OAAO,gBAAgB,WAAW,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AAChF,CAAC;AAED,SAAS,eAAe,CAAC,IAAY;IACnC,IAAI,CAAC;QACH,OAAO,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACpC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAAC,IAA6B;IAChD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,IAAI,eAAe,CAAC;IAC9C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,cAAc,IAAI,wBAAwB,EAAE,CAAC;QACnE,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC;QACjC,IAAI,KAAK,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC;YAAE,OAAO,KAAK,CAAC;IACtE,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,QAAQ,CAAC,IAA6B,EAAE,KAAa;IAC5D,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,IAAI,eAAe,CAAC;IAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,eAAe,EAAE,KAAK,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;AACzE,CAAC;AAQD,SAAS,YAAY,CAAC,IAAY;IAChC,IAAI,YAAY,GAAG,EAAE,CAAC;IACtB,IAAI,KAAK,GAAG,EAAE,CAAC;IACf,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,SAAS,GAAG,CAAC;YAAE,SAAS;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC;QAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/C,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,iBAAiB,CAAC;YAAE,YAAY,GAAG,KAAK,CAAC;QAChG,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,KAAK,YAAY,IAAI,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,UAAU,CAAC;YAAE,KAAK,GAAG,KAAK,CAAC;QACpG,IAAI,CAAC,aAAa,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5B,IAAI,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC;gBAAE,aAAa,GAAG,KAAK,CAAC;QACtE,CAAC;IACH,CAAC;IACD,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;AAChD,CAAC;AAED,SAAS,QAAQ,CAAC,IAA6B;IAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,IAAI,eAAe,CAAC;IAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,IAAI,uBAAuB,CAAC,CAAC;IACpE,IAAI,CAAC,OAAO;QAAE,OAAO,EAAE,CAAC;IACxB,MAAM,KAAK,GAAG,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClD,OAAO,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;AACxD,CAAC;AAED,SAAS,eAAe,CAAC,IAA6B;IAIpD,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB;QACvC,CAAC,CAAC,IAAI,CAAC,iBAAiB,EAAE;QAC1B,CAAC,CAAE,mBAAmB,EAA2D,CAAC;IACpF,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC;SAC3C,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,GAAG,IAAI,KAAK,CAAC,GAAG,KAAK,QAAQ,CAAC,CAAC;SACxH,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,GAAa,CAAC;SACnC,IAAI,EAAE,CAAC;IACV,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,EAAE,CAAC;AAC1E,CAAC;AAED,KAAK,UAAU,6BAA6B,CAAC,IAA6B;IACxE,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,CAAC,SAAS;YAAE,OAAO,IAAI,CAAC;QAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,IAAI,eAAe,CAAC;QAC9C,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,oBAAoB,CAAC,IAAI,EAAE,CAAC,CAAC;QAC/E,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,UAAU,EAAE,CAAC;QAC/C,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;QACtC,MAAM,OAAO,GAAG;YACd,MAAM,EAAE;gBACN,YAAY,EAAE,QAAQ,CAAC,IAAI,EAAE,YAAY,CAAC;gBAC1C,KAAK,EAAE,QAAQ,CAAC,IAAI,EAAE,cAAc,CAAC;gBACrC,MAAM,EAAE,QAAQ,CAAC,IAAI,EAAE,gBAAgB,CAAC;gBACxC,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,cAAc,CAAC;aACrC;YACD,GAAG,EAAE;gBACH,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,KAAK,EAAE,GAAG,CAAC,KAAK;gBAChB,KAAK,EAAE,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC,EAAE;gBACnD,aAAa,EAAE,GAAG,CAAC,aAAa;aACjC;YACD,EAAE,EAAE;gBACF,QAAQ;gBACR,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC;gBACtB,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,MAAM,EAAE;gBAC3B,QAAQ,EAAE,CAAC,IAAI,CAAC,QAAQ,IAAI,UAAU,CAAC,EAAE;aAC1C;YACD,OAAO,EAAE;gBACP,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,OAAO,CAAC,OAAO;gBAChD,QAAQ;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,MAAM,EAAE;gBAC3B,KAAK,EAAE,4BAA4B,CAAC,GAAG,CAAC,KAAK,CAAC;gBAC9C,QAAQ,EAAE,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC,EAAE;aACvD;YACD,OAAO;YACP,SAAS;YACT,kBAAkB,EAAE,KAAK;SAC1B,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACxF,OAAO,YAAY,MAAM,EAAE,CAAC;IAC9B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,IAAI,iBAA8C,CAAC;AAEnD,KAAK,UAAU,4BAA4B,CAAC,IAA6B;IACvE,OAAO,CAAC,MAAM,6BAA6B,CAAC,IAAI,CAAC,CAAC,IAAI,iCAAiC,EAAE,CAAC;AAC5F,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,OAAgC,EAAE;IACtE,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,iBAAiB,GAAG,4BAA4B,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,iCAAiC,EAAE,CAAC,CAAC;IAC1G,CAAC;IACD,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,wBAAwB;IACtC,iBAAiB,GAAG,SAAS,CAAC;AAChC,CAAC"}

package/dist/llm/freebuff-models.d.ts

@@ -1,8 +1,29 @@
-import { type FreebuffSessionResponse } from "./freebuff-session-api.js";
-export declare const FREEBUFF_CATALOG_VERSION = 1;
+export declare const FREEBUFF_CATALOG_PROTOCOL = "1";
+export declare const FREEBUFF_CATALOG_PROTOCOL_HEADER = "x-freebuff-catalog-protocol";
+export declare const FREEBUFF_CATALOG_FETCH_HEADER = "x-freebuff-catalog-fetch";
+export declare const FREEBUFF_CATALOG_CLIENT = "cli";
 export declare const FREEBUFF_STATIC_MODEL_IDS: readonly string[];
-export declare function visibleFreebuffModelIds(response: FreebuffSessionResponse): string[];
+export interface FreebuffCatalogRow {
+    readonly key: string;
+    readonly handle: string;
+    readonly access: string | undefined;
+    readonly opensAt: number | undefined;
+}
+export interface FreebuffCatalogSnapshot {
+    readonly version: string | undefined;
+    readonly fetchId: string | undefined;
+    readonly rows: readonly FreebuffCatalogRow[];
+}
+export interface FreebuffCatalogDeps {
+    readonly fetch?: typeof fetch | undefined;
+    readonly baseUrl?: string | undefined;
+    readonly now?: (() => number) | undefined;
+}
+export declare function freebuffCatalogSnapshot(token: string, deps?: FreebuffCatalogDeps): FreebuffCatalogSnapshot | undefined;
+export declare function freebuffCatalogHandle(token: string, model: string, deps?: FreebuffCatalogDeps): string | undefined;
+export declare function refreshFreebuffCatalog(token: string, signal?: AbortSignal, deps?: FreebuffCatalogDeps): Promise<FreebuffCatalogSnapshot | null>;
+export declare function visibleFreebuffCatalogModelIds(snapshot: FreebuffCatalogSnapshot, now?: number): string[];
 export declare function mergeFreebuffModelIds(visible: readonly string[]): string[];
-export declare function resetFreebuffCatalogCache(): void;
-export declare function fetchFreebuffModelIds(token: string, signal?: AbortSignal): Promise<string[]>;
 export declare function freebuffStaticModelIds(): string[];
+export declare function resetFreebuffCatalogCache(): void;
+export declare function fetchFreebuffModelIds(token: string, signal?: AbortSignal, deps?: FreebuffCatalogDeps): Promise<string[]>;

package/dist/llm/freebuff-models.js

@@ -1,7 +1,19 @@
 import { createHash } from "node:crypto";
-import { callFreebuffSession } from "./freebuff-session-api.js";
-export const FREEBUFF_CATALOG_VERSION = 1;
+import { FREEBUFF_API_BASE_URL } from "./freebuff-auth.js";
+import { freebuffDeviceHeaders, freebuffForgetDeviceKeyRegistration, isFreebuffDeviceKeyError, } from "./freebuff-device-key.js";
+import { fallbackModelList, MODEL_LIST_TTL_MS, onModelListInvalidated } from "./model-list-cache.js";
+const CATALOG_PATH = "/api/v1/freebuff/models";
+export const FREEBUFF_CATALOG_PROTOCOL = "1";
+export const FREEBUFF_CATALOG_PROTOCOL_HEADER = "x-freebuff-catalog-protocol";
+export const FREEBUFF_CATALOG_FETCH_HEADER = "x-freebuff-catalog-fetch";
+export const FREEBUFF_CATALOG_CLIENT = "cli";
 const CACHE_TTL_MS = 30 * 60 * 1000;
+let refreshListing = false;
+onModelListInvalidated("freebuff", () => {
+    refreshListing = true;
+});
+const CATALOG_TIMEOUT_MS = 20_000;
+const MAX_CACHED_ACCOUNTS = 10;
 export const FREEBUFF_STATIC_MODEL_IDS = [
     "anthropic/claude-fable-5.1",
     "anthropic/claude-opus-4.8",
@@ -66,112 +78,166 @@ export const FREEBUFF_STATIC_MODEL_IDS = [
     "z-ai/glm-5.3-flashx",
     "z-ai/glm-5.3-prime",
 ];
-function collectKeys(value, into) {
-    if (!value || typeof value !== "object" || Array.isArray(value))
-        return;
-    for (const key of Object.keys(value)) {
-        if (key.trim())
-            into.add(key.trim());
-    }
-}
-function planRequiredIds(freebucks) {
-    const required = new Set();
-    const ids = freebucks?.planRequiredModelIds;
-    if (Array.isArray(ids)) {
-        for (const id of ids) {
-            if (typeof id === "string" && id.trim())
-                required.add(id.trim());
-        }
-    }
-    return required;
-}
-function zeroPriceIds(freebucks) {
-    const free = new Set();
-    const prices = freebucks?.prices;
-    if (!prices || typeof prices !== "object" || Array.isArray(prices))
-        return free;
-    for (const [id, price] of Object.entries(prices)) {
-        if (id.trim() && price === 0)
-            free.add(id.trim());
-    }
-    return free;
-}
-export function visibleFreebuffModelIds(response) {
-    const record = response;
-    const blocked = planRequiredIds(record.freebucks);
-    const ids = new Set();
-    if (typeof record.model === "string" && record.model.trim())
-        ids.add(record.model.trim());
-    collectKeys(record.rateLimitsByModel, ids);
-    for (const id of zeroPriceIds(record.freebucks))
-        ids.add(id);
-    for (const id of blocked)
-        ids.delete(id);
-    return [...ids];
-}
-export function mergeFreebuffModelIds(visible) {
-    const merged = new Set(FREEBUFF_STATIC_MODEL_IDS);
-    for (const id of visible) {
-        const trimmed = id.trim();
-        if (trimmed)
-            merged.add(trimmed);
-    }
-    return [...merged].sort((a, b) => a.localeCompare(b));
-}
 const cached = new Map();
 const requestsInFlight = new Map();
-const MAX_CACHED_ACCOUNTS = 10;
-export function resetFreebuffCatalogCache() {
-    cached.clear();
-    requestsInFlight.clear();
-}
 function tokenCacheKey(token) {
     return createHash("sha256").update(token).digest("hex");
 }
-function cacheCatalog(tokenHash, modelIds, fetchedAt) {
+function catalogUrl(base) {
+    return `${base.replace(/\/$/, "")}${CATALOG_PATH}`;
+}
+function recordOf(value) {
+    return value !== null && typeof value === "object" && !Array.isArray(value)
+        ? value
+        : undefined;
+}
+function parseSnapshot(payload) {
+    const record = recordOf(payload);
+    if (!record || record.protocol !== 1 || !Array.isArray(record.rows))
+        return null;
+    const rows = [];
+    for (const raw of record.rows) {
+        const row = recordOf(raw);
+        const key = row?.key;
+        const handle = row?.handle;
+        if (typeof key !== "string" || !key.trim() || typeof handle !== "string" || !handle.trim())
+            continue;
+        rows.push({
+            key: key.trim(),
+            handle: handle.trim(),
+            access: typeof row?.access === "string" ? row.access : undefined,
+            opensAt: typeof row?.opensAt === "number" && Number.isFinite(row.opensAt) ? row.opensAt : undefined,
+        });
+    }
+    return {
+        version: typeof record.version === "string" ? record.version : undefined,
+        fetchId: typeof record.fetchId === "string" ? record.fetchId : undefined,
+        rows,
+    };
+}
+function cacheSnapshot(tokenHash, snapshot, fetchedAt) {
     cached.delete(tokenHash);
-    cached.set(tokenHash, { tokenHash, modelIds, fetchedAt });
+    cached.set(tokenHash, { snapshot, fetchedAt });
     if (cached.size > MAX_CACHED_ACCOUNTS) {
         const oldest = cached.keys().next().value;
         if (oldest)
             cached.delete(oldest);
     }
 }
-export async function fetchFreebuffModelIds(token, signal) {
+function pruneStaleEntries(now) {
+    for (const [key, entry] of cached) {
+        if (now - entry.fetchedAt >= CACHE_TTL_MS)
+            cached.delete(key);
+    }
+}
+async function requestCatalog(token, deps, signal, register) {
+    const url = catalogUrl(deps.baseUrl ?? FREEBUFF_API_BASE_URL);
+    const timeout = AbortSignal.timeout(CATALOG_TIMEOUT_MS);
+    const combined = signal ? AbortSignal.any([signal, timeout]) : timeout;
+    const headers = {
+        authorization: `Bearer ${token}`,
+        [FREEBUFF_CATALOG_PROTOCOL_HEADER]: FREEBUFF_CATALOG_PROTOCOL,
+        "x-freebuff-client": FREEBUFF_CATALOG_CLIENT,
+        "user-agent": "Bun/1.3.14",
+        accept: "*/*",
+        "accept-encoding": "gzip, deflate, br",
+    };
+    const current = cached.get(tokenCacheKey(token))?.snapshot;
+    Object.assign(headers, await freebuffDeviceHeaders({
+        token,
+        method: "GET",
+        url,
+        ...(current?.fetchId !== undefined ? { fetchId: current.fetchId } : {}),
+        register,
+    }));
+    const response = await (deps.fetch ?? fetch)(url, { method: "GET", headers, signal: combined });
+    if (!response.ok) {
+        const payload = (await response.json().catch(() => null));
+        const error = (recordOf(payload)?.error);
+        if (typeof error === "string") {
+            if (isFreebuffDeviceKeyError(error))
+                freebuffForgetDeviceKeyRegistration(token, url);
+        }
+        return null;
+    }
+    return parseSnapshot(await response.json().catch(() => null));
+}
+async function loadCatalog(token, deps, signal, force) {
     const tokenHash = tokenCacheKey(token);
-    const now = Date.now();
+    const now = deps.now?.() ?? Date.now();
+    pruneStaleEntries(now);
     const previous = cached.get(tokenHash);
-    if (previous && now - previous.fetchedAt < CACHE_TTL_MS) {
-        cached.delete(tokenHash);
-        cached.set(tokenHash, previous);
-        return [...previous.modelIds];
-    }
-    const inFlight = requestsInFlight.get(tokenHash);
-    if (inFlight)
-        return [...(await inFlight)];
+    if (!force && previous && now - previous.fetchedAt < CACHE_TTL_MS)
+        return previous.snapshot;
+    const inflight = requestsInFlight.get(tokenHash);
+    if (inflight)
+        return inflight;
     const request = (async () => {
         try {
-            const response = await callFreebuffSession("GET", token, { signal });
-            const visible = visibleFreebuffModelIds(response).sort((a, b) => a.localeCompare(b));
-            const modelIds = visible.length > 0 ? visible : freebuffStaticModelIds();
-            cacheCatalog(tokenHash, modelIds, Date.now());
-            return modelIds;
+            const snapshot = await requestCatalog(token, deps, signal, Boolean(previous));
+            if (snapshot)
+                cacheSnapshot(tokenHash, snapshot, deps.now?.() ?? Date.now());
+            return snapshot ?? previous?.snapshot ?? null;
         }
         catch {
-            return previous ? [...previous.modelIds] : freebuffStaticModelIds();
+            return previous?.snapshot ?? null;
         }
     })();
     requestsInFlight.set(tokenHash, request);
     try {
-        return [...(await request)];
+        return await request;
     }
     finally {
-        if (requestsInFlight.get(tokenHash) === request) {
+        if (requestsInFlight.get(tokenHash) === request)
             requestsInFlight.delete(tokenHash);
-        }
     }
 }
+export function freebuffCatalogSnapshot(token, deps = {}) {
+    const now = deps.now?.() ?? Date.now();
+    pruneStaleEntries(now);
+    return cached.get(tokenCacheKey(token))?.snapshot;
+}
+export function freebuffCatalogHandle(token, model, deps = {}) {
+    const snapshot = freebuffCatalogSnapshot(token, deps);
+    return snapshot?.rows.find((row) => row.key === model)?.handle;
+}
+export function refreshFreebuffCatalog(token, signal, deps = {}) {
+    return loadCatalog(token, deps, signal, true);
+}
+export function visibleFreebuffCatalogModelIds(snapshot, now = Date.now()) {
+    const ids = new Set();
+    for (const row of snapshot.rows) {
+        if (row.access === "locked")
+            continue;
+        if (row.opensAt !== undefined && row.opensAt > now)
+            continue;
+        ids.add(row.key);
+    }
+    return [...ids];
+}
+export function mergeFreebuffModelIds(visible) {
+    const merged = new Set(FREEBUFF_STATIC_MODEL_IDS);
+    for (const id of visible) {
+        const trimmed = id.trim();
+        if (trimmed)
+            merged.add(trimmed);
+    }
+    return [...merged].sort((a, b) => a.localeCompare(b));
+}
 export function freebuffStaticModelIds() {
     return mergeFreebuffModelIds([]);
 }
+export function resetFreebuffCatalogCache() {
+    cached.clear();
+    requestsInFlight.clear();
+}
+export async function fetchFreebuffModelIds(token, signal, deps = {}) {
+    const previous = cached.get(tokenCacheKey(token));
+    const age = previous ? (deps.now?.() ?? Date.now()) - previous.fetchedAt : Number.POSITIVE_INFINITY;
+    const force = refreshListing || age >= MODEL_LIST_TTL_MS;
+    refreshListing = false;
+    const snapshot = await loadCatalog(token, deps, signal, force);
+    const ids = snapshot ? visibleFreebuffCatalogModelIds(snapshot).sort((a, b) => a.localeCompare(b)) : [];
+    return ids.length > 0 ? ids : fallbackModelList(freebuffStaticModelIds());
+}
 //# sourceMappingURL=freebuff-models.js.map

package/dist/llm/freebuff-models.js.map

@@ -1 +1 @@
-{"version":3,"file":"freebuff-models.js","sourceRoot":"","sources":["../../src/llm/freebuff-models.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAgC,MAAM,2BAA2B,CAAC;AAE9F,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC;AAE1C,MAAM,YAAY,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAEpC,MAAM,CAAC,MAAM,yBAAyB,GAAsB;IAC1D,4BAA4B;IAC5B,2BAA2B;IAC3B,yBAAyB;IACzB,2BAA2B;IAC3B,6BAA6B;IAC7B,2BAA2B;IAC3B,kBAAkB;IAClB,4BAA4B;IAC5B,0BAA0B;IAC1B,8BAA8B;IAC9B,+BAA+B;IAC/B,yBAAyB;IACzB,yBAAyB;IACzB,yBAAyB;IACzB,yBAAyB;IACzB,6BAA6B;IAC7B,iCAAiC;IACjC,iCAAiC;IACjC,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,oBAAoB;IACpB,0BAA0B;IAC1B,yBAAyB;IACzB,oBAAoB;IACpB,oBAAoB;IACpB,gBAAgB;IAChB,oBAAoB;IACpB,qBAAqB;IACrB,wBAAwB;IACxB,yBAAyB;IACzB,oBAAoB;IACpB,wBAAwB;IACxB,sBAAsB;IACtB,0BAA0B;IAC1B,oBAAoB;IACpB,wBAAwB;IACxB,mBAAmB;IACnB,uBAAuB;IACvB,kBAAkB;IAClB,sBAAsB;IACtB,eAAe;IACf,0BAA0B;IAC1B,mBAAmB;IACnB,kBAAkB;IAClB,mBAAmB;IACnB,kBAAkB;IAClB,oBAAoB;IACpB,uBAAuB;IACvB,wBAAwB;IACxB,kBAAkB;IAClB,2BAA2B;IAC3B,gBAAgB;IAChB,eAAe;IACf,eAAe;IACf,eAAe;IACf,kBAAkB;IAClB,cAAc;IACd,cAAc;IACd,oBAAoB;IACpB,qBAAqB;IACrB,oBAAoB;CACrB,CAAC;AAEF,SAAS,WAAW,CAAC,KAAc,EAAE,IAAiB;IACpD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO;IACxE,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,KAAgC,CAAC,EAAE,CAAC;QAChE,IAAI,GAAG,CAAC,IAAI,EAAE;YAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IACvC,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,SAAkB;IACzC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IACnC,MAAM,GAAG,GAAI,SAAmE,EAAE,oBAAoB,CAAC;IACvG,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACvB,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;YACrB,IAAI,OAAO,EAAE,KAAK,QAAQ,IAAI,EAAE,CAAC,IAAI,EAAE;gBAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,YAAY,CAAC,SAAkB;IACtC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,MAAM,GAAI,SAAqD,EAAE,MAAM,CAAC;IAC9E,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,IAAI,CAAC;IAChF,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAiC,CAAC,EAAE,CAAC;QAC5E,IAAI,EAAE,CAAC,IAAI,EAAE,IAAI,KAAK,KAAK,CAAC;YAAE,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,QAAiC;IACvE,MAAM,MAAM,GAAG,QAId,CAAC;IACF,MAAM,OAAO,GAAG,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAClD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAU,CAAC;IAC9B,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE;QAAE,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IAC1F,WAAW,CAAC,MAAM,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;IAC3C,KAAK,MAAM,EAAE,IAAI,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC;QAAE,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC7D,KAAK,MAAM,EAAE,IAAI,OAAO;QAAE,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACzC,OAAO,CAAC,GAAG,GAAG,CAAC,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,OAA0B;IAC9D,MAAM,MAAM,GAAG,IAAI,GAAG,CAAS,yBAAyB,CAAC,CAAC;IAC1D,KAAK,MAAM,EAAE,IAAI,OAAO,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;QAC1B,IAAI,OAAO;YAAE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;AACxD,CAAC;AAQD,MAAM,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;AAChD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAA6B,CAAC;AAC9D,MAAM,mBAAmB,GAAG,EAAE,CAAC;AAE/B,MAAM,UAAU,yBAAyB;IACvC,MAAM,CAAC,KAAK,EAAE,CAAC;IACf,gBAAgB,CAAC,KAAK,EAAE,CAAC;AAC3B,CAAC;AAED,SAAS,aAAa,CAAC,KAAa;IAClC,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC1D,CAAC;AAED,SAAS,YAAY,CAAC,SAAiB,EAAE,QAA2B,EAAE,SAAiB;IACrF,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACzB,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;IAC1D,IAAI,MAAM,CAAC,IAAI,GAAG,mBAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;QAC1C,IAAI,MAAM;YAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,KAAa,EACb,MAAoB;IAEpB,MAAM,SAAS,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;IACvC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACvC,IAAI,QAAQ,IAAI,GAAG,GAAG,QAAQ,CAAC,SAAS,GAAG,YAAY,EAAE,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACzB,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAChC,OAAO,CAAC,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IACD,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACjD,IAAI,QAAQ;QAAE,OAAO,CAAC,GAAG,CAAC,MAAM,QAAQ,CAAC,CAAC,CAAC;IAE3C,MAAM,OAAO,GAAG,CAAC,KAAK,IAAuB,EAAE;QAC7C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YACrE,MAAM,OAAO,GAAG,uBAAuB,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;YACrF,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,sBAAsB,EAAE,CAAC;YACzE,YAAY,CAAC,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;YAC9C,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,sBAAsB,EAAE,CAAC;QACtE,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IACL,gBAAgB,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IACzC,IAAI,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,MAAM,OAAO,CAAC,CAAC,CAAC;IAC9B,CAAC;YAAS,CAAC;QACT,IAAI,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,OAAO,EAAE,CAAC;YAChD,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,sBAAsB;IACpC,OAAO,qBAAqB,CAAC,EAAE,CAAC,CAAC;AACnC,CAAC"}
+{"version":3,"file":"freebuff-models.js","sourceRoot":"","sources":["../../src/llm/freebuff-models.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EACL,qBAAqB,EACrB,mCAAmC,EACnC,wBAAwB,GACzB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAErG,MAAM,YAAY,GAAG,yBAAyB,CAAC;AAC/C,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAG,CAAC;AAC7C,MAAM,CAAC,MAAM,gCAAgC,GAAG,6BAA6B,CAAC;AAC9E,MAAM,CAAC,MAAM,6BAA6B,GAAG,0BAA0B,CAAC;AACxE,MAAM,CAAC,MAAM,uBAAuB,GAAG,KAAK,CAAC;AAE7C,MAAM,YAAY,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AACpC,IAAI,cAAc,GAAG,KAAK,CAAC;AAE3B,sBAAsB,CAAC,UAAU,EAAE,GAAG,EAAE;IACtC,cAAc,GAAG,IAAI,CAAC;AACxB,CAAC,CAAC,CAAC;AACH,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAClC,MAAM,mBAAmB,GAAG,EAAE,CAAC;AAE/B,MAAM,CAAC,MAAM,yBAAyB,GAAsB;IAC1D,4BAA4B;IAC5B,2BAA2B;IAC3B,yBAAyB;IACzB,2BAA2B;IAC3B,6BAA6B;IAC7B,2BAA2B;IAC3B,kBAAkB;IAClB,4BAA4B;IAC5B,0BAA0B;IAC1B,8BAA8B;IAC9B,+BAA+B;IAC/B,yBAAyB;IACzB,yBAAyB;IACzB,yBAAyB;IACzB,yBAAyB;IACzB,6BAA6B;IAC7B,iCAAiC;IACjC,iCAAiC;IACjC,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,oBAAoB;IACpB,0BAA0B;IAC1B,yBAAyB;IACzB,oBAAoB;IACpB,oBAAoB;IACpB,gBAAgB;IAChB,oBAAoB;IACpB,qBAAqB;IACrB,wBAAwB;IACxB,yBAAyB;IACzB,oBAAoB;IACpB,wBAAwB;IACxB,sBAAsB;IACtB,0BAA0B;IAC1B,oBAAoB;IACpB,wBAAwB;IACxB,mBAAmB;IACnB,uBAAuB;IACvB,kBAAkB;IAClB,sBAAsB;IACtB,eAAe;IACf,0BAA0B;IAC1B,mBAAmB;IACnB,kBAAkB;IAClB,mBAAmB;IACnB,kBAAkB;IAClB,oBAAoB;IACpB,uBAAuB;IACvB,wBAAwB;IACxB,kBAAkB;IAClB,2BAA2B;IAC3B,gBAAgB;IAChB,eAAe;IACf,eAAe;IACf,eAAe;IACf,kBAAkB;IAClB,cAAc;IACd,cAAc;IACd,oBAAoB;IACpB,qBAAqB;IACrB,oBAAoB;CACrB,CAAC;AA0BF,MAAM,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;AAChD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAmD,CAAC;AAEpF,SAAS,aAAa,CAAC,KAAa;IAClC,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC1D,CAAC;AAED,SAAS,UAAU,CAAC,IAAY;IAC9B,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,YAAY,EAAE,CAAC;AACrD,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QACzE,CAAC,CAAE,KAAiC;QACpC,CAAC,CAAC,SAAS,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,OAAgB;IACrC,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;IACjC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACjF,MAAM,IAAI,GAAyB,EAAE,CAAC;IACtC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QAC9B,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;QAC1B,MAAM,GAAG,GAAG,GAAG,EAAE,GAAG,CAAC;QACrB,MAAM,MAAM,GAAG,GAAG,EAAE,MAAM,CAAC;QAC3B,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE;YAAE,SAAS;QACrG,IAAI,CAAC,IAAI,CAAC;YACR,GAAG,EAAE,GAAG,CAAC,IAAI,EAAE;YACf,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;YACrB,MAAM,EAAE,OAAO,GAAG,EAAE,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;YAChE,OAAO,EAAE,OAAO,GAAG,EAAE,OAAO,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;SACpG,CAAC,CAAC;IACL,CAAC;IACD,OAAO;QACL,OAAO,EAAE,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;QACxE,OAAO,EAAE,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;QACxE,IAAI;KACL,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,SAAiB,EAAE,QAAiC,EAAE,SAAiB;IAC5F,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACzB,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;IAC/C,IAAI,MAAM,CAAC,IAAI,GAAG,mBAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;QAC1C,IAAI,MAAM;YAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,GAAW;IACpC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;QAClC,IAAI,GAAG,GAAG,KAAK,CAAC,SAAS,IAAI,YAAY;YAAE,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAChE,CAAC;AACH,CAAC;AAED,KAAK,UAAU,cAAc,CAC3B,KAAa,EACb,IAAyB,EACzB,MAA+B,EAC/B,QAAiB;IAEjB,MAAM,GAAG,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,IAAI,qBAAqB,CAAC,CAAC;IAC9D,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACxD,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;IACvE,MAAM,OAAO,GAA2B;QACtC,aAAa,EAAE,UAAU,KAAK,EAAE;QAChC,CAAC,gCAAgC,CAAC,EAAE,yBAAyB;QAC7D,mBAAmB,EAAE,uBAAuB;QAC5C,YAAY,EAAE,YAAY;QAC1B,MAAM,EAAE,KAAK;QACb,iBAAiB,EAAE,mBAAmB;KACvC,CAAC;IACF,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC;IAC3D,MAAM,CAAC,MAAM,CACX,OAAO,EACP,MAAM,qBAAqB,CAAC;QAC1B,KAAK;QACL,MAAM,EAAE,KAAK;QACb,GAAG;QACH,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvE,QAAQ;KACT,CAAC,CACH,CAAC;IACF,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IAChG,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAY,CAAC;QACrE,MAAM,KAAK,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,KAAK,CAAC,CAAC;QACzC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,IAAI,wBAAwB,CAAC,KAAK,CAAC;gBAAE,mCAAmC,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QACvF,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,aAAa,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;AAChE,CAAC;AAED,KAAK,UAAU,WAAW,CACxB,KAAa,EACb,IAAyB,EACzB,MAA+B,EAC/B,KAAc;IAEd,MAAM,SAAS,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;IACvC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;IACvC,iBAAiB,CAAC,GAAG,CAAC,CAAC;IACvB,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACvC,IAAI,CAAC,KAAK,IAAI,QAAQ,IAAI,GAAG,GAAG,QAAQ,CAAC,SAAS,GAAG,YAAY;QAAE,OAAO,QAAQ,CAAC,QAAQ,CAAC;IAC5F,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACjD,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC9B,MAAM,OAAO,GAAG,CAAC,KAAK,IAA6C,EAAE;QACnE,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC9E,IAAI,QAAQ;gBAAE,aAAa,CAAC,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;YAC7E,OAAO,QAAQ,IAAI,QAAQ,EAAE,QAAQ,IAAI,IAAI,CAAC;QAChD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,QAAQ,EAAE,QAAQ,IAAI,IAAI,CAAC;QACpC,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IACL,gBAAgB,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IACzC,IAAI,CAAC;QACH,OAAO,MAAM,OAAO,CAAC;IACvB,CAAC;YAAS,CAAC;QACT,IAAI,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,OAAO;YAAE,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACtF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,KAAa,EACb,OAA4B,EAAE;IAE9B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;IACvC,iBAAiB,CAAC,GAAG,CAAC,CAAC;IACvB,OAAO,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,KAAa,EACb,KAAa,EACb,OAA4B,EAAE;IAE9B,MAAM,QAAQ,GAAG,uBAAuB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACtD,OAAO,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,KAAK,KAAK,CAAC,EAAE,MAAM,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,KAAa,EACb,MAAoB,EACpB,OAA4B,EAAE;IAE9B,OAAO,WAAW,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,8BAA8B,CAAC,QAAiC,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IAChG,MAAM,GAAG,GAAG,IAAI,GAAG,EAAU,CAAC;IAC9B,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;QAChC,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ;YAAE,SAAS;QACtC,IAAI,GAAG,CAAC,OAAO,KAAK,SAAS,IAAI,GAAG,CAAC,OAAO,GAAG,GAAG;YAAE,SAAS;QAC7D,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACnB,CAAC;IACD,OAAO,CAAC,GAAG,GAAG,CAAC,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,OAA0B;IAC9D,MAAM,MAAM,GAAG,IAAI,GAAG,CAAS,yBAAyB,CAAC,CAAC;IAC1D,KAAK,MAAM,EAAE,IAAI,OAAO,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;QAC1B,IAAI,OAAO;YAAE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,sBAAsB;IACpC,OAAO,qBAAqB,CAAC,EAAE,CAAC,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,yBAAyB;IACvC,MAAM,CAAC,KAAK,EAAE,CAAC;IACf,gBAAgB,CAAC,KAAK,EAAE,CAAC;AAC3B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,KAAa,EACb,MAAoB,EACpB,OAA4B,EAAE;IAE9B,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IAClD,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC;IACpG,MAAM,KAAK,GAAG,cAAc,IAAI,GAAG,IAAI,iBAAiB,CAAC;IACzD,cAAc,GAAG,KAAK,CAAC;IACvB,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IAC/D,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,8BAA8B,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACxG,OAAO,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,CAAC,CAAC;AAC5E,CAAC"}

package/dist/llm/freebuff-run.d.ts

@@ -1,4 +1,6 @@
+export declare const FREEBUFF_CATALOG_AGENT_ID = "base3-free-catalog";
 export declare function freebuffAgentId(model: string): string;
+export declare function freebuffAgentIdForToken(token: string, model: string): string;
 export interface FreebuffRun {
     readonly runId: string;
     readonly clientId: string;

package/dist/llm/freebuff-run.js

@@ -1,6 +1,8 @@
 import { randomUUID } from "node:crypto";
 import { ProviderError } from "./http.js";
 import { FREEBUFF_API_BASE_URL } from "./freebuff-auth.js";
+import { freebuffCatalogSnapshot } from "./freebuff-models.js";
+import { freebuffActingUserHeaders } from "./freebuff-wire.js";
 const RUN_PATH = "/api/v1/agent-runs";
 const RUN_TIMEOUT_MS = 30_000;
 const BASE3_AGENT_BY_MODEL = {
@@ -89,11 +91,15 @@ const BASE2_AGENT_BY_MODEL = {
     "z-ai/glm-5.3-flashx": "base2-free-glm-5-3-flashx",
     "z-ai/glm-5.3-prime": "base2-free-glm-5-3-prime",
 };
-const FALLBACK_AGENT_ID = "codebuff/base2-free";
+const FALLBACK_AGENT_ID = "base2-free";
+export const FREEBUFF_CATALOG_AGENT_ID = "base3-free-catalog";
 export function freebuffAgentId(model) {
     const id = model.trim();
-    const agent = BASE3_AGENT_BY_MODEL[id] ?? BASE2_AGENT_BY_MODEL[id];
-    return agent ? `codebuff/${agent}` : FALLBACK_AGENT_ID;
+    return BASE3_AGENT_BY_MODEL[id] ?? BASE2_AGENT_BY_MODEL[id] ?? FALLBACK_AGENT_ID;
+}
+export function freebuffAgentIdForToken(token, model) {
+    const row = freebuffCatalogSnapshot(token)?.rows.find((candidate) => candidate.key === model.trim());
+    return row ? FREEBUFF_CATALOG_AGENT_ID : freebuffAgentId(model);
 }
 function runUrl(base) {
     return `${base.replace(/\/$/, "")}${RUN_PATH}`;
@@ -102,31 +108,78 @@ function runSignal(signal) {
     const timeout = AbortSignal.timeout(RUN_TIMEOUT_MS);
     return signal ? AbortSignal.any([signal, timeout]) : timeout;
 }
-async function postRun(token, body, action, deps) {
-    const response = await (deps.fetch ?? fetch)(runUrl(deps.baseUrl ?? FREEBUFF_API_BASE_URL), {
-        method: "POST",
-        headers: {
-            authorization: `Bearer ${token}`,
-            "content-type": "application/json",
-        },
-        body: JSON.stringify(body),
-        signal: runSignal(deps.signal),
+const RUN_RETRIES = 3;
+const RUN_RETRY_BASE_MS = 1_000;
+const RUN_RETRY_CAP_MS = 8_000;
+const RETRYABLE_RUN_STATUSES = new Set([408, 429, 500, 502, 503, 504]);
+function runRetryDelay(attempt) {
+    return Math.min(RUN_RETRY_BASE_MS * 2 ** attempt, RUN_RETRY_CAP_MS);
+}
+function delay(ms, signal) {
+    return new Promise((resolve, reject) => {
+        if (signal?.aborted) {
+            reject(signal.reason ?? new DOMException("Aborted", "AbortError"));
+            return;
+        }
+        const timer = setTimeout(() => {
+            signal?.removeEventListener("abort", onAbort);
+            resolve();
+        }, ms);
+        function onAbort() {
+            clearTimeout(timer);
+            reject(signal?.reason ?? new DOMException("Aborted", "AbortError"));
+        }
+        signal?.addEventListener("abort", onAbort, { once: true });
     });
-    const text = await response.text().catch(() => "");
-    if (!response.ok) {
-        throw new ProviderError(`Freebuff ${action} failed (${response.status}${text ? `: ${text.slice(0, 300)}` : ""})`, response.status, text.slice(0, 2_000));
-    }
-    if (!text)
-        return undefined;
-    try {
-        return JSON.parse(text);
-    }
-    catch {
-        return undefined;
+}
+async function postRun(token, body, action, deps) {
+    const url = runUrl(deps.baseUrl ?? FREEBUFF_API_BASE_URL);
+    const requestBody = JSON.stringify(body);
+    let lastError;
+    for (let attempt = 0; attempt <= RUN_RETRIES; attempt += 1) {
+        let response;
+        try {
+            response = await (deps.fetch ?? fetch)(url, {
+                method: "POST",
+                headers: {
+                    authorization: `Bearer ${token}`,
+                    "content-type": "application/json",
+                    "user-agent": "Bun/1.3.14",
+                    accept: "*/*",
+                    "accept-encoding": "gzip, deflate, br",
+                    ...freebuffActingUserHeaders(),
+                },
+                body: requestBody,
+                signal: runSignal(deps.signal),
+            });
+        }
+        catch (error) {
+            lastError = error;
+            if (attempt === RUN_RETRIES || deps.signal?.aborted)
+                throw error;
+            await delay(runRetryDelay(attempt), deps.signal);
+            continue;
+        }
+        const text = await response.text().catch(() => "");
+        if (response.ok) {
+            if (!text)
+                return undefined;
+            try {
+                return JSON.parse(text);
+            }
+            catch {
+                return undefined;
+            }
+        }
+        lastError = new ProviderError(`Freebuff ${action} failed (${response.status}${text ? `: ${text.slice(0, 300)}` : ""})`, response.status, text.slice(0, 2_000));
+        if (!RETRYABLE_RUN_STATUSES.has(response.status) || attempt === RUN_RETRIES)
+            throw lastError;
+        await delay(runRetryDelay(attempt), deps.signal);
     }
+    throw lastError instanceof Error ? lastError : new ProviderError(`Freebuff ${action} failed after retries.`);
 }
 export async function startFreebuffRun(token, model, deps = {}) {
-    const agentId = freebuffAgentId(model);
+    const agentId = freebuffAgentIdForToken(token, model);
     const payload = await postRun(token, { action: "START", agentId, ancestorRunIds: [] }, "run start", deps);
     const runId = payload?.["runId"];
     if (typeof runId !== "string" || !runId) {