parakeet.ts 1.0.0

package/dist/audio.js

@@ -0,0 +1,324 @@
+/**
+ * Backend-agnostic audio front-end — pure TypeScript, no tensor library.
+ *
+ * This is a port of the MLX front-end in `src/mlx/audio.ts`, with the MxArray
+ * tail (filterbank matmul, log, per-feature normalize) rewritten over plain
+ * typed arrays so it runs identically under the MLX and ONNX backends.
+ *
+ * It intentionally reproduces parakeet-mlx's magnitude convention
+ * (`|re| + |im|`, not `sqrt(re^2 + im^2)`) so both backends see the same
+ * features. See docs/cuda.md for the parity analysis.
+ */
+import { spawnSync } from 'node:child_process';
+// ---------------------------------------------------------------------------
+// Audio loading
+// ---------------------------------------------------------------------------
+/** Decode any ffmpeg-readable file to mono float32 PCM in [-1, 1]. */
+export function loadAudioRaw(filename, samplingRate) {
+    const result = spawnSync('ffmpeg', [
+        '-nostdin', '-i', filename,
+        '-threads', '0',
+        '-f', 's16le',
+        '-ac', '1',
+        '-acodec', 'pcm_s16le',
+        '-ar', String(samplingRate),
+        '-',
+    ], { maxBuffer: 512 * 1024 * 1024 });
+    if (result.error)
+        throw new Error(`FFmpeg not found: ${result.error.message}`);
+    if (result.status !== 0) {
+        throw new Error(`FFmpeg failed: ${result.stderr?.toString() ?? 'unknown error'}`);
+    }
+    const buf = result.stdout;
+    const int16 = new Int16Array(buf.buffer, buf.byteOffset, buf.byteLength / 2);
+    const float32 = new Float32Array(int16.length);
+    for (let i = 0; i < int16.length; i++)
+        float32[i] = int16[i] / 32768.0;
+    return float32;
+}
+// ---------------------------------------------------------------------------
+// Windows
+// ---------------------------------------------------------------------------
+function hanningWindow(size) {
+    const w = new Float32Array(size);
+    for (let i = 0; i < size; i++)
+        w[i] = 0.5 - 0.5 * Math.cos((2 * Math.PI * i) / size);
+    return w;
+}
+function hammingWindow(size) {
+    const w = new Float32Array(size);
+    for (let i = 0; i < size; i++)
+        w[i] = 0.54 - 0.46 * Math.cos((2 * Math.PI * i) / size);
+    return w;
+}
+function blackmanWindow(size) {
+    const w = new Float32Array(size);
+    for (let i = 0; i < size; i++) {
+        w[i] = 0.42 - 0.5 * Math.cos((2 * Math.PI * i) / size) + 0.08 * Math.cos((4 * Math.PI * i) / size);
+    }
+    return w;
+}
+function bartlettWindow(size) {
+    const w = new Float32Array(size);
+    for (let i = 0; i < size; i++)
+        w[i] = 1.0 - Math.abs((2 * i - size) / size);
+    return w;
+}
+export function getWindow(name, size) {
+    switch (name.toLowerCase()) {
+        case 'hann':
+        case 'hanning': return hanningWindow(size);
+        case 'hamming': return hammingWindow(size);
+        case 'blackman': return blackmanWindow(size);
+        case 'bartlett': return bartlettWindow(size);
+        default: return hanningWindow(size);
+    }
+}
+// ---------------------------------------------------------------------------
+// FFT / STFT
+// ---------------------------------------------------------------------------
+/** Radix-2 Cooley-Tukey FFT. Returns interleaved [re, im] for the first nFft/2+1 bins. */
+function rfft(input, nFft, re, im, out) {
+    const N = nFft;
+    re.set(input.subarray(0, Math.min(input.length, N)));
+    if (input.length < N)
+        re.fill(0, input.length);
+    im.fill(0);
+    let j = 0;
+    for (let i = 1; i < N; i++) {
+        let bit = N >> 1;
+        for (; j & bit; bit >>= 1)
+            j ^= bit;
+        j ^= bit;
+        if (i < j) {
+            const tr = re[i];
+            re[i] = re[j];
+            re[j] = tr;
+            const ti = im[i];
+            im[i] = im[j];
+            im[j] = ti;
+        }
+    }
+    for (let len = 2; len <= N; len <<= 1) {
+        const ang = (-2 * Math.PI) / len;
+        const wRe = Math.cos(ang), wIm = Math.sin(ang);
+        const half = len >> 1;
+        for (let i = 0; i < N; i += len) {
+            let curRe = 1.0, curIm = 0.0;
+            for (let k = 0; k < half; k++) {
+                const uRe = re[i + k], uIm = im[i + k];
+                const aRe = re[i + k + half], aIm = im[i + k + half];
+                const vRe = aRe * curRe - aIm * curIm;
+                const vIm = aRe * curIm + aIm * curRe;
+                re[i + k] = uRe + vRe;
+                im[i + k] = uIm + vIm;
+                re[i + k + half] = uRe - vRe;
+                im[i + k + half] = uIm - vIm;
+                const n = curRe * wRe - curIm * wIm;
+                curIm = curRe * wIm + curIm * wRe;
+                curRe = n;
+            }
+        }
+    }
+    const numBins = (N >> 1) + 1;
+    for (let i = 0; i < numBins; i++) {
+        out[i * 2] = re[i];
+        out[i * 2 + 1] = im[i];
+    }
+}
+/** STFT magnitude, [numFrames, nFft/2+1]. */
+function computeStft(signal, nFft, hopLength, winLength, window, magPower) {
+    const padding = Math.floor(nFft / 2);
+    const padded = new Float32Array(signal.length + 2 * padding);
+    for (let i = 0; i < padding; i++)
+        padded[padding - 1 - i] = signal[i + 1] ?? 0;
+    padded.set(signal, padding);
+    for (let i = 0; i < padding; i++) {
+        const srcIdx = signal.length - 2 - i;
+        padded[padding + signal.length + i] = signal[Math.max(0, srcIdx)] ?? 0;
+    }
+    const numBins = Math.floor(nFft / 2) + 1;
+    const numFrames = Math.max(0, Math.floor((padded.length - winLength) / hopLength) + 1);
+    // window is zero-padded to nFft at the END (matches parakeet-mlx)
+    const win = new Float32Array(nFft);
+    win.set(window.subarray(0, Math.min(winLength, nFft)));
+    const magnitude = new Float32Array(numFrames * numBins);
+    const frame = new Float32Array(nFft);
+    const re = new Float32Array(nFft);
+    const im = new Float32Array(nFft);
+    const spectrum = new Float32Array(numBins * 2);
+    for (let t = 0; t < numFrames; t++) {
+        const start = t * hopLength;
+        for (let i = 0; i < nFft; i++)
+            frame[i] = (padded[start + i] ?? 0) * win[i];
+        rfft(frame, nFft, re, im, spectrum);
+        const base = t * numBins;
+        if (magPower === 1.0) {
+            for (let k = 0; k < numBins; k++) {
+                magnitude[base + k] = Math.abs(spectrum[k * 2]) + Math.abs(spectrum[k * 2 + 1]);
+            }
+        }
+        else {
+            for (let k = 0; k < numBins; k++) {
+                const mag = Math.abs(spectrum[k * 2]) + Math.abs(spectrum[k * 2 + 1]);
+                magnitude[base + k] = Math.pow(mag, magPower);
+            }
+        }
+    }
+    return { magnitude, numFrames, numBins };
+}
+// ---------------------------------------------------------------------------
+// Mel filterbanks
+// ---------------------------------------------------------------------------
+/**
+ * Filterbank as parakeet-mlx builds it: mel points snapped to FFT bins with
+ * `floor`. NOTE: at 16 kHz / nFft 512 / 128 mels this collapses 13 rows to all
+ * zero — see docs/cuda.md. Kept as the default for bit-compatibility with the
+ * existing MLX path and the current test fixtures.
+ */
+export function computeMelFilterbanks(sr, nFft, nMels, fMin, fMax) {
+    const numBins = Math.floor(nFft / 2) + 1;
+    const hzToMel = (hz) => 2595 * Math.log10(1 + hz / 700);
+    const melToHz = (mel) => 700 * (Math.pow(10, mel / 2595) - 1);
+    const melMin = hzToMel(fMin), melMax = hzToMel(fMax);
+    const melPoints = new Float32Array(nMels + 2);
+    for (let i = 0; i < nMels + 2; i++)
+        melPoints[i] = melMin + (i / (nMels + 1)) * (melMax - melMin);
+    const hzPoints = melPoints.map(m => melToHz(m));
+    const binPoints = hzPoints.map(hz => Math.floor((hz * (nFft + 1)) / sr));
+    const fb = new Float32Array(nMels * numBins);
+    for (let m = 0; m < nMels; m++) {
+        const fLeft = binPoints[m], fCenter = binPoints[m + 1], fRight = binPoints[m + 2];
+        for (let k = fLeft; k < fCenter; k++) {
+            if (k >= 0 && k < numBins)
+                fb[m * numBins + k] = (k - fLeft) / Math.max(1, fCenter - fLeft);
+        }
+        for (let k = fCenter; k <= fRight; k++) {
+            if (k >= 0 && k < numBins)
+                fb[m * numBins + k] = (fRight - k) / Math.max(1, fRight - fCenter);
+        }
+    }
+    for (let m = 0; m < nMels; m++) {
+        const norm = 2.0 / Math.max(1e-8, hzPoints[m + 2] - hzPoints[m]);
+        for (let k = 0; k < numBins; k++)
+            fb[m * numBins + k] *= norm;
+    }
+    return fb;
+}
+/**
+ * librosa/NeMo-style filterbank: triangles evaluated on continuous frequencies,
+ * so no rows collapse. Matches NVIDIA's reference preprocessor. Opt in via
+ * `makePreprocessArgs({ filterbank: 'interpolated' })`.
+ */
+export function computeMelFilterbanksInterpolated(sr, nFft, nMels, fMin, fMax) {
+    const numBins = Math.floor(nFft / 2) + 1;
+    const hzToMel = (hz) => 2595 * Math.log10(1 + hz / 700);
+    const melToHz = (mel) => 700 * (Math.pow(10, mel / 2595) - 1);
+    const melMin = hzToMel(fMin), melMax = hzToMel(fMax);
+    const hzPoints = new Float64Array(nMels + 2);
+    for (let i = 0; i < nMels + 2; i++) {
+        hzPoints[i] = melToHz(melMin + (i / (nMels + 1)) * (melMax - melMin));
+    }
+    const fftFreqs = new Float64Array(numBins);
+    for (let k = 0; k < numBins; k++)
+        fftFreqs[k] = (k * sr) / nFft;
+    const fb = new Float32Array(nMels * numBins);
+    for (let m = 0; m < nMels; m++) {
+        const l = hzPoints[m], c = hzPoints[m + 1], r = hzPoints[m + 2];
+        const norm = 2.0 / Math.max(1e-8, r - l);
+        for (let k = 0; k < numBins; k++) {
+            const f = fftFreqs[k];
+            const up = (f - l) / Math.max(1e-8, c - l);
+            const dn = (r - f) / Math.max(1e-8, r - c);
+            fb[m * numBins + k] = Math.max(0, Math.min(up, dn)) * norm;
+        }
+    }
+    return fb;
+}
+// ---------------------------------------------------------------------------
+// Log-mel
+// ---------------------------------------------------------------------------
+export function makePreprocessArgs(args) {
+    const winLength = Math.round(args.windowSize * args.sampleRate);
+    const hopLength = Math.round(args.windowStride * args.sampleRate);
+    const build = args.filterbank === 'interpolated'
+        ? computeMelFilterbanksInterpolated
+        : computeMelFilterbanks;
+    const filterbanks = build(args.sampleRate, args.nFft, args.features, 0, args.sampleRate / 2);
+    return {
+        ...args,
+        padTo: args.padTo ?? 0,
+        padValue: args.padValue ?? 0,
+        preemph: args.preemph !== undefined ? args.preemph : 0.97,
+        magPower: args.magPower ?? 2.0,
+        winLength,
+        hopLength,
+        filterbanks,
+    };
+}
+/** Log-mel spectrogram from mono PCM. Output is [nMels, numFrames]. */
+export function getLogMel(signalIn, args) {
+    let signal = signalIn;
+    if (args.padTo > 0 && signal.length < args.padTo) {
+        const padded = new Float32Array(args.padTo).fill(args.padValue);
+        padded.set(signal);
+        signal = padded;
+    }
+    if (args.preemph !== null) {
+        const p = new Float32Array(signal.length);
+        p[0] = signal[0];
+        for (let i = 1; i < signal.length; i++)
+            p[i] = signal[i] - args.preemph * signal[i - 1];
+        signal = p;
+    }
+    const win = getWindow(args.window, args.winLength);
+    const { magnitude, numFrames, numBins } = computeStft(signal, args.nFft, args.hopLength, args.winLength, win, args.magPower);
+    const nMels = args.features;
+    const fb = args.filterbanks;
+    // mel[m, t] = log(sum_k fb[m, k] * mag[t, k] + 1e-5)
+    const out = new Float32Array(nMels * numFrames);
+    for (let m = 0; m < nMels; m++) {
+        const fbRow = m * numBins;
+        const outRow = m * numFrames;
+        for (let t = 0; t < numFrames; t++) {
+            let acc = 0;
+            const magRow = t * numBins;
+            for (let k = 0; k < numBins; k++)
+                acc += fb[fbRow + k] * magnitude[magRow + k];
+            out[outRow + t] = Math.log(acc + 1e-5);
+        }
+    }
+    if (args.normalize === 'per_feature') {
+        for (let m = 0; m < nMels; m++) {
+            const base = m * numFrames;
+            let mean = 0;
+            for (let t = 0; t < numFrames; t++)
+                mean += out[base + t];
+            mean /= numFrames;
+            let varr = 0;
+            for (let t = 0; t < numFrames; t++) {
+                const d = out[base + t] - mean;
+                varr += d * d;
+            }
+            const inv = 1 / (Math.sqrt(varr / numFrames) + 1e-5);
+            for (let t = 0; t < numFrames; t++)
+                out[base + t] = (out[base + t] - mean) * inv;
+        }
+    }
+    else {
+        let mean = 0;
+        for (let i = 0; i < out.length; i++)
+            mean += out[i];
+        mean /= out.length;
+        let varr = 0;
+        for (let i = 0; i < out.length; i++) {
+            const d = out[i] - mean;
+            varr += d * d;
+        }
+        const inv = 1 / (Math.sqrt(varr / out.length) + 1e-5);
+        for (let i = 0; i < out.length; i++)
+            out[i] = (out[i] - mean) * inv;
+    }
+    return { data: out, nMels, numFrames };
+}
+//# sourceMappingURL=audio.js.map

package/dist/audio.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"audio.js","sourceRoot":"","sources":["../src/audio.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AACH,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AA6B/C,8EAA8E;AAC9E,gBAAgB;AAChB,8EAA8E;AAE9E,sEAAsE;AACtE,MAAM,UAAU,YAAY,CAAC,QAAgB,EAAE,YAAoB;IACjE,MAAM,MAAM,GAAG,SAAS,CAAC,QAAQ,EAAE;QACjC,UAAU,EAAE,IAAI,EAAE,QAAQ;QAC1B,UAAU,EAAE,GAAG;QACf,IAAI,EAAE,OAAO;QACb,KAAK,EAAE,GAAG;QACV,SAAS,EAAE,WAAW;QACtB,KAAK,EAAE,MAAM,CAAC,YAAY,CAAC;QAC3B,GAAG;KACJ,EAAE,EAAE,SAAS,EAAE,GAAG,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC,CAAC;IAErC,IAAI,MAAM,CAAC,KAAK;QAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IAC/E,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,kBAAkB,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,eAAe,EAAE,CAAC,CAAC;IACpF,CAAC;IAED,MAAM,GAAG,GAAG,MAAM,CAAC,MAAgB,CAAC;IACpC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;IAC7E,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE;QAAE,OAAO,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC;IACvE,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E,SAAS,aAAa,CAAC,IAAY;IACjC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE;QAAE,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IACrF,OAAO,CAAC,CAAC;AACX,CAAC;AACD,SAAS,aAAa,CAAC,IAAY;IACjC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE;QAAE,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IACvF,OAAO,CAAC,CAAC;AACX,CAAC;AACD,SAAS,cAAc,CAAC,IAAY;IAClC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IACrG,CAAC;IACD,OAAO,CAAC,CAAC;AACX,CAAC;AACD,SAAS,cAAc,CAAC,IAAY;IAClC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE;QAAE,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;IAC5E,OAAO,CAAC,CAAC;AACX,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,IAAY,EAAE,IAAY;IAClD,QAAQ,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;QAC3B,KAAK,MAAM,CAAC;QACZ,KAAK,SAAS,CAAC,CAAC,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC;QAC3C,KAAK,SAAS,CAAC,CAAC,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC;QAC3C,KAAK,UAAU,CAAC,CAAC,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC;QAC7C,KAAK,UAAU,CAAC,CAAC,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC;QAC7C,OAAO,CAAC,CAAC,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC;AACH,CAAC;AAED,8EAA8E;AAC9E,aAAa;AACb,8EAA8E;AAE9E,0FAA0F;AAC1F,SAAS,IAAI,CAAC,KAAmB,EAAE,IAAY,EAAE,EAAgB,EAAE,EAAgB,EAAE,GAAiB;IACpG,MAAM,CAAC,GAAG,IAAI,CAAC;IACf,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACrD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC;QAAE,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/C,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAEX,IAAI,CAAC,GAAG,CAAC,CAAC;IACV,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC;QACjB,OAAO,CAAC,GAAG,GAAG,EAAE,GAAG,KAAK,CAAC;YAAE,CAAC,IAAI,GAAG,CAAC;QACpC,CAAC,IAAI,GAAG,CAAC;QACT,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YACV,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;YAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;YAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;YAC5C,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;YAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;YAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,IAAI,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;QACtC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC;QACjC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,GAAG,IAAI,CAAC,CAAC;QACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC;YAChC,IAAI,KAAK,GAAG,GAAG,EAAE,KAAK,GAAG,GAAG,CAAC;YAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9B,MAAM,GAAG,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;gBACvC,MAAM,GAAG,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;gBACrD,MAAM,GAAG,GAAG,GAAG,GAAG,KAAK,GAAG,GAAG,GAAG,KAAK,CAAC;gBACtC,MAAM,GAAG,GAAG,GAAG,GAAG,KAAK,GAAG,GAAG,GAAG,KAAK,CAAC;gBACtC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC;gBAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC;gBAC7C,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC;gBAAC,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC;gBAC3D,MAAM,CAAC,GAAG,KAAK,GAAG,GAAG,GAAG,KAAK,GAAG,GAAG,CAAC;gBACpC,KAAK,GAAG,KAAK,GAAG,GAAG,GAAG,KAAK,GAAG,GAAG,CAAC;gBAClC,KAAK,GAAG,CAAC,CAAC;YACZ,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;QAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;QAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;IAAC,CAAC;AACnF,CAAC;AAED,6CAA6C;AAC7C,SAAS,WAAW,CAClB,MAAoB,EACpB,IAAY,EACZ,SAAiB,EACjB,SAAiB,EACjB,MAAoB,EACpB,QAAgB;IAEhB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC;IAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE;QAAE,MAAM,CAAC,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;IAC/E,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;QACjC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC;IACzE,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;IACzC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,MAAM,GAAG,SAAS,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;IAEvF,kEAAkE;IAClE,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IACnC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;IAEvD,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,SAAS,GAAG,OAAO,CAAC,CAAC;IACxD,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,EAAE,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,EAAE,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;IAE/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,CAAC,GAAG,SAAS,CAAC;QAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE;YAAE,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QAC5E,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,QAAQ,CAAC,CAAC;QACpC,MAAM,IAAI,GAAG,CAAC,GAAG,OAAO,CAAC;QACzB,IAAI,QAAQ,KAAK,GAAG,EAAE,CAAC;YACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;gBACjC,SAAS,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAClF,CAAC;QACH,CAAC;aAAM,CAAC;YACN,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;gBACjC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBACtE,SAAS,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;AAC3C,CAAC;AAED,8EAA8E;AAC9E,kBAAkB;AAClB,8EAA8E;AAE9E;;;;;GAKG;AACH,MAAM,UAAU,qBAAqB,CACnC,EAAU,EAAE,IAAY,EAAE,KAAa,EAAE,IAAY,EAAE,IAAY;IAEnE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;IACzC,MAAM,OAAO,GAAG,CAAC,EAAU,EAAE,EAAE,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,CAAC;IAChE,MAAM,OAAO,GAAG,CAAC,GAAW,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IACtE,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAErD,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;IAC9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,GAAG,CAAC,EAAE,CAAC,EAAE;QAAE,SAAS,CAAC,CAAC,CAAC,GAAG,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC;IAClG,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IAEzE,MAAM,EAAE,GAAG,IAAI,YAAY,CAAC,KAAK,GAAG,OAAO,CAAC,CAAC;IAC7C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/B,MAAM,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAClF,KAAK,IAAI,CAAC,GAAG,KAAK,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,OAAO;gBAAE,EAAE,CAAC,CAAC,GAAG,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,GAAG,KAAK,CAAC,CAAC;QAC9F,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,OAAO,EAAE,CAAC,IAAI,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,OAAO;gBAAE,EAAE,CAAC,CAAC,GAAG,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC;QAChG,CAAC;IACH,CAAC;IACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QACjE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE;YAAE,EAAE,CAAC,CAAC,GAAG,OAAO,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC;IAChE,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,iCAAiC,CAC/C,EAAU,EAAE,IAAY,EAAE,KAAa,EAAE,IAAY,EAAE,IAAY;IAEnE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;IACzC,MAAM,OAAO,GAAG,CAAC,EAAU,EAAE,EAAE,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,CAAC;IAChE,MAAM,OAAO,GAAG,CAAC,GAAW,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IACtE,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAErD,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;IAC7C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,QAAQ,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC;IACxE,CAAC;IACD,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC;IAC3C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE;QAAE,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC;IAEhE,MAAM,EAAE,GAAG,IAAI,YAAY,CAAC,KAAK,GAAG,OAAO,CAAC,CAAC;IAC7C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/B,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAChE,MAAM,IAAI,GAAG,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;YACjC,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YACtB,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;YAC3C,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;YAC3C,EAAE,CAAC,CAAC,GAAG,OAAO,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC;QAC7D,CAAC;IACH,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E,MAAM,UAAU,kBAAkB,CAAC,IAelC;IACC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC;IAChE,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC;IAClE,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,KAAK,cAAc;QAC9C,CAAC,CAAC,iCAAiC;QACnC,CAAC,CAAC,qBAAqB,CAAC;IAC1B,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,EAAE,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;IAE7F,OAAO;QACL,GAAG,IAAI;QACP,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,CAAC;QACtB,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,CAAC;QAC5B,OAAO,EAAE,IAAI,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;QACzD,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,GAAG;QAC9B,SAAS;QACT,SAAS;QACT,WAAW;KACZ,CAAC;AACJ,CAAC;AAED,uEAAuE;AACvE,MAAM,UAAU,SAAS,CAAC,QAAsB,EAAE,IAAoB;IACpE,IAAI,MAAM,GAAG,QAAQ,CAAC;IAEtB,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;QACjD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACnB,MAAM,GAAG,MAAM,CAAC;IAClB,CAAC;IAED,IAAI,IAAI,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;QAC1B,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE;YAAE,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACxF,MAAM,GAAG,CAAC,CAAC;IACb,CAAC;IAED,MAAM,GAAG,GAAG,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IACnD,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,GACrC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IAErF,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC;IAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC;IAE5B,qDAAqD;IACrD,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,KAAK,GAAG,SAAS,CAAC,CAAC;IAChD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/B,MAAM,KAAK,GAAG,CAAC,GAAG,OAAO,CAAC;QAC1B,MAAM,MAAM,GAAG,CAAC,GAAG,SAAS,CAAC;QAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,IAAI,GAAG,GAAG,CAAC,CAAC;YACZ,MAAM,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC;YAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE;gBAAE,GAAG,IAAI,EAAE,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC/E,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,IAAI,IAAI,CAAC,SAAS,KAAK,aAAa,EAAE,CAAC;QACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,MAAM,IAAI,GAAG,CAAC,GAAG,SAAS,CAAC;YAC3B,IAAI,IAAI,GAAG,CAAC,CAAC;YACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE;gBAAE,IAAI,IAAI,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;YAC1D,IAAI,IAAI,SAAS,CAAC;YAClB,IAAI,IAAI,GAAG,CAAC,CAAC;YACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;gBAAC,MAAM,CAAC,GAAG,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC;gBAAC,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC;YAAC,CAAC;YACtF,MAAM,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC;YACrD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE;gBAAE,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,GAAG,CAAC;QACnF,CAAC;IACH,CAAC;SAAM,CAAC;QACN,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE;YAAE,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;QACpD,IAAI,IAAI,GAAG,CAAC,MAAM,CAAC;QACnB,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAAC,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;YAAC,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC;QAAC,CAAC;QAChF,MAAM,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC;QACtD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE;YAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,GAAG,CAAC;IACtE,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;AACzC,CAAC"}

package/dist/backend.d.ts

@@ -0,0 +1,61 @@
+/**
+ * The backend boundary.
+ *
+ * The library is abstracted at the *model* boundary, not the tensor boundary:
+ * everything except the encoder forward and the prediction+joint step is plain
+ * TypeScript and shared by all backends.
+ *
+ * Two calls are enough. The ONNX export fuses the prediction network and the
+ * joint network into a single `decoder_joint` graph that cannot be split, so
+ * `decodeStep` is the unit of work — the MLX backend simply calls its own
+ * `predict` and `joint` back to back behind it.
+ */
+/**
+ * Which axis is contiguous in `EncoderOutput.data`.
+ * ONNX emits [1, dim, time] ("dim-major"); the MLX encoder emits
+ * [1, time, dim] ("time-major"). Declaring it avoids a transpose copy.
+ */
+export type EncoderLayout = 'dim-major' | 'time-major';
+/** Encoder output for a single utterance, flattened row-major. */
+export interface EncoderOutput {
+    /** length dim * timeStride */
+    data: Float32Array;
+    /** encoder hidden size */
+    dim: number;
+    /** number of valid time frames */
+    frames: number;
+    /**
+     * Allocated extent of the time axis, which can exceed `frames` when the graph
+     * pads. Used as the row stride for 'dim-major'.
+     */
+    stride: number;
+    layout: EncoderLayout;
+}
+/**
+ * Opaque, backend-owned decoder state (the prediction network's LSTM state).
+ * The decode loop only threads it through; it never inspects it.
+ */
+export type DecoderStateHandle = unknown;
+export interface DecodeStepResult {
+    /** joint logits: vocab + blank + durations */
+    logits: Float32Array;
+    state: DecoderStateHandle;
+}
+export interface ParakeetBackend {
+    /** Human-readable backend id, e.g. "mlx" or "onnx-cuda". */
+    readonly name: string;
+    /** Encoder hidden size. */
+    readonly encoderDim: number;
+    /** Run the conformer encoder over a log-mel spectrogram laid out [nMels, numFrames]. */
+    encode(mel: Float32Array, nMels: number, numFrames: number): Promise<EncoderOutput>;
+    /**
+     * One autoregressive decode step.
+     * @param encFrame  a single encoder frame, length `encoderDim`
+     * @param token     previously emitted token id, or `null` at sequence start
+     * @param state     previous decoder state, or `null` at sequence start
+     */
+    decodeStep(encFrame: Float32Array, token: number | null, state: DecoderStateHandle | null): Promise<DecodeStepResult>;
+    /** Release any native resources. */
+    dispose?(): void | Promise<void>;
+}
+//# sourceMappingURL=backend.d.ts.map

package/dist/backend.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"backend.d.ts","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH;;;;GAIG;AACH,MAAM,MAAM,aAAa,GAAG,WAAW,GAAG,YAAY,CAAC;AAEvD,kEAAkE;AAClE,MAAM,WAAW,aAAa;IAC5B,8BAA8B;IAC9B,IAAI,EAAE,YAAY,CAAC;IACnB,0BAA0B;IAC1B,GAAG,EAAE,MAAM,CAAC;IACZ,kCAAkC;IAClC,MAAM,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,aAAa,CAAC;CACvB;AAED;;;GAGG;AACH,MAAM,MAAM,kBAAkB,GAAG,OAAO,CAAC;AAEzC,MAAM,WAAW,gBAAgB;IAC/B,8CAA8C;IAC9C,MAAM,EAAE,YAAY,CAAC;IACrB,KAAK,EAAE,kBAAkB,CAAC;CAC3B;AAED,MAAM,WAAW,eAAe;IAC9B,4DAA4D;IAC5D,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAEtB,2BAA2B;IAC3B,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAE5B,wFAAwF;IACxF,MAAM,CAAC,GAAG,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;IAEpF;;;;;OAKG;IACH,UAAU,CACR,QAAQ,EAAE,YAAY,EACtB,KAAK,EAAE,MAAM,GAAG,IAAI,EACpB,KAAK,EAAE,kBAAkB,GAAG,IAAI,GAC/B,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAE7B,oCAAoC;IACpC,OAAO,CAAC,IAAI,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAClC"}

package/dist/backend.js

@@ -0,0 +1,14 @@
+/**
+ * The backend boundary.
+ *
+ * The library is abstracted at the *model* boundary, not the tensor boundary:
+ * everything except the encoder forward and the prediction+joint step is plain
+ * TypeScript and shared by all backends.
+ *
+ * Two calls are enough. The ONNX export fuses the prediction network and the
+ * joint network into a single `decoder_joint` graph that cannot be split, so
+ * `decodeStep` is the unit of work — the MLX backend simply calls its own
+ * `predict` and `joint` back to back behind it.
+ */
+export {};
+//# sourceMappingURL=backend.js.map

package/dist/backend.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"backend.js","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG"}

package/dist/decode.d.ts

@@ -0,0 +1,38 @@
+/**
+ * Backend-agnostic greedy decoding.
+ *
+ * These loops previously lived inside the MLX backend and operated on MxArray.
+ * They are plain TypeScript over typed arrays now, and drive any
+ * `ParakeetBackend`. Semantics (blank handling, duration stepping, maxSymbols
+ * guard, entropy confidence, timestamps) are unchanged from the MLX versions in
+ * `src/mlx/rnnt.ts`.
+ */
+import type { ParakeetBackend, EncoderOutput, DecoderStateHandle } from './backend.js';
+import { AlignedToken } from './alignment.js';
+export interface DecoderState {
+    lastToken: number | null;
+    hiddenState: DecoderStateHandle | null;
+}
+export interface GreedyOptions {
+    vocabulary: string[];
+    maxSymbols: number | null;
+    timeRatio: number;
+    /** Absolute time offset (seconds) added to emitted token starts. */
+    timeOffset?: number;
+    /** Decode only frames in [from, to). Defaults to the whole encoder output. */
+    from?: number;
+    to?: number;
+}
+/**
+ * Greedy TDT decoding. `durations` is the TDT duration table, e.g. [0,1,2,3,4].
+ */
+export declare function decodeTDTGreedy(backend: ParakeetBackend, enc: EncoderOutput, durations: number[], state: DecoderState, opts: GreedyOptions): Promise<{
+    tokens: AlignedToken[];
+    state: DecoderState;
+}>;
+/** Greedy RNN-T decoding (no duration head). */
+export declare function decodeRNNTGreedy(backend: ParakeetBackend, enc: EncoderOutput, state: DecoderState, opts: GreedyOptions): Promise<{
+    tokens: AlignedToken[];
+    state: DecoderState;
+}>;
+//# sourceMappingURL=decode.d.ts.map

package/dist/decode.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"decode.d.ts","sourceRoot":"","sources":["../src/decode.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,KAAK,EAAE,eAAe,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AACvF,OAAO,EAAoB,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAGhE,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,WAAW,EAAE,kBAAkB,GAAG,IAAI,CAAC;CACxC;AAwCD,MAAM,WAAW,aAAa;IAC5B,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,oEAAoE;IACpE,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,8EAA8E;IAC9E,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,EAAE,CAAC,EAAE,MAAM,CAAC;CACb;AAED;;GAEG;AACH,wBAAsB,eAAe,CACnC,OAAO,EAAE,eAAe,EACxB,GAAG,EAAE,aAAa,EAClB,SAAS,EAAE,MAAM,EAAE,EACnB,KAAK,EAAE,YAAY,EACnB,IAAI,EAAE,aAAa,GAClB,OAAO,CAAC;IAAE,MAAM,EAAE,YAAY,EAAE,CAAC;IAAC,KAAK,EAAE,YAAY,CAAA;CAAE,CAAC,CA+C1D;AAED,gDAAgD;AAChD,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,eAAe,EACxB,GAAG,EAAE,aAAa,EAClB,KAAK,EAAE,YAAY,EACnB,IAAI,EAAE,aAAa,GAClB,OAAO,CAAC;IAAE,MAAM,EAAE,YAAY,EAAE,CAAC;IAAC,KAAK,EAAE,YAAY,CAAA;CAAE,CAAC,CAyC1D"}

package/dist/decode.js

@@ -0,0 +1,124 @@
+import { makeAlignedToken } from './alignment.js';
+import { decode as decodeTokens } from './tokenizer.js';
+/** Copy encoder frame `t` into `out`, honouring the backend's layout. */
+function readFrame(enc, t, out) {
+    if (enc.layout === 'time-major') {
+        const base = t * enc.dim;
+        for (let d = 0; d < enc.dim; d++)
+            out[d] = enc.data[base + d];
+    }
+    else {
+        for (let d = 0; d < enc.dim; d++)
+            out[d] = enc.data[d * enc.stride + t];
+    }
+    return out;
+}
+/** Softmax in place over `n` values starting at `offset`, returning the array. */
+function softmax(src, offset, n) {
+    const out = new Float32Array(n);
+    let max = -Infinity;
+    for (let i = 0; i < n; i++) {
+        const v = src[offset + i];
+        if (v > max)
+            max = v;
+    }
+    let sum = 0;
+    for (let i = 0; i < n; i++) {
+        const e = Math.exp(src[offset + i] - max);
+        out[i] = e;
+        sum += e;
+    }
+    for (let i = 0; i < n; i++)
+        out[i] /= sum;
+    return out;
+}
+function argmax(a) {
+    let best = 0, bv = a[0];
+    for (let i = 1; i < a.length; i++)
+        if (a[i] > bv) {
+            bv = a[i];
+            best = i;
+        }
+    return best;
+}
+/** Normalized-entropy confidence, matching the MLX implementation. */
+function confidenceOf(probs) {
+    let entropy = 0;
+    for (let i = 0; i < probs.length; i++) {
+        const p = probs[i] + 1e-10;
+        entropy -= p * Math.log(p);
+    }
+    return 1.0 - entropy / Math.log(probs.length);
+}
+/**
+ * Greedy TDT decoding. `durations` is the TDT duration table, e.g. [0,1,2,3,4].
+ */
+export async function decodeTDTGreedy(backend, enc, durations, state, opts) {
+    const { vocabulary, maxSymbols, timeRatio, timeOffset = 0 } = opts;
+    const from = opts.from ?? 0;
+    const to = Math.min(opts.to ?? enc.frames, enc.frames);
+    const vocabSize = vocabulary.length;
+    const tokens = [];
+    const frame = new Float32Array(enc.dim);
+    let { lastToken, hiddenState } = state;
+    let step = from;
+    let newSymbols = 0;
+    while (step < to) {
+        readFrame(enc, step, frame);
+        const { logits, state: nextState } = await backend.decodeStep(frame, lastToken, hiddenState);
+        const tokenProbs = softmax(logits, 0, vocabSize + 1);
+        const durationProbs = softmax(logits, vocabSize + 1, durations.length);
+        const predToken = argmax(tokenProbs);
+        const decision = argmax(durationProbs);
+        const isBlank = predToken === vocabSize;
+        if (!isBlank) {
+            tokens.push(makeAlignedToken(predToken, decodeTokens([predToken], vocabulary), timeOffset + step * timeRatio, durations[decision] * timeRatio, confidenceOf(tokenProbs)));
+            lastToken = predToken;
+            hiddenState = nextState;
+        }
+        step += durations[decision];
+        newSymbols += 1;
+        if (durations[decision] !== 0) {
+            newSymbols = 0;
+        }
+        else if (maxSymbols !== null && newSymbols >= maxSymbols) {
+            step += 1;
+            newSymbols = 0;
+        }
+    }
+    return { tokens, state: { lastToken, hiddenState } };
+}
+/** Greedy RNN-T decoding (no duration head). */
+export async function decodeRNNTGreedy(backend, enc, state, opts) {
+    const { vocabulary, maxSymbols, timeRatio, timeOffset = 0 } = opts;
+    const from = opts.from ?? 0;
+    const to = Math.min(opts.to ?? enc.frames, enc.frames);
+    const vocabSize = vocabulary.length;
+    const tokens = [];
+    const frame = new Float32Array(enc.dim);
+    let { lastToken, hiddenState } = state;
+    let step = from;
+    let newSymbols = 0;
+    while (step < to) {
+        readFrame(enc, step, frame);
+        const { logits, state: nextState } = await backend.decodeStep(frame, lastToken, hiddenState);
+        const tokenProbs = softmax(logits, 0, vocabSize + 1);
+        const predToken = argmax(tokenProbs);
+        if (predToken !== vocabSize) {
+            tokens.push(makeAlignedToken(predToken, decodeTokens([predToken], vocabulary), timeOffset + step * timeRatio, timeRatio, confidenceOf(tokenProbs)));
+            lastToken = predToken;
+            hiddenState = nextState;
+            newSymbols += 1;
+            if (maxSymbols !== null && newSymbols >= maxSymbols) {
+                step += 1;
+                newSymbols = 0;
+            }
+        }
+        else {
+            step += 1;
+            newSymbols = 0;
+        }
+    }
+    return { tokens, state: { lastToken, hiddenState } };
+}
+//# sourceMappingURL=decode.js.map

package/dist/decode.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"decode.js","sourceRoot":"","sources":["../src/decode.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,gBAAgB,EAAgB,MAAM,gBAAgB,CAAC;AAChE,OAAO,EAAE,MAAM,IAAI,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAOxD,yEAAyE;AACzE,SAAS,SAAS,CAAC,GAAkB,EAAE,CAAS,EAAE,GAAiB;IACjE,IAAI,GAAG,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC;QACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE;YAAE,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;IAChE,CAAC;SAAM,CAAC;QACN,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE;YAAE,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC1E,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,kFAAkF;AAClF,SAAS,OAAO,CAAC,GAAiB,EAAE,MAAc,EAAE,CAAS;IAC3D,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC;IACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAAC,MAAM,CAAC,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAAC,IAAI,CAAC,GAAG,GAAG;YAAE,GAAG,GAAG,CAAC,CAAC;IAAC,CAAC;IAChF,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAAC,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC;QAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAAC,GAAG,IAAI,CAAC,CAAC;IAAC,CAAC;IAChG,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;QAAE,GAAG,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC;IAC1C,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,MAAM,CAAC,CAAe;IAC7B,IAAI,IAAI,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACxB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE;QAAE,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;YAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAAC,IAAI,GAAG,CAAC,CAAC;QAAC,CAAC;IAC1E,OAAO,IAAI,CAAC;AACd,CAAC;AAED,sEAAsE;AACtE,SAAS,YAAY,CAAC,KAAmB;IACvC,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;QAC3B,OAAO,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IAC7B,CAAC;IACD,OAAO,GAAG,GAAG,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AAChD,CAAC;AAaD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,OAAwB,EACxB,GAAkB,EAClB,SAAmB,EACnB,KAAmB,EACnB,IAAmB;IAEnB,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,UAAU,GAAG,CAAC,EAAE,GAAG,IAAI,CAAC;IACnE,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC;IAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,IAAI,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACvD,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC;IAEpC,MAAM,MAAM,GAAmB,EAAE,CAAC;IAClC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACxC,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IACvC,IAAI,IAAI,GAAG,IAAI,CAAC;IAChB,IAAI,UAAU,GAAG,CAAC,CAAC;IAEnB,OAAO,IAAI,GAAG,EAAE,EAAE,CAAC;QACjB,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QAC5B,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,KAAK,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;QAE7F,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;QACrD,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,EAAE,SAAS,GAAG,CAAC,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;QAEvE,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;QACrC,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC;QACvC,MAAM,OAAO,GAAG,SAAS,KAAK,SAAS,CAAC;QAExC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAC1B,SAAS,EACT,YAAY,CAAC,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,EACrC,UAAU,GAAG,IAAI,GAAG,SAAS,EAC7B,SAAS,CAAC,QAAQ,CAAC,GAAG,SAAS,EAC/B,YAAY,CAAC,UAAU,CAAC,CACzB,CAAC,CAAC;YACH,SAAS,GAAG,SAAS,CAAC;YACtB,WAAW,GAAG,SAAS,CAAC;QAC1B,CAAC;QAED,IAAI,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC5B,UAAU,IAAI,CAAC,CAAC;QAEhB,IAAI,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,UAAU,GAAG,CAAC,CAAC;QACjB,CAAC;aAAM,IAAI,UAAU,KAAK,IAAI,IAAI,UAAU,IAAI,UAAU,EAAE,CAAC;YAC3D,IAAI,IAAI,CAAC,CAAC;YACV,UAAU,GAAG,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,CAAC;AACvD,CAAC;AAED,gDAAgD;AAChD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,OAAwB,EACxB,GAAkB,EAClB,KAAmB,EACnB,IAAmB;IAEnB,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,UAAU,GAAG,CAAC,EAAE,GAAG,IAAI,CAAC;IACnE,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC;IAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,IAAI,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACvD,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC;IAEpC,MAAM,MAAM,GAAmB,EAAE,CAAC;IAClC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACxC,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IACvC,IAAI,IAAI,GAAG,IAAI,CAAC;IAChB,IAAI,UAAU,GAAG,CAAC,CAAC;IAEnB,OAAO,IAAI,GAAG,EAAE,EAAE,CAAC;QACjB,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QAC5B,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,KAAK,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;QAE7F,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;QACrD,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;QAErC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAC1B,SAAS,EACT,YAAY,CAAC,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,EACrC,UAAU,GAAG,IAAI,GAAG,SAAS,EAC7B,SAAS,EACT,YAAY,CAAC,UAAU,CAAC,CACzB,CAAC,CAAC;YACH,SAAS,GAAG,SAAS,CAAC;YACtB,WAAW,GAAG,SAAS,CAAC;YACxB,UAAU,IAAI,CAAC,CAAC;YAChB,IAAI,UAAU,KAAK,IAAI,IAAI,UAAU,IAAI,UAAU,EAAE,CAAC;gBACpD,IAAI,IAAI,CAAC,CAAC;gBACV,UAAU,GAAG,CAAC,CAAC;YACjB,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,IAAI,CAAC,CAAC;YACV,UAAU,GAAG,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,CAAC;AACvD,CAAC"}

package/dist/hub.d.ts

@@ -0,0 +1,18 @@
+export type ProgressFn = (downloaded: number, total: number) => void;
+/** Default HF cache root, honouring HF_HOME / HF_HUB_CACHE like huggingface_hub. */
+export declare function defaultCacheDir(): string;
+/** Local directory a repo's files are cached in. */
+export declare function repoDir(repoId: string, cacheDir?: string): string;
+/**
+ * Fetch one file from a repo, returning its local path. No-op if already cached.
+ */
+export declare function downloadFromHub(repoId: string, filename: string, cacheDir?: string, onProgress?: ProgressFn): Promise<string>;
+/**
+ * Fetch several files, skipping any marked optional that 404.
+ * Returns the repo's local directory.
+ */
+export declare function downloadRepoFiles(repoId: string, files: Array<{
+    name: string;
+    optional?: boolean;
+}>, cacheDir?: string, onProgress?: (file: string, downloaded: number, total: number) => void): Promise<string>;
+//# sourceMappingURL=hub.d.ts.map

package/dist/hub.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hub.d.ts","sourceRoot":"","sources":["../src/hub.ts"],"names":[],"mappings":"AAUA,MAAM,MAAM,UAAU,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;AAErE,oFAAoF;AACpF,wBAAgB,eAAe,IAAI,MAAM,CAOxC;AAED,oDAAoD;AACpD,wBAAgB,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,CAEjE;AAED;;GAEG;AACH,wBAAsB,eAAe,CACnC,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,EAChB,QAAQ,CAAC,EAAE,MAAM,EACjB,UAAU,CAAC,EAAE,UAAU,GACtB,OAAO,CAAC,MAAM,CAAC,CA8CjB;AAED;;;GAGG;AACH,wBAAsB,iBAAiB,CACrC,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,KAAK,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC,EAClD,QAAQ,CAAC,EAAE,MAAM,EACjB,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,GACrE,OAAO,CAAC,MAAM,CAAC,CAUjB"}