@atomic-chat/core 0.5.1

package/dist/models/gguf/reader.js

@@ -0,0 +1,271 @@
+/**
+ * GGUF header reader. Port of `tauri-plugin-llamacpp-upstream/src/gguf/{helpers,types}.rs`.
+ *
+ * Reads magic, version, tensor count and the metadata KV block only — never the tensor table.
+ * Every value is stringified the way the Rust reader does (`to_string()`, `[a, b]` for arrays of
+ * up to 24 elements, a placeholder for longer ones), so downstream code sees identical strings.
+ *
+ * Pure over a byte buffer; callers feed it a file prefix and grow it while it asks for more.
+ */
+import { formatRustF32, formatRustF64 } from '../../util/index.js';
+export const GGUF_VALUE_TYPE_NAMES = [
+    'Uint8',
+    'Int8',
+    'Uint16',
+    'Int16',
+    'Uint32',
+    'Int32',
+    'Float32',
+    'Bool',
+    'String',
+    'Array',
+    'Uint64',
+    'Int64',
+    'Float64',
+];
+export const MAX_GGUF_STRING_BYTES = 1024 * 1024;
+export const MAX_GGUF_ARRAY_LEN = 1_000_000;
+export const INLINE_ARRAY_MAX_LEN = 24;
+/** The buffer ended before the header did; feed more bytes and retry. */
+export class GgufNeedMoreData extends Error {
+    constructor() {
+        super('GGUF header incomplete');
+        this.name = 'GgufNeedMoreData';
+    }
+}
+/** The bytes are not a GGUF header the reader understands (same messages as the Rust reader). */
+export class GgufParseError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'GgufParseError';
+    }
+}
+class Cursor {
+    buf;
+    pos = 0;
+    view;
+    constructor(buf) {
+        this.buf = buf;
+        this.view = new DataView(buf.buffer, buf.byteOffset, buf.byteLength);
+    }
+    need(n) {
+        if (this.pos + n > this.buf.byteLength)
+            throw new GgufNeedMoreData();
+    }
+    u8() {
+        this.need(1);
+        return this.view.getUint8(this.pos++);
+    }
+    i8() {
+        this.need(1);
+        return this.view.getInt8(this.pos++);
+    }
+    u16() {
+        this.need(2);
+        const v = this.view.getUint16(this.pos, true);
+        this.pos += 2;
+        return v;
+    }
+    i16() {
+        this.need(2);
+        const v = this.view.getInt16(this.pos, true);
+        this.pos += 2;
+        return v;
+    }
+    u32() {
+        this.need(4);
+        const v = this.view.getUint32(this.pos, true);
+        this.pos += 4;
+        return v;
+    }
+    i32() {
+        this.need(4);
+        const v = this.view.getInt32(this.pos, true);
+        this.pos += 4;
+        return v;
+    }
+    f32() {
+        this.need(4);
+        const v = this.view.getFloat32(this.pos, true);
+        this.pos += 4;
+        return v;
+    }
+    u64() {
+        this.need(8);
+        const v = this.view.getBigUint64(this.pos, true);
+        this.pos += 8;
+        return v;
+    }
+    i64() {
+        this.need(8);
+        const v = this.view.getBigInt64(this.pos, true);
+        this.pos += 8;
+        return v;
+    }
+    f64() {
+        this.need(8);
+        const v = this.view.getFloat64(this.pos, true);
+        this.pos += 8;
+        return v;
+    }
+    bytes(n) {
+        this.need(n);
+        const out = this.buf.subarray(this.pos, this.pos + n);
+        this.pos += n;
+        return out;
+    }
+    skip(n) {
+        this.need(n);
+        this.pos += n;
+    }
+}
+const utf8 = new TextDecoder('utf-8', { fatal: true });
+function readString(c) {
+    const len = c.u64();
+    if (len > BigInt(MAX_GGUF_STRING_BYTES))
+        throw new GgufParseError(`String length ${len} is unreasonably large`);
+    const raw = c.bytes(Number(len));
+    try {
+        return utf8.decode(raw);
+    }
+    catch {
+        throw new GgufParseError('invalid utf-8 sequence');
+    }
+}
+function valueType(n) {
+    if (n < 0 || n >= GGUF_VALUE_TYPE_NAMES.length)
+        throw new GgufParseError(`Unknown GGUF value type: ${n}`);
+    return n;
+}
+function readValue(c, type) {
+    switch (type) {
+        case 0:
+            return String(c.u8());
+        case 1:
+            return String(c.i8());
+        case 2:
+            return String(c.u16());
+        case 3:
+            return String(c.i16());
+        case 4:
+            return String(c.u32());
+        case 5:
+            return String(c.i32());
+        case 6:
+            return formatRustF32(c.f32());
+        case 7:
+            return c.u8() !== 0 ? 'true' : 'false';
+        case 8:
+            return readString(c);
+        case 10:
+            return c.u64().toString();
+        case 11:
+            return c.i64().toString();
+        case 12:
+            return formatRustF64(c.f64());
+        case 9: {
+            const elemType = valueType(c.u32());
+            const len = c.u64();
+            if (len > BigInt(MAX_GGUF_ARRAY_LEN))
+                throw new GgufParseError(`Array length ${len} is unreasonably large`);
+            const n = Number(len);
+            if (n > INLINE_ARRAY_MAX_LEN) {
+                skipArray(c, elemType, n);
+                return `<Array of type ${GGUF_VALUE_TYPE_NAMES[elemType]} with ${n} elements, data skipped>`;
+            }
+            const elems = [];
+            for (let i = 0; i < n; i++)
+                elems.push(readValue(c, elemType));
+            return `[${elems.join(', ')}]`;
+        }
+        default:
+            throw new GgufParseError(`Unknown GGUF value type: ${type}`);
+    }
+}
+function skipArray(c, elemType, len) {
+    switch (elemType) {
+        case 0:
+        case 1:
+        case 7:
+            c.skip(len);
+            return;
+        case 2:
+        case 3:
+            c.skip(len * 2);
+            return;
+        case 4:
+        case 5:
+        case 6:
+            c.skip(len * 4);
+            return;
+        case 10:
+        case 11:
+        case 12:
+            c.skip(len * 8);
+            return;
+        case 8:
+            for (let i = 0; i < len; i++)
+                c.skip(Number(c.u64()));
+            return;
+        case 9:
+            // Nested arrays are read (not skipped), exactly like the Rust reader.
+            for (let i = 0; i < len; i++)
+                readValue(c, elemType);
+            return;
+        default:
+            throw new GgufParseError(`Unknown GGUF value type: ${elemType}`);
+    }
+}
+/**
+ * Parse the header from `bytes`. Throws `GgufNeedMoreData` when the buffer is too short and
+ * `GgufParseError` when the bytes are not a valid header.
+ */
+export function readGgufMetadata(bytes) {
+    const c = new Cursor(bytes);
+    const magic = c.bytes(4);
+    if (!(magic[0] === 0x47 && magic[1] === 0x47 && magic[2] === 0x55 && magic[3] === 0x46)) {
+        throw new GgufParseError('Not a GGUF file');
+    }
+    const version = c.u32();
+    const tensorCount = c.u64();
+    const metadataCount = c.u64();
+    const metadata = {};
+    for (let i = 0n; i < metadataCount; i++) {
+        try {
+            const key = readString(c);
+            const type = valueType(c.u32());
+            metadata[key] = readValue(c, type);
+        }
+        catch (e) {
+            if (e instanceof GgufNeedMoreData)
+                throw e;
+            throw new GgufParseError(`Error reading metadata entry ${i}: ${e.message}`);
+        }
+    }
+    return { version, tensor_count: Number(tensorCount), metadata };
+}
+/**
+ * Drive `readGgufMetadata` over a source that yields the file in growing prefixes (a local file
+ * read in chunks, or HTTP range requests). `readPrefix(byteLength)` returns at least what it can;
+ * a prefix shorter than requested means EOF.
+ */
+export async function readGgufMetadataChunked(readPrefix, options = {}) {
+    const chunk = options.chunkSize ?? 2 * 1024 * 1024;
+    const max = options.maxBytes ?? 120 * 1024 * 1024;
+    let want = chunk;
+    for (;;) {
+        const bytes = await readPrefix(Math.min(want, max));
+        try {
+            return readGgufMetadata(bytes);
+        }
+        catch (e) {
+            if (!(e instanceof GgufNeedMoreData))
+                throw e;
+            if (bytes.byteLength < Math.min(want, max) || want >= max) {
+                throw new GgufParseError('Could not parse GGUF metadata from downloaded data');
+            }
+            want += chunk;
+        }
+    }
+}
+//# sourceMappingURL=reader.js.map

package/dist/models/gguf/reader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reader.js","sourceRoot":"","sources":["../../../src/models/gguf/reader.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAA;AAQlE,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,OAAO;IACP,MAAM;IACN,QAAQ;IACR,OAAO;IACP,QAAQ;IACR,OAAO;IACP,SAAS;IACT,MAAM;IACN,QAAQ;IACR,OAAO;IACP,QAAQ;IACR,OAAO;IACP,SAAS;CACD,CAAA;AAEV,MAAM,CAAC,MAAM,qBAAqB,GAAG,IAAI,GAAG,IAAI,CAAA;AAChD,MAAM,CAAC,MAAM,kBAAkB,GAAG,SAAS,CAAA;AAC3C,MAAM,CAAC,MAAM,oBAAoB,GAAG,EAAE,CAAA;AAEtC,yEAAyE;AACzE,MAAM,OAAO,gBAAiB,SAAQ,KAAK;IACzC;QACE,KAAK,CAAC,wBAAwB,CAAC,CAAA;QAC/B,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAA;IAChC,CAAC;CACF;AAED,iGAAiG;AACjG,MAAM,OAAO,cAAe,SAAQ,KAAK;IACvC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAA;QACd,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAA;IAC9B,CAAC;CACF;AAED,MAAM,MAAM;IAGmB;IAFrB,GAAG,GAAG,CAAC,CAAA;IACE,IAAI,CAAU;IAC/B,YAA6B,GAAe;QAAf,QAAG,GAAH,GAAG,CAAY;QAC1C,IAAI,CAAC,IAAI,GAAG,IAAI,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,UAAU,CAAC,CAAA;IACtE,CAAC;IACO,IAAI,CAAC,CAAS;QACpB,IAAI,IAAI,CAAC,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU;YAAE,MAAM,IAAI,gBAAgB,EAAE,CAAA;IACtE,CAAC;IACD,EAAE;QACA,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAA;IACvC,CAAC;IACD,EAAE;QACA,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAA;IACtC,CAAC;IACD,GAAG;QACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAA;QAC7C,IAAI,CAAC,GAAG,IAAI,CAAC,CAAA;QACb,OAAO,CAAC,CAAA;IACV,CAAC;IACD,GAAG;QACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAA;QAC5C,IAAI,CAAC,GAAG,IAAI,CAAC,CAAA;QACb,OAAO,CAAC,CAAA;IACV,CAAC;IACD,GAAG;QACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAA;QAC7C,IAAI,CAAC,GAAG,IAAI,CAAC,CAAA;QACb,OAAO,CAAC,CAAA;IACV,CAAC;IACD,GAAG;QACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAA;QAC5C,IAAI,CAAC,GAAG,IAAI,CAAC,CAAA;QACb,OAAO,CAAC,CAAA;IACV,CAAC;IACD,GAAG;QACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAA;QAC9C,IAAI,CAAC,GAAG,IAAI,CAAC,CAAA;QACb,OAAO,CAAC,CAAA;IACV,CAAC;IACD,GAAG;QACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAA;QAChD,IAAI,CAAC,GAAG,IAAI,CAAC,CAAA;QACb,OAAO,CAAC,CAAA;IACV,CAAC;IACD,GAAG;QACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAA;QAC/C,IAAI,CAAC,GAAG,IAAI,CAAC,CAAA;QACb,OAAO,CAAC,CAAA;IACV,CAAC;IACD,GAAG;QACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAA;QAC9C,IAAI,CAAC,GAAG,IAAI,CAAC,CAAA;QACb,OAAO,CAAC,CAAA;IACV,CAAC;IACD,KAAK,CAAC,CAAS;QACb,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC,CAAA;QACrD,IAAI,CAAC,GAAG,IAAI,CAAC,CAAA;QACb,OAAO,GAAG,CAAA;IACZ,CAAC;IACD,IAAI,CAAC,CAAS;QACZ,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACZ,IAAI,CAAC,GAAG,IAAI,CAAC,CAAA;IACf,CAAC;CACF;AAED,MAAM,IAAI,GAAG,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;AAEtD,SAAS,UAAU,CAAC,CAAS;IAC3B,MAAM,GAAG,GAAG,CAAC,CAAC,GAAG,EAAE,CAAA;IACnB,IAAI,GAAG,GAAG,MAAM,CAAC,qBAAqB,CAAC;QACrC,MAAM,IAAI,cAAc,CAAC,iBAAiB,GAAG,wBAAwB,CAAC,CAAA;IACxE,MAAM,GAAG,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAA;IAChC,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;IACzB,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,cAAc,CAAC,wBAAwB,CAAC,CAAA;IACpD,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,CAAS;IAC1B,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,qBAAqB,CAAC,MAAM;QAAE,MAAM,IAAI,cAAc,CAAC,4BAA4B,CAAC,EAAE,CAAC,CAAA;IACzG,OAAO,CAAC,CAAA;AACV,CAAC;AAED,SAAS,SAAS,CAAC,CAAS,EAAE,IAAY;IACxC,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,CAAC;YACJ,OAAO,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QACvB,KAAK,CAAC;YACJ,OAAO,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QACvB,KAAK,CAAC;YACJ,OAAO,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAA;QACxB,KAAK,CAAC;YACJ,OAAO,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAA;QACxB,KAAK,CAAC;YACJ,OAAO,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAA;QACxB,KAAK,CAAC;YACJ,OAAO,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAA;QACxB,KAAK,CAAC;YACJ,OAAO,aAAa,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAA;QAC/B,KAAK,CAAC;YACJ,OAAO,CAAC,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAA;QACxC,KAAK,CAAC;YACJ,OAAO,UAAU,CAAC,CAAC,CAAC,CAAA;QACtB,KAAK,EAAE;YACL,OAAO,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAA;QAC3B,KAAK,EAAE;YACL,OAAO,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAA;QAC3B,KAAK,EAAE;YACL,OAAO,aAAa,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAA;QAC/B,KAAK,CAAC,CAAC,CAAC,CAAC;YACP,MAAM,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAA;YACnC,MAAM,GAAG,GAAG,CAAC,CAAC,GAAG,EAAE,CAAA;YACnB,IAAI,GAAG,GAAG,MAAM,CAAC,kBAAkB,CAAC;gBAClC,MAAM,IAAI,cAAc,CAAC,gBAAgB,GAAG,wBAAwB,CAAC,CAAA;YACvE,MAAM,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAA;YACrB,IAAI,CAAC,GAAG,oBAAoB,EAAE,CAAC;gBAC7B,SAAS,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAA;gBACzB,OAAO,kBAAkB,qBAAqB,CAAC,QAAQ,CAAC,SAAS,CAAC,0BAA0B,CAAA;YAC9F,CAAC;YACD,MAAM,KAAK,GAAa,EAAE,CAAA;YAC1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;gBAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAA;YAC9D,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAA;QAChC,CAAC;QACD;YACE,MAAM,IAAI,cAAc,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAA;IAChE,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,CAAS,EAAE,QAAgB,EAAE,GAAW;IACzD,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,CAAC,CAAC;QACP,KAAK,CAAC,CAAC;QACP,KAAK,CAAC;YACJ,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;YACX,OAAM;QACR,KAAK,CAAC,CAAC;QACP,KAAK,CAAC;YACJ,CAAC,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC,CAAA;YACf,OAAM;QACR,KAAK,CAAC,CAAC;QACP,KAAK,CAAC,CAAC;QACP,KAAK,CAAC;YACJ,CAAC,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC,CAAA;YACf,OAAM;QACR,KAAK,EAAE,CAAC;QACR,KAAK,EAAE,CAAC;QACR,KAAK,EAAE;YACL,CAAC,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC,CAAA;YACf,OAAM;QACR,KAAK,CAAC;YACJ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE;gBAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAA;YACrD,OAAM;QACR,KAAK,CAAC;YACJ,sEAAsE;YACtE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE;gBAAE,SAAS,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAA;YACpD,OAAM;QACR;YACE,MAAM,IAAI,cAAc,CAAC,4BAA4B,QAAQ,EAAE,CAAC,CAAA;IACpE,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAiB;IAChD,MAAM,CAAC,GAAG,IAAI,MAAM,CAAC,KAAK,CAAC,CAAA;IAC3B,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;IACxB,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,EAAE,CAAC;QACxF,MAAM,IAAI,cAAc,CAAC,iBAAiB,CAAC,CAAA;IAC7C,CAAC;IACD,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,EAAE,CAAA;IACvB,MAAM,WAAW,GAAG,CAAC,CAAC,GAAG,EAAE,CAAA;IAC3B,MAAM,aAAa,GAAG,CAAC,CAAC,GAAG,EAAE,CAAA;IAC7B,MAAM,QAAQ,GAA2B,EAAE,CAAA;IAC3C,KAAK,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,UAAU,CAAC,CAAC,CAAC,CAAA;YACzB,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAA;YAC/B,QAAQ,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;QACpC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,YAAY,gBAAgB;gBAAE,MAAM,CAAC,CAAA;YAC1C,MAAM,IAAI,cAAc,CAAC,gCAAgC,CAAC,KAAM,CAAW,CAAC,OAAO,EAAE,CAAC,CAAA;QACxF,CAAC;IACH,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,CAAA;AACjE,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,UAAuD,EACvD,UAAqD,EAAE;IAEvD,MAAM,KAAK,GAAG,OAAO,CAAC,SAAS,IAAI,CAAC,GAAG,IAAI,GAAG,IAAI,CAAA;IAClD,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,IAAI,GAAG,GAAG,IAAI,GAAG,IAAI,CAAA;IACjD,IAAI,IAAI,GAAG,KAAK,CAAA;IAChB,SAAS,CAAC;QACR,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAA;QACnD,IAAI,CAAC;YACH,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAChC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,CAAC,CAAC,YAAY,gBAAgB,CAAC;gBAAE,MAAM,CAAC,CAAA;YAC7C,IAAI,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,IAAI,GAAG,EAAE,CAAC;gBAC1D,MAAM,IAAI,cAAc,CAAC,oDAAoD,CAAC,CAAA;YAChF,CAAC;YACD,IAAI,IAAI,KAAK,CAAA;QACf,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/models/gguf/support.d.ts

@@ -0,0 +1,47 @@
+/**
+ * "Will this model fit?" — port of `gguf/commands.rs` (`memory_budget`, `is_model_supported`).
+ * Pure over numbers; reading the file size and the hardware facts is the caller's job.
+ */
+export type ModelSupportStatus = 'RED' | 'YELLOW' | 'GREEN';
+/**
+ * Headroom left to the OS, the app and llama.cpp's own allocations on top of weights + KV cache:
+ * ~2.13 GiB per memory pool. Inherited, not measured (see the Rust comment).
+ */
+export declare const RESERVE_BYTES = 2288490189;
+export interface GpuMemory {
+    total_bytes: number;
+    /** Memory carved out of system RAM (an integrated GPU) rather than its own pool. */
+    integrated: boolean;
+}
+export interface MemoryBudget {
+    /** Fits here → runs entirely on the GPU. */
+    usable_vram: number;
+    /** Fits here → runs at all, spilling to system RAM. */
+    usable_total: number;
+}
+/** Vulkan's `IntegratedGpu` is the only trusted signal; anything else counts as discrete. */
+export declare function isIntegratedGpu(gpu: {
+    vulkan_info?: {
+        device_type?: string;
+    } | null;
+}): boolean;
+/**
+ * Integrated graphics shares system RAM, so adding it to RAM would count it twice (the 16 GB
+ * laptop that was told a 17 GB model fits). Only discrete GPUs add memory of their own. With no
+ * GPU enumerated (Apple Silicon) RAM *is* the VRAM: one pool.
+ */
+export declare function memoryBudget(totalRamBytes: number, gpus: GpuMemory[]): MemoryBudget;
+/** RED = cannot run at all; GREEN = fits in VRAM; YELLOW = runs with CPU/GPU spill. */
+export declare function modelSupportStatus(requiredBytes: number, budget: MemoryBudget): ModelSupportStatus;
+export interface SystemMemoryFacts {
+    /** MiB, as the hardware probe reports it. */
+    total_memory: number;
+    gpus: Array<{
+        total_memory: number;
+        vulkan_info?: {
+            device_type?: string;
+        } | null;
+    }>;
+}
+/** `is_model_supported` after the file size and KV estimate are known. */
+export declare function isModelSupported(modelSizeBytes: number, kvCacheBytes: number, system: SystemMemoryFacts): ModelSupportStatus;

package/dist/models/gguf/support.js

@@ -0,0 +1,48 @@
+/**
+ * "Will this model fit?" — port of `gguf/commands.rs` (`memory_budget`, `is_model_supported`).
+ * Pure over numbers; reading the file size and the hardware facts is the caller's job.
+ */
+import { saturatingSub } from '../../util/index.js';
+/**
+ * Headroom left to the OS, the app and llama.cpp's own allocations on top of weights + KV cache:
+ * ~2.13 GiB per memory pool. Inherited, not measured (see the Rust comment).
+ */
+export const RESERVE_BYTES = 2_288_490_189;
+/** Vulkan's `IntegratedGpu` is the only trusted signal; anything else counts as discrete. */
+export function isIntegratedGpu(gpu) {
+    return gpu.vulkan_info?.device_type?.toLowerCase() === 'integratedgpu';
+}
+/**
+ * Integrated graphics shares system RAM, so adding it to RAM would count it twice (the 16 GB
+ * laptop that was told a 17 GB model fits). Only discrete GPUs add memory of their own. With no
+ * GPU enumerated (Apple Silicon) RAM *is* the VRAM: one pool.
+ */
+export function memoryBudget(totalRamBytes, gpus) {
+    if (gpus.length === 0) {
+        const usable = saturatingSub(totalRamBytes, RESERVE_BYTES);
+        return { usable_vram: usable, usable_total: usable };
+    }
+    const totalVram = gpus.reduce((n, g) => n + g.total_bytes, 0);
+    const discreteVram = gpus.filter((g) => !g.integrated).reduce((n, g) => n + g.total_bytes, 0);
+    return {
+        usable_vram: saturatingSub(totalVram, RESERVE_BYTES),
+        usable_total: saturatingSub(totalRamBytes, RESERVE_BYTES) + saturatingSub(discreteVram, RESERVE_BYTES),
+    };
+}
+/** RED = cannot run at all; GREEN = fits in VRAM; YELLOW = runs with CPU/GPU spill. */
+export function modelSupportStatus(requiredBytes, budget) {
+    if (requiredBytes > budget.usable_total)
+        return 'RED';
+    if (requiredBytes <= budget.usable_vram)
+        return 'GREEN';
+    return 'YELLOW';
+}
+/** `is_model_supported` after the file size and KV estimate are known. */
+export function isModelSupported(modelSizeBytes, kvCacheBytes, system) {
+    const gpus = system.gpus.map((g) => ({
+        total_bytes: g.total_memory * 1024 * 1024,
+        integrated: isIntegratedGpu(g),
+    }));
+    return modelSupportStatus(modelSizeBytes + kvCacheBytes, memoryBudget(system.total_memory * 1024 * 1024, gpus));
+}
+//# sourceMappingURL=support.js.map

package/dist/models/gguf/support.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"support.js","sourceRoot":"","sources":["../../../src/models/gguf/support.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAA;AAInD;;;GAGG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,aAAa,CAAA;AAe1C,6FAA6F;AAC7F,MAAM,UAAU,eAAe,CAAC,GAAsD;IACpF,OAAO,GAAG,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,eAAe,CAAA;AACxE,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,YAAY,CAAC,aAAqB,EAAE,IAAiB;IACnE,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtB,MAAM,MAAM,GAAG,aAAa,CAAC,aAAa,EAAE,aAAa,CAAC,CAAA;QAC1D,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,CAAA;IACtD,CAAC;IACD,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAA;IAC7D,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAA;IAC7F,OAAO;QACL,WAAW,EAAE,aAAa,CAAC,SAAS,EAAE,aAAa,CAAC;QACpD,YAAY,EAAE,aAAa,CAAC,aAAa,EAAE,aAAa,CAAC,GAAG,aAAa,CAAC,YAAY,EAAE,aAAa,CAAC;KACvG,CAAA;AACH,CAAC;AAED,uFAAuF;AACvF,MAAM,UAAU,kBAAkB,CAAC,aAAqB,EAAE,MAAoB;IAC5E,IAAI,aAAa,GAAG,MAAM,CAAC,YAAY;QAAE,OAAO,KAAK,CAAA;IACrD,IAAI,aAAa,IAAI,MAAM,CAAC,WAAW;QAAE,OAAO,OAAO,CAAA;IACvD,OAAO,QAAQ,CAAA;AACjB,CAAC;AAQD,0EAA0E;AAC1E,MAAM,UAAU,gBAAgB,CAC9B,cAAsB,EACtB,YAAoB,EACpB,MAAyB;IAEzB,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACnC,WAAW,EAAE,CAAC,CAAC,YAAY,GAAG,IAAI,GAAG,IAAI;QACzC,UAAU,EAAE,eAAe,CAAC,CAAC,CAAC;KAC/B,CAAC,CAAC,CAAA;IACH,OAAO,kBAAkB,CACvB,cAAc,GAAG,YAAY,EAC7B,YAAY,CAAC,MAAM,CAAC,YAAY,GAAG,IAAI,GAAG,IAAI,EAAE,IAAI,CAAC,CACtD,CAAA;AACH,CAAC"}

package/dist/models/hf.d.ts

@@ -0,0 +1,35 @@
+/**
+ * Hugging Face GGUF discovery and import used by `serve owner/repo`. The HTTP metadata call is
+ * separate from the downloader so selection is pure and tests never need the public service.
+ */
+import type { DataLayout } from '../config/index.js';
+import type { DownloaderEmit } from '../downloads/index.js';
+import type { ModelRegistry } from './registry.js';
+export interface HfFileInfo {
+    filename: string;
+    size: number;
+    sha256?: string;
+    downloadUrl: string;
+}
+export interface HfDownloadOptions {
+    layout: DataLayout;
+    registry: ModelRegistry;
+    repoId: string;
+    file: HfFileInfo;
+    fetch?: typeof fetch;
+    env?: NodeJS.ProcessEnv;
+    emit?: DownloaderEmit;
+}
+/** Exactly `owner/repo`, with the same conservative character set as the legacy Rust CLI. */
+export declare function looksLikeHfRepo(value: string): boolean;
+/** `HF_TOKEN` is the Hub-wide convention; retain the older alias as a fallback. */
+export declare function hfToken(env?: NodeJS.ProcessEnv): string | undefined;
+/** Fetch and normalize the GGUF siblings, smallest first. */
+export declare function fetchHfGgufFiles(repoId: string, options?: {
+    fetch?: typeof fetch;
+    token?: string;
+}): Promise<HfFileInfo[]>;
+/** Prefer the Rust CLI's default quantization, then the largest known file. */
+export declare function chooseDefaultHfFile(files: HfFileInfo[]): HfFileInfo;
+/** Download into the app's shared model tree, validate, then atomically publish via `model.yml`. */
+export declare function downloadHfModel(options: HfDownloadOptions): Promise<string>;

package/dist/models/hf.js

@@ -0,0 +1,140 @@
+/**
+ * Hugging Face GGUF discovery and import used by `serve owner/repo`. The HTTP metadata call is
+ * separate from the downloader so selection is pure and tests never need the public service.
+ */
+import { stat } from 'node:fs/promises';
+import { isAbsolute, join, relative, sep } from 'node:path';
+import { AtomicCoreError } from '../contracts/index.js';
+import { modelDirFromId } from '../config/index.js';
+import { Downloader } from '../downloads/index.js';
+/** Exactly `owner/repo`, with the same conservative character set as the legacy Rust CLI. */
+export function looksLikeHfRepo(value) {
+    if (value.startsWith('/') || value.startsWith('.') || value.startsWith('~'))
+        return false;
+    const parts = value.split('/');
+    return (parts.length === 2 &&
+        parts.every((part) => part.length > 0 && part !== '.' && part !== '..' && /^[\p{L}\p{N}._-]+$/u.test(part)));
+}
+/** `HF_TOKEN` is the Hub-wide convention; retain the older alias as a fallback. */
+export function hfToken(env = process.env) {
+    return nonEmpty(env['HF_TOKEN']) ?? nonEmpty(env['HUGGING_FACE_HUB_TOKEN']);
+}
+/** Fetch and normalize the GGUF siblings, smallest first. */
+export async function fetchHfGgufFiles(repoId, options = {}) {
+    if (!looksLikeHfRepo(repoId)) {
+        throw new AtomicCoreError('INVALID_ARGUMENT', `Invalid Hugging Face repository id "${repoId}".`);
+    }
+    const fetchImpl = options.fetch ?? fetch;
+    const headers = options.token ? { authorization: `Bearer ${options.token}` } : undefined;
+    const url = `https://huggingface.co/api/models/${repoId}?blobs=true&files_metadata=true`;
+    const response = await fetchImpl(url, { ...(headers ? { headers } : {}) });
+    if (!response.ok) {
+        const details = response.status === 401 || response.status === 403
+            ? 'The repository may be gated; set HF_TOKEN and accept its access conditions.'
+            : response.status === 404
+                ? 'Check the owner/repository spelling.'
+                : undefined;
+        throw new AtomicCoreError('IO_ERROR', `Hugging Face returned HTTP ${response.status} for "${repoId}".`, details);
+    }
+    const body = (await response.json());
+    if (!Array.isArray(body.siblings)) {
+        throw new AtomicCoreError('IO_ERROR', 'Unexpected Hugging Face model metadata response.');
+    }
+    const files = body.siblings
+        .map(parseSibling(repoId))
+        .filter((file) => file !== undefined)
+        .sort((a, b) => a.size - b.size || a.filename.localeCompare(b.filename));
+    if (files.length === 0) {
+        throw new AtomicCoreError('MODEL_NOT_FOUND', `No GGUF files found in Hugging Face repo "${repoId}".`);
+    }
+    return files;
+}
+/** Prefer the Rust CLI's default quantization, then the largest known file. */
+export function chooseDefaultHfFile(files) {
+    const preferred = files.find((file) => file.filename.includes('Q4_K_XL'));
+    const chosen = preferred ??
+        files.reduce((best, file) => {
+            return !best || file.size > best.size ? file : best;
+        }, undefined);
+    if (!chosen)
+        throw new AtomicCoreError('MODEL_NOT_FOUND', 'No GGUF file is available to download.');
+    return chosen;
+}
+/** Download into the app's shared model tree, validate, then atomically publish via `model.yml`. */
+export async function downloadHfModel(options) {
+    const { layout, registry, repoId, file } = options;
+    if (!looksLikeHfRepo(repoId)) {
+        throw new AtomicCoreError('INVALID_ARGUMENT', `Invalid Hugging Face repository id "${repoId}".`);
+    }
+    assertSafeRepoFile(file.filename);
+    const destination = join(modelDirFromId(registry.modelsDir, repoId), ...file.filename.split('/'));
+    const savePath = relative(layout.root, destination).split(sep).join('/');
+    const token = hfToken(options.env);
+    const downloader = new Downloader({
+        dataFolder: layout.root,
+        platform: process.platform,
+        fetch: options.fetch ?? fetch,
+        emit: options.emit ?? (() => { }),
+    });
+    await downloader.download(`hf:${repoId}`, [
+        {
+            url: file.downloadUrl,
+            save_path: savePath,
+            size: file.size || null,
+            sha256: file.sha256 ?? null,
+            model_id: repoId,
+        },
+    ], { headers: token ? { authorization: `Bearer ${token}` } : {}, resume: true });
+    const actualSize = (await stat(destination)).size;
+    await registry.write(repoId, {
+        model_path: savePath,
+        name: repoId.split('/')[1],
+        size_bytes: actualSize,
+        model_size_bytes: actualSize,
+        ...(file.sha256 ? { model_sha256: file.sha256 } : {}),
+        embedding: false,
+    });
+    return repoId;
+}
+function parseSibling(repoId) {
+    return (value) => {
+        if (!value || typeof value !== 'object')
+            return undefined;
+        const sibling = value;
+        const filename = sibling['rfilename'];
+        if (typeof filename !== 'string' || !filename.toLowerCase().endsWith('.gguf'))
+            return undefined;
+        try {
+            assertSafeRepoFile(filename);
+        }
+        catch {
+            return undefined;
+        }
+        const lfs = sibling['lfs'] && typeof sibling['lfs'] === 'object' ? sibling['lfs'] : {};
+        const size = finiteSize(lfs['size']) ?? finiteSize(sibling['size']) ?? 0;
+        const sha256 = typeof lfs['sha256'] === 'string' && lfs['sha256'] ? lfs['sha256'] : undefined;
+        const encoded = filename.split('/').map(encodeURIComponent).join('/');
+        return {
+            filename,
+            size,
+            ...(sha256 ? { sha256 } : {}),
+            downloadUrl: `https://huggingface.co/${repoId}/resolve/main/${encoded}`,
+        };
+    };
+}
+function assertSafeRepoFile(filename) {
+    const parts = filename.split('/');
+    if (filename.includes('\\') ||
+        isAbsolute(filename) ||
+        parts.some((part) => !part || part === '.' || part === '..')) {
+        throw new AtomicCoreError('INVALID_ARGUMENT', `Unsafe Hugging Face filename "${filename}".`);
+    }
+}
+function finiteSize(value) {
+    return typeof value === 'number' && Number.isSafeInteger(value) && value >= 0 ? value : undefined;
+}
+function nonEmpty(value) {
+    const trimmed = value?.trim();
+    return trimmed ? trimmed : undefined;
+}
+//# sourceMappingURL=hf.js.map

package/dist/models/hf.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hf.js","sourceRoot":"","sources":["../../src/models/hf.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAA;AACvC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,WAAW,CAAA;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAA;AAEvD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAA;AACnD,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAA;AAqBlD,6FAA6F;AAC7F,MAAM,UAAU,eAAe,CAAC,KAAa;IAC3C,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,KAAK,CAAA;IACzF,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAC9B,OAAO,CACL,KAAK,CAAC,MAAM,KAAK,CAAC;QAClB,KAAK,CAAC,KAAK,CACT,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,IAAI,IAAI,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,CAC/F,CACF,CAAA;AACH,CAAC;AAED,mFAAmF;AACnF,MAAM,UAAU,OAAO,CAAC,MAAyB,OAAO,CAAC,GAAG;IAC1D,OAAO,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,QAAQ,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC,CAAA;AAC7E,CAAC;AAED,6DAA6D;AAC7D,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,MAAc,EACd,UAAoD,EAAE;IAEtD,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,eAAe,CAAC,kBAAkB,EAAE,uCAAuC,MAAM,IAAI,CAAC,CAAA;IAClG,CAAC;IACD,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,IAAI,KAAK,CAAA;IACxC,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAA;IACxF,MAAM,GAAG,GAAG,qCAAqC,MAAM,iCAAiC,CAAA;IACxF,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAA;IAC1E,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,OAAO,GACX,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG;YAChD,CAAC,CAAC,6EAA6E;YAC/E,CAAC,CAAC,QAAQ,CAAC,MAAM,KAAK,GAAG;gBACvB,CAAC,CAAC,sCAAsC;gBACxC,CAAC,CAAC,SAAS,CAAA;QACjB,MAAM,IAAI,eAAe,CACvB,UAAU,EACV,8BAA8B,QAAQ,CAAC,MAAM,SAAS,MAAM,IAAI,EAChE,OAAO,CACR,CAAA;IACH,CAAC;IACD,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA2B,CAAA;IAC9D,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QAClC,MAAM,IAAI,eAAe,CAAC,UAAU,EAAE,kDAAkD,CAAC,CAAA;IAC3F,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ;SACxB,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;SACzB,MAAM,CAAC,CAAC,IAAI,EAAsB,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC;SACxD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAA;IAC1E,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,6CAA6C,MAAM,IAAI,CAAC,CAAA;IACvG,CAAC;IACD,OAAO,KAAK,CAAA;AACd,CAAC;AAED,+EAA+E;AAC/E,MAAM,UAAU,mBAAmB,CAAC,KAAmB;IACrD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAA;IACzE,MAAM,MAAM,GACV,SAAS;QACT,KAAK,CAAC,MAAM,CAAyB,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE;YAClD,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAA;QACrD,CAAC,EAAE,SAAS,CAAC,CAAA;IACf,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,wCAAwC,CAAC,CAAA;IACnG,OAAO,MAAM,CAAA;AACf,CAAC;AAED,oGAAoG;AACpG,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,OAA0B;IAC9D,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAA;IAClD,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,eAAe,CAAC,kBAAkB,EAAE,uCAAuC,MAAM,IAAI,CAAC,CAAA;IAClG,CAAC;IACD,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IACjC,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAA;IACjG,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACxE,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;IAClC,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC;QAChC,UAAU,EAAE,MAAM,CAAC,IAAI;QACvB,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,KAAK;QAC7B,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;KACjC,CAAC,CAAA;IACF,MAAM,UAAU,CAAC,QAAQ,CACvB,MAAM,MAAM,EAAE,EACd;QACE;YACE,GAAG,EAAE,IAAI,CAAC,WAAW;YACrB,SAAS,EAAE,QAAQ;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI;YACvB,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,IAAI;YAC3B,QAAQ,EAAE,MAAM;SACjB;KACF,EACD,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAC7E,CAAA;IACD,MAAM,UAAU,GAAG,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAA;IACjD,MAAM,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE;QAC3B,UAAU,EAAE,QAAQ;QACpB,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAW;QACpC,UAAU,EAAE,UAAU;QACtB,gBAAgB,EAAE,UAAU;QAC5B,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrD,SAAS,EAAE,KAAK;KACjB,CAAC,CAAA;IACF,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAS,YAAY,CAAC,MAAc;IAClC,OAAO,CAAC,KAAK,EAAE,EAAE;QACf,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,OAAO,SAAS,CAAA;QACzD,MAAM,OAAO,GAAG,KAAgC,CAAA;QAChD,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,CAAA;QACrC,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,SAAS,CAAA;QAC/F,IAAI,CAAC;YACH,kBAAkB,CAAC,QAAQ,CAAC,CAAA;QAC9B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,SAAS,CAAA;QAClB,CAAC;QACD,MAAM,GAAG,GACP,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,OAAO,CAAC,KAAK,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAE,OAAO,CAAC,KAAK,CAA6B,CAAC,CAAC,CAAC,EAAE,CAAA;QACzG,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAA;QACxE,MAAM,MAAM,GAAG,OAAO,GAAG,CAAC,QAAQ,CAAC,KAAK,QAAQ,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;QAC7F,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QACrE,OAAO;YACL,QAAQ;YACR,IAAI;YACJ,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7B,WAAW,EAAE,0BAA0B,MAAM,iBAAiB,OAAO,EAAE;SACxE,CAAA;IACH,CAAC,CAAA;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,QAAgB;IAC1C,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IACjC,IACE,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;QACvB,UAAU,CAAC,QAAQ,CAAC;QACpB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,IAAI,CAAC,EAC5D,CAAC;QACD,MAAM,IAAI,eAAe,CAAC,kBAAkB,EAAE,iCAAiC,QAAQ,IAAI,CAAC,CAAA;IAC9F,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CAAC,KAAc;IAChC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAA;AACnG,CAAC;AAED,SAAS,QAAQ,CAAC,KAAyB;IACzC,MAAM,OAAO,GAAG,KAAK,EAAE,IAAI,EAAE,CAAA;IAC7B,OAAO,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAA;AACtC,CAAC"}

package/dist/models/index.d.ts

@@ -0,0 +1,14 @@
+/**
+ * model.yml codec, shard helpers, GGUF metadata (reader, KV-cache estimate, fit check,
+ * classification). Registry/scan, import and HF land in phase 1.
+ *
+ * Ported from: guest-js/types.ts, extensions/llamacpp-upstream-extension/src/{index,util}.ts,
+ * tauri-plugin-llamacpp-upstream/src/gguf. See PLAN.md §3.2.
+ */
+export * from './model-yml.js';
+export * from './shards.js';
+export * from './gguf/index.js';
+export * from './capabilities.js';
+export * from './embed.js';
+export * from './registry.js';
+export * from './hf.js';

package/dist/models/index.js

@@ -0,0 +1,15 @@
+/**
+ * model.yml codec, shard helpers, GGUF metadata (reader, KV-cache estimate, fit check,
+ * classification). Registry/scan, import and HF land in phase 1.
+ *
+ * Ported from: guest-js/types.ts, extensions/llamacpp-upstream-extension/src/{index,util}.ts,
+ * tauri-plugin-llamacpp-upstream/src/gguf. See PLAN.md §3.2.
+ */
+export * from './model-yml.js';
+export * from './shards.js';
+export * from './gguf/index.js';
+export * from './capabilities.js';
+export * from './embed.js';
+export * from './registry.js';
+export * from './hf.js';
+//# sourceMappingURL=index.js.map

package/dist/models/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/models/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,cAAc,gBAAgB,CAAA;AAC9B,cAAc,aAAa,CAAA;AAC3B,cAAc,iBAAiB,CAAA;AAC/B,cAAc,mBAAmB,CAAA;AACjC,cAAc,YAAY,CAAA;AAC1B,cAAc,eAAe,CAAA;AAC7B,cAAc,SAAS,CAAA"}

package/dist/models/model-yml.d.ts

@@ -0,0 +1,22 @@
+/**
+ * `model.yml` codec. The app writes it with serde_yaml (`write_yaml`) and reads it back with
+ * `read_yaml`; the shape is `ModelYml` in `contracts/model-yml.ts`. Unknown keys are preserved on
+ * round-trip (PLAN.md §5.1); known keys are written in the app's order, `undefined` ones omitted.
+ */
+import type { ModelYml } from '../contracts/index.js';
+export declare class ModelYmlError extends Error {
+    constructor(message: string);
+}
+/** A parsed file: the typed fields plus whatever else was in it. */
+export type ModelYmlDocument = ModelYml & Record<string, unknown>;
+export declare function parseModelYml(text: string): ModelYmlDocument;
+/** Anything with a `model_path`; other known keys are optional and may be `undefined` (omitted). */
+export type ModelYmlInput = {
+    model_path: string;
+} & {
+    [key: string]: unknown;
+};
+/** Known keys first in the app's order, then unknown keys in their original order. */
+export declare function serializeModelYml(doc: ModelYmlInput): string;
+/** Rename the model id inside the path fields, as `update()` does when a model is renamed. */
+export declare function renameModelPaths(doc: ModelYmlDocument, modelsRootPrefix: string, oldId: string, newId: string): ModelYmlDocument;