@caupulican/pi-adaptative 0.81.38 → 0.81.40

package/dist/core/models/fitness-store.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fitness-store.d.ts","sourceRoot":"","sources":["../../../src/core/models/fitness-store.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAEvE;;;;;;GAMG;AAEH,MAAM,WAAW,eAAe;IAC/B,8DAA8D;IAC9D,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;CACnB;AAED,wBAAgB,sBAAsB,IAAI,eAAe,CAWxD;AAED,MAAM,WAAW,mBAAmB;IACnC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,kBAAkB,CAAC;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,eAAe,CAAC;CACtB;AAQD,qBAAa,YAAY;IACxB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAwB;IAEpD,YAAY,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,MAAM,eAAe,CAAA;KAAE,EAG9E;IAED,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,MAAM,eAAe,CAAA;KAAE,GAAG,YAAY,CAEpG;IAED,OAAO,CAAC,IAAI;IAaZ,iGAAiG;IACjG,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,kBAAkB,EAAE,EAAE,CAAC,EAAE,MAAM,GAAG,mBAAmB,CAQhF;IAED,yFAAyF;IACzF,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAO1B;IAED,qEAAqE;IACrE,UAAU,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,mBAAmB,EAAE,CAGjD;IAED,4EAA4E;IAC5E,MAAM,IAAI,mBAAmB,EAAE,CAG9B;CACD","sourcesContent":["import { existsSync, mkdirSync, readFileSync, writeFileSync } from \"node:fs\";\nimport { cpus, totalmem } from \"node:os\";\nimport { dirname, join } from \"node:path\";\nimport type { ModelFitnessReport } from \"../research/model-fitness.ts\";\n\n/**\n * Durable, HOST-KEYED storage for model fitness reports. Fitness is a property of a model ON a\n * host (tok/s and latency-driven failures do not travel between machines), so reports are keyed\n * by a hardware fingerprint: the same model can be \"the heavy lifter\" on one machine and\n * \"waiting for better hardware\" on another, and pi remembers both without confusing them —\n * including when settings/dotfiles are synced across machines.\n */\n\nexport interface HostFingerprint {\n\t/** Stable, human-readable id derived from the specs below. */\n\tid: string;\n\tcpu: string;\n\tcores: number;\n\ttotalMemGb: number;\n}\n\nexport function currentHostFingerprint(): HostFingerprint {\n\tconst cpuList = cpus();\n\tconst cpu = (cpuList[0]?.model ?? \"unknown-cpu\").trim();\n\tconst cores = Math.max(1, cpuList.length);\n\tconst totalMemGb = Math.round(totalmem() / 1024 ** 3);\n\tconst id = `${cpu\n\t\t.toLowerCase()\n\t\t.replace(/[^a-z0-9]+/g, \"-\")\n\t\t.replace(/^-+|-+$/g, \"\")\n\t\t.slice(0, 48)}-${cores}c-${totalMemGb}g`;\n\treturn { id, cpu, cores, totalMemGb };\n}\n\nexport interface StoredFitnessReport {\n\tmodel: string;\n\treport: ModelFitnessReport;\n\tat: string;\n\thost: HostFingerprint;\n}\n\ninterface FitnessStoreFile {\n\tversion: 1;\n\t/** hostId -> modelRef -> latest stored report. */\n\thosts: Record<string, Record<string, StoredFitnessReport>>;\n}\n\nexport class FitnessStore {\n\tprivate readonly filePath: string;\n\tprivate readonly fingerprint: () => HostFingerprint;\n\n\tconstructor(filePath: string, options?: { fingerprint?: () => HostFingerprint }) {\n\t\tthis.filePath = filePath;\n\t\tthis.fingerprint = options?.fingerprint ?? currentHostFingerprint;\n\t}\n\n\tstatic forAgentDir(agentDir: string, options?: { fingerprint?: () => HostFingerprint }): FitnessStore {\n\t\treturn new FitnessStore(join(agentDir, \"state\", \"model-fitness.json\"), options);\n\t}\n\n\tprivate load(): FitnessStoreFile {\n\t\ttry {\n\t\t\tif (!existsSync(this.filePath)) return { version: 1, hosts: {} };\n\t\t\tconst parsed = JSON.parse(readFileSync(this.filePath, \"utf-8\")) as FitnessStoreFile;\n\t\t\tif (parsed && parsed.version === 1 && parsed.hosts && typeof parsed.hosts === \"object\") {\n\t\t\t\treturn parsed;\n\t\t\t}\n\t\t} catch {\n\t\t\t// Unreadable/corrupt store: start fresh in memory; the next save rewrites the file.\n\t\t}\n\t\treturn { version: 1, hosts: {} };\n\t}\n\n\t/** Persist the latest report for a model on the CURRENT host. Best-effort, returns the entry. */\n\tsave(model: string, report: ModelFitnessReport, at?: string): StoredFitnessReport {\n\t\tconst host = this.fingerprint();\n\t\tconst entry: StoredFitnessReport = { model, report, at: at ?? new Date().toISOString(), host };\n\t\tconst file = this.load();\n\t\tfile.hosts[host.id] = { ...(file.hosts[host.id] ?? {}), [model]: entry };\n\t\tmkdirSync(dirname(this.filePath), { recursive: true });\n\t\twriteFileSync(this.filePath, `${JSON.stringify(file, null, \"\\t\")}\\n`, \"utf-8\");\n\t\treturn entry;\n\t}\n\n\t/** Drop a model's report for the CURRENT host (uninstall cleanup). No-op when absent. */\n\tremove(model: string): void {\n\t\tconst host = this.fingerprint();\n\t\tconst file = this.load();\n\t\tif (!file.hosts[host.id]?.[model]) return;\n\t\tdelete file.hosts[host.id][model];\n\t\tmkdirSync(dirname(this.filePath), { recursive: true });\n\t\twriteFileSync(this.filePath, `${JSON.stringify(file, null, \"\\t\")}\\n`, \"utf-8\");\n\t}\n\n\t/** Reports for the current host (default) or an explicit host id. */\n\tgetForHost(hostId?: string): StoredFitnessReport[] {\n\t\tconst file = this.load();\n\t\treturn Object.values(file.hosts[hostId ?? this.fingerprint().id] ?? {});\n\t}\n\n\t/** Every stored report across all hosts (for cross-machine comparisons). */\n\tgetAll(): StoredFitnessReport[] {\n\t\tconst file = this.load();\n\t\treturn Object.values(file.hosts).flatMap((models) => Object.values(models));\n\t}\n}\n"]}
+{"version":3,"file":"fitness-store.d.ts","sourceRoot":"","sources":["../../../src/core/models/fitness-store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAGvE,OAAO,EAAE,KAAK,eAAe,EAAqC,MAAM,uBAAuB,CAAC;AAEhG;;;;;;GAMG;AAEH,MAAM,WAAW,mBAAmB;IACnC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,kBAAkB,CAAC;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,eAAe,CAAC;CACtB;AAyED,qBAAa,YAAY;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAsD;IAE9E,YAAY,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,MAAM,eAAe,CAAC;QAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,EAQlG;IAED,MAAM,CAAC,WAAW,CACjB,QAAQ,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,MAAM,eAAe,CAAC;QAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GACnE,YAAY,CAEd;IAED,iGAAiG;IACjG,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,kBAAkB,EAAE,EAAE,CAAC,EAAE,MAAM,GAAG,mBAAmB,CAShF;IAED,yFAAyF;IACzF,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAS1B;IAED,qEAAqE;IACrE,UAAU,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,mBAAmB,EAAE,CAEjD;IAED,4EAA4E;IAC5E,MAAM,IAAI,mBAAmB,EAAE,CAE9B;CACD","sourcesContent":["import { stateFile } from \"../agent-paths.ts\";\nimport type { ModelFitnessReport } from \"../research/model-fitness.ts\";\nimport { isWorkerSession } from \"../session-role.ts\";\nimport { isPlainRecord } from \"../util/value-guards.ts\";\nimport { type HostFingerprint, HostStateStore, isHostFingerprint } from \"./host-state-store.ts\";\n\n/**\n * Durable, HOST-KEYED storage for model fitness reports. Fitness is a property of a model ON a\n * host (tok/s and latency-driven failures do not travel between machines), so reports are keyed\n * by a hardware fingerprint: the same model can be \"the heavy lifter\" on one machine and\n * \"waiting for better hardware\" on another, and pi remembers both without confusing them —\n * including when settings/dotfiles are synced across machines.\n */\n\nexport interface StoredFitnessReport {\n\tmodel: string;\n\treport: ModelFitnessReport;\n\tat: string;\n\thost: HostFingerprint;\n}\n\nfunction isFiniteNumber(value: unknown): value is number {\n\treturn typeof value === \"number\" && Number.isFinite(value);\n}\n\nfunction isLaneScore(value: unknown): boolean {\n\treturn (\n\t\tisPlainRecord(value) &&\n\t\tisFiniteNumber(value.succeeded) &&\n\t\tisFiniteNumber(value.total) &&\n\t\tArray.isArray(value.outcomes) &&\n\t\tvalue.outcomes.every((outcome) => typeof outcome === \"string\") &&\n\t\tisFiniteNumber(value.meanMs) &&\n\t\t(value.tokensPerSecond === undefined || isFiniteNumber(value.tokensPerSecond))\n\t);\n}\n\nfunction isFitnessReport(value: unknown): value is ModelFitnessReport {\n\treturn (\n\t\tisPlainRecord(value) &&\n\t\tisFiniteNumber(value.trials) &&\n\t\t(value.tokensPerSecond === undefined || isFiniteNumber(value.tokensPerSecond)) &&\n\t\tisLaneScore(value.research) &&\n\t\tisLaneScore(value.worker) &&\n\t\tisLaneScore(value.search) &&\n\t\tisLaneScore(value.toolCall) &&\n\t\tisLaneScore(value.digest) &&\n\t\tisPlainRecord(value.judge) &&\n\t\tisFiniteNumber(value.judge.parsed) &&\n\t\tisFiniteNumber(value.judge.planningElevated) &&\n\t\tisFiniteNumber(value.judge.planningTotal) &&\n\t\tisFiniteNumber(value.judge.trivialCheap) &&\n\t\tisFiniteNumber(value.judge.trivialTotal) &&\n\t\tisFiniteNumber(value.judge.total) &&\n\t\tArray.isArray(value.judge.outcomes) &&\n\t\tvalue.judge.outcomes.every((outcome) => typeof outcome === \"string\") &&\n\t\tisFiniteNumber(value.judge.meanMs) &&\n\t\t(value.judge.tokensPerSecond === undefined || isFiniteNumber(value.judge.tokensPerSecond)) &&\n\t\t(value.capacity === undefined ||\n\t\t\t(isPlainRecord(value.capacity) &&\n\t\t\t\tisFiniteNumber(value.capacity.registeredContextWindow) &&\n\t\t\t\tisFiniteNumber(value.capacity.servedContextWindow) &&\n\t\t\t\tArray.isArray(value.capacity.outcomes) &&\n\t\t\t\tvalue.capacity.outcomes.every((outcome) => typeof outcome === \"string\") &&\n\t\t\t\tisFiniteNumber(value.capacity.meanMs))) &&\n\t\tisFiniteNumber(value.totalCostUsd)\n\t);\n}\n\nfunction parseFitnessHost(value: unknown, hostId: string): Record<string, StoredFitnessReport> | undefined {\n\tif (!isPlainRecord(value)) return undefined;\n\tconst reports: Record<string, StoredFitnessReport> = {};\n\tfor (const [model, candidate] of Object.entries(value)) {\n\t\tif (\n\t\t\t!isPlainRecord(candidate) ||\n\t\t\tcandidate.model !== model ||\n\t\t\ttypeof candidate.at !== \"string\" ||\n\t\t\t!isHostFingerprint(candidate.host) ||\n\t\t\tcandidate.host.id !== hostId ||\n\t\t\t!isFitnessReport(candidate.report)\n\t\t)\n\t\t\tcontinue;\n\t\treports[model] = {\n\t\t\tmodel,\n\t\t\treport: candidate.report,\n\t\t\tat: candidate.at,\n\t\t\thost: candidate.host,\n\t\t};\n\t}\n\treturn reports;\n}\n\nexport class FitnessStore {\n\tprivate readonly storage: HostStateStore<Record<string, StoredFitnessReport>>;\n\n\tconstructor(filePath: string, options?: { fingerprint?: () => HostFingerprint; readOnly?: boolean }) {\n\t\tthis.storage = new HostStateStore({\n\t\t\tfilePath,\n\t\t\tversion: 1,\n\t\t\tfingerprint: options?.fingerprint,\n\t\t\treadOnly: options?.readOnly ?? isWorkerSession(),\n\t\t\tparseHost: parseFitnessHost,\n\t\t});\n\t}\n\n\tstatic forAgentDir(\n\t\tagentDir: string,\n\t\toptions?: { fingerprint?: () => HostFingerprint; readOnly?: boolean },\n\t): FitnessStore {\n\t\treturn new FitnessStore(stateFile(agentDir, \"model-fitness.json\"), options);\n\t}\n\n\t/** Persist the latest report for a model on the CURRENT host. Best-effort, returns the entry. */\n\tsave(model: string, report: ModelFitnessReport, at?: string): StoredFitnessReport {\n\t\treturn this.storage.mutateCurrentHost(\n\t\t\t() => ({}),\n\t\t\t(reports, host) => {\n\t\t\t\tconst entry: StoredFitnessReport = { model, report, at: at ?? new Date().toISOString(), host };\n\t\t\t\treports[model] = entry;\n\t\t\t\treturn { result: entry, changed: true };\n\t\t\t},\n\t\t);\n\t}\n\n\t/** Drop a model's report for the CURRENT host (uninstall cleanup). No-op when absent. */\n\tremove(model: string): void {\n\t\tthis.storage.mutateCurrentHost(\n\t\t\t() => ({}),\n\t\t\t(reports) => {\n\t\t\t\tif (!reports[model]) return { result: undefined, changed: false };\n\t\t\t\tdelete reports[model];\n\t\t\t\treturn { result: undefined, changed: true };\n\t\t\t},\n\t\t);\n\t}\n\n\t/** Reports for the current host (default) or an explicit host id. */\n\tgetForHost(hostId?: string): StoredFitnessReport[] {\n\t\treturn Object.values(this.storage.getHost(hostId) ?? {});\n\t}\n\n\t/** Every stored report across all hosts (for cross-machine comparisons). */\n\tgetAll(): StoredFitnessReport[] {\n\t\treturn this.storage.getAllHosts().flatMap((reports) => Object.values(reports));\n\t}\n}\n"]}

package/dist/core/models/fitness-store.js

@@ -1,71 +1,107 @@
-import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
-import { cpus, totalmem } from "node:os";
-import { dirname, join } from "node:path";
-export function currentHostFingerprint() {
-    const cpuList = cpus();
-    const cpu = (cpuList[0]?.model ?? "unknown-cpu").trim();
-    const cores = Math.max(1, cpuList.length);
-    const totalMemGb = Math.round(totalmem() / 1024 ** 3);
-    const id = `${cpu
-        .toLowerCase()
-        .replace(/[^a-z0-9]+/g, "-")
-        .replace(/^-+|-+$/g, "")
-        .slice(0, 48)}-${cores}c-${totalMemGb}g`;
-    return { id, cpu, cores, totalMemGb };
+import { stateFile } from "../agent-paths.js";
+import { isWorkerSession } from "../session-role.js";
+import { isPlainRecord } from "../util/value-guards.js";
+import { HostStateStore, isHostFingerprint } from "./host-state-store.js";
+function isFiniteNumber(value) {
+    return typeof value === "number" && Number.isFinite(value);
+}
+function isLaneScore(value) {
+    return (isPlainRecord(value) &&
+        isFiniteNumber(value.succeeded) &&
+        isFiniteNumber(value.total) &&
+        Array.isArray(value.outcomes) &&
+        value.outcomes.every((outcome) => typeof outcome === "string") &&
+        isFiniteNumber(value.meanMs) &&
+        (value.tokensPerSecond === undefined || isFiniteNumber(value.tokensPerSecond)));
+}
+function isFitnessReport(value) {
+    return (isPlainRecord(value) &&
+        isFiniteNumber(value.trials) &&
+        (value.tokensPerSecond === undefined || isFiniteNumber(value.tokensPerSecond)) &&
+        isLaneScore(value.research) &&
+        isLaneScore(value.worker) &&
+        isLaneScore(value.search) &&
+        isLaneScore(value.toolCall) &&
+        isLaneScore(value.digest) &&
+        isPlainRecord(value.judge) &&
+        isFiniteNumber(value.judge.parsed) &&
+        isFiniteNumber(value.judge.planningElevated) &&
+        isFiniteNumber(value.judge.planningTotal) &&
+        isFiniteNumber(value.judge.trivialCheap) &&
+        isFiniteNumber(value.judge.trivialTotal) &&
+        isFiniteNumber(value.judge.total) &&
+        Array.isArray(value.judge.outcomes) &&
+        value.judge.outcomes.every((outcome) => typeof outcome === "string") &&
+        isFiniteNumber(value.judge.meanMs) &&
+        (value.judge.tokensPerSecond === undefined || isFiniteNumber(value.judge.tokensPerSecond)) &&
+        (value.capacity === undefined ||
+            (isPlainRecord(value.capacity) &&
+                isFiniteNumber(value.capacity.registeredContextWindow) &&
+                isFiniteNumber(value.capacity.servedContextWindow) &&
+                Array.isArray(value.capacity.outcomes) &&
+                value.capacity.outcomes.every((outcome) => typeof outcome === "string") &&
+                isFiniteNumber(value.capacity.meanMs))) &&
+        isFiniteNumber(value.totalCostUsd));
+}
+function parseFitnessHost(value, hostId) {
+    if (!isPlainRecord(value))
+        return undefined;
+    const reports = {};
+    for (const [model, candidate] of Object.entries(value)) {
+        if (!isPlainRecord(candidate) ||
+            candidate.model !== model ||
+            typeof candidate.at !== "string" ||
+            !isHostFingerprint(candidate.host) ||
+            candidate.host.id !== hostId ||
+            !isFitnessReport(candidate.report))
+            continue;
+        reports[model] = {
+            model,
+            report: candidate.report,
+            at: candidate.at,
+            host: candidate.host,
+        };
+    }
+    return reports;
 }
 export class FitnessStore {
-    filePath;
-    fingerprint;
+    storage;
     constructor(filePath, options) {
-        this.filePath = filePath;
-        this.fingerprint = options?.fingerprint ?? currentHostFingerprint;
+        this.storage = new HostStateStore({
+            filePath,
+            version: 1,
+            fingerprint: options?.fingerprint,
+            readOnly: options?.readOnly ?? isWorkerSession(),
+            parseHost: parseFitnessHost,
+        });
     }
     static forAgentDir(agentDir, options) {
-        return new FitnessStore(join(agentDir, "state", "model-fitness.json"), options);
-    }
-    load() {
-        try {
-            if (!existsSync(this.filePath))
-                return { version: 1, hosts: {} };
-            const parsed = JSON.parse(readFileSync(this.filePath, "utf-8"));
-            if (parsed && parsed.version === 1 && parsed.hosts && typeof parsed.hosts === "object") {
-                return parsed;
-            }
-        }
-        catch {
-            // Unreadable/corrupt store: start fresh in memory; the next save rewrites the file.
-        }
-        return { version: 1, hosts: {} };
+        return new FitnessStore(stateFile(agentDir, "model-fitness.json"), options);
     }
     /** Persist the latest report for a model on the CURRENT host. Best-effort, returns the entry. */
     save(model, report, at) {
-        const host = this.fingerprint();
-        const entry = { model, report, at: at ?? new Date().toISOString(), host };
-        const file = this.load();
-        file.hosts[host.id] = { ...(file.hosts[host.id] ?? {}), [model]: entry };
-        mkdirSync(dirname(this.filePath), { recursive: true });
-        writeFileSync(this.filePath, `${JSON.stringify(file, null, "\t")}\n`, "utf-8");
-        return entry;
+        return this.storage.mutateCurrentHost(() => ({}), (reports, host) => {
+            const entry = { model, report, at: at ?? new Date().toISOString(), host };
+            reports[model] = entry;
+            return { result: entry, changed: true };
+        });
     }
     /** Drop a model's report for the CURRENT host (uninstall cleanup). No-op when absent. */
     remove(model) {
-        const host = this.fingerprint();
-        const file = this.load();
-        if (!file.hosts[host.id]?.[model])
-            return;
-        delete file.hosts[host.id][model];
-        mkdirSync(dirname(this.filePath), { recursive: true });
-        writeFileSync(this.filePath, `${JSON.stringify(file, null, "\t")}\n`, "utf-8");
+        this.storage.mutateCurrentHost(() => ({}), (reports) => {
+            if (!reports[model])
+                return { result: undefined, changed: false };
+            delete reports[model];
+            return { result: undefined, changed: true };
+        });
     }
     /** Reports for the current host (default) or an explicit host id. */
     getForHost(hostId) {
-        const file = this.load();
-        return Object.values(file.hosts[hostId ?? this.fingerprint().id] ?? {});
+        return Object.values(this.storage.getHost(hostId) ?? {});
     }
     /** Every stored report across all hosts (for cross-machine comparisons). */
     getAll() {
-        const file = this.load();
-        return Object.values(file.hosts).flatMap((models) => Object.values(models));
+        return this.storage.getAllHosts().flatMap((reports) => Object.values(reports));
     }
 }
 //# sourceMappingURL=fitness-store.js.map

package/dist/core/models/fitness-store.js.map

@@ -1 +1 @@
-{"version":3,"file":"fitness-store.js","sourceRoot":"","sources":["../../../src/core/models/fitness-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC7E,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAmB1C,MAAM,UAAU,sBAAsB,GAAoB;IACzD,MAAM,OAAO,GAAG,IAAI,EAAE,CAAC;IACvB,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,aAAa,CAAC,CAAC,IAAI,EAAE,CAAC;IACxD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC;IACtD,MAAM,EAAE,GAAG,GAAG,GAAG;SACf,WAAW,EAAE;SACb,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC;SAC3B,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SACvB,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,KAAK,UAAU,GAAG,CAAC;IAC1C,OAAO,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;AAAA,CACtC;AAeD,MAAM,OAAO,YAAY;IACP,QAAQ,CAAS;IACjB,WAAW,CAAwB;IAEpD,YAAY,QAAgB,EAAE,OAAiD,EAAE;QAChF,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,WAAW,GAAG,OAAO,EAAE,WAAW,IAAI,sBAAsB,CAAC;IAAA,CAClE;IAED,MAAM,CAAC,WAAW,CAAC,QAAgB,EAAE,OAAiD,EAAgB;QACrG,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,oBAAoB,CAAC,EAAE,OAAO,CAAC,CAAC;IAAA,CAChF;IAEO,IAAI,GAAqB;QAChC,IAAI,CAAC;YACJ,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAAE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YACjE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAqB,CAAC;YACpF,IAAI,MAAM,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC,IAAI,MAAM,CAAC,KAAK,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACxF,OAAO,MAAM,CAAC;YACf,CAAC;QACF,CAAC;QAAC,MAAM,CAAC;YACR,oFAAoF;QACrF,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IAAA,CACjC;IAED,iGAAiG;IACjG,IAAI,CAAC,KAAa,EAAE,MAA0B,EAAE,EAAW,EAAuB;QACjF,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAChC,MAAM,KAAK,GAAwB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC;QAC/F,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QACzB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC;QACzE,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC/E,OAAO,KAAK,CAAC;IAAA,CACb;IAED,yFAAyF;IACzF,MAAM,CAAC,KAAa,EAAQ;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;YAAE,OAAO;QAC1C,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;QAClC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAAA,CAC/E;IAED,qEAAqE;IACrE,UAAU,CAAC,MAAe,EAAyB;QAClD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QACzB,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;IAAA,CACxE;IAED,4EAA4E;IAC5E,MAAM,GAA0B;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QACzB,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;IAAA,CAC5E;CACD","sourcesContent":["import { existsSync, mkdirSync, readFileSync, writeFileSync } from \"node:fs\";\nimport { cpus, totalmem } from \"node:os\";\nimport { dirname, join } from \"node:path\";\nimport type { ModelFitnessReport } from \"../research/model-fitness.ts\";\n\n/**\n * Durable, HOST-KEYED storage for model fitness reports. Fitness is a property of a model ON a\n * host (tok/s and latency-driven failures do not travel between machines), so reports are keyed\n * by a hardware fingerprint: the same model can be \"the heavy lifter\" on one machine and\n * \"waiting for better hardware\" on another, and pi remembers both without confusing them —\n * including when settings/dotfiles are synced across machines.\n */\n\nexport interface HostFingerprint {\n\t/** Stable, human-readable id derived from the specs below. */\n\tid: string;\n\tcpu: string;\n\tcores: number;\n\ttotalMemGb: number;\n}\n\nexport function currentHostFingerprint(): HostFingerprint {\n\tconst cpuList = cpus();\n\tconst cpu = (cpuList[0]?.model ?? \"unknown-cpu\").trim();\n\tconst cores = Math.max(1, cpuList.length);\n\tconst totalMemGb = Math.round(totalmem() / 1024 ** 3);\n\tconst id = `${cpu\n\t\t.toLowerCase()\n\t\t.replace(/[^a-z0-9]+/g, \"-\")\n\t\t.replace(/^-+|-+$/g, \"\")\n\t\t.slice(0, 48)}-${cores}c-${totalMemGb}g`;\n\treturn { id, cpu, cores, totalMemGb };\n}\n\nexport interface StoredFitnessReport {\n\tmodel: string;\n\treport: ModelFitnessReport;\n\tat: string;\n\thost: HostFingerprint;\n}\n\ninterface FitnessStoreFile {\n\tversion: 1;\n\t/** hostId -> modelRef -> latest stored report. */\n\thosts: Record<string, Record<string, StoredFitnessReport>>;\n}\n\nexport class FitnessStore {\n\tprivate readonly filePath: string;\n\tprivate readonly fingerprint: () => HostFingerprint;\n\n\tconstructor(filePath: string, options?: { fingerprint?: () => HostFingerprint }) {\n\t\tthis.filePath = filePath;\n\t\tthis.fingerprint = options?.fingerprint ?? currentHostFingerprint;\n\t}\n\n\tstatic forAgentDir(agentDir: string, options?: { fingerprint?: () => HostFingerprint }): FitnessStore {\n\t\treturn new FitnessStore(join(agentDir, \"state\", \"model-fitness.json\"), options);\n\t}\n\n\tprivate load(): FitnessStoreFile {\n\t\ttry {\n\t\t\tif (!existsSync(this.filePath)) return { version: 1, hosts: {} };\n\t\t\tconst parsed = JSON.parse(readFileSync(this.filePath, \"utf-8\")) as FitnessStoreFile;\n\t\t\tif (parsed && parsed.version === 1 && parsed.hosts && typeof parsed.hosts === \"object\") {\n\t\t\t\treturn parsed;\n\t\t\t}\n\t\t} catch {\n\t\t\t// Unreadable/corrupt store: start fresh in memory; the next save rewrites the file.\n\t\t}\n\t\treturn { version: 1, hosts: {} };\n\t}\n\n\t/** Persist the latest report for a model on the CURRENT host. Best-effort, returns the entry. */\n\tsave(model: string, report: ModelFitnessReport, at?: string): StoredFitnessReport {\n\t\tconst host = this.fingerprint();\n\t\tconst entry: StoredFitnessReport = { model, report, at: at ?? new Date().toISOString(), host };\n\t\tconst file = this.load();\n\t\tfile.hosts[host.id] = { ...(file.hosts[host.id] ?? {}), [model]: entry };\n\t\tmkdirSync(dirname(this.filePath), { recursive: true });\n\t\twriteFileSync(this.filePath, `${JSON.stringify(file, null, \"\\t\")}\\n`, \"utf-8\");\n\t\treturn entry;\n\t}\n\n\t/** Drop a model's report for the CURRENT host (uninstall cleanup). No-op when absent. */\n\tremove(model: string): void {\n\t\tconst host = this.fingerprint();\n\t\tconst file = this.load();\n\t\tif (!file.hosts[host.id]?.[model]) return;\n\t\tdelete file.hosts[host.id][model];\n\t\tmkdirSync(dirname(this.filePath), { recursive: true });\n\t\twriteFileSync(this.filePath, `${JSON.stringify(file, null, \"\\t\")}\\n`, \"utf-8\");\n\t}\n\n\t/** Reports for the current host (default) or an explicit host id. */\n\tgetForHost(hostId?: string): StoredFitnessReport[] {\n\t\tconst file = this.load();\n\t\treturn Object.values(file.hosts[hostId ?? this.fingerprint().id] ?? {});\n\t}\n\n\t/** Every stored report across all hosts (for cross-machine comparisons). */\n\tgetAll(): StoredFitnessReport[] {\n\t\tconst file = this.load();\n\t\treturn Object.values(file.hosts).flatMap((models) => Object.values(models));\n\t}\n}\n"]}
+{"version":3,"file":"fitness-store.js","sourceRoot":"","sources":["../../../src/core/models/fitness-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAE9C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAwB,cAAc,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAiBhG,SAAS,cAAc,CAAC,KAAc,EAAmB;IACxD,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;AAAA,CAC3D;AAED,SAAS,WAAW,CAAC,KAAc,EAAW;IAC7C,OAAO,CACN,aAAa,CAAC,KAAK,CAAC;QACpB,cAAc,CAAC,KAAK,CAAC,SAAS,CAAC;QAC/B,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC;QAC3B,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC7B,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,OAAO,KAAK,QAAQ,CAAC;QAC9D,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC;QAC5B,CAAC,KAAK,CAAC,eAAe,KAAK,SAAS,IAAI,cAAc,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,CAC9E,CAAC;AAAA,CACF;AAED,SAAS,eAAe,CAAC,KAAc,EAA+B;IACrE,OAAO,CACN,aAAa,CAAC,KAAK,CAAC;QACpB,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC;QAC5B,CAAC,KAAK,CAAC,eAAe,KAAK,SAAS,IAAI,cAAc,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;QAC9E,WAAW,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC3B,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC;QACzB,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC;QACzB,WAAW,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC3B,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC;QACzB,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC;QAC1B,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;QAClC,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,gBAAgB,CAAC;QAC5C,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC;QACzC,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;QACxC,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;QACxC,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC;QACjC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC;QACnC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,OAAO,KAAK,QAAQ,CAAC;QACpE,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;QAClC,CAAC,KAAK,CAAC,KAAK,CAAC,eAAe,KAAK,SAAS,IAAI,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;QAC1F,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS;YAC5B,CAAC,aAAa,CAAC,KAAK,CAAC,QAAQ,CAAC;gBAC7B,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,uBAAuB,CAAC;gBACtD,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,mBAAmB,CAAC;gBAClD,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACtC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,OAAO,KAAK,QAAQ,CAAC;gBACvE,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;QACzC,cAAc,CAAC,KAAK,CAAC,YAAY,CAAC,CAClC,CAAC;AAAA,CACF;AAED,SAAS,gBAAgB,CAAC,KAAc,EAAE,MAAc,EAAmD;IAC1G,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC5C,MAAM,OAAO,GAAwC,EAAE,CAAC;IACxD,KAAK,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACxD,IACC,CAAC,aAAa,CAAC,SAAS,CAAC;YACzB,SAAS,CAAC,KAAK,KAAK,KAAK;YACzB,OAAO,SAAS,CAAC,EAAE,KAAK,QAAQ;YAChC,CAAC,iBAAiB,CAAC,SAAS,CAAC,IAAI,CAAC;YAClC,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,MAAM;YAC5B,CAAC,eAAe,CAAC,SAAS,CAAC,MAAM,CAAC;YAElC,SAAS;QACV,OAAO,CAAC,KAAK,CAAC,GAAG;YAChB,KAAK;YACL,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,EAAE,EAAE,SAAS,CAAC,EAAE;YAChB,IAAI,EAAE,SAAS,CAAC,IAAI;SACpB,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AAAA,CACf;AAED,MAAM,OAAO,YAAY;IACP,OAAO,CAAsD;IAE9E,YAAY,QAAgB,EAAE,OAAqE,EAAE;QACpG,IAAI,CAAC,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,QAAQ;YACR,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,OAAO,EAAE,WAAW;YACjC,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,eAAe,EAAE;YAChD,SAAS,EAAE,gBAAgB;SAC3B,CAAC,CAAC;IAAA,CACH;IAED,MAAM,CAAC,WAAW,CACjB,QAAgB,EAChB,OAAqE,EACtD;QACf,OAAO,IAAI,YAAY,CAAC,SAAS,CAAC,QAAQ,EAAE,oBAAoB,CAAC,EAAE,OAAO,CAAC,CAAC;IAAA,CAC5E;IAED,iGAAiG;IACjG,IAAI,CAAC,KAAa,EAAE,MAA0B,EAAE,EAAW,EAAuB;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,iBAAiB,CACpC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EACV,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC;YAClB,MAAM,KAAK,GAAwB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC;YAC/F,OAAO,CAAC,KAAK,CAAC,GAAG,KAAK,CAAC;YACvB,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAAA,CACxC,CACD,CAAC;IAAA,CACF;IAED,yFAAyF;IACzF,MAAM,CAAC,KAAa,EAAQ;QAC3B,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAC7B,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EACV,CAAC,OAAO,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;gBAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAClE,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC;YACtB,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAAA,CAC5C,CACD,CAAC;IAAA,CACF;IAED,qEAAqE;IACrE,UAAU,CAAC,MAAe,EAAyB;QAClD,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IAAA,CACzD;IAED,4EAA4E;IAC5E,MAAM,GAA0B;QAC/B,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IAAA,CAC/E;CACD","sourcesContent":["import { stateFile } from \"../agent-paths.ts\";\nimport type { ModelFitnessReport } from \"../research/model-fitness.ts\";\nimport { isWorkerSession } from \"../session-role.ts\";\nimport { isPlainRecord } from \"../util/value-guards.ts\";\nimport { type HostFingerprint, HostStateStore, isHostFingerprint } from \"./host-state-store.ts\";\n\n/**\n * Durable, HOST-KEYED storage for model fitness reports. Fitness is a property of a model ON a\n * host (tok/s and latency-driven failures do not travel between machines), so reports are keyed\n * by a hardware fingerprint: the same model can be \"the heavy lifter\" on one machine and\n * \"waiting for better hardware\" on another, and pi remembers both without confusing them —\n * including when settings/dotfiles are synced across machines.\n */\n\nexport interface StoredFitnessReport {\n\tmodel: string;\n\treport: ModelFitnessReport;\n\tat: string;\n\thost: HostFingerprint;\n}\n\nfunction isFiniteNumber(value: unknown): value is number {\n\treturn typeof value === \"number\" && Number.isFinite(value);\n}\n\nfunction isLaneScore(value: unknown): boolean {\n\treturn (\n\t\tisPlainRecord(value) &&\n\t\tisFiniteNumber(value.succeeded) &&\n\t\tisFiniteNumber(value.total) &&\n\t\tArray.isArray(value.outcomes) &&\n\t\tvalue.outcomes.every((outcome) => typeof outcome === \"string\") &&\n\t\tisFiniteNumber(value.meanMs) &&\n\t\t(value.tokensPerSecond === undefined || isFiniteNumber(value.tokensPerSecond))\n\t);\n}\n\nfunction isFitnessReport(value: unknown): value is ModelFitnessReport {\n\treturn (\n\t\tisPlainRecord(value) &&\n\t\tisFiniteNumber(value.trials) &&\n\t\t(value.tokensPerSecond === undefined || isFiniteNumber(value.tokensPerSecond)) &&\n\t\tisLaneScore(value.research) &&\n\t\tisLaneScore(value.worker) &&\n\t\tisLaneScore(value.search) &&\n\t\tisLaneScore(value.toolCall) &&\n\t\tisLaneScore(value.digest) &&\n\t\tisPlainRecord(value.judge) &&\n\t\tisFiniteNumber(value.judge.parsed) &&\n\t\tisFiniteNumber(value.judge.planningElevated) &&\n\t\tisFiniteNumber(value.judge.planningTotal) &&\n\t\tisFiniteNumber(value.judge.trivialCheap) &&\n\t\tisFiniteNumber(value.judge.trivialTotal) &&\n\t\tisFiniteNumber(value.judge.total) &&\n\t\tArray.isArray(value.judge.outcomes) &&\n\t\tvalue.judge.outcomes.every((outcome) => typeof outcome === \"string\") &&\n\t\tisFiniteNumber(value.judge.meanMs) &&\n\t\t(value.judge.tokensPerSecond === undefined || isFiniteNumber(value.judge.tokensPerSecond)) &&\n\t\t(value.capacity === undefined ||\n\t\t\t(isPlainRecord(value.capacity) &&\n\t\t\t\tisFiniteNumber(value.capacity.registeredContextWindow) &&\n\t\t\t\tisFiniteNumber(value.capacity.servedContextWindow) &&\n\t\t\t\tArray.isArray(value.capacity.outcomes) &&\n\t\t\t\tvalue.capacity.outcomes.every((outcome) => typeof outcome === \"string\") &&\n\t\t\t\tisFiniteNumber(value.capacity.meanMs))) &&\n\t\tisFiniteNumber(value.totalCostUsd)\n\t);\n}\n\nfunction parseFitnessHost(value: unknown, hostId: string): Record<string, StoredFitnessReport> | undefined {\n\tif (!isPlainRecord(value)) return undefined;\n\tconst reports: Record<string, StoredFitnessReport> = {};\n\tfor (const [model, candidate] of Object.entries(value)) {\n\t\tif (\n\t\t\t!isPlainRecord(candidate) ||\n\t\t\tcandidate.model !== model ||\n\t\t\ttypeof candidate.at !== \"string\" ||\n\t\t\t!isHostFingerprint(candidate.host) ||\n\t\t\tcandidate.host.id !== hostId ||\n\t\t\t!isFitnessReport(candidate.report)\n\t\t)\n\t\t\tcontinue;\n\t\treports[model] = {\n\t\t\tmodel,\n\t\t\treport: candidate.report,\n\t\t\tat: candidate.at,\n\t\t\thost: candidate.host,\n\t\t};\n\t}\n\treturn reports;\n}\n\nexport class FitnessStore {\n\tprivate readonly storage: HostStateStore<Record<string, StoredFitnessReport>>;\n\n\tconstructor(filePath: string, options?: { fingerprint?: () => HostFingerprint; readOnly?: boolean }) {\n\t\tthis.storage = new HostStateStore({\n\t\t\tfilePath,\n\t\t\tversion: 1,\n\t\t\tfingerprint: options?.fingerprint,\n\t\t\treadOnly: options?.readOnly ?? isWorkerSession(),\n\t\t\tparseHost: parseFitnessHost,\n\t\t});\n\t}\n\n\tstatic forAgentDir(\n\t\tagentDir: string,\n\t\toptions?: { fingerprint?: () => HostFingerprint; readOnly?: boolean },\n\t): FitnessStore {\n\t\treturn new FitnessStore(stateFile(agentDir, \"model-fitness.json\"), options);\n\t}\n\n\t/** Persist the latest report for a model on the CURRENT host. Best-effort, returns the entry. */\n\tsave(model: string, report: ModelFitnessReport, at?: string): StoredFitnessReport {\n\t\treturn this.storage.mutateCurrentHost(\n\t\t\t() => ({}),\n\t\t\t(reports, host) => {\n\t\t\t\tconst entry: StoredFitnessReport = { model, report, at: at ?? new Date().toISOString(), host };\n\t\t\t\treports[model] = entry;\n\t\t\t\treturn { result: entry, changed: true };\n\t\t\t},\n\t\t);\n\t}\n\n\t/** Drop a model's report for the CURRENT host (uninstall cleanup). No-op when absent. */\n\tremove(model: string): void {\n\t\tthis.storage.mutateCurrentHost(\n\t\t\t() => ({}),\n\t\t\t(reports) => {\n\t\t\t\tif (!reports[model]) return { result: undefined, changed: false };\n\t\t\t\tdelete reports[model];\n\t\t\t\treturn { result: undefined, changed: true };\n\t\t\t},\n\t\t);\n\t}\n\n\t/** Reports for the current host (default) or an explicit host id. */\n\tgetForHost(hostId?: string): StoredFitnessReport[] {\n\t\treturn Object.values(this.storage.getHost(hostId) ?? {});\n\t}\n\n\t/** Every stored report across all hosts (for cross-machine comparisons). */\n\tgetAll(): StoredFitnessReport[] {\n\t\treturn this.storage.getAllHosts().flatMap((reports) => Object.values(reports));\n\t}\n}\n"]}

package/dist/core/models/host-state-store.d.ts

@@ -0,0 +1,34 @@
+export interface HostFingerprint {
+    id: string;
+    cpu: string;
+    cores: number;
+    totalMemGb: number;
+}
+export declare function currentHostFingerprint(): HostFingerprint;
+export declare function isHostFingerprint(value: unknown): value is HostFingerprint;
+export interface HostStateMutation<TResult> {
+    result: TResult;
+    changed: boolean;
+}
+export interface HostStateStoreOptions<THostData> {
+    filePath: string;
+    version: number;
+    fingerprint?: () => HostFingerprint;
+    readOnly?: boolean;
+    parseHost(value: unknown, hostId: string): THostData | undefined;
+}
+/** Shared lock-safe, atomic substrate for versioned state partitioned by host fingerprint. */
+export declare class HostStateStore<THostData> {
+    private readonly filePath;
+    private readonly version;
+    private readonly fingerprint;
+    private readonly readOnly;
+    private readonly parseHost;
+    constructor(options: HostStateStoreOptions<THostData>);
+    private currentHost;
+    getHost(hostId?: string): THostData | undefined;
+    getAllHosts(): THostData[];
+    mutateCurrentHost<TResult>(create: (host: HostFingerprint) => THostData, mutate: (data: THostData, host: HostFingerprint) => HostStateMutation<TResult>): TResult;
+    private load;
+}
+//# sourceMappingURL=host-state-store.d.ts.map

package/dist/core/models/host-state-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"host-state-store.d.ts","sourceRoot":"","sources":["../../../src/core/models/host-state-store.ts"],"names":[],"mappings":"AAKA,MAAM,WAAW,eAAe;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;CACnB;AAED,wBAAgB,sBAAsB,IAAI,eAAe,CAWxD;AAED,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,eAAe,CAY1E;AAOD,MAAM,WAAW,iBAAiB,CAAC,OAAO;IACzC,MAAM,EAAE,OAAO,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,qBAAqB,CAAC,SAAS;IAC/C,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,eAAe,CAAC;IACpC,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,SAAS,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,CAAC;CACjE;AAED,8FAA8F;AAC9F,qBAAa,cAAc,CAAC,SAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAwB;IACpD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAU;IACnC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA4D;IAEtF,YAAY,OAAO,EAAE,qBAAqB,CAAC,SAAS,CAAC,EAMpD;IAED,OAAO,CAAC,WAAW;IAInB,OAAO,CAAC,MAAM,SAAwB,GAAG,SAAS,GAAG,SAAS,CAE7D;IAED,WAAW,IAAI,SAAS,EAAE,CAEzB;IAED,iBAAiB,CAAC,OAAO,EACxB,MAAM,EAAE,CAAC,IAAI,EAAE,eAAe,KAAK,SAAS,EAC5C,MAAM,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,eAAe,KAAK,iBAAiB,CAAC,OAAO,CAAC,GAC5E,OAAO,CAaT;IAED,OAAO,CAAC,IAAI;CAiBZ","sourcesContent":["import { existsSync, readFileSync } from \"node:fs\";\nimport { cpus, totalmem } from \"node:os\";\nimport { withFileLockSync, writeFileAtomicSync } from \"../util/atomic-file.ts\";\nimport { isPlainRecord } from \"../util/value-guards.ts\";\n\nexport interface HostFingerprint {\n\tid: string;\n\tcpu: string;\n\tcores: number;\n\ttotalMemGb: number;\n}\n\nexport function currentHostFingerprint(): HostFingerprint {\n\tconst cpuList = cpus();\n\tconst cpu = (cpuList[0]?.model ?? \"unknown-cpu\").trim();\n\tconst cores = Math.max(1, cpuList.length);\n\tconst totalMemGb = Math.round(totalmem() / 1024 ** 3);\n\tconst id = `${cpu\n\t\t.toLowerCase()\n\t\t.replace(/[^a-z0-9]+/g, \"-\")\n\t\t.replace(/^-+|-+$/g, \"\")\n\t\t.slice(0, 48)}-${cores}c-${totalMemGb}g`;\n\treturn { id, cpu, cores, totalMemGb };\n}\n\nexport function isHostFingerprint(value: unknown): value is HostFingerprint {\n\treturn (\n\t\tisPlainRecord(value) &&\n\t\ttypeof value.id === \"string\" &&\n\t\tvalue.id.length > 0 &&\n\t\ttypeof value.cpu === \"string\" &&\n\t\tNumber.isSafeInteger(value.cores) &&\n\t\tNumber(value.cores) > 0 &&\n\t\ttypeof value.totalMemGb === \"number\" &&\n\t\tNumber.isFinite(value.totalMemGb) &&\n\t\tvalue.totalMemGb >= 0\n\t);\n}\n\ninterface HostStateFile<THostData> {\n\tversion: number;\n\thosts: Record<string, THostData>;\n}\n\nexport interface HostStateMutation<TResult> {\n\tresult: TResult;\n\tchanged: boolean;\n}\n\nexport interface HostStateStoreOptions<THostData> {\n\tfilePath: string;\n\tversion: number;\n\tfingerprint?: () => HostFingerprint;\n\treadOnly?: boolean;\n\tparseHost(value: unknown, hostId: string): THostData | undefined;\n}\n\n/** Shared lock-safe, atomic substrate for versioned state partitioned by host fingerprint. */\nexport class HostStateStore<THostData> {\n\tprivate readonly filePath: string;\n\tprivate readonly version: number;\n\tprivate readonly fingerprint: () => HostFingerprint;\n\tprivate readonly readOnly: boolean;\n\tprivate readonly parseHost: (value: unknown, hostId: string) => THostData | undefined;\n\n\tconstructor(options: HostStateStoreOptions<THostData>) {\n\t\tthis.filePath = options.filePath;\n\t\tthis.version = options.version;\n\t\tthis.fingerprint = options.fingerprint ?? currentHostFingerprint;\n\t\tthis.readOnly = options.readOnly ?? false;\n\t\tthis.parseHost = options.parseHost;\n\t}\n\n\tprivate currentHost(): HostFingerprint {\n\t\treturn this.fingerprint();\n\t}\n\n\tgetHost(hostId = this.currentHost().id): THostData | undefined {\n\t\treturn this.load().hosts[hostId];\n\t}\n\n\tgetAllHosts(): THostData[] {\n\t\treturn Object.values(this.load().hosts);\n\t}\n\n\tmutateCurrentHost<TResult>(\n\t\tcreate: (host: HostFingerprint) => THostData,\n\t\tmutate: (data: THostData, host: HostFingerprint) => HostStateMutation<TResult>,\n\t): TResult {\n\t\tconst host = this.currentHost();\n\t\tconst execute = (): TResult => {\n\t\t\tconst file = this.load();\n\t\t\tconst data = file.hosts[host.id] ?? create(host);\n\t\t\tfile.hosts[host.id] = data;\n\t\t\tconst mutation = mutate(data, host);\n\t\t\tif (mutation.changed && !this.readOnly) {\n\t\t\t\twriteFileAtomicSync(this.filePath, `${JSON.stringify(file, null, \"\\t\")}\\n`);\n\t\t\t}\n\t\t\treturn mutation.result;\n\t\t};\n\t\treturn this.readOnly ? execute() : withFileLockSync(this.filePath, execute);\n\t}\n\n\tprivate load(): HostStateFile<THostData> {\n\t\tif (!existsSync(this.filePath)) return { version: this.version, hosts: {} };\n\t\ttry {\n\t\t\tconst parsed: unknown = JSON.parse(readFileSync(this.filePath, \"utf-8\"));\n\t\t\tif (!isPlainRecord(parsed) || parsed.version !== this.version || !isPlainRecord(parsed.hosts)) {\n\t\t\t\treturn { version: this.version, hosts: {} };\n\t\t\t}\n\t\t\tconst hosts: Record<string, THostData> = {};\n\t\t\tfor (const [hostId, value] of Object.entries(parsed.hosts)) {\n\t\t\t\tconst host = this.parseHost(value, hostId);\n\t\t\t\tif (host !== undefined) hosts[hostId] = host;\n\t\t\t}\n\t\t\treturn { version: this.version, hosts };\n\t\t} catch {\n\t\t\treturn { version: this.version, hosts: {} };\n\t\t}\n\t}\n}\n"]}

package/dist/core/models/host-state-store.js

@@ -0,0 +1,86 @@
+import { existsSync, readFileSync } from "node:fs";
+import { cpus, totalmem } from "node:os";
+import { withFileLockSync, writeFileAtomicSync } from "../util/atomic-file.js";
+import { isPlainRecord } from "../util/value-guards.js";
+export function currentHostFingerprint() {
+    const cpuList = cpus();
+    const cpu = (cpuList[0]?.model ?? "unknown-cpu").trim();
+    const cores = Math.max(1, cpuList.length);
+    const totalMemGb = Math.round(totalmem() / 1024 ** 3);
+    const id = `${cpu
+        .toLowerCase()
+        .replace(/[^a-z0-9]+/g, "-")
+        .replace(/^-+|-+$/g, "")
+        .slice(0, 48)}-${cores}c-${totalMemGb}g`;
+    return { id, cpu, cores, totalMemGb };
+}
+export function isHostFingerprint(value) {
+    return (isPlainRecord(value) &&
+        typeof value.id === "string" &&
+        value.id.length > 0 &&
+        typeof value.cpu === "string" &&
+        Number.isSafeInteger(value.cores) &&
+        Number(value.cores) > 0 &&
+        typeof value.totalMemGb === "number" &&
+        Number.isFinite(value.totalMemGb) &&
+        value.totalMemGb >= 0);
+}
+/** Shared lock-safe, atomic substrate for versioned state partitioned by host fingerprint. */
+export class HostStateStore {
+    filePath;
+    version;
+    fingerprint;
+    readOnly;
+    parseHost;
+    constructor(options) {
+        this.filePath = options.filePath;
+        this.version = options.version;
+        this.fingerprint = options.fingerprint ?? currentHostFingerprint;
+        this.readOnly = options.readOnly ?? false;
+        this.parseHost = options.parseHost;
+    }
+    currentHost() {
+        return this.fingerprint();
+    }
+    getHost(hostId = this.currentHost().id) {
+        return this.load().hosts[hostId];
+    }
+    getAllHosts() {
+        return Object.values(this.load().hosts);
+    }
+    mutateCurrentHost(create, mutate) {
+        const host = this.currentHost();
+        const execute = () => {
+            const file = this.load();
+            const data = file.hosts[host.id] ?? create(host);
+            file.hosts[host.id] = data;
+            const mutation = mutate(data, host);
+            if (mutation.changed && !this.readOnly) {
+                writeFileAtomicSync(this.filePath, `${JSON.stringify(file, null, "\t")}\n`);
+            }
+            return mutation.result;
+        };
+        return this.readOnly ? execute() : withFileLockSync(this.filePath, execute);
+    }
+    load() {
+        if (!existsSync(this.filePath))
+            return { version: this.version, hosts: {} };
+        try {
+            const parsed = JSON.parse(readFileSync(this.filePath, "utf-8"));
+            if (!isPlainRecord(parsed) || parsed.version !== this.version || !isPlainRecord(parsed.hosts)) {
+                return { version: this.version, hosts: {} };
+            }
+            const hosts = {};
+            for (const [hostId, value] of Object.entries(parsed.hosts)) {
+                const host = this.parseHost(value, hostId);
+                if (host !== undefined)
+                    hosts[hostId] = host;
+            }
+            return { version: this.version, hosts };
+        }
+        catch {
+            return { version: this.version, hosts: {} };
+        }
+    }
+}
+//# sourceMappingURL=host-state-store.js.map

package/dist/core/models/host-state-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"host-state-store.js","sourceRoot":"","sources":["../../../src/core/models/host-state-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC/E,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AASxD,MAAM,UAAU,sBAAsB,GAAoB;IACzD,MAAM,OAAO,GAAG,IAAI,EAAE,CAAC;IACvB,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,aAAa,CAAC,CAAC,IAAI,EAAE,CAAC;IACxD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC;IACtD,MAAM,EAAE,GAAG,GAAG,GAAG;SACf,WAAW,EAAE;SACb,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC;SAC3B,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SACvB,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,KAAK,UAAU,GAAG,CAAC;IAC1C,OAAO,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;AAAA,CACtC;AAED,MAAM,UAAU,iBAAiB,CAAC,KAAc,EAA4B;IAC3E,OAAO,CACN,aAAa,CAAC,KAAK,CAAC;QACpB,OAAO,KAAK,CAAC,EAAE,KAAK,QAAQ;QAC5B,KAAK,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC;QACnB,OAAO,KAAK,CAAC,GAAG,KAAK,QAAQ;QAC7B,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC;QACvB,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ;QACpC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC;QACjC,KAAK,CAAC,UAAU,IAAI,CAAC,CACrB,CAAC;AAAA,CACF;AAoBD,8FAA8F;AAC9F,MAAM,OAAO,cAAc;IACT,QAAQ,CAAS;IACjB,OAAO,CAAS;IAChB,WAAW,CAAwB;IACnC,QAAQ,CAAU;IAClB,SAAS,CAA4D;IAEtF,YAAY,OAAyC,EAAE;QACtD,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/B,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,sBAAsB,CAAC;QACjE,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,KAAK,CAAC;QAC1C,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IAAA,CACnC;IAEO,WAAW,GAAoB;QACtC,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;IAAA,CAC1B;IAED,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,EAAE,EAAyB;QAC9D,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAAA,CACjC;IAED,WAAW,GAAgB;QAC1B,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC;IAAA,CACxC;IAED,iBAAiB,CAChB,MAA4C,EAC5C,MAA8E,EACpE;QACV,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,GAAY,EAAE,CAAC;YAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YACzB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC;YACjD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;YAC3B,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACpC,IAAI,QAAQ,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACxC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7E,CAAC;YACD,OAAO,QAAQ,CAAC,MAAM,CAAC;QAAA,CACvB,CAAC;QACF,OAAO,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAAA,CAC5E;IAEO,IAAI,GAA6B;QACxC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC;YAAE,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QAC5E,IAAI,CAAC;YACJ,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;YACzE,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,KAAK,IAAI,CAAC,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC/F,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YAC7C,CAAC;YACD,MAAM,KAAK,GAA8B,EAAE,CAAC;YAC5C,KAAK,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5D,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC3C,IAAI,IAAI,KAAK,SAAS;oBAAE,KAAK,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;YAC9C,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QAC7C,CAAC;IAAA,CACD;CACD","sourcesContent":["import { existsSync, readFileSync } from \"node:fs\";\nimport { cpus, totalmem } from \"node:os\";\nimport { withFileLockSync, writeFileAtomicSync } from \"../util/atomic-file.ts\";\nimport { isPlainRecord } from \"../util/value-guards.ts\";\n\nexport interface HostFingerprint {\n\tid: string;\n\tcpu: string;\n\tcores: number;\n\ttotalMemGb: number;\n}\n\nexport function currentHostFingerprint(): HostFingerprint {\n\tconst cpuList = cpus();\n\tconst cpu = (cpuList[0]?.model ?? \"unknown-cpu\").trim();\n\tconst cores = Math.max(1, cpuList.length);\n\tconst totalMemGb = Math.round(totalmem() / 1024 ** 3);\n\tconst id = `${cpu\n\t\t.toLowerCase()\n\t\t.replace(/[^a-z0-9]+/g, \"-\")\n\t\t.replace(/^-+|-+$/g, \"\")\n\t\t.slice(0, 48)}-${cores}c-${totalMemGb}g`;\n\treturn { id, cpu, cores, totalMemGb };\n}\n\nexport function isHostFingerprint(value: unknown): value is HostFingerprint {\n\treturn (\n\t\tisPlainRecord(value) &&\n\t\ttypeof value.id === \"string\" &&\n\t\tvalue.id.length > 0 &&\n\t\ttypeof value.cpu === \"string\" &&\n\t\tNumber.isSafeInteger(value.cores) &&\n\t\tNumber(value.cores) > 0 &&\n\t\ttypeof value.totalMemGb === \"number\" &&\n\t\tNumber.isFinite(value.totalMemGb) &&\n\t\tvalue.totalMemGb >= 0\n\t);\n}\n\ninterface HostStateFile<THostData> {\n\tversion: number;\n\thosts: Record<string, THostData>;\n}\n\nexport interface HostStateMutation<TResult> {\n\tresult: TResult;\n\tchanged: boolean;\n}\n\nexport interface HostStateStoreOptions<THostData> {\n\tfilePath: string;\n\tversion: number;\n\tfingerprint?: () => HostFingerprint;\n\treadOnly?: boolean;\n\tparseHost(value: unknown, hostId: string): THostData | undefined;\n}\n\n/** Shared lock-safe, atomic substrate for versioned state partitioned by host fingerprint. */\nexport class HostStateStore<THostData> {\n\tprivate readonly filePath: string;\n\tprivate readonly version: number;\n\tprivate readonly fingerprint: () => HostFingerprint;\n\tprivate readonly readOnly: boolean;\n\tprivate readonly parseHost: (value: unknown, hostId: string) => THostData | undefined;\n\n\tconstructor(options: HostStateStoreOptions<THostData>) {\n\t\tthis.filePath = options.filePath;\n\t\tthis.version = options.version;\n\t\tthis.fingerprint = options.fingerprint ?? currentHostFingerprint;\n\t\tthis.readOnly = options.readOnly ?? false;\n\t\tthis.parseHost = options.parseHost;\n\t}\n\n\tprivate currentHost(): HostFingerprint {\n\t\treturn this.fingerprint();\n\t}\n\n\tgetHost(hostId = this.currentHost().id): THostData | undefined {\n\t\treturn this.load().hosts[hostId];\n\t}\n\n\tgetAllHosts(): THostData[] {\n\t\treturn Object.values(this.load().hosts);\n\t}\n\n\tmutateCurrentHost<TResult>(\n\t\tcreate: (host: HostFingerprint) => THostData,\n\t\tmutate: (data: THostData, host: HostFingerprint) => HostStateMutation<TResult>,\n\t): TResult {\n\t\tconst host = this.currentHost();\n\t\tconst execute = (): TResult => {\n\t\t\tconst file = this.load();\n\t\t\tconst data = file.hosts[host.id] ?? create(host);\n\t\t\tfile.hosts[host.id] = data;\n\t\t\tconst mutation = mutate(data, host);\n\t\t\tif (mutation.changed && !this.readOnly) {\n\t\t\t\twriteFileAtomicSync(this.filePath, `${JSON.stringify(file, null, \"\\t\")}\\n`);\n\t\t\t}\n\t\t\treturn mutation.result;\n\t\t};\n\t\treturn this.readOnly ? execute() : withFileLockSync(this.filePath, execute);\n\t}\n\n\tprivate load(): HostStateFile<THostData> {\n\t\tif (!existsSync(this.filePath)) return { version: this.version, hosts: {} };\n\t\ttry {\n\t\t\tconst parsed: unknown = JSON.parse(readFileSync(this.filePath, \"utf-8\"));\n\t\t\tif (!isPlainRecord(parsed) || parsed.version !== this.version || !isPlainRecord(parsed.hosts)) {\n\t\t\t\treturn { version: this.version, hosts: {} };\n\t\t\t}\n\t\t\tconst hosts: Record<string, THostData> = {};\n\t\t\tfor (const [hostId, value] of Object.entries(parsed.hosts)) {\n\t\t\t\tconst host = this.parseHost(value, hostId);\n\t\t\t\tif (host !== undefined) hosts[hostId] = host;\n\t\t\t}\n\t\t\treturn { version: this.version, hosts };\n\t\t} catch {\n\t\t\treturn { version: this.version, hosts: {} };\n\t\t}\n\t}\n}\n"]}

package/dist/core/models/llamacpp-runtime.d.ts

@@ -0,0 +1,180 @@
+import { type ChildProcess } from "node:child_process";
+import type { Readable } from "node:stream";
+/**
+ * Managed runtime for prism-ml 1-bit GGUF models (Bonsai-27B first). The provider requires their
+ * llama.cpp fork (Q1_0_g128 hybrid-attention kernels) — stock llama.cpp/Ollama cannot serve these
+ * weights — so this module owns runtime install (prebuilt release download+extract), GGUF
+ * downloads, and llama-server lifecycle. Mirrors local-runtime.ts's OllamaRuntime: injectable
+ * seams for fetch/spawn/exists, pi-owned directories under agentDir (runtimes/, models/),
+ * detached+tracked child processes, onProgress as best-effort UI feedback, honest error taxonomies
+ * instead of silent fallbacks.
+ *
+ * The fork publishes precompiled per-platform archives on GitHub Releases (no local build): pin a
+ * release tag, download the matching asset, extract it, and locate the `llama-server` binary once
+ * at install time (release archive layouts vary — bin/ vs build/bin/ at the top level — so this is
+ * a deterministic one-time scan, not a fallback). The found relative path and backend (cpu/cuda)
+ * are persisted to a manifest so later `detect()`/`serve()` calls never need to re-scan.
+ */
+/**
+ * Pinned Prism ML llama.cpp fork release — verified via
+ * `GET https://api.github.com/repos/PrismML-Eng/llama.cpp/releases/tags/prism-b9594-38c66ad` on
+ * 2026-07-18 (release exists, all assets referenced below are present on it). Bump here (and
+ * re-verify with the same API call) when the provider ships new kernels.
+ */
+export declare const PRISM_LLAMACPP_PINNED_RELEASE = "prism-b9594-38c66ad";
+export declare const PRISM_LLAMACPP_RELEASES_BASE_URL = "https://github.com/PrismML-Eng/llama.cpp/releases/download";
+export interface PrismModelDescriptor {
+    repo: string;
+    file: string;
+    mmprojFile?: string;
+    displayName: string;
+}
+/** Curated first-model descriptor; the `/models add` wiring task consumes this. */
+export declare const BONSAI_27B: PrismModelDescriptor;
+export type PrismBackend = "cpu" | "cuda";
+export type PrismAssetKind = "tar-gz" | "zip";
+export interface PrismLlamaAsset {
+    name: string;
+    kind: PrismAssetKind;
+    backend: PrismBackend;
+}
+/**
+ * Maps a platform/arch/GPU triple to the exact Prism llama.cpp release asset for
+ * {@link PRISM_LLAMACPP_PINNED_RELEASE} — verified against the real GitHub release, not guessed.
+ * CPU asset by default; the CUDA 12.4 variant is only offered for linux x64 with an NVIDIA GPU
+ * present (a 12.8 variant also exists upstream but has no wired caller yet). Windows CUDA needs a
+ * companion `cudart-*` archive and is out of scope for v1 — Windows always resolves to the CPU zip
+ * regardless of `hasNvidiaGpu`. Pure and exported so it's independently testable.
+ */
+export declare function resolvePrismLlamaAsset(plat: string, architecture: string, hasNvidiaGpu: boolean): PrismLlamaAsset | undefined;
+export interface PrismDetectResult {
+    runtimeInstalled: boolean;
+    binaryPath?: string;
+    release?: string;
+}
+export interface PrismDownloadResult {
+    ok: boolean;
+    path?: string;
+    skipped?: boolean;
+    error?: string;
+}
+export type PrismServeResult = {
+    ok: true;
+    baseUrl: string;
+} | {
+    ok: false;
+    error: string;
+};
+type PrismSpawnOptions = {
+    detached?: boolean;
+    stdio?: "ignore";
+    env?: NodeJS.ProcessEnv;
+};
+type PrismSpawnFn = (command: string, args: string[], options: PrismSpawnOptions) => Pick<ChildProcess, "pid" | "kill" | "unref" | "on">;
+type PrismExtractArchiveFn = (input: Readable, destDir: string, kind: PrismAssetKind) => Promise<{
+    ok: boolean;
+    error?: string;
+}>;
+export interface PrismLlamaCppDeps {
+    fetchFn?: typeof fetch;
+    spawnFn?: PrismSpawnFn;
+    existsFn?: (path: string) => boolean;
+    sleepFn?: (ms: number) => Promise<void>;
+    /** Whether a named command exists on PATH (nvidia-smi, tar — for zip extraction on Windows). */
+    hasCommand?: (command: string) => boolean;
+    /** Decided once at install time and persisted as `backend` — serve() reads the persisted value
+     * so `-ngl 99` reflects what was actually installed, not the host's current GPU state. */
+    hasNvidiaGpu?: () => boolean;
+    platform?: () => string;
+    arch?: () => string;
+    /** Runs the extraction step for a downloaded archive. Injectable so installManaged's
+     * download->extract->scan orchestration is testable without a real tar/unzip pipeline; defaults
+     * to the real spawn-based extractor. */
+    extractArchive?: PrismExtractArchiveFn;
+    /** Health-poll bounds for serve(); default ~120s (240 * 500ms), both overridable for tests. */
+    healthPollAttempts?: number;
+    healthPollIntervalMs?: number;
+}
+export declare class PrismLlamaCppRuntime {
+    private readonly _agentDir;
+    private readonly _fetch;
+    private readonly _spawn;
+    private readonly _exists;
+    private readonly _sleep;
+    private readonly _hasCommand;
+    private readonly _hasNvidiaGpu;
+    private readonly _platform;
+    private readonly _arch;
+    private readonly _extractArchiveOverride;
+    private readonly _healthPollAttempts;
+    private readonly _healthPollIntervalMs;
+    private _child;
+    constructor(args: {
+        agentDir: string;
+        deps?: PrismLlamaCppDeps;
+    });
+    runtimeDir(): string;
+    modelsDir(): string;
+    private _binaryName;
+    private _manifestPath;
+    private _readManifest;
+    private _writeManifest;
+    detect(): Promise<PrismDetectResult>;
+    /** One-time recursive scan for the `llama-server` binary inside a freshly extracted release
+     * archive — layouts vary (bin/ vs build/bin/ at the top level) across platforms/backends, so
+     * this locates it deterministically instead of assuming a fixed path, then the result is
+     * persisted so later calls never need to re-scan. */
+    private _findBinaryRelPath;
+    /**
+     * Download the pinned release asset for this host and extract it (consent-gated by the caller,
+     * same contract as OllamaRuntime#installManaged — this method only does the mechanical
+     * download+extract+locate and reports the outcome honestly). No compiler toolchain required:
+     * the fork ships prebuilt binaries.
+     */
+    installManaged(onProgress?: (status: string) => void): Promise<{
+        ok: boolean;
+        error?: string;
+    }>;
+    private _extractArchive;
+    private _extractTarGz;
+    private _extractZip;
+    private _remoteContentLength;
+    private _fileSizeBytes;
+    private _cleanupPartial;
+    /**
+     * Wait until a write stream's underlying fd is actually closed. `pipeline()` destroys the
+     * stream on error but its promise can settle before the fd close completes (Windows in
+     * particular keeps the file handle open a tick longer) — unlinking before that races with the
+     * OS and can leave the "partial" file behind or fail the delete outright. Resolves immediately
+     * if the stream already reports `closed`.
+     */
+    private _waitForStreamClosed;
+    /**
+     * Stream a GGUF (or mmproj) file from Hugging Face into pi's owned models dir. Skips a re-download
+     * when the local file already matches the remote size; verifies size when the response reports
+     * `content-length` and deletes the partial file on any failure or mismatch — never leaves a
+     * corrupt/truncated weight file behind for a later load to silently misread.
+     */
+    downloadModel(args: {
+        repo: string;
+        file: string;
+    }, onProgress?: (status: string) => void): Promise<PrismDownloadResult>;
+    private _healthUp;
+    /**
+     * Spawn `llama-server` detached+tracked (killed on parent shutdown even if pi crashes without
+     * calling stop()) and poll `/health` until ready. `-ngl 99` is only added when the installed
+     * asset's persisted `backend` is `"cuda"`, not from the host's current GPU state.
+     */
+    serve(args: {
+        modelPath: string;
+        mmprojPath?: string;
+        port: number;
+        numCtx: number;
+    }): Promise<PrismServeResult>;
+    stop(): {
+        stopped: boolean;
+    };
+    isRunning(): boolean;
+}
+export {};
+//# sourceMappingURL=llamacpp-runtime.d.ts.map