@atomic-chat/core 0.5.1 → 0.13.2

package/dist/runtime/managed-models/delete.js

@@ -0,0 +1,60 @@
+/**
+ * The files of a deleted `tensorrt-llm` model (task 2.24, design D12a, spec `tensorrt-llm-models`
+ * "Модель удаляется через core"): every engine cache of the model, from any descriptor it was ever
+ * loaded with, then its own folder. Caches go first: a failure there leaves the model listed, so the
+ * deletion can simply be retried, instead of leaving gigabytes of cache no model points at.
+ *
+ * The caller (`tensorrtLlmModelDeleter` in `src/core/tensorrt-llm.ts`) has already stopped the model
+ * with Docker's confirmation and holds its loads off; nothing here asks Docker anything.
+ */
+import { lstat, readdir, rm } from 'node:fs/promises';
+import { join } from 'node:path';
+import { engineCacheDirsOf, removeEngineCaches } from '../managed-text/index.js';
+/**
+ * Bytes of the regular files under `path`, symlinks not followed, a hard-linked file counted once
+ * across one deletion (`seen`). A missing path is 0.
+ */
+async function bytesUnder(path, seen) {
+    const info = await lstat(path).catch(() => undefined);
+    if (info === undefined)
+        return 0;
+    if (info.isFile()) {
+        const inode = `${info.dev}:${info.ino}`;
+        if (seen.has(inode))
+            return 0;
+        seen.add(inode);
+        return info.size;
+    }
+    if (!info.isDirectory())
+        return 0;
+    const children = await readdir(path).catch(() => []);
+    let total = 0;
+    for (const child of children)
+        total += await bytesUnder(join(path, child), seen);
+    return total;
+}
+/**
+ * `files` (change `add-tensorrt-llm-windows`, task 2.8): on Windows the model and its caches are in the
+ * WSL guest, and are sized and removed by commands there (`guestModelFiles`) — all caches and the model
+ * in one call each, the caches listed first. Without it, the walk below, as on Linux.
+ */
+export async function deleteManagedModelFiles(paths, model, files) {
+    if (files !== undefined) {
+        const caches = await engineCacheDirsOf(paths, model.id);
+        const sizes = await files.sizes([...caches, model.dir]);
+        await files.remove([...caches, model.dir]);
+        const freed = [...sizes.values()].reduce((sum, bytes) => sum + bytes, 0);
+        return { freedBytes: freed, engineCachesRemoved: caches.length };
+    }
+    const seen = new Set();
+    const caches = await engineCacheDirsOf(paths, model.id);
+    const sizes = new Map();
+    for (const dir of caches)
+        sizes.set(dir, await bytesUnder(dir, seen));
+    const modelBytes = await bytesUnder(model.dir, seen);
+    const removed = await removeEngineCaches(paths, { modelId: model.id });
+    await rm(model.dir, { recursive: true, force: true });
+    const cacheBytes = removed.reduce((sum, dir) => sum + (sizes.get(dir) ?? 0), 0);
+    return { freedBytes: cacheBytes + modelBytes, engineCachesRemoved: removed.length };
+}
+//# sourceMappingURL=delete.js.map

package/dist/runtime/managed-models/delete.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"delete.js","sourceRoot":"","sources":["../../../src/runtime/managed-models/delete.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AACrD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAEhC,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAA;AAQhF;;;GAGG;AACH,KAAK,UAAU,UAAU,CAAC,IAAY,EAAE,IAAiB;IACvD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAA;IACrD,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO,CAAC,CAAA;IAChC,IAAI,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,CAAA;QACvC,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,OAAO,CAAC,CAAA;QAC7B,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QACf,OAAO,IAAI,CAAC,IAAI,CAAA;IAClB,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;QAAE,OAAO,CAAC,CAAA;IACjC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAA;IACpD,IAAI,KAAK,GAAG,CAAC,CAAA;IACb,KAAK,MAAM,KAAK,IAAI,QAAQ;QAAE,KAAK,IAAI,MAAM,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,CAAA;IAChF,OAAO,KAAK,CAAA;AACd,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,KAAwB,EACxB,KAAkC,EAClC,KAAoB;IAEpB,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,CAAA;QACvD,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC,GAAG,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAA;QACvD,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAA;QAC1C,MAAM,KAAK,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,EAAE,CAAC,CAAC,CAAA;QACxE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,mBAAmB,EAAE,MAAM,CAAC,MAAM,EAAE,CAAA;IAClE,CAAC;IACD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAA;IAC9B,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,CAAA;IACvD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAA;IACvC,KAAK,MAAM,GAAG,IAAI,MAAM;QAAE,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAA;IACrE,MAAM,UAAU,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,CAAC,CAAA;IAEpD,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAA;IACtE,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;IACrD,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;IAC/E,OAAO,EAAE,UAAU,EAAE,UAAU,GAAG,UAAU,EAAE,mBAAmB,EAAE,OAAO,CAAC,MAAM,EAAE,CAAA;AACrF,CAAC"}

package/dist/runtime/managed-models/guest-files.d.ts

@@ -0,0 +1,8 @@
+import { type GuestMount, type WslDistributionTransport } from '../wsl/index.js';
+/** Sizing and removing model and cache folders; the Linux default walks the file system itself. */
+export interface ModelFileOps {
+    /** Bytes under each path; a path that is not there is 0. */
+    sizes(paths: string[]): Promise<Map<string, number>>;
+    remove(paths: string[]): Promise<void>;
+}
+export declare function guestModelFiles(transport: WslDistributionTransport, mount?: GuestMount): ModelFileOps;

package/dist/runtime/managed-models/guest-files.js

@@ -0,0 +1,49 @@
+/**
+ * The files of `tensorrt-llm` models and engine caches on Windows (change `add-tensorrt-llm-windows`,
+ * task 2.8, design D5): they live in the WSL guest's ext4, and core sees them through
+ * `\\wsl.localhost`. Listing and reading a `model.yml` through that path is fine; sizing or deleting
+ * tens of gigabytes through it is not — thousands of 9p round trips for what `du` and `rm` do in the
+ * guest at once. So sizes and removals run as commands in the guest, as root, with the guest paths
+ * behind the UNC ones (`guestPathFor`, which refuses anything outside the distribution). One `du`
+ * call for every path counts a hard-linked file once, as the Linux walk does.
+ */
+import { AtomicCoreError } from '../../contracts/index.js';
+import { WSL_LOCALHOST_MOUNT } from '../wsl/index.js';
+const TIMEOUT_MS = 30 * 60_000;
+export function guestModelFiles(transport, mount = WSL_LOCALHOST_MOUNT) {
+    const toGuest = (paths) => paths.map((path) => mount.guestPath(transport.name, path));
+    return {
+        sizes: async (paths) => {
+            const sizes = new Map(paths.map((path) => [path, 0]));
+            if (paths.length === 0)
+                return sizes;
+            const guest = toGuest(paths);
+            const answer = await transport.exec(['du', '-s', '-b', '--', ...guest], {
+                user: 'root',
+                timeoutMs: TIMEOUT_MS,
+            });
+            // `du` exits 1 when one path is missing, and still prints every other.
+            for (const line of answer.stdout.split('\n')) {
+                const match = /^(\d+)\t(.+)$/.exec(line);
+                if (match === null)
+                    continue;
+                const index = guest.indexOf(match[2]);
+                if (index !== -1)
+                    sizes.set(paths[index], Number(match[1]));
+            }
+            return sizes;
+        },
+        remove: async (paths) => {
+            if (paths.length === 0)
+                return;
+            const answer = await transport.exec(['rm', '-rf', '--', ...toGuest(paths)], {
+                user: 'root',
+                timeoutMs: TIMEOUT_MS,
+            });
+            if (answer.code !== 0) {
+                throw new AtomicCoreError('IO_ERROR', 'The files could not be removed in the Atomic Chat distribution.', answer.stderr.trim());
+            }
+        },
+    };
+}
+//# sourceMappingURL=guest-files.js.map

package/dist/runtime/managed-models/guest-files.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"guest-files.js","sourceRoot":"","sources":["../../../src/runtime/managed-models/guest-files.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAA;AAC1D,OAAO,EAAE,mBAAmB,EAAkD,MAAM,iBAAiB,CAAA;AASrG,MAAM,UAAU,GAAG,EAAE,GAAG,MAAM,CAAA;AAE9B,MAAM,UAAU,eAAe,CAC7B,SAAmC,EACnC,QAAoB,mBAAmB;IAEvC,MAAM,OAAO,GAAG,CAAC,KAAe,EAAY,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAA;IACzG,OAAO;QACL,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YACrB,MAAM,KAAK,GAAG,IAAI,GAAG,CAAiB,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YACrE,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,KAAK,CAAA;YACpC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAA;YAC5B,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC,EAAE;gBACtE,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,UAAU;aACtB,CAAC,CAAA;YACF,uEAAuE;YACvE,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC7C,MAAM,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBACxC,IAAI,KAAK,KAAK,IAAI;oBAAE,SAAQ;gBAC5B,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAW,CAAC,CAAA;gBAC/C,IAAI,KAAK,KAAK,CAAC,CAAC;oBAAE,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAW,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;YACvE,CAAC;YACD,OAAO,KAAK,CAAA;QACd,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YACtB,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAM;YAC9B,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE;gBAC1E,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,UAAU;aACtB,CAAC,CAAA;YACF,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACtB,MAAM,IAAI,eAAe,CACvB,UAAU,EACV,iEAAiE,EACjE,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CACrB,CAAA;YACH,CAAC;QACH,CAAC;KACF,CAAA;AACH,CAAC"}

package/dist/runtime/managed-models/index.d.ts

@@ -0,0 +1,17 @@
+/**
+ * What every managed engine shares about the models it runs (change `add-vllm-runtime`): the one
+ * quantization naming rule (`quant-format.ts`, design D6), the compatibility check skeleton with
+ * each engine's hooks (`compatibility.ts`, D7) and the check route behind
+ * `POST /atomic/v1/models/:provider/check` (`check.ts`).
+ *
+ * Public API of this module is exported from this file only.
+ */
+export * from './quant-format.js';
+export * from './compatibility.js';
+export * from './check.js';
+export * from './model-dir.js';
+export * from './registry.js';
+export * from './delete.js';
+export * from './location.js';
+export * from './guest-files.js';
+export * from './migrate.js';

package/dist/runtime/managed-models/index.js

@@ -0,0 +1,18 @@
+/**
+ * What every managed engine shares about the models it runs (change `add-vllm-runtime`): the one
+ * quantization naming rule (`quant-format.ts`, design D6), the compatibility check skeleton with
+ * each engine's hooks (`compatibility.ts`, D7) and the check route behind
+ * `POST /atomic/v1/models/:provider/check` (`check.ts`).
+ *
+ * Public API of this module is exported from this file only.
+ */
+export * from './quant-format.js';
+export * from './compatibility.js';
+export * from './check.js';
+export * from './model-dir.js';
+export * from './registry.js';
+export * from './delete.js';
+export * from './location.js';
+export * from './guest-files.js';
+export * from './migrate.js';
+//# sourceMappingURL=index.js.map

package/dist/runtime/managed-models/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/runtime/managed-models/index.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,cAAc,mBAAmB,CAAA;AACjC,cAAc,oBAAoB,CAAA;AAClC,cAAc,YAAY,CAAA;AAC1B,cAAc,gBAAgB,CAAA;AAC9B,cAAc,eAAe,CAAA;AAC7B,cAAc,aAAa,CAAA;AAC3B,cAAc,eAAe,CAAA;AAC7B,cAAc,kBAAkB,CAAA;AAChC,cAAc,cAAc,CAAA"}

package/dist/runtime/managed-models/location.d.ts

@@ -0,0 +1,22 @@
+import type { ManagedModelLocation } from '../../contracts/index.js';
+import type { WindowsEnvironmentRecord } from '../environment/index.js';
+import { type GuestMount, type WslDistributionTransport } from '../wsl/index.js';
+/** Linux: `<data>/managed-models`, with the free space of the volume it is (or will be) on. */
+export declare function linuxModelLocation(root: string, freeBytes?: (path: string) => Promise<number | null>): Promise<ManagedModelLocation>;
+export interface WindowsModelLocationDeps {
+    records: {
+        read(): Promise<WindowsEnvironmentRecord | null>;
+    };
+    /** This scope's `scope_key` (`readOrCreateGuestScopeKey`). */
+    scopeKey: () => Promise<string>;
+    transport: (distribution: string) => WslDistributionTransport;
+    /** Free bytes on the Windows volume holding `path` (`WindowsHost.freeDiskBytes`). */
+    volumeFreeBytes: (path: string) => Promise<number | null>;
+    /** How core reaches the guest's files; `\\wsl.localhost` unless a test says otherwise. */
+    mount?: GuestMount;
+}
+/** The guest folder of this scope's model store, every managed engine's. */
+export declare function guestModelsRoot(scopeKey: string): string;
+/** Where this scope's TensorRT-LLM models lived in the guest before the store (design D5). */
+export declare function legacyGuestModelsRoot(scopeKey: string): string;
+export declare function windowsModelLocation(deps: WindowsModelLocationDeps): Promise<ManagedModelLocation>;

package/dist/runtime/managed-models/location.js

@@ -0,0 +1,53 @@
+/**
+ * Where the managed engines' models live, as clients must use it (spec `managed-model-store`, "Core
+ * сообщает расположение хранилища"; change `add-vllm-runtime`, design D4 — before it, `tensorrt-llm`'s
+ * own folder, change `add-tensorrt-llm-windows`, task 2.8). One answer, from core, on every platform,
+ * so a client never derives the folder itself: on Linux the data folder's `<data>/managed-models`; on
+ * Windows the scope's folder in Atomic Chat's WSL distribution,
+ * `\\wsl.localhost\<distribution>\var\lib\atomic-chat\scopes\<scope_key>\managed-models`, created
+ * (and handed to uid 1000) before a client writes into it. The free space on Windows is the smaller of
+ * the guest's own and the Windows volume's that holds the distribution's disk image: the guest's disk
+ * can grow only as far as that volume lets it. Before the distribution exists there is no such
+ * folder: `MANAGED_ADAPTER_UNAVAILABLE`, and no download starts.
+ */
+import { AtomicCoreError } from '../../contracts/index.js';
+import { freeBytesAtNearest, parseDfAvail } from '../environment/index.js';
+import { ensureGuestScope, guestScopeRoot, WSL_LOCALHOST_MOUNT, } from '../wsl/index.js';
+/** Linux: `<data>/managed-models`, with the free space of the volume it is (or will be) on. */
+export async function linuxModelLocation(root, freeBytes = freeBytesAtNearest) {
+    return { root, free_bytes: await freeBytes(root).catch(() => null) };
+}
+/** The guest folder of this scope's model store, every managed engine's. */
+export function guestModelsRoot(scopeKey) {
+    return `${guestScopeRoot(scopeKey)}/managed-models`;
+}
+/** Where this scope's TensorRT-LLM models lived in the guest before the store (design D5). */
+export function legacyGuestModelsRoot(scopeKey) {
+    return `${guestScopeRoot(scopeKey)}/models/tensorrt-llm`;
+}
+export async function windowsModelLocation(deps) {
+    const record = await deps.records.read();
+    if (record === null) {
+        throw new AtomicCoreError('MANAGED_ADAPTER_UNAVAILABLE', 'TensorRT-LLM models live in Atomic Chat’s WSL distribution, which is not set up yet.');
+    }
+    const key = await deps.scopeKey();
+    const transport = deps.transport(record.distribution.name);
+    const guestRoot = guestModelsRoot(key);
+    // A distribution unregistered behind the app's back is no environment at all for a client.
+    await ensureGuestScope(transport, key).catch((error) => {
+        throw new AtomicCoreError('MANAGED_ADAPTER_UNAVAILABLE', 'Atomic Chat’s WSL distribution does not answer; set the environment up again.', error instanceof Error ? error.message : String(error));
+    });
+    const [guest, volume] = await Promise.all([
+        transport
+            .exec(['df', '--output=avail', '-B1', guestRoot], { user: 'root', timeoutMs: 120_000 })
+            .then(parseDfAvail)
+            .catch(() => null),
+        deps.volumeFreeBytes(record.distribution.path).catch(() => null),
+    ]);
+    const known = [guest, volume].filter((value) => value !== null);
+    return {
+        root: (deps.mount ?? WSL_LOCALHOST_MOUNT).hostPath(record.distribution.name, guestRoot),
+        free_bytes: known.length === 0 ? null : Math.min(...known),
+    };
+}
+//# sourceMappingURL=location.js.map

package/dist/runtime/managed-models/location.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"location.js","sourceRoot":"","sources":["../../../src/runtime/managed-models/location.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AACH,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAA;AAE1D,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAA;AAE1E,OAAO,EACL,gBAAgB,EAChB,cAAc,EACd,mBAAmB,GAGpB,MAAM,iBAAiB,CAAA;AAExB,+FAA+F;AAC/F,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,IAAY,EACZ,YAAsD,kBAAkB;IAExE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,CAAA;AACtE,CAAC;AAaD,4EAA4E;AAC5E,MAAM,UAAU,eAAe,CAAC,QAAgB;IAC9C,OAAO,GAAG,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAAA;AACrD,CAAC;AAED,8FAA8F;AAC9F,MAAM,UAAU,qBAAqB,CAAC,QAAgB;IACpD,OAAO,GAAG,cAAc,CAAC,QAAQ,CAAC,sBAAsB,CAAA;AAC1D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,IAA8B;IACvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAA;IACxC,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;QACpB,MAAM,IAAI,eAAe,CACvB,6BAA6B,EAC7B,sFAAsF,CACvF,CAAA;IACH,CAAC;IACD,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAA;IACjC,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;IAC1D,MAAM,SAAS,GAAG,eAAe,CAAC,GAAG,CAAC,CAAA;IACtC,2FAA2F;IAC3F,MAAM,gBAAgB,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;QAC9D,MAAM,IAAI,eAAe,CACvB,6BAA6B,EAC7B,+EAA+E,EAC/E,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACvD,CAAA;IACH,CAAC,CAAC,CAAA;IACF,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QACxC,SAAS;aACN,IAAI,CAAC,CAAC,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,SAAS,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;aACtF,IAAI,CAAC,YAAY,CAAC;aAClB,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;QACpB,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;KACjE,CAAC,CAAA;IACF,MAAM,KAAK,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,CAAA;IAChF,OAAO;QACL,IAAI,EAAE,CAAC,IAAI,CAAC,KAAK,IAAI,mBAAmB,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,CAAC;QACvF,UAAU,EAAE,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;KAC3D,CAAA;AACH,CAAC"}

package/dist/runtime/managed-models/migrate.d.ts

@@ -0,0 +1,30 @@
+import type { ManagedStoreMigration } from '../../contracts/index.js';
+import type { WslDistributionTransport } from '../wsl/index.js';
+/** The file operations a migration needs, on the host's file system or in the WSL guest. */
+export interface StoreMigrationFs {
+    /** Ids (`/`-separated, relative to `root`) of the folders under it that hold `model.yml`, sorted. */
+    modelIds(root: string): Promise<string[]>;
+    exists(path: string): Promise<boolean>;
+    /** Renames `from` to `to` on the same file system, creating `to`'s parent first. */
+    move(from: string, to: string): Promise<void>;
+    /**
+     * Removes `dir` when it is empty, then its parent, and so on up to and including `root`, stopping at
+     * the first that is not empty — never a file, and never a folder no move emptied.
+     */
+    pruneEmptyUpTo(root: string, dir: string): Promise<void>;
+    /** How a model path is joined: the host's separator, or `/` in the guest. */
+    join(root: string, id: string): string;
+}
+/** Moves every model folder of `from` into `to`; see the file banner. */
+export declare function migrateTensorrtLlmModels(options: {
+    from: string;
+    to: string;
+    fs: StoreMigrationFs;
+}): Promise<ManagedStoreMigration>;
+/** The host's own file system: Linux (and the data folder of any platform). */
+export declare const nodeStoreMigrationFs: StoreMigrationFs;
+/**
+ * The WSL guest's file system (Windows): every step a command in the guest, as root, as an argv —
+ * never a shell — over the guest's own paths, so nothing crosses `\\wsl.localhost`.
+ */
+export declare function guestStoreMigrationFs(transport: WslDistributionTransport): StoreMigrationFs;

package/dist/runtime/managed-models/migrate.js

@@ -0,0 +1,119 @@
+/**
+ * Moving TensorRT-LLM's models into the managed model store (change `add-vllm-runtime`, design D5;
+ * spec `managed-model-store`, "Модели TensorRT-LLM переезжают в общий корень"). Before the store, a
+ * model of `tensorrt-llm` lived in `<data>/tensorrt-llm/models/<id>` (in the WSL guest,
+ * `…/scopes/<key>/models/tensorrt-llm/<id>`); now every managed engine reads `managed-models/<id>`.
+ *
+ * Core does the move, at startup, before anything lists or loads a model: the layout is its own, and
+ * it starts before the app lists anything. Each folder holding `model.yml` is renamed to the same id
+ * under the store — one rename on one file system, so it is cheap and atomic per folder, and an engine
+ * cache (`caches/<descriptor_id>/<model id>`) stays valid since the id does not change. A folder
+ * without `model.yml` is a download still in progress and stays where it is: the client that started
+ * it cleans it up. An id already in the store is a conflict: neither folder is touched, and the
+ * conflict is reported (environment diagnostics). The folders a move emptied go — the moved model's
+ * parents, the old root once nothing is in it; a folder left in place is never touched. Every step is idempotent: on the next start there is nothing to move.
+ *
+ * What runs it is the core's own lock on its data folder (one owner per canonical data folder): the
+ * old root and the store are both in that folder, or in that scope's guest folder.
+ */
+import { mkdir, readdir, readFile, rename, rmdir, stat } from 'node:fs/promises';
+import { dirname, join } from 'node:path';
+import { MODEL_YML, modelIdFromDir } from '../../config/index.js';
+import { AtomicCoreError } from '../../contracts/index.js';
+/** Moves every model folder of `from` into `to`; see the file banner. */
+export async function migrateTensorrtLlmModels(options) {
+    const { from, to, fs } = options;
+    const result = { from, to, moved: [], conflicts: [] };
+    if (!(await fs.exists(from)))
+        return result;
+    for (const id of await fs.modelIds(from)) {
+        const source = fs.join(from, id);
+        const target = fs.join(to, id);
+        if (await fs.exists(target)) {
+            result.conflicts.push({ model_id: id, source, target });
+            continue;
+        }
+        await fs.move(source, target);
+        result.moved.push(id);
+        // Only what this move emptied: the moved folder's parents, never a folder left in place.
+        await fs.pruneEmptyUpTo(from, parentOf(source));
+    }
+    return result;
+}
+/** The parent of a path, by either separator (host paths or the guest's `/`). */
+const parentOf = (path) => path.slice(0, Math.max(path.lastIndexOf('/'), path.lastIndexOf('\\')));
+const exists = (path) => stat(path).then(() => true, () => false);
+/** The host's own file system: Linux (and the data folder of any platform). */
+export const nodeStoreMigrationFs = {
+    async modelIds(root) {
+        const ids = [];
+        const stack = [root];
+        while (stack.length > 0) {
+            const dir = stack.pop();
+            if (await readFile(join(dir, MODEL_YML)).then(() => true, () => false)) {
+                if (dir !== root)
+                    ids.push(modelIdFromDir(root, dir));
+                continue; // a model folder is a leaf
+            }
+            const children = await readdir(dir, { withFileTypes: true }).catch(() => []);
+            for (const child of children)
+                if (child.isDirectory())
+                    stack.push(join(dir, child.name));
+        }
+        return ids.sort();
+    },
+    exists,
+    async move(from, to) {
+        await mkdir(dirname(to), { recursive: true });
+        await rename(from, to);
+    },
+    async pruneEmptyUpTo(root, dir) {
+        for (let current = dir; current.length >= root.length; current = dirname(current)) {
+            if (!(await rmdir(current).then(() => true, () => false)))
+                return;
+            if (current === root)
+                return;
+        }
+    },
+    join: (root, id) => join(root, ...id.split('/')),
+};
+const GUEST_TIMEOUT_MS = 5 * 60_000;
+/**
+ * The WSL guest's file system (Windows): every step a command in the guest, as root, as an argv —
+ * never a shell — over the guest's own paths, so nothing crosses `\\wsl.localhost`.
+ */
+export function guestStoreMigrationFs(transport) {
+    const run = async (argv) => transport.exec(argv, { user: 'root', timeoutMs: GUEST_TIMEOUT_MS });
+    return {
+        async modelIds(root) {
+            const answer = await run(['find', root, '-name', MODEL_YML, '-type', 'f', '-printf', '%h\n']);
+            const dirs = answer.stdout
+                .split('\n')
+                .map((line) => line.trim())
+                .filter((line) => line.startsWith(`${root}/`))
+                .sort();
+            // A model folder is a leaf: a model.yml inside another model's folder is that model's file.
+            const leaves = dirs.filter((dir) => !dirs.some((other) => other !== dir && dir.startsWith(`${other}/`)));
+            return leaves.map((dir) => dir.slice(root.length + 1));
+        },
+        exists: async (path) => (await run(['test', '-e', path])).code === 0,
+        async move(from, to) {
+            const parent = to.slice(0, to.lastIndexOf('/'));
+            await run(['mkdir', '-p', parent]);
+            const answer = await run(['mv', '-T', '--', from, to]);
+            if (answer.code !== 0) {
+                throw new AtomicCoreError('IO_ERROR', 'A model could not be moved into the model store in the Atomic Chat distribution.', answer.stderr.trim());
+            }
+        },
+        async pruneEmptyUpTo(root, dir) {
+            for (let current = dir; current.length >= root.length; current = current.slice(0, current.lastIndexOf('/'))) {
+                if ((await run(['rmdir', '--', current])).code !== 0)
+                    return;
+                if (current === root)
+                    return;
+            }
+        },
+        join: (root, id) => `${root}/${id}`,
+    };
+}
+//# sourceMappingURL=migrate.js.map

package/dist/runtime/managed-models/migrate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"migrate.js","sourceRoot":"","sources":["../../../src/runtime/managed-models/migrate.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;GAiBG;AACH,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAA;AAChF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AACzC,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAA;AACjE,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAA;AAoB1D,yEAAyE;AACzE,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,OAI9C;IACC,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,OAAO,CAAA;IAChC,MAAM,MAAM,GAA0B,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAA;IAC5E,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAAE,OAAO,MAAM,CAAA;IAC3C,KAAK,MAAM,EAAE,IAAI,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACzC,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAA;QAChC,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;QAC9B,IAAI,MAAM,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5B,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAA;YACvD,SAAQ;QACV,CAAC;QACD,MAAM,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QAC7B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QACrB,yFAAyF;QACzF,MAAM,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAA;IACjD,CAAC;IACD,OAAO,MAAM,CAAA;AACf,CAAC;AAED,iFAAiF;AACjF,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAU,EAAE,CACxC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;AAExE,MAAM,MAAM,GAAG,CAAC,IAAY,EAAoB,EAAE,CAChD,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CACb,GAAG,EAAE,CAAC,IAAI,EACV,GAAG,EAAE,CAAC,KAAK,CACZ,CAAA;AAEH,+EAA+E;AAC/E,MAAM,CAAC,MAAM,oBAAoB,GAAqB;IACpD,KAAK,CAAC,QAAQ,CAAC,IAAI;QACjB,MAAM,GAAG,GAAa,EAAE,CAAA;QACxB,MAAM,KAAK,GAAG,CAAC,IAAI,CAAC,CAAA;QACpB,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,EAAY,CAAA;YACjC,IACE,MAAM,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CACvC,GAAG,EAAE,CAAC,IAAI,EACV,GAAG,EAAE,CAAC,KAAK,CACZ,EACD,CAAC;gBACD,IAAI,GAAG,KAAK,IAAI;oBAAE,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAA;gBACrD,SAAQ,CAAC,2BAA2B;YACtC,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAA;YAC5E,KAAK,MAAM,KAAK,IAAI,QAAQ;gBAAE,IAAI,KAAK,CAAC,WAAW,EAAE;oBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAA;QAC1F,CAAC;QACD,OAAO,GAAG,CAAC,IAAI,EAAE,CAAA;IACnB,CAAC;IACD,MAAM;IACN,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE;QACjB,MAAM,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;QAC7C,MAAM,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC,CAAA;IACxB,CAAC;IACD,KAAK,CAAC,cAAc,CAAC,IAAI,EAAE,GAAG;QAC5B,KAAK,IAAI,OAAO,GAAG,GAAG,EAAE,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAClF,IACE,CAAC,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CACzB,GAAG,EAAE,CAAC,IAAI,EACV,GAAG,EAAE,CAAC,KAAK,CACZ,CAAC;gBAEF,OAAM;YACR,IAAI,OAAO,KAAK,IAAI;gBAAE,OAAM;QAC9B,CAAC;IACH,CAAC;IACD,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;CACjD,CAAA;AAED,MAAM,gBAAgB,GAAG,CAAC,GAAG,MAAM,CAAA;AAEnC;;;GAGG;AACH,MAAM,UAAU,qBAAqB,CAAC,SAAmC;IACvE,MAAM,GAAG,GAAG,KAAK,EAAE,IAAc,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC,CAAA;IACzG,OAAO;QACL,KAAK,CAAC,QAAQ,CAAC,IAAI;YACjB,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAA;YAC7F,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM;iBACvB,KAAK,CAAC,IAAI,CAAC;iBACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;iBAC1B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC;iBAC7C,IAAI,EAAE,CAAA;YACT,4FAA4F;YAC5F,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAA;YACxG,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAA;QACxD,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC;QACpE,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE;YACjB,MAAM,MAAM,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAA;YAC/C,MAAM,GAAG,CAAC,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC,CAAA;YAClC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAA;YACtD,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACtB,MAAM,IAAI,eAAe,CACvB,UAAU,EACV,kFAAkF,EAClF,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CACrB,CAAA;YACH,CAAC;QACH,CAAC;QACD,KAAK,CAAC,cAAc,CAAC,IAAI,EAAE,GAAG;YAC5B,KACE,IAAI,OAAO,GAAG,GAAG,EACjB,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,EAC7B,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,EACpD,CAAC;gBACD,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC;oBAAE,OAAM;gBAC5D,IAAI,OAAO,KAAK,IAAI;oBAAE,OAAM;YAC9B,CAAC;QACH,CAAC;QACD,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC,GAAG,IAAI,IAAI,EAAE,EAAE;KACpC,CAAA;AACH,CAAC"}

package/dist/runtime/managed-models/model-dir.d.ts

@@ -0,0 +1,36 @@
+import type { CheckpointFile } from './compatibility.js';
+/** `model.yml`'s own fields, parsed and typed; nothing here is derived. */
+export interface ManagedModelYmlDocument {
+    /** `null` when `model.yml` does not carry one (tolerated: `readManagedModel`'s own contract). */
+    repository: string | null;
+    revision: string | null;
+    architectures: string[];
+    files: CheckpointFile[];
+}
+export interface ManagedModel {
+    id: string;
+    /** The checkpoint directory as core sees it; mounted read-only into the container. */
+    dir: string;
+    repository: string | null;
+    revision: string | null;
+    /** `architectures[0]`: the Hugging Face class name the descriptor's `model_families` is keyed by. */
+    architecture: string | null;
+    /** The revision's file listing exactly as `model.yml` recorded it (the pre-launch check's input). */
+    files: CheckpointFile[];
+    /** Weight bytes of `files` (`weightBytes`), 0 when `model.yml` lists none. */
+    weightBytes: number;
+}
+/**
+ * `model.yml`'s codec (`parse` only — the app and the CLI write it, core never does, spec
+ * `tensorrt-llm-models`): `MANAGED_METADATA_INVALID` when `text` is not valid YAML or not a mapping.
+ * `repository`/`revision`/`architectures` tolerate absence or the wrong type by
+ * reading as `null`/`[]`, the same way `readManagedModel`'s own tests already expect for a bare
+ * `model.yml` — a model missing optional metadata is still a model, not a broken one. `files` is the
+ * one field held to a stricter rule (task 2.16w round 1, finding 4): the array itself may be absent
+ * (`[]`), but once present every entry must be well-formed (`fileList`/`parseFileEntry`) —
+ * silently dropping a malformed entry would leave a shard unverified, a checkpoint's weight bytes
+ * under-counted, or an `hf_quant_config.json` entry quietly lost.
+ */
+export declare function parseManagedModelYml(text: string, path: string): ManagedModelYmlDocument;
+/** `MODEL_NOT_FOUND` without a readable `model.yml`; `MANAGED_METADATA_INVALID` when it is not a mapping. */
+export declare function readManagedModel(modelsDir: string, modelId: string): Promise<ManagedModel>;

package/dist/runtime/managed-models/model-dir.js

@@ -0,0 +1,136 @@
+/**
+ * Reading a model of the managed engines' store (`<data>/managed-models/`, spec `managed-model-store`):
+ * its `model.yml` (`parseManagedModelYml` — `repository`, `revision`, `architectures`, `files` with
+ * `path`/`size`/`sha256`, documented in `docs/contracts.md`) and the few derived facts a load needs —
+ * the first architecture (for the descriptor's `model_families` entry) and the weight bytes (for the
+ * readiness timeout). `model.yml` names nothing that depends on an engine: a `quantization` an older
+ * app wrote is read past and ignored, since every engine names the format itself from the files on
+ * disk (spec "model.yml не зависит от движка"). The app and the CLI write the directory and its
+ * `model.yml` last; a directory without one is a download still in progress and is not a model.
+ *
+ * `readManagedModel` is the single-id lookup `ManagedTextRuntime.load` uses; the full scan across
+ * every model (`registry.ts`) shares `parseManagedModelYml` with it, so both agree about what
+ * `model.yml` means.
+ */
+import { readFile } from 'node:fs/promises';
+import { join, relative, isAbsolute } from 'node:path';
+import { parse } from 'yaml';
+import { AtomicCoreError } from '../../contracts/index.js';
+import { MODEL_YML, modelDirFromId } from '../../config/index.js';
+import { weightBytes } from './compatibility.js';
+/** An id is one or more `/`-separated path segments inside the models directory, never out of it. */
+function assertModelId(modelsDir, modelId) {
+    const segments = modelId.split('/');
+    if (modelId === '' || segments.some((s) => s === '' || s === '.' || s === '..' || s.includes('\\'))) {
+        throw new AtomicCoreError('INVALID_ARGUMENT', 'Not a managed model id.', modelId);
+    }
+    const dir = modelDirFromId(modelsDir, modelId);
+    const inside = relative(modelsDir, dir);
+    if (inside === '' || inside.startsWith('..') || isAbsolute(inside)) {
+        throw new AtomicCoreError('INVALID_ARGUMENT', 'Not a managed model id.', modelId);
+    }
+    return dir;
+}
+/**
+ * A relative file path safe to join onto the model directory and mount read-only into a container:
+ * forward-slash separated, never absolute, no empty/`.`/`..` segment and no backslash (which some
+ * shells would treat as an escape on the file's own leaf name).
+ */
+function isSafeRelativePath(path) {
+    if (path === '' || path.startsWith('/') || path.includes('\\'))
+        return false;
+    return path.split('/').every((segment) => segment !== '' && segment !== '.' && segment !== '..');
+}
+/**
+ * One `model.yml` `files[]` entry, strictly: `MANAGED_METADATA_INVALID` for anything malformed
+ * (task 2.16w round 1, finding 4) — a non-string or unsafe `path`, a `size` that is not a finite
+ * non-negative integer, a `sha256` that is neither a string nor `null`. The previous, lenient
+ * parsing silently dropped a malformed entry instead: an unverified shard the pre-launch check would
+ * never even look for, a checkpoint's weight bytes under-counted, or an `hf_quant_config.json` entry
+ * quietly lost — every one of those is worse than refusing the model outright and naming why.
+ */
+function parseFileEntry(raw, index, path) {
+    if (raw === null || typeof raw !== 'object' || Array.isArray(raw)) {
+        throw new AtomicCoreError('MANAGED_METADATA_INVALID', `${path}: files[${index}] is not a mapping.`, JSON.stringify(raw));
+    }
+    const entry = raw;
+    if (typeof entry.path !== 'string' || !isSafeRelativePath(entry.path)) {
+        throw new AtomicCoreError('MANAGED_METADATA_INVALID', `${path}: files[${index}].path is not a safe relative path.`, String(entry.path));
+    }
+    if (typeof entry.size !== 'number' || !Number.isInteger(entry.size) || entry.size < 0) {
+        throw new AtomicCoreError('MANAGED_METADATA_INVALID', `${path}: files[${index}].size must be a non-negative integer.`, String(entry.size));
+    }
+    if (entry.sha256 !== undefined && entry.sha256 !== null && typeof entry.sha256 !== 'string') {
+        throw new AtomicCoreError('MANAGED_METADATA_INVALID', `${path}: files[${index}].sha256 must be a string or null.`, String(entry.sha256));
+    }
+    return {
+        path: entry.path,
+        size: entry.size,
+        sha256: typeof entry.sha256 === 'string' ? entry.sha256 : null,
+    };
+}
+/** `files` absent entirely is tolerated as `[]` (a bare `model.yml` is still a model); once present, every entry must be well-formed. */
+function fileList(raw, path) {
+    if (raw === undefined)
+        return [];
+    if (!Array.isArray(raw)) {
+        throw new AtomicCoreError('MANAGED_METADATA_INVALID', `${path}: files must be an array.`, JSON.stringify(raw));
+    }
+    return raw.map((entry, index) => parseFileEntry(entry, index, path));
+}
+/**
+ * `model.yml`'s codec (`parse` only — the app and the CLI write it, core never does, spec
+ * `tensorrt-llm-models`): `MANAGED_METADATA_INVALID` when `text` is not valid YAML or not a mapping.
+ * `repository`/`revision`/`architectures` tolerate absence or the wrong type by
+ * reading as `null`/`[]`, the same way `readManagedModel`'s own tests already expect for a bare
+ * `model.yml` — a model missing optional metadata is still a model, not a broken one. `files` is the
+ * one field held to a stricter rule (task 2.16w round 1, finding 4): the array itself may be absent
+ * (`[]`), but once present every entry must be well-formed (`fileList`/`parseFileEntry`) —
+ * silently dropping a malformed entry would leave a shard unverified, a checkpoint's weight bytes
+ * under-counted, or an `hf_quant_config.json` entry quietly lost.
+ */
+export function parseManagedModelYml(text, path) {
+    let yml;
+    try {
+        yml = parse(text);
+    }
+    catch (error) {
+        throw new AtomicCoreError('MANAGED_METADATA_INVALID', `${path} is not valid YAML.`, String(error));
+    }
+    if (yml === null || typeof yml !== 'object' || Array.isArray(yml)) {
+        throw new AtomicCoreError('MANAGED_METADATA_INVALID', `${path} does not describe a model.`, path);
+    }
+    const doc = yml;
+    const architectures = Array.isArray(doc['architectures'])
+        ? doc['architectures'].filter((entry) => typeof entry === 'string' && entry !== '')
+        : [];
+    return {
+        repository: typeof doc['repository'] === 'string' ? doc['repository'] : null,
+        revision: typeof doc['revision'] === 'string' ? doc['revision'] : null,
+        architectures,
+        files: fileList(doc['files'], path),
+    };
+}
+/** `MODEL_NOT_FOUND` without a readable `model.yml`; `MANAGED_METADATA_INVALID` when it is not a mapping. */
+export async function readManagedModel(modelsDir, modelId) {
+    const dir = assertModelId(modelsDir, modelId);
+    const path = join(dir, MODEL_YML);
+    let text;
+    try {
+        text = await readFile(path, 'utf8');
+    }
+    catch {
+        throw new AtomicCoreError('MODEL_NOT_FOUND', `The model '${modelId}' is not in the managed model store.`, path);
+    }
+    const yml = parseManagedModelYml(text, path);
+    return {
+        id: modelId,
+        dir,
+        repository: yml.repository,
+        revision: yml.revision,
+        architecture: yml.architectures[0] ?? null,
+        files: yml.files,
+        weightBytes: weightBytes(yml.files),
+    };
+}
+//# sourceMappingURL=model-dir.js.map

package/dist/runtime/managed-models/model-dir.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-dir.js","sourceRoot":"","sources":["../../../src/runtime/managed-models/model-dir.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AACH,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAA;AAC3C,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,WAAW,CAAA;AACtD,OAAO,EAAE,KAAK,EAAE,MAAM,MAAM,CAAA;AAC5B,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAA;AAC1D,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAA;AACjE,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AA0BhD,qGAAqG;AACrG,SAAS,aAAa,CAAC,SAAiB,EAAE,OAAe;IACvD,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IACnC,IAAI,OAAO,KAAK,EAAE,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;QACpG,MAAM,IAAI,eAAe,CAAC,kBAAkB,EAAE,yBAAyB,EAAE,OAAO,CAAC,CAAA;IACnF,CAAC;IACD,MAAM,GAAG,GAAG,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;IAC9C,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,EAAE,GAAG,CAAC,CAAA;IACvC,IAAI,MAAM,KAAK,EAAE,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QACnE,MAAM,IAAI,eAAe,CAAC,kBAAkB,EAAE,yBAAyB,EAAE,OAAO,CAAC,CAAA;IACnF,CAAC;IACD,OAAO,GAAG,CAAA;AACZ,CAAC;AAED;;;;GAIG;AACH,SAAS,kBAAkB,CAAC,IAAY;IACtC,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,KAAK,CAAA;IAC5E,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,KAAK,EAAE,IAAI,OAAO,KAAK,GAAG,IAAI,OAAO,KAAK,IAAI,CAAC,CAAA;AAClG,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,cAAc,CAAC,GAAY,EAAE,KAAa,EAAE,IAAY;IAC/D,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QAClE,MAAM,IAAI,eAAe,CACvB,0BAA0B,EAC1B,GAAG,IAAI,WAAW,KAAK,qBAAqB,EAC5C,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CACpB,CAAA;IACH,CAAC;IACD,MAAM,KAAK,GAAG,GAA2D,CAAA;IACzE,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACtE,MAAM,IAAI,eAAe,CACvB,0BAA0B,EAC1B,GAAG,IAAI,WAAW,KAAK,qCAAqC,EAC5D,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CACnB,CAAA;IACH,CAAC;IACD,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QACtF,MAAM,IAAI,eAAe,CACvB,0BAA0B,EAC1B,GAAG,IAAI,WAAW,KAAK,wCAAwC,EAC/D,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CACnB,CAAA;IACH,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,IAAI,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC5F,MAAM,IAAI,eAAe,CACvB,0BAA0B,EAC1B,GAAG,IAAI,WAAW,KAAK,oCAAoC,EAC3D,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CACrB,CAAA;IACH,CAAC;IACD,OAAO;QACL,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,MAAM,EAAE,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI;KAC/D,CAAA;AACH,CAAC;AAED,yIAAyI;AACzI,SAAS,QAAQ,CAAC,GAAY,EAAE,IAAY;IAC1C,IAAI,GAAG,KAAK,SAAS;QAAE,OAAO,EAAE,CAAA;IAChC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,eAAe,CACvB,0BAA0B,EAC1B,GAAG,IAAI,2BAA2B,EAClC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CACpB,CAAA;IACH,CAAC;IACD,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,cAAc,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC,CAAA;AACtE,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,UAAU,oBAAoB,CAAC,IAAY,EAAE,IAAY;IAC7D,IAAI,GAAY,CAAA;IAChB,IAAI,CAAC;QACH,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,CAAA;IACnB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,eAAe,CAAC,0BAA0B,EAAE,GAAG,IAAI,qBAAqB,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAA;IACpG,CAAC;IACD,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QAClE,MAAM,IAAI,eAAe,CAAC,0BAA0B,EAAE,GAAG,IAAI,6BAA6B,EAAE,IAAI,CAAC,CAAA;IACnG,CAAC;IACD,MAAM,GAAG,GAAG,GAA8B,CAAA;IAC1C,MAAM,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QACvD,CAAC,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,EAAE,CAAC;QACpG,CAAC,CAAC,EAAE,CAAA;IACN,OAAO;QACL,UAAU,EAAE,OAAO,GAAG,CAAC,YAAY,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI;QAC5E,QAAQ,EAAE,OAAO,GAAG,CAAC,UAAU,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI;QACtE,aAAa;QACb,KAAK,EAAE,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC;KACpC,CAAA;AACH,CAAC;AAED,6GAA6G;AAC7G,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,SAAiB,EAAE,OAAe;IACvE,MAAM,GAAG,GAAG,aAAa,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;IAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAA;IACjC,IAAI,IAAY,CAAA;IAChB,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;IACrC,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,eAAe,CACvB,iBAAiB,EACjB,cAAc,OAAO,sCAAsC,EAC3D,IAAI,CACL,CAAA;IACH,CAAC;IACD,MAAM,GAAG,GAAG,oBAAoB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;IAC5C,OAAO;QACL,EAAE,EAAE,OAAO;QACX,GAAG;QACH,UAAU,EAAE,GAAG,CAAC,UAAU;QAC1B,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,YAAY,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,IAAI;QAC1C,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,WAAW,EAAE,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC;KACpC,CAAA;AACH,CAAC"}