@elizaos/ui 2.0.0-beta.2 → 2.0.3-beta.2

package/services/local-inference/device-bridge.d.ts

@@ -30,6 +30,7 @@
  */
 import type { Server as HttpServer } from "node:http";
 import type { AgentRuntime } from "@elizaos/core";
+import { type GenerationThroughput } from "@elizaos/shared/local-inference";
 import type { LocalInferenceLoadArgs } from "./active-model";
 interface DeviceCapabilities {
     platform: "ios" | "android" | "web" | "electrobun" | "desktop";
@@ -49,8 +50,8 @@ interface DeviceCapabilities {
     gpuSupported?: boolean;
     lowPowerMode?: boolean;
     thermalState?: "nominal" | "fair" | "serious" | "critical" | "unknown";
-    dflashSupported?: boolean;
-    dflashReason?: string;
+    mtpSupported?: boolean;
+    mtpReason?: string;
 }
 export interface DeviceSummary {
     deviceId: string;
@@ -74,6 +75,24 @@ export interface DeviceBridgeStatus {
     loadedPath: string | null;
     connectedSince: string | null;
 }
+/**
+ * One on-device generation's measured resource signal, emitted to
+ * `subscribeGenerationMetrics` listeners after every successful `generateResult`.
+ * The Mobile Resource Workbench folds these into a `DeviceResourceMetrics`
+ * accumulator (prefill/decode tok/s, TTFT, per-tier aggregation). All
+ * throughput fields are `null` when the device could not measure the inputs.
+ */
+export interface DeviceGenerationMetrics {
+    deviceId: string;
+    platform: DeviceCapabilities["platform"] | null;
+    /** Device model identifier (e.g. `iPhone17,2`) for per-device baselines. */
+    deviceModel: string | null;
+    promptTokens: number;
+    outputTokens: number;
+    durationMs: number;
+    ttftMs: number | null;
+    throughput: GenerationThroughput;
+}
 export declare class DeviceBridge {
     private readonly devices;
     private wss;
@@ -83,11 +102,23 @@ export declare class DeviceBridge {
     private readonly pendingGenerates;
     private readonly pendingEmbeds;
     private readonly statusListeners;
+    private readonly generationMetricsListeners;
+    /** The most recent successful generation's metrics, or null. */
+    private lastGenerationMetrics;
     private readonly expectedPairingToken;
     status(): DeviceBridgeStatus;
     private countRouted;
     subscribeStatus(listener: (status: DeviceBridgeStatus) => void): () => void;
     private emitStatus;
+    /**
+     * Subscribe to per-generation throughput metrics. Fires once per successful
+     * on-device generation with the differenced prefill/decode tok/s. Returns an
+     * unsubscribe function.
+     */
+    subscribeGenerationMetrics(listener: (metrics: DeviceGenerationMetrics) => void): () => void;
+    /** The most recent successful generation's measured metrics, or null. */
+    latestGenerationMetrics(): DeviceGenerationMetrics | null;
+    private emitGenerationMetrics;
     attachToHttpServer(server: HttpServer): Promise<void>;
     private handleConnection;
     private onDeviceRegistered;
@@ -122,8 +153,8 @@ export declare class DeviceBridge {
     /**
      * On startup, read persisted pending requests back into memory. Their
      * promises are gone (the original caller's process is dead) so they can
-     * only be resolved externally — for now we just re-queue them with a
-     * fresh timeout, and the first device that connects will process them.
+     * only be resolved externally, so we re-queue them with a fresh timeout.
+     * The first connected device that can handle generation will process them.
      * If nothing consumes them within the timeout they reject quietly.
      *
      * Stale entries older than 24h are purged rather than resurrected.

package/services/local-inference/device-bridge.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"device-bridge.d.ts","sourceRoot":"","sources":["../../../../../../src/services/local-inference/device-bridge.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AAIH,OAAO,KAAK,EAAE,MAAM,IAAI,UAAU,EAAmB,MAAM,WAAW,CAAC;AAGvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAElD,OAAO,KAAK,EACV,sBAAsB,EAEvB,MAAM,gBAAgB,CAAC;AAQxB,UAAU,kBAAkB;IAC1B,QAAQ,EAAE,KAAK,GAAG,SAAS,GAAG,KAAK,GAAG,YAAY,GAAG,SAAS,CAAC;IAC/D,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,GAAG,EAAE;QACH,OAAO,EAAE,OAAO,GAAG,QAAQ,GAAG,cAAc,GAAG,MAAM,CAAC;QACtD,SAAS,EAAE,OAAO,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,IAAI,CAAC;IACT,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,YAAY,CAAC,EAAE,SAAS,GAAG,MAAM,GAAG,SAAS,GAAG,UAAU,GAAG,SAAS,CAAC;IACvE,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AA0KD,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,kBAAkB,CAAC;IACjC,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,cAAc,EAAE,MAAM,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IACjC,iDAAiD;IACjD,SAAS,EAAE,OAAO,CAAC;IACnB,OAAO,EAAE,aAAa,EAAE,CAAC;IACzB,qEAAqE;IACrE,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,oFAAoF;IACpF,eAAe,EAAE,MAAM,CAAC;IAGxB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,YAAY,EAAE,kBAAkB,GAAG,IAAI,CAAC;IACxC,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,cAAc,EAAE,MAAM,GAAG,IAAI,CAAC;CAC/B;AA+CD,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAsC;IAC9D,OAAO,CAAC,GAAG,CAA4B;IACvC,OAAO,CAAC,QAAQ,CAAS;IAEzB,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAkC;IAC/D,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAoC;IACnE,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAsC;IACvE,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAmC;IAEjE,OAAO,CAAC,QAAQ,CAAC,eAAe,CAE5B;IAEJ,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CACoB;IAEzD,MAAM,IAAI,kBAAkB;IA0C5B,OAAO,CAAC,WAAW;IAWnB,eAAe,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,kBAAkB,KAAK,IAAI,GAAG,MAAM,IAAI;IAO3E,OAAO,CAAC,UAAU;IAWZ,kBAAkB,CAAC,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IA8B3D,OAAO,CAAC,gBAAgB;IAiExB,OAAO,CAAC,kBAAkB;IA6G1B,OAAO,CAAC,oBAAoB;IA2D5B,OAAO,CAAC,mBAAmB;IAoE3B,OAAO,CAAC,YAAY;IAMpB,wFAAwF;IACxF,OAAO,CAAC,cAAc;IAiBhB,SAAS,CAAC,IAAI,EAAE,sBAAsB,GAAG,OAAO,CAAC,IAAI,CAAC;IAsCtD,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAuClC,gBAAgB,IAAI,MAAM,GAAG,IAAI;IAO3B,KAAK,CAAC,IAAI,EAAE;QAChB,KAAK,EAAE,MAAM,CAAC;KACf,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;IA6D9C,QAAQ,CAAC,IAAI,EAAE;QACnB,MAAM,EAAE,MAAM,CAAC;QACf,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;QACzB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,MAAM,CAAC;IA+DnB,OAAO,CAAC,cAAc;IAItB;;;;;OAKG;YACW,uBAAuB;IAqBrC;;;;;;;;OAQG;YACW,uBAAuB;CA2DtC;AAED,eAAO,MAAM,YAAY,cAAqB,CAAC;AAE/C,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,YAAY,GAAG;IACtB,eAAe,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,KAAK,OAAO,CAAC;CAC5D,GACA,IAAI,CAuBN"}
+{"version":3,"file":"device-bridge.d.ts","sourceRoot":"","sources":["../../../../../../src/services/local-inference/device-bridge.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AAIH,OAAO,KAAK,EAAE,MAAM,IAAI,UAAU,EAAmB,MAAM,WAAW,CAAC;AAGvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAElD,OAAO,EAEL,KAAK,oBAAoB,EAC1B,MAAM,iCAAiC,CAAC;AACzC,OAAO,KAAK,EACV,sBAAsB,EAEvB,MAAM,gBAAgB,CAAC;AAQxB,UAAU,kBAAkB;IAC1B,QAAQ,EAAE,KAAK,GAAG,SAAS,GAAG,KAAK,GAAG,YAAY,GAAG,SAAS,CAAC;IAC/D,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,GAAG,EAAE;QACH,OAAO,EAAE,OAAO,GAAG,QAAQ,GAAG,cAAc,GAAG,MAAM,CAAC;QACtD,SAAS,EAAE,OAAO,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,IAAI,CAAC;IACT,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,YAAY,CAAC,EAAE,SAAS,GAAG,MAAM,GAAG,SAAS,GAAG,UAAU,GAAG,SAAS,CAAC;IACvE,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAgLD,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,kBAAkB,CAAC;IACjC,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,cAAc,EAAE,MAAM,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IACjC,iDAAiD;IACjD,SAAS,EAAE,OAAO,CAAC;IACnB,OAAO,EAAE,aAAa,EAAE,CAAC;IACzB,qEAAqE;IACrE,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,oFAAoF;IACpF,eAAe,EAAE,MAAM,CAAC;IAGxB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,YAAY,EAAE,kBAAkB,GAAG,IAAI,CAAC;IACxC,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,cAAc,EAAE,MAAM,GAAG,IAAI,CAAC;CAC/B;AAQD;;;;;;GAMG;AACH,MAAM,WAAW,uBAAuB;IACtC,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,kBAAkB,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC;IAChD,4EAA4E;IAC5E,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,UAAU,EAAE,oBAAoB,CAAC;CAClC;AAyCD,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAsC;IAC9D,OAAO,CAAC,GAAG,CAA4B;IACvC,OAAO,CAAC,QAAQ,CAAS;IAEzB,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAkC;IAC/D,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAoC;IACnE,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAsC;IACvE,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAmC;IAEjE,OAAO,CAAC,QAAQ,CAAC,eAAe,CAE5B;IAEJ,OAAO,CAAC,QAAQ,CAAC,0BAA0B,CAEvC;IAEJ,gEAAgE;IAChE,OAAO,CAAC,qBAAqB,CAAwC;IAErE,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CACoB;IAEzD,MAAM,IAAI,kBAAkB;IA0C5B,OAAO,CAAC,WAAW;IAWnB,eAAe,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,kBAAkB,KAAK,IAAI,GAAG,MAAM,IAAI;IAO3E,OAAO,CAAC,UAAU;IAWlB;;;;OAIG;IACH,0BAA0B,CACxB,QAAQ,EAAE,CAAC,OAAO,EAAE,uBAAuB,KAAK,IAAI,GACnD,MAAM,IAAI;IAOb,yEAAyE;IACzE,uBAAuB,IAAI,uBAAuB,GAAG,IAAI;IAIzD,OAAO,CAAC,qBAAqB;IAWvB,kBAAkB,CAAC,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IA8B3D,OAAO,CAAC,gBAAgB;IAiExB,OAAO,CAAC,kBAAkB;IA6G1B,OAAO,CAAC,oBAAoB;IA2D5B,OAAO,CAAC,mBAAmB;IA2F3B,OAAO,CAAC,YAAY;IAMpB,wFAAwF;IACxF,OAAO,CAAC,cAAc;IAiBhB,SAAS,CAAC,IAAI,EAAE,sBAAsB,GAAG,OAAO,CAAC,IAAI,CAAC;IAsCtD,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAuClC,gBAAgB,IAAI,MAAM,GAAG,IAAI;IAO3B,KAAK,CAAC,IAAI,EAAE;QAChB,KAAK,EAAE,MAAM,CAAC;KACf,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;IA6D9C,QAAQ,CAAC,IAAI,EAAE;QACnB,MAAM,EAAE,MAAM,CAAC;QACf,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;QACzB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,MAAM,CAAC;IA+DnB,OAAO,CAAC,cAAc;IAItB;;;;;OAKG;YACW,uBAAuB;IAqBrC;;;;;;;;OAQG;YACW,uBAAuB;CA2DtC;AAED,eAAO,MAAM,YAAY,cAAqB,CAAC;AAE/C,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,YAAY,GAAG;IACtB,eAAe,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,KAAK,OAAO,CAAC;CAC5D,GACA,IAAI,CAuBN"}

package/services/local-inference/device-bridge.js

@@ -31,7 +31,8 @@
 import { randomUUID } from "node:crypto";
 import fs from "node:fs/promises";
 import path from "node:path";
-import { logger } from "@elizaos/core";
+import { logger } from "@elizaos/logger";
+import { computeGenerationThroughput, } from "@elizaos/shared/local-inference";
 import { localInferenceRoot } from "./paths.js";
 const DEFAULT_CALL_TIMEOUT_MS = 60_000;
 const DEFAULT_LOAD_TIMEOUT_MS = 120_000;
@@ -93,6 +94,9 @@ export class DeviceBridge {
     pendingGenerates = new Map();
     pendingEmbeds = new Map();
     statusListeners = new Set();
+    generationMetricsListeners = new Set();
+    /** The most recent successful generation's metrics, or null. */
+    lastGenerationMetrics = null;
     expectedPairingToken = process.env.ELIZA_DEVICE_PAIRING_TOKEN?.trim() || null;
     status() {
         const summaries = [];
@@ -157,6 +161,32 @@ export class DeviceBridge {
             }
         }
     }
+    /**
+     * Subscribe to per-generation throughput metrics. Fires once per successful
+     * on-device generation with the differenced prefill/decode tok/s. Returns an
+     * unsubscribe function.
+     */
+    subscribeGenerationMetrics(listener) {
+        this.generationMetricsListeners.add(listener);
+        return () => {
+            this.generationMetricsListeners.delete(listener);
+        };
+    }
+    /** The most recent successful generation's measured metrics, or null. */
+    latestGenerationMetrics() {
+        return this.lastGenerationMetrics;
+    }
+    emitGenerationMetrics(metrics) {
+        this.lastGenerationMetrics = metrics;
+        for (const listener of this.generationMetricsListeners) {
+            try {
+                listener(metrics);
+            }
+            catch {
+                this.generationMetricsListeners.delete(listener);
+            }
+        }
+    }
     async attachToHttpServer(server) {
         if (this.wss)
             return;
@@ -440,6 +470,29 @@ export class DeviceBridge {
                 pending.reject(new Error(msg.error));
             }
             else {
+                // Difference the raw counters into prefill/decode tok/s and surface
+                // them to profiling subscribers. The loader contract is unchanged —
+                // callers still get the text; metrics are a side channel.
+                const ttftMs = typeof msg.ttftMs === "number" ? msg.ttftMs : null;
+                const throughput = computeGenerationThroughput({
+                    promptTokens: msg.promptTokens,
+                    outputTokens: msg.outputTokens,
+                    durationMs: msg.durationMs,
+                    ttftMs,
+                });
+                const device = pending.routedDeviceId
+                    ? this.devices.get(pending.routedDeviceId)
+                    : null;
+                this.emitGenerationMetrics({
+                    deviceId: pending.routedDeviceId ?? "unknown",
+                    platform: device?.capabilities.platform ?? null,
+                    deviceModel: device?.capabilities.deviceModel ?? null,
+                    promptTokens: msg.promptTokens,
+                    outputTokens: msg.outputTokens,
+                    durationMs: msg.durationMs,
+                    ttftMs,
+                    throughput,
+                });
                 pending.resolve(msg.text);
             }
             return;
@@ -681,8 +734,8 @@ export class DeviceBridge {
     /**
      * On startup, read persisted pending requests back into memory. Their
      * promises are gone (the original caller's process is dead) so they can
-     * only be resolved externally — for now we just re-queue them with a
-     * fresh timeout, and the first device that connects will process them.
+     * only be resolved externally, so we re-queue them with a fresh timeout.
+     * The first connected device that can handle generation will process them.
      * If nothing consumes them within the timeout they reject quietly.
      *
      * Stale entries older than 24h are purged rather than resurrected.

package/services/local-inference/disk-space.d.ts

@@ -0,0 +1,15 @@
+export type DiskSpaceWarning = "low-disk" | "critical-disk";
+export interface DiskSpace {
+    path: string;
+    totalBytes: number;
+    freeBytes: number;
+    availableBytes: number;
+}
+export interface DiskSpaceAdvice {
+    warning?: DiskSpaceWarning;
+    requiredBytes: number;
+    recommendedFreeBytes: number;
+}
+export declare function probeDiskSpace(path: string): Promise<DiskSpace>;
+export declare function adviseDiskSpace(probe: DiskSpace, modelSizeBytes: number): DiskSpaceAdvice;
+//# sourceMappingURL=disk-space.d.ts.map

package/services/local-inference/disk-space.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"disk-space.d.ts","sourceRoot":"","sources":["../../../../../../src/services/local-inference/disk-space.ts"],"names":[],"mappings":"AAMA,MAAM,MAAM,gBAAgB,GAAG,UAAU,GAAG,eAAe,CAAC;AAE5D,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,EAAE,gBAAgB,CAAC;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,oBAAoB,EAAE,MAAM,CAAC;CAC9B;AAED,wBAAsB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAYrE;AAED,wBAAgB,eAAe,CAC7B,KAAK,EAAE,SAAS,EAChB,cAAc,EAAE,MAAM,GACrB,eAAe,CAcjB"}

package/services/local-inference/disk-space.js

@@ -0,0 +1,29 @@
+import { statfsSync } from "node:fs";
+const BYTES_PER_GB = 1024 ** 3;
+const DOWNLOAD_SAFETY_MARGIN_BYTES = 2 * BYTES_PER_GB;
+const LOW_DISK_MULTIPLIER = 1.5;
+export async function probeDiskSpace(path) {
+    const stats = statfsSync(path);
+    const blockSize = Number(stats.bsize);
+    const totalBytes = Number(stats.blocks) * blockSize;
+    const _freeBytes = Number(stats.bfree) * blockSize;
+    const availableBytes = Number(stats.bavail) * blockSize;
+    return {
+        path,
+        totalBytes,
+        freeBytes: availableBytes,
+        availableBytes,
+    };
+}
+export function adviseDiskSpace(probe, modelSizeBytes) {
+    const requiredBytes = modelSizeBytes + DOWNLOAD_SAFETY_MARGIN_BYTES;
+    const recommendedFreeBytes = Math.ceil(modelSizeBytes * LOW_DISK_MULTIPLIER);
+    const freeBytes = probe.freeBytes;
+    if (freeBytes < requiredBytes) {
+        return { warning: "critical-disk", requiredBytes, recommendedFreeBytes };
+    }
+    if (freeBytes < recommendedFreeBytes) {
+        return { warning: "low-disk", requiredBytes, recommendedFreeBytes };
+    }
+    return { requiredBytes, recommendedFreeBytes };
+}

package/services/local-inference/downloader.d.ts

@@ -14,14 +14,48 @@
  * The runtime `fetch` follows HuggingFace redirects and still gives us a body
  * stream that can be piped into a Node WriteStream.
  */
+import { type Eliza1DeviceCaps } from "./manifest";
 import type { CatalogModel, DownloadEvent, DownloadJob } from "./types";
 type DownloadListener = (event: DownloadEvent) => void;
+/**
+ * Thrown before any weight byte is fetched when an Eliza-1 bundle's manifest
+ * is incompatible with this device — wrong schema version, no overlapping
+ * verified backend, or a RAM budget that exceeds the device's memory. Per
+ * `packages/inference/AGENTS.md` §7 there is no "download anyway" path.
+ */
+export declare class BundleIncompatibleError extends Error {
+    readonly code: "ELIZA1_BUNDLE_INCOMPATIBLE";
+    constructor(message: string);
+}
+/**
+ * One-time verify-on-device pass per `packages/inference/AGENTS.md` §7:
+ * load → 1-token text generation → 1-phrase voice generation → barge-in
+ * cancel. The downloader stays decoupled from the engine — the service
+ * layer injects this; when absent the bundle is materialized and registered
+ * but its `bundleVerifiedAt` stays unset and it does NOT auto-fill an empty
+ * default slot (an unverified bundle must not become the recommended
+ * default).
+ */
+export type VerifyBundleOnDevice = (args: {
+    modelId: string;
+    bundleRoot: string;
+    manifestPath: string;
+    textGgufPath: string;
+}) => Promise<void>;
+export interface DownloaderOptions {
+    /** Override the device-capability probe (tests / headless environments). */
+    probeDeviceCaps?: () => Promise<Eliza1DeviceCaps>;
+    /** Verify-on-device smoke run; see {@link VerifyBundleOnDevice}. */
+    verifyOnDevice?: VerifyBundleOnDevice;
+}
 export declare class Downloader {
     private readonly active;
     private readonly terminal;
     private readonly listeners;
     private readonly lastEmit;
-    constructor();
+    private readonly probeDeviceCaps;
+    private readonly verifyOnDevice?;
+    constructor(options?: DownloaderOptions);
     subscribe(listener: DownloadListener): () => void;
     snapshot(): DownloadJob[];
     isActive(modelId: string): boolean;

package/services/local-inference/downloader.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"downloader.d.ts","sourceRoot":"","sources":["../../../../../../src/services/local-inference/downloader.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAoBH,OAAO,KAAK,EACV,YAAY,EACZ,aAAa,EACb,WAAW,EAGZ,MAAM,SAAS,CAAC;AAUjB,KAAK,gBAAgB,GAAG,CAAC,KAAK,EAAE,aAAa,KAAK,IAAI,CAAC;AAsPvD,qBAAa,UAAU;IACrB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAgC;IACvD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAkC;IAC3D,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA+B;IACzD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA6B;;IAMtD,SAAS,CAAC,QAAQ,EAAE,gBAAgB,GAAG,MAAM,IAAI;IAOjD,QAAQ,IAAI,WAAW,EAAE;IAUzB,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQlC;;;;OAIG;IACG,KAAK,CAAC,aAAa,EAAE,MAAM,GAAG,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IA2DvE,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAchC,OAAO,CAAC,IAAI;IAWZ,OAAO,CAAC,WAAW;IAKnB,OAAO,CAAC,qBAAqB;IAwB7B,OAAO,CAAC,wBAAwB;IAkBhC,OAAO,CAAC,wBAAwB;IAYhC,OAAO,CAAC,qBAAqB;IAK7B,OAAO,CAAC,YAAY;YAQN,MAAM;YAgKN,YAAY;YAgJZ,kBAAkB;YAiHlB,cAAc;CAkC7B"}
+{"version":3,"file":"downloader.d.ts","sourceRoot":"","sources":["../../../../../../src/services/local-inference/downloader.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAgBH,OAAO,EACL,KAAK,gBAAgB,EAMtB,MAAM,YAAY,CAAC;AAOpB,OAAO,KAAK,EACV,YAAY,EACZ,aAAa,EACb,WAAW,EAGZ,MAAM,SAAS,CAAC;AAUjB,KAAK,gBAAgB,GAAG,CAAC,KAAK,EAAE,aAAa,KAAK,IAAI,CAAC;AAGvD;;;;;GAKG;AACH,qBAAa,uBAAwB,SAAQ,KAAK;IAChD,QAAQ,CAAC,IAAI,EAAG,4BAA4B,CAAU;gBAC1C,OAAO,EAAE,MAAM;CAI5B;AAED;;;;;;;;GAQG;AACH,MAAM,MAAM,oBAAoB,GAAG,CAAC,IAAI,EAAE;IACxC,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;CACtB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;AAEpB,MAAM,WAAW,iBAAiB;IAChC,4EAA4E;IAC5E,eAAe,CAAC,EAAE,MAAM,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAClD,oEAAoE;IACpE,cAAc,CAAC,EAAE,oBAAoB,CAAC;CACvC;AAqLD,qBAAa,UAAU;IACrB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAgC;IACvD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAkC;IAC3D,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA+B;IACzD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA6B;IACtD,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAkC;IAClE,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAuB;gBAE3C,OAAO,GAAE,iBAAsB;IAM3C,SAAS,CAAC,QAAQ,EAAE,gBAAgB,GAAG,MAAM,IAAI;IAOjD,QAAQ,IAAI,WAAW,EAAE;IAUzB,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAQlC;;;;OAIG;IACG,KAAK,CAAC,aAAa,EAAE,MAAM,GAAG,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IA2DvE,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAchC,OAAO,CAAC,IAAI;IAWZ,OAAO,CAAC,WAAW;IAKnB,OAAO,CAAC,qBAAqB;IAwB7B,OAAO,CAAC,wBAAwB;IAkBhC,OAAO,CAAC,wBAAwB;IAYhC,OAAO,CAAC,qBAAqB;IAK7B,OAAO,CAAC,YAAY;YAQN,MAAM;YAsIN,YAAY;YAwIZ,kBAAkB;YAiHlB,cAAc;CAkC7B"}

package/services/local-inference/downloader.js

@@ -21,10 +21,52 @@ import path from "node:path";
 import { Readable } from "node:stream";
 import { pipeline } from "node:stream/promises";
 import { ensureDefaultAssignment } from "./assignments.js";
-import { buildHuggingFaceResolveUrl, buildHuggingFaceResolveUrlForPath, findCatalogModel, } from "./catalog.js";
+import { buildHuggingFaceResolveUrl, buildHuggingFaceResolveUrlForPath, findCatalogModel, isDefaultEligibleId, } from "./catalog.js";
+import { deviceCapsFromProbe, probeHardware } from "./hardware.js";
+import { parseManifestOrThrow, SUPPORTED_BACKENDS_BY_TIER, } from "./manifest.js";
 import { downloadsStagingDir, elizaModelsDir, localInferenceRoot, } from "./paths.js";
 import { upsertElizaModel } from "./registry.js";
 import { hashFile } from "./verify.js";
+/**
+ * Thrown before any weight byte is fetched when an Eliza-1 bundle's manifest
+ * is incompatible with this device — wrong schema version, no overlapping
+ * verified backend, or a RAM budget that exceeds the device's memory. Per
+ * `packages/inference/AGENTS.md` §7 there is no "download anyway" path.
+ */
+export class BundleIncompatibleError extends Error {
+    code = "ELIZA1_BUNDLE_INCOMPATIBLE";
+    constructor(message) {
+        super(message);
+        this.name = "BundleIncompatibleError";
+    }
+}
+async function defaultProbeDeviceCaps() {
+    return deviceCapsFromProbe(await probeHardware());
+}
+/**
+ * Reject bundles this device cannot run — runs against the manifest before
+ * any weight byte is fetched. Mirrors the publish-side `canSetAsDefault`
+ * device check, minus the `defaultEligible` flag (a user may explicitly
+ * install a non-default bundle, but only one the device can actually load).
+ */
+function assertBundleInstallable(manifest, device) {
+    // Schema version is enforced upstream by `parseManifestOrThrow` -- the Zod
+    // schema only accepts the current `$schema` URL, so a manifest with an
+    // unsupported schema version is rejected before we get here.
+    if (manifest.ramBudgetMb.min > device.ramMb) {
+        throw new BundleIncompatibleError(`Eliza-1 bundle ${manifest.id} needs at least ${manifest.ramBudgetMb.min} MB RAM; this device has ${device.ramMb} MB`);
+    }
+    const tierBackends = new Set(SUPPORTED_BACKENDS_BY_TIER[manifest.tier]);
+    const usable = device.availableBackends.filter((b) => tierBackends.has(b) &&
+        manifest.kernels.verifiedBackends[b].status === "pass");
+    if (usable.length === 0) {
+        const verified = Object.entries(manifest.kernels.verifiedBackends)
+            .filter(([, v]) => v.status === "pass")
+            .map(([b]) => b);
+        throw new BundleIncompatibleError(`Eliza-1 bundle ${manifest.id}: no required-kernel backend is available on this device. ` +
+            `bundle verified [${verified.join(", ") || "none"}], device has [${device.availableBackends.join(", ")}], tier ${manifest.tier} supports [${[...tierBackends].join(", ")}]`);
+    }
+}
 const PROGRESS_THROTTLE_MS = 250;
 const TERMINAL_DOWNLOADS_FILENAME = "download-status.json";
 const TERMINAL_DOWNLOAD_LIMIT = 32;
@@ -75,80 +117,15 @@ function bundleTargetPath(root, filePath) {
     }
     return target;
 }
-function parseBundleFileEntry(value, kind) {
-    if (!value || typeof value !== "object") {
-        throw new Error(`Invalid Eliza-1 manifest file entry in files.${kind}`);
-    }
-    const entry = value;
-    if (typeof entry.path !== "string" || entry.path.length === 0) {
-        throw new Error(`Invalid Eliza-1 manifest file path in files.${kind}`);
-    }
-    if (typeof entry.sha256 !== "string" ||
-        !/^[a-f0-9]{64}$/.test(entry.sha256)) {
-        throw new Error(`Invalid Eliza-1 manifest sha256 for ${entry.path}`);
-    }
-    const parsed = {
-        path: entry.path,
-        sha256: entry.sha256,
-    };
-    if (typeof entry.ctx === "number")
-        parsed.ctx = entry.ctx;
-    return parsed;
-}
 function parseBundleManifestOrThrow(input, catalogEntry) {
-    if (!input || typeof input !== "object") {
-        throw new Error("Invalid Eliza-1 manifest: expected object");
-    }
-    const raw = input;
-    if (raw.id !== catalogEntry.id) {
-        throw new Error(`Invalid Eliza-1 manifest: id ${String(raw.id)} does not match ${catalogEntry.id}`);
-    }
-    if (typeof raw.version !== "string" || raw.version.length === 0) {
-        throw new Error("Invalid Eliza-1 manifest: missing version");
-    }
-    if (raw.defaultEligible !== true) {
-        throw new Error("Invalid Eliza-1 manifest: defaultEligible must be true");
-    }
-    if (!raw.files || typeof raw.files !== "object") {
-        throw new Error("Invalid Eliza-1 manifest: missing files");
-    }
-    const filesRaw = raw.files;
-    const files = {};
-    for (const kind of [
-        "text",
-        "voice",
-        "asr",
-        "vision",
-        "dflash",
-        "cache",
-        "embedding",
-        "vad",
-        "wakeword",
-    ]) {
-        const value = filesRaw[kind];
-        if (value === undefined &&
-            (kind === "embedding" || kind === "vad" || kind === "wakeword")) {
-            files[kind] = [];
-            continue;
-        }
-        if (!Array.isArray(value)) {
-            throw new Error(`Invalid Eliza-1 manifest: files.${kind} must be an array`);
-        }
-        files[kind] = value.map((entry) => parseBundleFileEntry(entry, kind));
+    const manifest = parseManifestOrThrow(input);
+    if (manifest.id !== catalogEntry.id) {
+        throw new Error(`Invalid Eliza-1 manifest: id ${manifest.id} does not match ${catalogEntry.id}`);
     }
-    for (const kind of ["text", "voice", "dflash", "cache"]) {
-        if (files[kind].length === 0) {
-            throw new Error(`Invalid Eliza-1 manifest: files.${kind} must be non-empty`);
-        }
-    }
-    if (!files.text.some((entry) => entry.path === catalogEntry.ggufFile)) {
+    if (!manifest.files.text.some((entry) => entry.path === catalogEntry.ggufFile)) {
         throw new Error(`Invalid Eliza-1 manifest: primary text file ${catalogEntry.ggufFile} is missing`);
     }
-    return {
-        id: catalogEntry.id,
-        version: raw.version,
-        files,
-    };
+    return manifest;
 }
 function collectBundleFiles(manifest) {
     const seen = new Map();
@@ -157,13 +134,12 @@ function collectBundleFiles(manifest) {
         "voice",
         "asr",
         "vision",
-        "dflash",
         "cache",
         "embedding",
         "vad",
         "wakeword",
     ]) {
-        for (const entry of manifest.files[kind]) {
+        for (const entry of manifest.files[kind] ?? []) {
             const current = seen.get(entry.path);
             if (current && current.entry.sha256 !== entry.sha256) {
                 throw new Error(`Conflicting sha256 entries for bundle file ${entry.path}`);
@@ -194,7 +170,11 @@ export class Downloader {
     terminal = new Map();
     listeners = new Set();
     lastEmit = new Map();
-    constructor() {
+    probeDeviceCaps;
+    verifyOnDevice;
+    constructor(options = {}) {
+        this.probeDeviceCaps = options.probeDeviceCaps ?? defaultProbeDeviceCaps;
+        this.verifyOnDevice = options.verifyOnDevice;
         this.loadTerminalDownloads();
     }
     subscribe(listener) {
@@ -299,7 +279,7 @@ export class Downloader {
         try {
             const raw = fs.readFileSync(terminalDownloadsPath(), "utf8");
             const parsed = JSON.parse(raw);
-            if (!parsed || parsed.version !== 1 || !Array.isArray(parsed.jobs)) {
+            if (parsed?.version !== 1 || !Array.isArray(parsed.jobs)) {
                 return;
             }
             for (const job of parsed.jobs) {
@@ -356,8 +336,7 @@ export class Downloader {
     async runJob(catalogEntry, record) {
         try {
             this.updateState(record, "downloading");
-            if (catalogEntry.bundleManifestFile &&
-                catalogEntry.runtimeRole !== "dflash-drafter") {
+            if (catalogEntry.bundleManifestFile) {
                 await this.runBundleJob(catalogEntry, record);
                 return;
             }
@@ -376,17 +355,22 @@ export class Downloader {
                 signal: record.abortController.signal,
             });
             if (response.statusCode >= 400) {
-                throw new Error(`HTTP ${response.statusCode} from HuggingFace for ${catalogEntry.hfRepo}`);
+                throw new Error(`HTTP ${response.statusCode} from model hub for ${catalogEntry.hfRepo}`);
+            }
+            let effectiveStartByte = startByte;
+            if (effectiveStartByte > 0 && response.statusCode !== 206) {
+                effectiveStartByte = 0;
+                record.job.received = 0;
             }
             const contentLengthHeader = response.headers["content-length"];
             const contentLength = Array.isArray(contentLengthHeader)
                 ? Number.parseInt(contentLengthHeader[0] ?? "0", 10)
                 : Number.parseInt(contentLengthHeader ?? "0", 10);
             if (Number.isFinite(contentLength) && contentLength > 0) {
-                record.job.total = startByte + contentLength;
+                record.job.total = effectiveStartByte + contentLength;
             }
             const writeStream = fs.createWriteStream(record.stagingPath, {
-                flags: startByte > 0 ? "a" : "w",
+                flags: effectiveStartByte > 0 ? "a" : "w",
             });
             let lastSampleBytes = record.job.received;
             let lastSampleAt = Date.now();
@@ -432,41 +416,14 @@ export class Downloader {
                 ...(catalogEntry.runtimeRole
                     ? { runtimeRole: catalogEntry.runtimeRole }
                     : {}),
-                ...(catalogEntry.companionForModelId
-                    ? { companionFor: catalogEntry.companionForModelId }
-                    : {}),
             };
             await upsertElizaModel(installed);
-            // First-light convenience: assign the freshly-installed model to any
-            // empty slot so chat works without a Settings detour. Idempotent —
-            // ignores slots the user has already configured. See
-            // assignments.ts#ensureDefaultAssignment for the per-slot policy.
-            if (catalogEntry.runtimeRole !== "dflash-drafter") {
+            // First-light convenience: only default-eligible Eliza-1 downloads
+            // can fill empty slots. Ad-hoc Hugging Face downloads remain
+            // explicit opt-in even though Eliza owns the downloaded file.
+            if (isDefaultEligibleId(installed.id)) {
                 await ensureDefaultAssignment(installed.id);
             }
-            for (const companionId of catalogEntry.companionModelIds ?? []) {
-                if (!this.isActive(companionId)) {
-                    void this.start(companionId).catch((err) => {
-                        const job = {
-                            jobId: randomUUID(),
-                            modelId: companionId,
-                            state: "failed",
-                            received: 0,
-                            total: 0,
-                            bytesPerSec: 0,
-                            etaMs: null,
-                            startedAt: new Date().toISOString(),
-                            updatedAt: new Date().toISOString(),
-                            error: err instanceof Error ? err.message : String(err),
-                        };
-                        this.rememberTerminalDownload(job);
-                        this.emit({
-                            type: "failed",
-                            job,
-                        });
-                    });
-                }
-            }
             this.updateState(record, "completed");
             record.job.received = finalStat.size;
             record.job.total = finalStat.size;
@@ -497,8 +454,12 @@ export class Downloader {
         const bundleRoot = path.join(elizaModelsDir(), bundleDirname(catalogEntry.id));
         await fsp.mkdir(bundleRoot, { recursive: true });
         const manifestPath = bundleTargetPath(bundleRoot, catalogEntry.bundleManifestFile);
-        const manifestDownloaded = await this.downloadRemotePath(catalogEntry, catalogEntry.bundleManifestFile, path.join(downloadsStagingDir(), bundleStagingFilename(catalogEntry.id, catalogEntry.bundleManifestFile)), manifestPath, record, 0);
+        const manifestDownloaded = await this.downloadRemotePath(catalogEntry, catalogEntry.bundleManifestFile, path.join(downloadsStagingDir(), bundleStagingFilename(catalogEntry.id, catalogEntry.bundleManifestFile)), manifestPath, record, 0, catalogEntry.bundleManifestSha256);
         const manifest = parseBundleManifestOrThrow(JSON.parse(await fsp.readFile(manifestPath, "utf8")), catalogEntry);
+        // §7: schema version, RAM budget, and kernel-backend availability are
+        // checked against this device BEFORE any weight byte is fetched. An
+        // incompatible bundle aborts here — there is no "download anyway" path.
+        assertBundleInstallable(manifest, await this.probeDeviceCaps());
         let completedBytes = manifestDownloaded.sizeBytes;
         const downloaded = new Map();
         for (const { entry } of collectBundleFiles(manifest)) {
@@ -518,6 +479,21 @@ export class Downloader {
         if (!textFile) {
             throw new Error(`[local-inference] Bundle did not install text file ${textEntry.path}`);
         }
+        // §7: materialize the bundle, then run the one-time verify-on-device
+        // pass before the bundle is treated as ready. The hook is injected by
+        // the service layer so the downloader stays decoupled from the engine.
+        // When no hook is wired, `bundleVerifiedAt` stays unset and the bundle
+        // is registered but does NOT auto-fill an empty default slot.
+        let bundleVerifiedAt;
+        if (this.verifyOnDevice) {
+            await this.verifyOnDevice({
+                modelId: catalogEntry.id,
+                bundleRoot,
+                manifestPath,
+                textGgufPath: textFile.path,
+            });
+            bundleVerifiedAt = new Date().toISOString();
+        }
         const now = new Date().toISOString();
         const bundleMeta = {
             bundleRoot,
@@ -525,6 +501,7 @@ export class Downloader {
             manifestSha256: manifestDownloaded.sha256,
             bundleVersion: manifest.version,
             bundleSizeBytes: completedBytes,
+            ...(bundleVerifiedAt ? { bundleVerifiedAt } : {}),
         };
         const installed = {
             id: catalogEntry.id,
@@ -540,35 +517,12 @@ export class Downloader {
             ...bundleMeta,
         };
         await upsertElizaModel(installed);
-        const companionId = catalogEntry.runtime?.dflash?.drafterModelId ??
-            catalogEntry.companionModelIds?.[0];
-        const companion = companionId ? findCatalogModel(companionId) : undefined;
-        if (companion) {
-            const drafterEntry = manifest.files.dflash.find((entry) => entry.path === companion.ggufFile);
-            if (!drafterEntry) {
-                throw new Error(`[local-inference] Bundle missing DFlash companion ${companion.ggufFile}`);
-            }
-            const drafterFile = downloaded.get(drafterEntry.path);
-            if (!drafterFile) {
-                throw new Error(`[local-inference] Bundle did not install DFlash companion ${drafterEntry.path}`);
-            }
-            await upsertElizaModel({
-                id: companion.id,
-                displayName: companion.displayName,
-                path: drafterFile.path,
-                sizeBytes: drafterFile.sizeBytes,
-                hfRepo: companion.hfRepo,
-                installedAt: now,
-                lastUsedAt: null,
-                source: "eliza-download",
-                sha256: drafterFile.sha256,
-                lastVerifiedAt: now,
-                runtimeRole: "dflash-drafter",
-                companionFor: catalogEntry.id,
-                ...bundleMeta,
-            });
+        // An empty default slot is filled only after the on-device verify pass
+        // succeeds. Without a verify hook the bundle is installed and visible,
+        // but it is not allowed to auto-fill defaults.
+        if (isDefaultEligibleId(installed.id) && bundleVerifiedAt !== undefined) {
+            await ensureDefaultAssignment(installed.id);
         }
-        await ensureDefaultAssignment(installed.id);
         this.updateState(record, "completed");
         record.job.received = completedBytes;
         record.job.total = completedBytes;
@@ -618,7 +572,7 @@ export class Downloader {
             signal: record.abortController.signal,
         });
         if (response.statusCode >= 400) {
-            throw new Error(`HTTP ${response.statusCode} from HuggingFace for ${catalogEntry.hfRepo}/${remotePath}`);
+            throw new Error(`HTTP ${response.statusCode} from model hub for ${catalogEntry.hfRepo}/${remotePath}`);
         }
         if (startByte > 0 && response.statusCode !== 206) {
             startByte = 0;