@atomic-chat/core 0.5.1 → 0.13.2

package/dist/core/managed-environment.d.ts

@@ -0,0 +1,72 @@
+import type { DataLayout } from '../config/index.js';
+import type { CoreEvents } from '../contracts/index.js';
+import type { ManagedContainersHandle, ReconcileLogger } from '../runtime/container/index.js';
+import { WindowsEnvironmentRecordStore } from '../runtime/environment/index.js';
+import type { GuestRecipeRunner, LinuxHost, LinuxProvisionerParts, ManagedRuntimes, UnloadEngineSessions, WindowsHost, WindowsProvisionerParts } from '../runtime/environment/index.js';
+import { type DistributionKeeper, type GuestMount } from '../runtime/wsl/index.js';
+import type { DescriptorSource } from '../runtime/environment/index.js';
+import type { ManagedStoreMigration } from '../contracts/index.js';
+import type { WindowsManagedContext } from './managed-engines.js';
+export interface WireManagedEnvironmentOptions {
+    env: NodeJS.ProcessEnv;
+    platform: NodeJS.Platform;
+    layout: DataLayout;
+    instanceId: string;
+    emit: <K extends keyof CoreEvents>(name: K, payload: CoreEvents[K]) => void;
+    newId: () => string;
+    log: ReconcileLogger;
+    onWarn?: (message: string) => void;
+    fetch?: typeof fetch;
+    /** The managed engine registry's descriptor sources (change `add-vllm-runtime`, D2); TensorRT-LLM alone when omitted. */
+    engines?: readonly DescriptorSource[] | undefined;
+    /** The move of TensorRT-LLM's models into the managed model store, for the diagnostics (D5). */
+    storeMigration?: (() => ManagedStoreMigration | null) | undefined;
+    /** `AtomicCoreOptions.dockerPath`: the docker CLI, or null for none. Omitted: the system directories. */
+    dockerPath?: string | null;
+    unloadEngineSessions?: UnloadEngineSessions;
+}
+export interface ManagedEnvironment {
+    managed: ManagedRuntimes;
+    /** The one Docker executor of this core; `resolve()` it at startup, before `managed.recover()`. */
+    containers: ManagedContainersHandle;
+    /** The platform the managed runtime runs on: `linux` on the test host, whatever the real one is. */
+    platform: NodeJS.Platform;
+    /** The machine the environment probes; the `tensorrt-llm` provider asks it for its cards too. */
+    host: LinuxHost;
+    /** `process.arch`, or `x64` on the Windows test machine (which stands in for a Windows x64 PC). */
+    arch: string;
+    /** Windows only: what the `tensorrt-llm` provider reaches Atomic Chat's WSL distribution through. */
+    windows?: WindowsManagedContext;
+}
+/**
+ * The Linux recipe's host-side parts: the machine, the privileged recipe, the one Docker executor,
+ * and this scope's engine caches and models.
+ */
+export declare function linuxProvisionerParts(options: Pick<WireManagedEnvironmentOptions, 'layout' | 'unloadEngineSessions'>, host: LinuxHost, containers: ManagedContainersHandle): LinuxProvisionerParts;
+/**
+ * Tests only: the guest recipe as the Windows test machine applies it (`atomic-test-recipe` in
+ * `fake-wsl.mjs`), the way the Linux e2e's fake privileged executor stands in for the real recipe —
+ * the recipe itself is the Linux one, covered where it runs.
+ */
+export declare const testGuestRecipe: GuestRecipeRunner;
+export interface WindowsPartsInput {
+    layout: DataLayout;
+    fetch: typeof fetch;
+    unloadEngineSessions: UnloadEngineSessions;
+    host: WindowsHost;
+    records: Pick<WindowsEnvironmentRecordStore, 'read' | 'write' | 'remove'>;
+    containers: Pick<ManagedContainersHandle, 'current'>;
+    mount: GuestMount;
+    keeper: DistributionKeeper;
+    scopeKey: () => Promise<string>;
+    /** The `ATOMIC_MANAGED_TEST_WINDOWS` machine: its guest recipe is `testGuestRecipe`. */
+    test: boolean;
+}
+/**
+ * The Windows provisioner's parts over this core's objects (change `add-tensorrt-llm-windows`, task
+ * 2.10): the rootfs download, the guest recipe, the journal of the one executor (wired or not yet),
+ * and this scope's caches and models removed by the guest's own `rm` — only ever this scope's folder,
+ * through the same mount the provider uses.
+ */
+export declare function windowsProvisionerParts(input: WindowsPartsInput): WindowsProvisionerParts;
+export declare function wireManagedEnvironment(options: WireManagedEnvironmentOptions): ManagedEnvironment;

package/dist/core/managed-environment.js

@@ -0,0 +1,218 @@
+/**
+ * Composition of the managed container environment (task 2.6): the pieces `wireManagedRuntimes`
+ * needs from outside its own module — the Linux machine, the privileged recipe from `src/host`,
+ * the one Docker executor of this core, this scope's engine caches and models — put together once
+ * for `create.ts`.
+ *
+ * The Docker executor is the startup handle (`createManagedContainersHandle`): core startup resolves
+ * it before recovery runs (so a recovered pull has it), the setup uses it, and the managed-text
+ * provider (task 2.14) must take the same handle rather than wire a second one.
+ *
+ * `unloadEngineSessions` is the `tensorrt-llm` provider's (`tensorrtLlmSessionUnloader`, task 2.14):
+ * a removal unloads a loaded session of the engine first, with its container's stop confirmed. Left
+ * out, the default (`NOTHING_LOADED`) says nothing is loaded.
+ *
+ * `ATOMIC_MANAGED_TEST_HOST` (see `linux-host.ts`) swaps the whole Linux machine for a folder the
+ * e2e suite describes, including on a non-Linux test runner; production never sets it. This is the
+ * one place it is read: the returned `platform` and `host` carry it to the `tensorrt-llm` provider.
+ */
+import { rm } from 'node:fs/promises';
+import { join } from 'node:path';
+import { managedSharedPaths, managedSharedRoot, nodeDataFolderEnv } from '../config/index.js';
+import { createGuestRecipeRunner, ENABLE_WSL_BINDING, INSTALL_CONTAINER_RUNTIME_BINDING, } from '../host/recipes/index.js';
+import { createManagedContainersHandle } from '../runtime/container/index.js';
+import { ATOMIC_CHAT_DISTRIBUTION, downloadVerifiedRootfs, managedTestHostDir, managedTestWindowsDir, NOTHING_LOADED, realLinuxHost, realWindowsHost, testLinuxHost, testWindowsHost, WindowsEnvironmentRecordStore, wireManagedRuntimes, } from '../runtime/environment/index.js';
+import { removeEngineCaches } from '../runtime/managed-text/index.js';
+import { guestModelFiles, guestModelsRoot } from '../runtime/managed-models/index.js';
+import { createDistributionKeeper, directoryGuestMount, guestScopePaths, guestScopeKeyReader, WSL_LOCALHOST_MOUNT, } from '../runtime/wsl/index.js';
+/**
+ * The Linux recipe's host-side parts: the machine, the privileged recipe, the one Docker executor,
+ * and this scope's engine caches and models.
+ */
+export function linuxProvisionerParts(options, host, containers) {
+    return {
+        host,
+        recipe: INSTALL_CONTAINER_RUNTIME_BINDING,
+        docker: async () => {
+            const wired = await containers.resolve();
+            return wired === null
+                ? null
+                : { exec: wired.exec, socketPath: wired.socketPath, journal: wired.journal };
+        },
+        removeEngineCaches: async (descriptorId) => {
+            await removeEngineCaches(options.layout.managed, { descriptorId });
+        },
+        // The managed model store, every engine's (change `add-vllm-runtime`, D13): the provisioner asks only
+        // when no other engine's installation is left.
+        removeStoreModels: () => rm(options.layout.managedModelsDir, { recursive: true, force: true }),
+        unloadEngineSessions: options.unloadEngineSessions ?? NOTHING_LOADED,
+    };
+}
+/**
+ * Tests only: the guest recipe as the Windows test machine applies it (`atomic-test-recipe` in
+ * `fake-wsl.mjs`), the way the Linux e2e's fake privileged executor stands in for the real recipe —
+ * the recipe itself is the Linux one, covered where it runs.
+ */
+export const testGuestRecipe = async (transport, request) => {
+    const applied = await transport.exec(['atomic-test-recipe', ...request.parameters.components], {
+        user: 'root',
+    });
+    return {
+        outcome: applied.code === 0 ? 'completed' : 'failed',
+        log_tail: `${applied.stdout}${applied.stderr}`.trim(),
+    };
+};
+/**
+ * The Windows provisioner's parts over this core's objects (change `add-tensorrt-llm-windows`, task
+ * 2.10): the rootfs download, the guest recipe, the journal of the one executor (wired or not yet),
+ * and this scope's caches and models removed by the guest's own `rm` — only ever this scope's folder,
+ * through the same mount the provider uses.
+ */
+export function windowsProvisionerParts(input) {
+    const wsl = input.host.probeDeps.wsl;
+    /** This scope's guest folder, through the mount, for the guest's own `rm`; null before the import. */
+    const guestFiles = async () => {
+        const record = await input.records.read();
+        if (record === null)
+            return null;
+        const name = record.distribution.name;
+        const key = await input.scopeKey();
+        return {
+            paths: guestScopePaths(input.layout.managed, name, key, input.mount),
+            store: input.mount.hostPath(name, guestModelsRoot(key)),
+            files: guestModelFiles(wsl.distribution(name), input.mount),
+        };
+    };
+    return {
+        host: input.host,
+        records: input.records,
+        guestRecipe: INSTALL_CONTAINER_RUNTIME_BINDING,
+        enableWsl: ENABLE_WSL_BINDING,
+        downloadRootfs: (rootfs, destination, signal) => downloadVerifiedRootfs(input.fetch, rootfs, destination, signal),
+        removeFile: (path) => rm(path, { force: true }),
+        runGuestRecipe: input.test ? testGuestRecipe : createGuestRecipeRunner({ fetch: input.fetch }),
+        journal: {
+            list: () => input.containers.current()?.journal.list() ?? [],
+            remove: async (id) => {
+                await input.containers.current()?.journal.remove(id);
+            },
+        },
+        removeEngineCaches: async (descriptorId) => {
+            const files = await guestFiles();
+            if (files !== null)
+                await files.files.remove([files.paths.descriptorCachesDir(descriptorId)]);
+        },
+        removeStoreModels: async () => {
+            const files = await guestFiles();
+            if (files !== null)
+                await files.files.remove([files.store]);
+        },
+        unloadEngineSessions: input.unloadEngineSessions,
+        fetch: input.fetch,
+        keeper: input.keeper,
+    };
+}
+/**
+ * The Windows half (change `add-tensorrt-llm-windows`, task 2.10): the machine and WSL (the real ones,
+ * or the `ATOMIC_MANAGED_TEST_WINDOWS` stand-in), the environment record under the shared root, this
+ * scope's `scope_key`, one keeper per distribution, the one Docker executor through the guest, and the
+ * provisioner's parts — all over the same objects the `tensorrt-llm` provider gets in its context.
+ */
+function wireWindowsEnvironment(options, testDir) {
+    const host = testDir === null ? realWindowsHost(options.env) : testWindowsHost(testDir);
+    const wsl = host.probeDeps.wsl;
+    const mount = testDir === null ? WSL_LOCALHOST_MOUNT : directoryGuestMount(join(testDir, 'guest-fs'));
+    const shared = managedSharedPaths(managedSharedRoot(nodeDataFolderEnv(options.env)));
+    const records = new WindowsEnvironmentRecordStore(shared.environmentFile);
+    const keepers = new Map();
+    const keeper = (name) => {
+        let existing = keepers.get(name);
+        if (existing === undefined) {
+            existing = createDistributionKeeper(wsl.distribution(name));
+            keepers.set(name, existing);
+        }
+        return existing;
+    };
+    const scopeKey = guestScopeKeyReader(options.layout.managed.guestScopeFile);
+    const guest = async () => {
+        const record = await records.read().catch(() => null);
+        return record === null ? null : wsl.distribution(record.distribution.name);
+    };
+    const containers = createManagedContainersHandle({
+        platform: 'win32',
+        layout: options.layout,
+        instanceId: options.instanceId,
+        log: options.log,
+        guest,
+    });
+    const parts = windowsProvisionerParts({
+        layout: options.layout,
+        fetch: options.fetch ?? fetch,
+        unloadEngineSessions: options.unloadEngineSessions ?? NOTHING_LOADED,
+        host,
+        records,
+        containers,
+        mount,
+        keeper: keeper(ATOMIC_CHAT_DISTRIBUTION),
+        scopeKey,
+        test: testDir !== null,
+    });
+    const managed = wireManagedRuntimes({
+        env: nodeDataFolderEnv(options.env),
+        instanceId: options.instanceId,
+        platform: 'win32',
+        emit: options.emit,
+        newId: options.newId,
+        ...(options.onWarn === undefined ? {} : { onWarn: options.onWarn }),
+        ...(options.fetch === undefined ? {} : { fetch: options.fetch }),
+        engines: options.engines,
+        storeMigration: options.storeMigration,
+        windows: parts,
+    });
+    return {
+        managed,
+        containers,
+        platform: 'win32',
+        // Never asked on Windows (the provider reads the guest); built so the type stays one.
+        host: realLinuxHost(options.env),
+        arch: testDir === null ? process.arch : 'x64',
+        windows: { records, wsl, keeper, scopeKey, host, ...(testDir === null ? {} : { mount }) },
+    };
+}
+export function wireManagedEnvironment(options) {
+    // The test hooks first, whatever the real platform: each stands in for a whole machine of its own.
+    const testWindows = managedTestWindowsDir(options.env);
+    if (testWindows !== null)
+        return wireWindowsEnvironment(options, testWindows);
+    const testHost = managedTestHostDir(options.env);
+    if (testHost === null && options.platform === 'win32')
+        return wireWindowsEnvironment(options, null);
+    const platform = testHost === null ? options.platform : 'linux';
+    const pinnedDocker = options.dockerPath;
+    const host = testHost !== null
+        ? testLinuxHost(testHost, options.env)
+        : realLinuxHost(options.env, undefined, pinnedDocker === undefined ? undefined : async () => pinnedDocker);
+    const dockerPath = pinnedDocker !== undefined ? pinnedDocker : host.dockerPath;
+    const containers = createManagedContainersHandle({
+        platform,
+        layout: options.layout,
+        instanceId: options.instanceId,
+        log: options.log,
+        ...(dockerPath === undefined ? {} : { dockerPath }),
+        ...(host.dockerSocketPath === undefined ? {} : { dockerSocketPath: host.dockerSocketPath }),
+    });
+    const managed = wireManagedRuntimes({
+        env: nodeDataFolderEnv(options.env),
+        instanceId: options.instanceId,
+        platform,
+        emit: options.emit,
+        newId: options.newId,
+        ...(options.onWarn === undefined ? {} : { onWarn: options.onWarn }),
+        ...(options.fetch === undefined ? {} : { fetch: options.fetch }),
+        engines: options.engines,
+        storeMigration: options.storeMigration,
+        linux: linuxProvisionerParts(options, host, containers),
+    });
+    return { managed, containers, platform, host, arch: process.arch };
+}
+//# sourceMappingURL=managed-environment.js.map

package/dist/core/managed-environment.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"managed-environment.js","sourceRoot":"","sources":["../../src/core/managed-environment.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;GAiBG;AACH,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AACrC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAA;AAG7F,OAAO,EACL,uBAAuB,EACvB,kBAAkB,EAClB,iCAAiC,GAClC,MAAM,0BAA0B,CAAA;AACjC,OAAO,EAAE,6BAA6B,EAAE,MAAM,+BAA+B,CAAA;AAE7E,OAAO,EACL,wBAAwB,EACxB,sBAAsB,EACtB,kBAAkB,EAClB,qBAAqB,EACrB,cAAc,EACd,aAAa,EACb,eAAe,EACf,aAAa,EACb,eAAe,EACf,6BAA6B,EAC7B,mBAAmB,GACpB,MAAM,iCAAiC,CAAA;AAUxC,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAA;AACrE,OAAO,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,oCAAoC,CAAA;AACrF,OAAO,EACL,wBAAwB,EACxB,mBAAmB,EACnB,eAAe,EACf,mBAAmB,EACnB,mBAAmB,GAGpB,MAAM,yBAAyB,CAAA;AAsChC;;;GAGG;AACH,MAAM,UAAU,qBAAqB,CACnC,OAA+E,EAC/E,IAAe,EACf,UAAmC;IAEnC,OAAO;QACL,IAAI;QACJ,MAAM,EAAE,iCAAiC;QACzC,MAAM,EAAE,KAAK,IAAI,EAAE;YACjB,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,OAAO,EAAE,CAAA;YACxC,OAAO,KAAK,KAAK,IAAI;gBACnB,CAAC,CAAC,IAAI;gBACN,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAA;QAChF,CAAC;QACD,kBAAkB,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE;YACzC,MAAM,kBAAkB,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,YAAY,EAAE,CAAC,CAAA;QACpE,CAAC;QACD,sGAAsG;QACtG,+CAA+C;QAC/C,iBAAiB,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAC9F,oBAAoB,EAAE,OAAO,CAAC,oBAAoB,IAAI,cAAc;KACrE,CAAA;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,MAAM,eAAe,GAAsB,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE;IAC7E,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC,oBAAoB,EAAE,GAAG,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE;QAC7F,IAAI,EAAE,MAAM;KACb,CAAC,CAAA;IACF,OAAO;QACL,OAAO,EAAE,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;QACpD,QAAQ,EAAE,GAAG,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;KACtD,CAAA;AACH,CAAC,CAAA;AAgBD;;;;;GAKG;AACH,MAAM,UAAU,uBAAuB,CAAC,KAAwB;IAC9D,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAA;IACpC,sGAAsG;IACtG,MAAM,UAAU,GAAG,KAAK,IAAI,EAAE;QAC5B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAA;QACzC,IAAI,MAAM,KAAK,IAAI;YAAE,OAAO,IAAI,CAAA;QAChC,MAAM,IAAI,GAAG,MAAM,CAAC,YAAY,CAAC,IAAI,CAAA;QACrC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAA;QAClC,OAAO;YACL,KAAK,EAAE,eAAe,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,CAAC,KAAK,CAAC;YACpE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC;YACvD,KAAK,EAAE,eAAe,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC;SAC5D,CAAA;IACH,CAAC,CAAA;IACD,OAAO;QACL,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,WAAW,EAAE,iCAAiC;QAC9C,SAAS,EAAE,kBAAkB;QAC7B,cAAc,EAAE,CAAC,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,CAC9C,sBAAsB,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,CAAC;QAClE,UAAU,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAC/C,cAAc,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,uBAAuB,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;QAC9F,OAAO,EAAE;YACP,IAAI,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE;YAC5D,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;gBACnB,MAAM,KAAK,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;YACtD,CAAC;SACF;QACD,kBAAkB,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE;YACzC,MAAM,KAAK,GAAG,MAAM,UAAU,EAAE,CAAA;YAChC,IAAI,KAAK,KAAK,IAAI;gBAAE,MAAM,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/F,CAAC;QACD,iBAAiB,EAAE,KAAK,IAAI,EAAE;YAC5B,MAAM,KAAK,GAAG,MAAM,UAAU,EAAE,CAAA;YAChC,IAAI,KAAK,KAAK,IAAI;gBAAE,MAAM,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAA;QAC7D,CAAC;QACD,oBAAoB,EAAE,KAAK,CAAC,oBAAoB;QAChD,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,MAAM,EAAE,KAAK,CAAC,MAAM;KACrB,CAAA;AACH,CAAC;AAED;;;;;GAKG;AACH,SAAS,sBAAsB,CAC7B,OAAsC,EACtC,OAAsB;IAEtB,MAAM,IAAI,GAAgB,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;IACpG,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAA;IAC9B,MAAM,KAAK,GACT,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,CAAA;IACzF,MAAM,MAAM,GAAG,kBAAkB,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAA;IACpF,MAAM,OAAO,GAAG,IAAI,6BAA6B,CAAC,MAAM,CAAC,eAAe,CAAC,CAAA;IACzE,MAAM,OAAO,GAAG,IAAI,GAAG,EAA8B,CAAA;IACrD,MAAM,MAAM,GAAG,CAAC,IAAY,EAAsB,EAAE;QAClD,IAAI,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QAChC,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,QAAQ,GAAG,wBAAwB,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAA;YAC3D,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;QAC7B,CAAC;QACD,OAAO,QAAQ,CAAA;IACjB,CAAC,CAAA;IACD,MAAM,QAAQ,GAAG,mBAAmB,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;IAC3E,MAAM,KAAK,GAAG,KAAK,IAAI,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAA;QACrD,OAAO,MAAM,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;IAC5E,CAAC,CAAA;IACD,MAAM,UAAU,GAAG,6BAA6B,CAAC;QAC/C,QAAQ,EAAE,OAAO;QACjB,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,GAAG,EAAE,OAAO,CAAC,GAAG;QAChB,KAAK;KACN,CAAC,CAAA;IACF,MAAM,KAAK,GAAG,uBAAuB,CAAC;QACpC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,KAAK;QAC7B,oBAAoB,EAAE,OAAO,CAAC,oBAAoB,IAAI,cAAc;QACpE,IAAI;QACJ,OAAO;QACP,UAAU;QACV,KAAK;QACL,MAAM,EAAE,MAAM,CAAC,wBAAwB,CAAC;QACxC,QAAQ;QACR,IAAI,EAAE,OAAO,KAAK,IAAI;KACvB,CAAC,CAAA;IACF,MAAM,OAAO,GAAG,mBAAmB,CAAC;QAClC,GAAG,EAAE,iBAAiB,CAAC,OAAO,CAAC,GAAG,CAAC;QACnC,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,QAAQ,EAAE,OAAO;QACjB,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;QACnE,GAAG,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;QAChE,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,cAAc,EAAE,OAAO,CAAC,cAAc;QACtC,OAAO,EAAE,KAAK;KACf,CAAC,CAAA;IACF,OAAO;QACL,OAAO;QACP,UAAU;QACV,QAAQ,EAAE,OAAO;QACjB,sFAAsF;QACtF,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC;QAChC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK;QAC7C,OAAO,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;KAC1F,CAAA;AACH,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,OAAsC;IAC3E,mGAAmG;IACnG,MAAM,WAAW,GAAG,qBAAqB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;IACtD,IAAI,WAAW,KAAK,IAAI;QAAE,OAAO,sBAAsB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAA;IAC7E,MAAM,QAAQ,GAAG,kBAAkB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;IAChD,IAAI,QAAQ,KAAK,IAAI,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO;QAAE,OAAO,sBAAsB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAA;IACnG,MAAM,QAAQ,GAAoB,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAA;IAChF,MAAM,YAAY,GAAG,OAAO,CAAC,UAAU,CAAA;IACvC,MAAM,IAAI,GACR,QAAQ,KAAK,IAAI;QACf,CAAC,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC;QACtC,CAAC,CAAC,aAAa,CACX,OAAO,CAAC,GAAG,EACX,SAAS,EACT,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,YAAY,CAClE,CAAA;IACP,MAAM,UAAU,GAAG,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAA;IAC9E,MAAM,UAAU,GAAG,6BAA6B,CAAC;QAC/C,QAAQ;QACR,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,GAAG,EAAE,OAAO,CAAC,GAAG;QAChB,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC;QACnD,GAAG,CAAC,IAAI,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;KAC5F,CAAC,CAAA;IAEF,MAAM,OAAO,GAAG,mBAAmB,CAAC;QAClC,GAAG,EAAE,iBAAiB,CAAC,OAAO,CAAC,GAAG,CAAC;QACnC,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,QAAQ;QACR,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;QACnE,GAAG,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;QAChE,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,cAAc,EAAE,OAAO,CAAC,cAAc;QACtC,KAAK,EAAE,qBAAqB,CAAC,OAAO,EAAE,IAAI,EAAE,UAAU,CAAC;KACxD,CAAC,CAAA;IACF,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAA;AACpE,CAAC"}

package/dist/core/model-setup/index.d.ts

@@ -0,0 +1,4 @@
+export * from './plan.js';
+export * from './store.js';
+export * from './service.js';
+export * from './wiring.js';

package/dist/core/model-setup/index.js

@@ -0,0 +1,5 @@
+export * from './plan.js';
+export * from './store.js';
+export * from './service.js';
+export * from './wiring.js';
+//# sourceMappingURL=index.js.map

package/dist/core/model-setup/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/model-setup/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAA;AACzB,cAAc,YAAY,CAAA;AAC1B,cAAc,cAAc,CAAA;AAC5B,cAAc,aAAa,CAAA"}

package/dist/core/model-setup/plan.d.ts

@@ -0,0 +1,56 @@
+/**
+ * The model-setup plan, pure: given the compatibility verdict, the conf rule (when one matched), the
+ * Hub's file facts, the Prism manifest and this machine, what a setup would download and install,
+ * how many bytes that is, and what blocks it. `digest` hashes exactly what decides the work, so a
+ * start computed against an older plan is refused as stale instead of doing something else.
+ */
+import type { EngineCapability, LocalProviderId, ModelCompatibilityResponse, ModelSetupPlan, ModelSetupPlanRequest } from '../../contracts/index.js';
+import type { BackendFeatures, GpuProbeInfo, PrismManifest, PrismOfferOptions } from '../../backend/index.js';
+import type { RuleMatch } from '../../models/index.js';
+/** The engine a stock file is registered for when it needs no PrismML build. */
+export declare const STOCK_SETUP_PROVIDER: LocalProviderId;
+export interface PrismHost {
+    osType: string;
+    arch: string;
+    features: BackendFeatures;
+    gpus: readonly GpuProbeInfo[];
+}
+export interface PrismPackRef {
+    version: string;
+    backend: string;
+}
+export interface ModelSetupPlanInput {
+    request: ModelSetupPlanRequest;
+    verdict: ModelCompatibilityResponse;
+    rule?: RuleMatch;
+    /** The Hub's LFS facts for `request.file` when no rule pins it. */
+    hubFile?: {
+        size: number;
+        sha256?: string;
+    };
+    manifest: PrismManifest;
+    offer: PrismOfferOptions;
+    host: PrismHost;
+    installedPacks: readonly PrismPackRef[];
+    /** The pack a load on `atomic-prism` would use now. */
+    currentPack: PrismPackRef | null;
+    freeBytes: number | null;
+}
+/** `owner/<file name without .gguf>`: unique per file, and a leaf of the model tree. */
+export declare function defaultModelId(repo: string, file: string): string;
+/**
+ * The best offered PrismML pack for this machine that runs a file needing `requires` from build
+ * `minBuild` on; `null` when none does.
+ */
+export declare function choosePrismPack(input: {
+    manifest: PrismManifest;
+    offer: PrismOfferOptions;
+    host: PrismHost;
+    requires: readonly EngineCapability[];
+    minBuild?: number;
+}): (PrismPackRef & {
+    download_size: number;
+}) | null;
+/** What `digest` covers: everything that decides the work, nothing that drifts on its own. */
+export declare function planDigest(plan: Omit<ModelSetupPlan, 'digest'>): string;
+export declare function planModelSetup(input: ModelSetupPlanInput): ModelSetupPlan;

package/dist/core/model-setup/plan.js

@@ -0,0 +1,158 @@
+/**
+ * The model-setup plan, pure: given the compatibility verdict, the conf rule (when one matched), the
+ * Hub's file facts, the Prism manifest and this machine, what a setup would download and install,
+ * how many bytes that is, and what blocks it. `digest` hashes exactly what decides the work, so a
+ * start computed against an older plan is refused as stale instead of doing something else.
+ */
+import { determineBestPrismBackend, determinePrismSupportedBackends, filterPrismBackendsBySupport, findPrismRelease, prismArchiveSources, prismCatalogToBackends, prismTagBuild, } from '../../backend/index.js';
+import { canonicalDigest } from '../../runtime/environment/index.js';
+/** The engine a stock file is registered for when it needs no PrismML build. */
+export const STOCK_SETUP_PROVIDER = 'llamacpp-upstream';
+/** `owner/<file name without .gguf>`: unique per file, and a leaf of the model tree. */
+export function defaultModelId(repo, file) {
+    const owner = repo.split('/')[0] ?? repo;
+    const name = (file.split('/').pop() ?? file).replace(/\.gguf$/i, '');
+    return `${owner}/${name}`;
+}
+/**
+ * The best offered PrismML pack for this machine that runs a file needing `requires` from build
+ * `minBuild` on; `null` when none does.
+ */
+export function choosePrismPack(input) {
+    const { manifest, host } = input;
+    const fits = prismCatalogToBackends(manifest, input.offer).filter((pack) => {
+        const release = findPrismRelease(manifest, pack.version);
+        const build = prismTagBuild(pack.version) ?? 0;
+        if (input.minBuild !== undefined && build < input.minBuild)
+            return false;
+        return input.requires.every((capability) => release?.capabilities.includes(capability));
+    });
+    const supported = filterPrismBackendsBySupport(fits, determinePrismSupportedBackends(host.osType, host.arch, host.features));
+    const best = determineBestPrismBackend(supported, host.gpus);
+    if (!best)
+        return null;
+    const [version, backend] = best.split('/');
+    const sources = prismArchiveSources(manifest, version, backend) ?? [];
+    return { version, backend, download_size: sources.reduce((sum, s) => sum + s.size, 0) };
+}
+function engineOf(input, blockers) {
+    const { verdict } = input;
+    if (verdict.provider !== 'atomic-prism')
+        return null;
+    if (verdict.outcome === 'compatible') {
+        return input.currentPack
+            ? { provider: 'atomic-prism', ...input.currentPack, installed: true, download_size: 0 }
+            : null;
+    }
+    if (verdict.outcome !== 'engine_required' && verdict.outcome !== 'engine_update_required')
+        return null;
+    const pack = choosePrismPack({
+        manifest: input.manifest,
+        offer: input.offer,
+        host: input.host,
+        requires: verdict.requires,
+        ...(verdict.min_prism_build !== undefined ? { minBuild: verdict.min_prism_build } : {}),
+    });
+    if (!pack) {
+        blockers.push({
+            code: 'no_engine_build',
+            message: 'No approved PrismML llama.cpp build runs this model on this computer yet.',
+        });
+        return null;
+    }
+    const installed = input.installedPacks.some((p) => p.version === pack.version && p.backend === pack.backend);
+    return {
+        provider: 'atomic-prism',
+        version: pack.version,
+        backend: pack.backend,
+        installed,
+        download_size: installed ? 0 : pack.download_size,
+    };
+}
+function artifactsOf(input) {
+    const { request, rule } = input;
+    if (!rule) {
+        return {
+            model: {
+                repo: request.repo,
+                file: request.file,
+                revision: request.revision ?? 'main',
+                ...(input.hubFile?.sha256 ? { sha256: input.hubFile.sha256 } : {}),
+                size: input.hubFile?.size ?? 0,
+            },
+            projector: null,
+        };
+    }
+    const { family, file } = rule;
+    const projector = request.include_projector === false ? undefined : family.projectors?.find((p) => p.default);
+    return {
+        model: {
+            repo: family.repo,
+            file: file.file,
+            revision: family.revision,
+            sha256: file.sha256,
+            size: file.size,
+        },
+        projector: projector
+            ? {
+                repo: family.repo,
+                file: projector.file,
+                revision: family.revision,
+                sha256: projector.sha256,
+                size: projector.size,
+            }
+            : null,
+    };
+}
+/** What `digest` covers: everything that decides the work, nothing that drifts on its own. */
+export function planDigest(plan) {
+    return canonicalDigest({
+        model_id: plan.model_id,
+        provider: plan.provider,
+        outcome: plan.verdict.outcome,
+        engine: plan.engine && {
+            version: plan.engine.version,
+            backend: plan.engine.backend,
+            installed: plan.engine.installed,
+        },
+        model: plan.model,
+        projector: plan.projector,
+        blockers: plan.blockers.map((b) => b.code),
+    });
+}
+export function planModelSetup(input) {
+    const { verdict } = input;
+    const blockers = [];
+    if (verdict.outcome === 'unsupported')
+        blockers.push({ code: 'unsupported', message: verdict.reason });
+    if (verdict.outcome === 'legacy_artifact') {
+        blockers.push({
+            code: 'legacy_artifact',
+            message: verdict.reason,
+            ...(verdict.replacement ? { replacement: verdict.replacement } : {}),
+        });
+    }
+    const engine = engineOf(input, blockers);
+    const { model, projector } = artifactsOf(input);
+    const total = (engine?.download_size ?? 0) + model.size + (projector?.size ?? 0);
+    if (input.freeBytes !== null && total > input.freeBytes) {
+        blockers.push({
+            code: 'insufficient_disk_space',
+            message: `The setup needs ${total} bytes and ${input.freeBytes} are free.`,
+        });
+    }
+    const body = {
+        model_id: input.request.model_id ?? defaultModelId(model.repo, model.file),
+        provider: verdict.provider ?? STOCK_SETUP_PROVIDER,
+        verdict,
+        engine,
+        model,
+        projector,
+        total_download_bytes: total,
+        free_bytes: input.freeBytes,
+        blockers,
+        ...(verdict.defaults ? { defaults: verdict.defaults } : {}),
+    };
+    return { digest: planDigest(body), ...body };
+}
+//# sourceMappingURL=plan.js.map

package/dist/core/model-setup/plan.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"plan.js","sourceRoot":"","sources":["../../../src/core/model-setup/plan.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAYH,OAAO,EACL,yBAAyB,EACzB,+BAA+B,EAC/B,4BAA4B,EAC5B,gBAAgB,EAChB,mBAAmB,EACnB,sBAAsB,EACtB,aAAa,GACd,MAAM,wBAAwB,CAAA;AAG/B,OAAO,EAAE,eAAe,EAAE,MAAM,oCAAoC,CAAA;AAEpE,gFAAgF;AAChF,MAAM,CAAC,MAAM,oBAAoB,GAAoB,mBAAmB,CAAA;AA6BxE,wFAAwF;AACxF,MAAM,UAAU,cAAc,CAAC,IAAY,EAAE,IAAY;IACvD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAA;IACxC,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAA;IACpE,OAAO,GAAG,KAAK,IAAI,IAAI,EAAE,CAAA;AAC3B,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,KAM/B;IACC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,KAAK,CAAA;IAChC,MAAM,IAAI,GAAG,sBAAsB,CAAC,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;QACzE,MAAM,OAAO,GAAG,gBAAgB,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAA;QACxD,MAAM,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;QAC9C,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,KAAK,GAAG,KAAK,CAAC,QAAQ;YAAE,OAAO,KAAK,CAAA;QACxE,OAAO,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAA;IACzF,CAAC,CAAC,CAAA;IACF,MAAM,SAAS,GAAG,4BAA4B,CAC5C,IAAI,EACJ,+BAA+B,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,CACvE,CAAA;IACD,MAAM,IAAI,GAAG,yBAAyB,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,CAAA;IAC5D,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAA;IACtB,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAqB,CAAA;IAC9D,MAAM,OAAO,GAAG,mBAAmB,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,EAAE,CAAA;IACrE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAA;AACzF,CAAC;AAED,SAAS,QAAQ,CAAC,KAA0B,EAAE,QAA6B;IACzE,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAA;IACzB,IAAI,OAAO,CAAC,QAAQ,KAAK,cAAc;QAAE,OAAO,IAAI,CAAA;IACpD,IAAI,OAAO,CAAC,OAAO,KAAK,YAAY,EAAE,CAAC;QACrC,OAAO,KAAK,CAAC,WAAW;YACtB,CAAC,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,KAAK,CAAC,WAAW,EAAE,SAAS,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE;YACvF,CAAC,CAAC,IAAI,CAAA;IACV,CAAC;IACD,IAAI,OAAO,CAAC,OAAO,KAAK,iBAAiB,IAAI,OAAO,CAAC,OAAO,KAAK,wBAAwB;QAAE,OAAO,IAAI,CAAA;IACtG,MAAM,IAAI,GAAG,eAAe,CAAC;QAC3B,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,GAAG,CAAC,OAAO,CAAC,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACxF,CAAC,CAAA;IACF,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,iBAAiB;YACvB,OAAO,EAAE,2EAA2E;SACrF,CAAC,CAAA;QACF,OAAO,IAAI,CAAA;IACb,CAAC;IACD,MAAM,SAAS,GAAG,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,IAAI,CAAC,OAAO,IAAI,CAAC,CAAC,OAAO,KAAK,IAAI,CAAC,OAAO,CAAC,CAAA;IAC5G,OAAO;QACL,QAAQ,EAAE,cAAc;QACxB,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,SAAS;QACT,aAAa,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa;KAClD,CAAA;AACH,CAAC;AAED,SAAS,WAAW,CAAC,KAA0B;IAI7C,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,KAAK,CAAA;IAC/B,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,KAAK,EAAE;gBACL,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,MAAM;gBACpC,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClE,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,IAAI,CAAC;aAC/B;YACD,SAAS,EAAE,IAAI;SAChB,CAAA;IACH,CAAC;IACD,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,IAAI,CAAA;IAC7B,MAAM,SAAS,GACb,OAAO,CAAC,iBAAiB,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAA;IAC7F,OAAO;QACL,KAAK,EAAE;YACL,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;SAChB;QACD,SAAS,EAAE,SAAS;YAClB,CAAC,CAAC;gBACE,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,MAAM,EAAE,SAAS,CAAC,MAAM;gBACxB,IAAI,EAAE,SAAS,CAAC,IAAI;aACrB;YACH,CAAC,CAAC,IAAI;KACT,CAAA;AACH,CAAC;AAED,8FAA8F;AAC9F,MAAM,UAAU,UAAU,CAAC,IAAoC;IAC7D,OAAO,eAAe,CAAC;QACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;QAC7B,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI;YACrB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;YAC5B,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;YAC5B,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;SACjC;QACD,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;KAC3C,CAAC,CAAA;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,KAA0B;IACvD,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAA;IACzB,MAAM,QAAQ,GAAwB,EAAE,CAAA;IACxC,IAAI,OAAO,CAAC,OAAO,KAAK,aAAa;QAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;IACtG,IAAI,OAAO,CAAC,OAAO,KAAK,iBAAiB,EAAE,CAAC;QAC1C,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,iBAAiB;YACvB,OAAO,EAAE,OAAO,CAAC,MAAM;YACvB,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACrE,CAAC,CAAA;IACJ,CAAC;IACD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAA;IACxC,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,WAAW,CAAC,KAAK,CAAC,CAAA;IAC/C,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE,aAAa,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,GAAG,CAAC,SAAS,EAAE,IAAI,IAAI,CAAC,CAAC,CAAA;IAChF,IAAI,KAAK,CAAC,SAAS,KAAK,IAAI,IAAI,KAAK,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;QACxD,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,yBAAyB;YAC/B,OAAO,EAAE,mBAAmB,KAAK,cAAc,KAAK,CAAC,SAAS,YAAY;SAC3E,CAAC,CAAA;IACJ,CAAC;IACD,MAAM,IAAI,GAAmC;QAC3C,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,QAAQ,IAAI,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC;QAC1E,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,oBAAoB;QAClD,OAAO;QACP,MAAM;QACN,KAAK;QACL,SAAS;QACT,oBAAoB,EAAE,KAAK;QAC3B,UAAU,EAAE,KAAK,CAAC,SAAS;QAC3B,QAAQ;QACR,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC5D,CAAA;IACD,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,EAAE,CAAA;AAC9C,CAAC"}

package/dist/core/model-setup/service.d.ts

@@ -0,0 +1,92 @@
+/**
+ * The model-setup runner: `queued → installing_engine → downloading_model → downloading_projector →
+ * verifying → registering → ready`, or `failed | cancelled`; a setup a stopped core left mid-way is
+ * `interrupted` at the next start and continues on `resume`, from the files already on disk.
+ *
+ * The PrismML pack is shared: two setups that need the same pack wait on one install, and cancelling
+ * one of them stops its wait, never the install the other still needs. Credentials are never stored:
+ * the Hugging Face token goes into the download headers of the run that uses it.
+ */
+import { AtomicCoreError } from '../../contracts/index.js';
+import type { CompatibilityVerdict, ErrorBody, ModelSetup, ModelSetupEngine, ModelSetupPlan, ModelSetupPlanRequest, ModelSetupStartRequest, ProxyConfig } from '../../contracts/index.js';
+import type { DataLayout } from '../../config/index.js';
+import type { DownloadItem } from '../../downloads/index.js';
+import type { ModelYmlInput } from '../../models/index.js';
+import type { ModelSetupStore } from './store.js';
+export interface ModelSetupServiceDeps {
+    layout: DataLayout;
+    store: ModelSetupStore;
+    plan: (request: ModelSetupPlanRequest) => Promise<ModelSetupPlan>;
+    /** Install the pack (idempotent: a pack already on disk returns at once). */
+    installEngine: (engine: ModelSetupEngine, taskId: string, proxy: ProxyConfig | null) => Promise<void>;
+    /** Make `atomic-prism` load from this pack. */
+    selectEngine: (engine: ModelSetupEngine) => Promise<void>;
+    download: (taskId: string, items: DownloadItem[]) => Promise<void>;
+    cancelDownload: (taskId: string) => void;
+    /** The header verdict of a downloaded file. */
+    verify: (modelPath: string) => Promise<CompatibilityVerdict>;
+    register: (modelId: string, yml: ModelYmlInput) => Promise<void>;
+    emit: (record: ModelSetup) => void;
+    /** `model:imported` for the app's model list. */
+    imported?: (event: {
+        provider: ModelSetupPlan['provider'];
+        modelId: string;
+        modelPath: string;
+        mmprojPath?: string;
+    }) => void;
+    newId: () => string;
+    /** Where model files are downloaded from; Hugging Face unless a test points it elsewhere. */
+    hfEndpoint?: string;
+    now?: () => number;
+    log?: (level: 'info' | 'warn', message: string) => void;
+}
+/** A download task id the progress events can carry: no characters a path or event name chokes on. */
+export declare function setupTaskId(raw: string): string;
+/** The engine install's task id, one per pack, shared by every setup that needs it. */
+export declare function engineTaskId(engine: Pick<ModelSetupEngine, 'version' | 'backend'>): string;
+/** The error the wire carries for a stopped setup. */
+export declare function setupErrorBody(error: unknown): ErrorBody;
+/** The refusal a blocked plan answers `POST /model-setups` with. */
+export declare function blockedPlanError(plan: ModelSetupPlan): AtomicCoreError | null;
+export declare class ModelSetupService {
+    private readonly deps;
+    private readonly records;
+    private readonly runs;
+    private readonly engines;
+    constructor(deps: ModelSetupServiceDeps);
+    /** Load the records and mark what a stopped core left mid-way `interrupted`. Call once at start. */
+    recover(): Promise<void>;
+    /** In-memory view for the control snapshot. */
+    snapshot(): ModelSetup[];
+    plan(request: ModelSetupPlanRequest): Promise<ModelSetupPlan>;
+    list(): Promise<ModelSetup[]>;
+    get(setupId: string): Promise<ModelSetup>;
+    /**
+     * Start a setup from the plan the user saw. A retry with the same `request_id` returns the setup
+     * it started; a plan that changed since is `MODEL_SETUP_PLAN_STALE`; a model already being set up
+     * returns that setup.
+     */
+    start(input: ModelSetupStartRequest): Promise<ModelSetup>;
+    /** Stop a setup. Files downloaded so far stay, so `resume` continues from them. */
+    cancel(setupId: string): Promise<ModelSetup>;
+    /** Run a stopped setup again, planned afresh from its request. */
+    resume(setupId: string, options?: {
+        proxy?: ProxyConfig | null;
+    }): Promise<ModelSetup>;
+    /** Wait for a run in flight (tests and shutdown). */
+    settled(setupId: string): Promise<void>;
+    private taskIds;
+    private launch;
+    private run;
+    private sharedInstall;
+    /** Data-folder-relative paths, `/`-separated, as `model.yml` stores them. */
+    private pathsOf;
+    private item;
+    /** The header is the last word: a file that turned out to need another engine is not registered. */
+    private check;
+    private registerModel;
+    /** Commit a patch on top of the newest record, retrying if a concurrent write moved it. */
+    private update;
+    private remember;
+    private now;
+}