@seanhogg/builderforce-memory 2026.6.18

package/dist/runtime/SSMRuntime.d.ts

@@ -0,0 +1,167 @@
+/**
+ * SSMRuntime – core runtime class for SSM.js.
+ *
+ * Wraps a MambaSession and adds:
+ *   - Hybrid inference routing (SSM ↔ transformer bridge)
+ *   - Unified generate / stream API
+ *   - Perplexity-aware routing (InferenceRouter auto mode)
+ *   - Thin save/load pass-throughs for MemoryStore integration
+ *
+ * GPU lifecycle is owned entirely by this class via the underlying session.
+ */
+import { type MambaSessionOptions, type CompleteOptions, type AdaptOptions, type AdaptResult, type SaveOptions, type LoadOptions, type CreateCallbacks } from '../session/index.js';
+import { type RoutingStrategy, type RoutingAuditEntry } from '../router/InferenceRouter.js';
+import type { DistillationLog } from '../distillation/DistillationEngine.js';
+import type { TransformerBridge, BridgeGenerateOptions } from '../bridges/TransformerBridge.js';
+export type GenerateOptions = CompleteOptions & {
+    /**
+     * Options forwarded to the transformer bridge when the router selects it.
+     * Ignored when SSM is selected.
+     */
+    bridgeOpts?: BridgeGenerateOptions;
+    /**
+     * Stable system / context prefix, kept separate from the volatile `input`.
+     *
+     * Splitting it out is what makes transformer-side prompt caching possible:
+     * on the SSM path it is prepended to `input` (reproducing the single-string
+     * trained prompt format, so SSM behaviour is unchanged), while on the
+     * transformer path it is sent as the cacheable `system` field rather than
+     * being concatenated into the user message. A bridge can then cache it
+     * across turns; concatenating it into the per-turn input could never cache.
+     */
+    system?: string;
+};
+export interface SSMRuntimeOptions {
+    /**
+     * Options forwarded verbatim to MambaSession.create().
+     * Controls model size, SSM variant, tokenizer, and checkpoint URL.
+     */
+    session: MambaSessionOptions;
+    /**
+     * Optional transformer bridge.  When absent, all requests go to the SSM
+     * and distillation is unavailable.
+     */
+    bridge?: TransformerBridge;
+    /**
+     * Routing strategy when a bridge is present.
+     * Default: 'auto'
+     */
+    routingStrategy?: RoutingStrategy;
+    /**
+     * Character length above which auto-routing prefers the transformer.
+     * Default: 1200
+     */
+    longInputThreshold?: number;
+    /**
+     * SSM perplexity above which auto-routing prefers the transformer.
+     * Default: 80
+     */
+    perplexityThreshold?: number;
+    /**
+     * MambaSession.create() progress callbacks.
+     */
+    callbacks?: CreateCallbacks;
+}
+export declare class SSMRuntime {
+    private readonly _session;
+    private readonly _bridge;
+    private readonly _router;
+    private _destroyed;
+    private constructor();
+    /**
+     * Creates and initialises a new SSMRuntime.
+     *
+     * Delegates to MambaSession.create() — can throw SessionError for
+     * GPU / tokenizer failures.
+     */
+    static create(opts: SSMRuntimeOptions): Promise<SSMRuntime>;
+    /**
+     * Generates a full response for the given input.
+     * Routes to SSM or transformer bridge based on the configured strategy.
+     */
+    generate(input: string, opts?: GenerateOptions): Promise<string>;
+    /**
+     * Streaming token generation.
+     *
+     * Routing note: streaming always uses the SSM path for consistent
+     * latency characteristics.  Use `generate()` if transformer streaming
+     * is needed (bridgeOpts.stream, handled by the bridge's own `stream()` method).
+     */
+    stream(input: string, opts?: GenerateOptions): AsyncIterable<string>;
+    /**
+     * Hybrid streaming: routes to SSM or transformer bridge stream.
+     * Falls back to `generate()` for bridges that don't support streaming.
+     */
+    streamHybrid(input: string, opts?: GenerateOptions): AsyncIterable<string>;
+    /**
+     * Combines the stable `system` prefix with the volatile `input` for the SSM
+     * path, reproducing the single-string format the model was trained on.
+     */
+    private _ssmInput;
+    /**
+     * Maps the top-level `system` prefix onto the bridge's `systemPrompt` so the
+     * transformer receives it as a cacheable `system` field. An explicit
+     * `bridgeOpts.systemPrompt` wins if the caller set one.
+     */
+    private _bridgeOptsWithSystem;
+    /**
+     * Fine-tunes the SSM on the provided text.
+     * Pass-through to MambaSession.adapt().
+     */
+    adapt(data: string, opts?: AdaptOptions): Promise<AdaptResult>;
+    /**
+     * Evaluates SSM perplexity on the provided text.
+     * Exposed for power users and InferenceRouter's perplexity probe.
+     */
+    evaluate(text: string): Promise<number>;
+    /**
+     * Returns an L2-normalised SSM embedding for `text`.
+     * Pass-through to MambaSession.embed(). Consumed by MemoryStore.recallSimilar
+     * for SSM-based semantic search.
+     */
+    embed(text: string): Promise<Float32Array>;
+    /**
+     * Saves SSM weights. Pass-through to MambaSession.save().
+     * Accepts a `key` option used by MemoryStore to save under a custom key.
+     */
+    save(opts?: SaveOptions & {
+        key?: string;
+    }): Promise<void>;
+    /**
+     * Loads SSM weights. Pass-through to MambaSession.load().
+     * Returns false when no checkpoint exists.
+     */
+    load(opts?: LoadOptions & {
+        key?: string;
+    }): Promise<boolean>;
+    /**
+     * Returns a copy of the in-memory routing audit log.
+     * Delegates to the internal InferenceRouter.
+     */
+    getRoutingAuditLog(): RoutingAuditEntry[];
+    /**
+     * Returns the distillation log if a DistillationEngine has been attached.
+     * Returns an empty array when no distillation engine is available.
+     *
+     * Note: DistillationEngine is a separate object — attach it by calling
+     *   `new DistillationEngine(runtime, bridge)` and keeping a reference.
+     *   This method is a stub for future inline engine support.
+     */
+    getDistillationLog(): DistillationLog[];
+    /** The transformer bridge attached to this runtime, if any. */
+    get bridge(): TransformerBridge | undefined;
+    /** Whether this runtime has been destroyed. */
+    get destroyed(): boolean;
+    /**
+     * Escape hatch to the underlying MambaSession internals.
+     * Use sparingly — the SSMRuntime API covers the common cases.
+     */
+    get internals(): import("../session/session.js").SessionInternals;
+    /**
+     * Destroys the underlying MambaSession and releases all GPU resources.
+     * After calling this, all methods throw SSMError('RUNTIME_DESTROYED').
+     */
+    destroy(): void;
+    private _checkAlive;
+}
+//# sourceMappingURL=SSMRuntime.d.ts.map

package/dist/runtime/SSMRuntime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"SSMRuntime.d.ts","sourceRoot":"","sources":["../../src/runtime/SSMRuntime.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAEH,KAAK,mBAAmB,EACxB,KAAK,eAAe,EACpB,KAAK,YAAY,EACjB,KAAK,WAAW,EAChB,KAAK,WAAW,EAChB,KAAK,WAAW,EAChB,KAAK,eAAe,EACvB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAmB,KAAK,eAAe,EAAE,KAAK,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAC7G,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uCAAuC,CAAC;AAC7E,OAAO,KAAK,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAKhG,MAAM,MAAM,eAAe,GAAG,eAAe,GAAG;IAC5C;;;OAGG;IACH,UAAU,CAAC,EAAE,qBAAqB,CAAC;IAEnC;;;;;;;;;OASG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,MAAM,WAAW,iBAAiB;IAC9B;;;OAGG;IACH,OAAO,EAAgB,mBAAmB,CAAC;IAE3C;;;OAGG;IACH,MAAM,CAAC,EAAgB,iBAAiB,CAAC;IAEzC;;;OAGG;IACH,eAAe,CAAC,EAAO,eAAe,CAAC;IAEvC;;;OAGG;IACH,kBAAkB,CAAC,EAAI,MAAM,CAAC;IAE9B;;;OAGG;IACH,mBAAmB,CAAC,EAAG,MAAM,CAAC;IAE9B;;OAEG;IACH,SAAS,CAAC,EAAa,eAAe,CAAC;CAC1C;AAID,qBAAa,UAAU;IACnB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAiB;IAC1C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmC;IAC3D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;IAC7C,OAAO,CAAC,UAAU,CAAS;IAE3B,OAAO;IAYP;;;;;OAKG;WACU,MAAM,CAAC,IAAI,EAAE,iBAAiB,GAAG,OAAO,CAAC,UAAU,CAAC;IAmBjE;;;OAGG;IACG,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,eAAoB,GAAG,OAAO,CAAC,MAAM,CAAC;IAc1E;;;;;;OAMG;IACI,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,eAAoB,GAAG,aAAa,CAAC,MAAM,CAAC;IAM/E;;;OAGG;IACI,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,eAAoB,GAAG,aAAa,CAAC,MAAM,CAAC;IAqBrF;;;OAGG;IACH,OAAO,CAAC,SAAS;IAIjB;;;;OAIG;IACH,OAAO,CAAC,qBAAqB;IAS7B;;;OAGG;IACG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IAKpE;;;OAGG;IACG,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAK7C;;;;OAIG;IACG,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAOhD;;;OAGG;IACG,IAAI,CAAC,IAAI,CAAC,EAAE,WAAW,GAAG;QAAE,GAAG,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAKhE;;;OAGG;IACG,IAAI,CAAC,IAAI,CAAC,EAAE,WAAW,GAAG;QAAE,GAAG,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC;IAOnE;;;OAGG;IACH,kBAAkB,IAAI,iBAAiB,EAAE;IAIzC;;;;;;;OAOG;IACH,kBAAkB,IAAI,eAAe,EAAE;IAIvC,+DAA+D;IAC/D,IAAI,MAAM,IAAI,iBAAiB,GAAG,SAAS,CAAyB;IAEpE,+CAA+C;IAC/C,IAAI,SAAS,IAAI,OAAO,CAA4B;IAEpD;;;OAGG;IACH,IAAI,SAAS,qDAAsC;IAInD;;;OAGG;IACH,OAAO,IAAI,IAAI;IAKf,OAAO,CAAC,WAAW;CAKtB"}

package/dist/runtime/SSMRuntime.js

@@ -0,0 +1,199 @@
+/**
+ * SSMRuntime – core runtime class for SSM.js.
+ *
+ * Wraps a MambaSession and adds:
+ *   - Hybrid inference routing (SSM ↔ transformer bridge)
+ *   - Unified generate / stream API
+ *   - Perplexity-aware routing (InferenceRouter auto mode)
+ *   - Thin save/load pass-throughs for MemoryStore integration
+ *
+ * GPU lifecycle is owned entirely by this class via the underlying session.
+ */
+import { MambaSession, } from '../session/index.js';
+import { InferenceRouter } from '../router/InferenceRouter.js';
+import { SSMError } from '../errors/SSMError.js';
+// ── SSMRuntime ────────────────────────────────────────────────────────────────
+export class SSMRuntime {
+    _session;
+    _bridge;
+    _router;
+    _destroyed = false;
+    constructor(session, bridge, router) {
+        this._session = session;
+        this._bridge = bridge;
+        this._router = router;
+    }
+    // ── Factory ───────────────────────────────────────────────────────────────
+    /**
+     * Creates and initialises a new SSMRuntime.
+     *
+     * Delegates to MambaSession.create() — can throw SessionError for
+     * GPU / tokenizer failures.
+     */
+    static async create(opts) {
+        const session = await MambaSession.create(opts.session, opts.callbacks);
+        const router = new InferenceRouter({
+            strategy: opts.routingStrategy,
+            longInputThreshold: opts.longInputThreshold,
+            perplexityThreshold: opts.perplexityThreshold,
+            hasBridge: !!opts.bridge,
+            // Pass perplexity probe as callback — avoids circular import
+            perplexityProbe: opts.bridge
+                ? (text) => session.evaluate(text)
+                : undefined,
+        });
+        return new SSMRuntime(session, opts.bridge, router);
+    }
+    // ── Inference ─────────────────────────────────────────────────────────────
+    /**
+     * Generates a full response for the given input.
+     * Routes to SSM or transformer bridge based on the configured strategy.
+     */
+    async generate(input, opts = {}) {
+        this._checkAlive();
+        const decision = await this._router.route(input);
+        if (decision.target === 'transformer' && this._bridge) {
+            return this._bridge.generate(input, this._bridgeOptsWithSystem(opts));
+        }
+        // SSM path — extract only CompleteOptions fields
+        const { bridgeOpts: _b, system, ...completeOpts } = opts;
+        return this._session.complete(this._ssmInput(input, system), completeOpts);
+    }
+    /**
+     * Streaming token generation.
+     *
+     * Routing note: streaming always uses the SSM path for consistent
+     * latency characteristics.  Use `generate()` if transformer streaming
+     * is needed (bridgeOpts.stream, handled by the bridge's own `stream()` method).
+     */
+    async *stream(input, opts = {}) {
+        this._checkAlive();
+        const { bridgeOpts: _b, system, ...completeOpts } = opts;
+        yield* this._session.completeStream(this._ssmInput(input, system), completeOpts);
+    }
+    /**
+     * Hybrid streaming: routes to SSM or transformer bridge stream.
+     * Falls back to `generate()` for bridges that don't support streaming.
+     */
+    async *streamHybrid(input, opts = {}) {
+        this._checkAlive();
+        const decision = await this._router.route(input);
+        if (decision.target === 'transformer' && this._bridge) {
+            const bridgeOpts = this._bridgeOptsWithSystem(opts);
+            if (this._bridge.supportsStreaming && this._bridge.stream) {
+                yield* this._bridge.stream(input, bridgeOpts);
+            }
+            else {
+                yield await this._bridge.generate(input, bridgeOpts);
+            }
+            return;
+        }
+        const { bridgeOpts: _b, system, ...completeOpts } = opts;
+        yield* this._session.completeStream(this._ssmInput(input, system), completeOpts);
+    }
+    // ── Prompt-prefix helpers ─────────────────────────────────────────────────
+    /**
+     * Combines the stable `system` prefix with the volatile `input` for the SSM
+     * path, reproducing the single-string format the model was trained on.
+     */
+    _ssmInput(input, system) {
+        return system ? `${system}\n${input}` : input;
+    }
+    /**
+     * Maps the top-level `system` prefix onto the bridge's `systemPrompt` so the
+     * transformer receives it as a cacheable `system` field. An explicit
+     * `bridgeOpts.systemPrompt` wins if the caller set one.
+     */
+    _bridgeOptsWithSystem(opts) {
+        return {
+            ...opts.bridgeOpts,
+            systemPrompt: opts.bridgeOpts?.systemPrompt ?? opts.system,
+        };
+    }
+    // ── Adaptation ────────────────────────────────────────────────────────────
+    /**
+     * Fine-tunes the SSM on the provided text.
+     * Pass-through to MambaSession.adapt().
+     */
+    async adapt(data, opts) {
+        this._checkAlive();
+        return this._session.adapt(data, opts);
+    }
+    /**
+     * Evaluates SSM perplexity on the provided text.
+     * Exposed for power users and InferenceRouter's perplexity probe.
+     */
+    async evaluate(text) {
+        this._checkAlive();
+        return this._session.evaluate(text);
+    }
+    /**
+     * Returns an L2-normalised SSM embedding for `text`.
+     * Pass-through to MambaSession.embed(). Consumed by MemoryStore.recallSimilar
+     * for SSM-based semantic search.
+     */
+    async embed(text) {
+        this._checkAlive();
+        return this._session.embed(text);
+    }
+    // ── Persistence ───────────────────────────────────────────────────────────
+    /**
+     * Saves SSM weights. Pass-through to MambaSession.save().
+     * Accepts a `key` option used by MemoryStore to save under a custom key.
+     */
+    async save(opts) {
+        this._checkAlive();
+        await this._session.save(opts);
+    }
+    /**
+     * Loads SSM weights. Pass-through to MambaSession.load().
+     * Returns false when no checkpoint exists.
+     */
+    async load(opts) {
+        this._checkAlive();
+        return this._session.load(opts);
+    }
+    // ── Accessors ─────────────────────────────────────────────────────────────
+    /**
+     * Returns a copy of the in-memory routing audit log.
+     * Delegates to the internal InferenceRouter.
+     */
+    getRoutingAuditLog() {
+        return this._router.getAuditLog();
+    }
+    /**
+     * Returns the distillation log if a DistillationEngine has been attached.
+     * Returns an empty array when no distillation engine is available.
+     *
+     * Note: DistillationEngine is a separate object — attach it by calling
+     *   `new DistillationEngine(runtime, bridge)` and keeping a reference.
+     *   This method is a stub for future inline engine support.
+     */
+    getDistillationLog() {
+        return [];
+    }
+    /** The transformer bridge attached to this runtime, if any. */
+    get bridge() { return this._bridge; }
+    /** Whether this runtime has been destroyed. */
+    get destroyed() { return this._destroyed; }
+    /**
+     * Escape hatch to the underlying MambaSession internals.
+     * Use sparingly — the SSMRuntime API covers the common cases.
+     */
+    get internals() { return this._session.internals; }
+    // ── Lifecycle ─────────────────────────────────────────────────────────────
+    /**
+     * Destroys the underlying MambaSession and releases all GPU resources.
+     * After calling this, all methods throw SSMError('RUNTIME_DESTROYED').
+     */
+    destroy() {
+        this._session.destroy();
+        this._destroyed = true;
+    }
+    _checkAlive() {
+        if (this._destroyed) {
+            throw new SSMError('RUNTIME_DESTROYED', 'SSMRuntime has been destroyed.');
+        }
+    }
+}
+//# sourceMappingURL=SSMRuntime.js.map

package/dist/runtime/SSMRuntime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SSMRuntime.js","sourceRoot":"","sources":["../../src/runtime/SSMRuntime.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EACH,YAAY,GAQf,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,eAAe,EAAgD,MAAM,8BAA8B,CAAC;AAG7G,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AA6DjD,iFAAiF;AAEjF,MAAM,OAAO,UAAU;IACF,QAAQ,CAAiB;IACzB,OAAO,CAAmC;IAC1C,OAAO,CAAqB;IACrC,UAAU,GAAG,KAAK,CAAC;IAE3B,YACI,OAAsB,EACtB,MAAuC,EACvC,MAAyB;QAEzB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,IAAI,CAAC,OAAO,GAAI,MAAM,CAAC;QACvB,IAAI,CAAC,OAAO,GAAI,MAAM,CAAC;IAC3B,CAAC;IAED,6EAA6E;IAE7E;;;;;OAKG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAuB;QACvC,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAExE,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC;YAC/B,QAAQ,EAAc,IAAI,CAAC,eAAe;YAC1C,kBAAkB,EAAI,IAAI,CAAC,kBAAkB;YAC7C,mBAAmB,EAAG,IAAI,CAAC,mBAAmB;YAC9C,SAAS,EAAa,CAAC,CAAC,IAAI,CAAC,MAAM;YACnC,6DAA6D;YAC7D,eAAe,EAAO,IAAI,CAAC,MAAM;gBAC7B,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAClC,CAAC,CAAC,SAAS;SAClB,CAAC,CAAC;QAEH,OAAO,IAAI,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACxD,CAAC;IAED,6EAA6E;IAE7E;;;OAGG;IACH,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,OAAwB,EAAE;QACpD,IAAI,CAAC,WAAW,EAAE,CAAC;QAEnB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAEjD,IAAI,QAAQ,CAAC,MAAM,KAAK,aAAa,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACpD,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1E,CAAC;QAED,iDAAiD;QACjD,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,YAAY,EAAE,GAAG,IAAI,CAAC;QACzD,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,YAAY,CAAC,CAAC;IAC/E,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,CAAC,MAAM,CAAC,KAAa,EAAE,OAAwB,EAAE;QACnD,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,YAAY,EAAE,GAAG,IAAI,CAAC;QACzD,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,YAAY,CAAC,CAAC;IACrF,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,CAAC,YAAY,CAAC,KAAa,EAAE,OAAwB,EAAE;QACzD,IAAI,CAAC,WAAW,EAAE,CAAC;QAEnB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAEjD,IAAI,QAAQ,CAAC,MAAM,KAAK,aAAa,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACpD,MAAM,UAAU,GAAG,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;YACpD,IAAI,IAAI,CAAC,OAAO,CAAC,iBAAiB,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBACxD,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAClD,CAAC;iBAAM,CAAC;gBACJ,MAAM,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YACzD,CAAC;YACD,OAAO;QACX,CAAC;QAED,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,YAAY,EAAE,GAAG,IAAI,CAAC;QACzD,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,YAAY,CAAC,CAAC;IACrF,CAAC;IAED,6EAA6E;IAE7E;;;OAGG;IACK,SAAS,CAAC,KAAa,EAAE,MAA0B;QACvD,OAAO,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,KAAK,KAAK,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;IAClD,CAAC;IAED;;;;OAIG;IACK,qBAAqB,CAAC,IAAqB;QAC/C,OAAO;YACH,GAAG,IAAI,CAAC,UAAU;YAClB,YAAY,EAAE,IAAI,CAAC,UAAU,EAAE,YAAY,IAAI,IAAI,CAAC,MAAM;SAC7D,CAAC;IACN,CAAC;IAED,6EAA6E;IAE7E;;;OAGG;IACH,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,IAAmB;QACzC,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,QAAQ,CAAC,IAAY;QACvB,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,KAAK,CAAC,IAAY;QACpB,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC;IAED,6EAA6E;IAE7E;;;OAGG;IACH,KAAK,CAAC,IAAI,CAAC,IAAqC;QAC5C,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,IAAI,CAAC,IAAqC;QAC5C,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAqB,CAAC;IACxD,CAAC;IAED,6EAA6E;IAE7E;;;OAGG;IACH,kBAAkB;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;IACtC,CAAC;IAED;;;;;;;OAOG;IACH,kBAAkB;QACd,OAAO,EAAE,CAAC;IACd,CAAC;IAED,+DAA+D;IAC/D,IAAI,MAAM,KAAoC,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAEpE,+CAA+C;IAC/C,IAAI,SAAS,KAAc,OAAO,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;IAEpD;;;OAGG;IACH,IAAI,SAAS,KAAK,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IAEnD,6EAA6E;IAE7E;;;OAGG;IACH,OAAO;QACH,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;QACxB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;IAC3B,CAAC;IAEO,WAAW;QACf,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAClB,MAAM,IAAI,QAAQ,CAAC,mBAAmB,EAAE,gCAAgC,CAAC,CAAC;QAC9E,CAAC;IACL,CAAC;CACJ"}

package/dist/runtime/index.d.ts

@@ -0,0 +1,3 @@
+export { SSMRuntime } from './SSMRuntime.js';
+export type { SSMRuntimeOptions, GenerateOptions } from './SSMRuntime.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/runtime/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/runtime/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,YAAY,EAAE,iBAAiB,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/runtime/index.js

@@ -0,0 +1,2 @@
+export { SSMRuntime } from './SSMRuntime.js';
+//# sourceMappingURL=index.js.map

package/dist/runtime/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/runtime/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/session/errors.d.ts

@@ -0,0 +1,10 @@
+/**
+ * errors.ts – Typed error class for the MambaSession layer.
+ */
+export type SessionErrorCode = 'GPU_UNAVAILABLE' | 'TOKENIZER_LOAD_FAILED' | 'CHECKPOINT_FETCH_FAILED' | 'CHECKPOINT_INVALID' | 'INPUT_TOO_SHORT' | 'STORAGE_UNAVAILABLE' | 'SESSION_DESTROYED' | 'UNKNOWN';
+export declare class SessionError extends Error {
+    readonly code: SessionErrorCode;
+    readonly cause?: unknown | undefined;
+    constructor(code: SessionErrorCode, message: string, cause?: unknown | undefined);
+}
+//# sourceMappingURL=errors.d.ts.map

package/dist/session/errors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.d.ts","sourceRoot":"","sources":["../../src/session/errors.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,MAAM,MAAM,gBAAgB,GACtB,iBAAiB,GACjB,uBAAuB,GACvB,yBAAyB,GACzB,oBAAoB,GACpB,iBAAiB,GACjB,qBAAqB,GACrB,mBAAmB,GACnB,SAAS,CAAC;AAEhB,qBAAa,YAAa,SAAQ,KAAK;aAEf,IAAI,EAAE,gBAAgB;aAEtB,KAAK,CAAC,EAAE,OAAO;gBAFf,IAAI,EAAE,gBAAgB,EACtC,OAAO,EAAE,MAAM,EACC,KAAK,CAAC,EAAE,OAAO,YAAA;CAKtC"}

package/dist/session/errors.js

@@ -0,0 +1,14 @@
+/**
+ * errors.ts – Typed error class for the MambaSession layer.
+ */
+export class SessionError extends Error {
+    code;
+    cause;
+    constructor(code, message, cause) {
+        super(message);
+        this.code = code;
+        this.cause = cause;
+        this.name = 'SessionError';
+    }
+}
+//# sourceMappingURL=errors.js.map

package/dist/session/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../src/session/errors.ts"],"names":[],"mappings":"AAAA;;GAEG;AAYH,MAAM,OAAO,YAAa,SAAQ,KAAK;IAEf;IAEA;IAHpB,YACoB,IAAsB,EACtC,OAAe,EACC,KAAe;QAE/B,KAAK,CAAC,OAAO,CAAC,CAAC;QAJC,SAAI,GAAJ,IAAI,CAAkB;QAEtB,UAAK,GAAL,KAAK,CAAU;QAG/B,IAAI,CAAC,IAAI,GAAG,cAAc,CAAC;IAC/B,CAAC;CACJ"}

package/dist/session/index.d.ts

@@ -0,0 +1,11 @@
+/**
+ * session/index.ts – barrel export for the MambaSession layer.
+ */
+export { MambaSession } from './session.js';
+export { SessionError } from './errors.js';
+export { MODEL_PRESETS, resolveLayerSchedule, resolveModelConfig } from './presets.js';
+export type { SessionErrorCode } from './errors.js';
+export type { LayerSchedulePreset } from './presets.js';
+export type { MambaSessionOptions, CompleteOptions, AdaptOptions, AdaptResult, SaveOptions, LoadOptions, StorageTarget, CreateProgressEvent, CreateStage, CreateCallbacks, SessionInternals, GpuMode, } from './session.js';
+export type { Tokenizer } from './tokenizer.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/session/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/session/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAmB,cAAc,CAAC;AACzD,OAAO,EAAE,YAAY,EAAE,MAAmB,aAAa,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAEvF,YAAY,EAAE,gBAAgB,EAAE,MAAU,aAAa,CAAC;AACxD,YAAY,EAAE,mBAAmB,EAAE,MAAO,cAAc,CAAC;AACzD,YAAY,EACR,mBAAmB,EACnB,eAAe,EACf,YAAY,EACZ,WAAW,EACX,WAAW,EACX,WAAW,EACX,aAAa,EACb,mBAAmB,EACnB,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,OAAO,GACV,MAAM,cAAc,CAAC;AACtB,YAAY,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/session/index.js

@@ -0,0 +1,7 @@
+/**
+ * session/index.ts – barrel export for the MambaSession layer.
+ */
+export { MambaSession } from './session.js';
+export { SessionError } from './errors.js';
+export { MODEL_PRESETS, resolveLayerSchedule, resolveModelConfig } from './presets.js';
+//# sourceMappingURL=index.js.map

package/dist/session/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/session/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAmB,cAAc,CAAC;AACzD,OAAO,EAAE,YAAY,EAAE,MAAmB,aAAa,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC"}

package/dist/session/persistence.d.ts

@@ -0,0 +1,14 @@
+/**
+ * persistence.ts – Storage helpers for the MambaSession layer.
+ *
+ * Supports three storage targets:
+ *  - IndexedDB  (browser default)
+ *  - Download   (Blob URL trigger)
+ *  - File System Access API
+ */
+export declare function saveToIndexedDB(key: string, buffer: ArrayBuffer, idb?: IDBFactory): Promise<void>;
+export declare function loadFromIndexedDB(key: string, idb?: IDBFactory): Promise<ArrayBuffer | undefined>;
+export declare function triggerDownload(filename: string, buffer: ArrayBuffer): Promise<void>;
+export declare function saveViaFileSystemAPI(filename: string, buffer: ArrayBuffer): Promise<void>;
+export declare function loadViaFileSystemAPI(): Promise<ArrayBuffer>;
+//# sourceMappingURL=persistence.d.ts.map

package/dist/session/persistence.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"persistence.d.ts","sourceRoot":"","sources":["../../src/session/persistence.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAuCH,wBAAsB,eAAe,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAevG;AAED,wBAAsB,iBAAiB,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC,CAiBvG;AAOD,wBAAsB,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAe1F;AAID,wBAAsB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAiB/F;AAED,wBAAsB,oBAAoB,IAAI,OAAO,CAAC,WAAW,CAAC,CAgBjE"}

package/dist/session/persistence.js

@@ -0,0 +1,100 @@
+/**
+ * persistence.ts – Storage helpers for the MambaSession layer.
+ *
+ * Supports three storage targets:
+ *  - IndexedDB  (browser default)
+ *  - Download   (Blob URL trigger)
+ *  - File System Access API
+ */
+import { SessionError } from './errors.js';
+const DB_NAME = 'ssmjs-session';
+const DB_VERSION = 1;
+const STORE_NAME = 'checkpoints';
+// ── IndexedDB helpers ─────────────────────────────────────────────────────────
+function openDB(idb) {
+    return new Promise((resolve, reject) => {
+        const factory = idb ?? (typeof indexedDB !== 'undefined' ? indexedDB : undefined);
+        if (!factory) {
+            reject(new SessionError('STORAGE_UNAVAILABLE', 'IndexedDB is not available in this environment. Pass an idbFactory option (e.g. from fake-indexeddb) for Node.js support.'));
+            return;
+        }
+        const req = factory.open(DB_NAME, DB_VERSION);
+        req.onupgradeneeded = (e) => {
+            const db = e.target.result;
+            if (!db.objectStoreNames.contains(STORE_NAME)) {
+                db.createObjectStore(STORE_NAME);
+            }
+        };
+        req.onsuccess = () => resolve(req.result);
+        req.onerror = () => reject(new SessionError('STORAGE_UNAVAILABLE', `Failed to open IndexedDB database "${DB_NAME}": ${req.error?.message ?? 'unknown error'}`, req.error));
+    });
+}
+export async function saveToIndexedDB(key, buffer, idb) {
+    const db = await openDB(idb);
+    return new Promise((resolve, reject) => {
+        const tx = db.transaction(STORE_NAME, 'readwrite');
+        const store = tx.objectStore(STORE_NAME);
+        const req = store.put(buffer, key);
+        req.onsuccess = () => resolve();
+        req.onerror = () => reject(new SessionError('STORAGE_UNAVAILABLE', `Failed to write checkpoint to IndexedDB (key="${key}"): ${req.error?.message ?? 'unknown error'}`, req.error));
+        tx.oncomplete = () => db.close();
+    });
+}
+export async function loadFromIndexedDB(key, idb) {
+    const db = await openDB(idb);
+    return new Promise((resolve, reject) => {
+        const tx = db.transaction(STORE_NAME, 'readonly');
+        const store = tx.objectStore(STORE_NAME);
+        const req = store.get(key);
+        req.onsuccess = () => {
+            db.close();
+            resolve(req.result);
+        };
+        req.onerror = () => reject(new SessionError('STORAGE_UNAVAILABLE', `Failed to read checkpoint from IndexedDB (key="${key}"): ${req.error?.message ?? 'unknown error'}`, req.error));
+    });
+}
+/** Milliseconds a download Blob URL is kept alive before being revoked. */
+const DOWNLOAD_URL_TTL_MS = 10_000;
+// ── Download helper ───────────────────────────────────────────────────────────
+export async function triggerDownload(filename, buffer) {
+    const blob = new Blob([buffer], { type: 'application/octet-stream' });
+    const url = URL.createObjectURL(blob);
+    const anchor = document.createElement('a');
+    anchor.href = url;
+    anchor.download = filename;
+    anchor.style.display = 'none';
+    document.body.appendChild(anchor);
+    anchor.click();
+    document.body.removeChild(anchor);
+    // Release the object URL after the TTL to allow the download to start
+    setTimeout(() => URL.revokeObjectURL(url), DOWNLOAD_URL_TTL_MS);
+}
+// ── File System Access API helpers ────────────────────────────────────────────
+export async function saveViaFileSystemAPI(filename, buffer) {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const win = window;
+    if (typeof win.showSaveFilePicker !== 'function') {
+        throw new SessionError('STORAGE_UNAVAILABLE', 'File System Access API (showSaveFilePicker) is not available in this browser.');
+    }
+    const handle = await win.showSaveFilePicker({
+        suggestedName: filename,
+        types: [{ description: 'MambaSession Checkpoint', accept: { 'application/octet-stream': ['.bin'] } }],
+    });
+    const writable = await handle.createWritable();
+    await writable.write(buffer);
+    await writable.close();
+}
+export async function loadViaFileSystemAPI() {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const win = window;
+    if (typeof win.showOpenFilePicker !== 'function') {
+        throw new SessionError('STORAGE_UNAVAILABLE', 'File System Access API (showOpenFilePicker) is not available in this browser.');
+    }
+    const [handle] = await win.showOpenFilePicker({
+        types: [{ description: 'MambaSession Checkpoint', accept: { 'application/octet-stream': ['.bin'] } }],
+        multiple: false,
+    });
+    const file = await handle.getFile();
+    return file.arrayBuffer();
+}
+//# sourceMappingURL=persistence.js.map

package/dist/session/persistence.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"persistence.js","sourceRoot":"","sources":["../../src/session/persistence.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,MAAM,OAAO,GAAM,eAAe,CAAC;AACnC,MAAM,UAAU,GAAG,CAAC,CAAC;AACrB,MAAM,UAAU,GAAG,aAAa,CAAC;AAEjC,iFAAiF;AAEjF,SAAS,MAAM,CAAC,GAAgB;IAC5B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACnC,MAAM,OAAO,GAAG,GAAG,IAAI,CAAC,OAAO,SAAS,KAAK,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAClF,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,MAAM,CAAC,IAAI,YAAY,CACnB,qBAAqB,EACrB,2HAA2H,CAC9H,CAAC,CAAC;YACH,OAAO;QACX,CAAC;QAED,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAE9C,GAAG,CAAC,eAAe,GAAG,CAAC,CAAC,EAAE,EAAE;YACxB,MAAM,EAAE,GAAI,CAAC,CAAC,MAA2B,CAAC,MAAM,CAAC;YACjD,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC5C,EAAE,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YACrC,CAAC;QACL,CAAC,CAAC;QAEF,GAAG,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,GAAG,CAAC,OAAO,GAAK,GAAI,EAAE,CAAC,MAAM,CAAC,IAAI,YAAY,CAC1C,qBAAqB,EACrB,sCAAsC,OAAO,MAAM,GAAG,CAAC,KAAK,EAAE,OAAO,IAAI,eAAe,EAAE,EAC1F,GAAG,CAAC,KAAK,CACZ,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,GAAW,EAAE,MAAmB,EAAE,GAAgB;IACpF,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,CAAC;IAC7B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACnC,MAAM,EAAE,GAAM,EAAE,CAAC,WAAW,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACtD,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QACzC,MAAM,GAAG,GAAK,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAErC,GAAG,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;QAChC,GAAG,CAAC,OAAO,GAAK,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,YAAY,CACzC,qBAAqB,EACrB,iDAAiD,GAAG,OAAO,GAAG,CAAC,KAAK,EAAE,OAAO,IAAI,eAAe,EAAE,EAClG,GAAG,CAAC,KAAK,CACZ,CAAC,CAAC;QACH,EAAE,CAAC,UAAU,GAAG,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IACrC,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,GAAW,EAAE,GAAgB;IACjE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,CAAC;IAC7B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACnC,MAAM,EAAE,GAAM,EAAE,CAAC,WAAW,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QACrD,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QACzC,MAAM,GAAG,GAAK,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAE7B,GAAG,CAAC,SAAS,GAAG,GAAG,EAAE;YACjB,EAAE,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,CAAC,GAAG,CAAC,MAAiC,CAAC,CAAC;QACnD,CAAC,CAAC;QACF,GAAG,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,YAAY,CACvC,qBAAqB,EACrB,kDAAkD,GAAG,OAAO,GAAG,CAAC,KAAK,EAAE,OAAO,IAAI,eAAe,EAAE,EACnG,GAAG,CAAC,KAAK,CACZ,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC;AAED,2EAA2E;AAC3E,MAAM,mBAAmB,GAAG,MAAM,CAAC;AAEnC,iFAAiF;AAEjF,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,QAAgB,EAAE,MAAmB;IACvE,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,EAAE,0BAA0B,EAAE,CAAC,CAAC;IACtE,MAAM,GAAG,GAAI,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IAEvC,MAAM,MAAM,GAAS,QAAQ,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IACjD,MAAM,CAAC,IAAI,GAAU,GAAG,CAAC;IACzB,MAAM,CAAC,QAAQ,GAAM,QAAQ,CAAC;IAC9B,MAAM,CAAC,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC;IAE9B,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAClC,MAAM,CAAC,KAAK,EAAE,CAAC;IACf,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAElC,sEAAsE;IACtE,UAAU,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,mBAAmB,CAAC,CAAC;AACpE,CAAC;AAED,iFAAiF;AAEjF,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,QAAgB,EAAE,MAAmB;IAC5E,8DAA8D;IAC9D,MAAM,GAAG,GAAG,MAAa,CAAC;IAC1B,IAAI,OAAO,GAAG,CAAC,kBAAkB,KAAK,UAAU,EAAE,CAAC;QAC/C,MAAM,IAAI,YAAY,CAClB,qBAAqB,EACrB,+EAA+E,CAClF,CAAC;IACN,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC;QACxC,aAAa,EAAE,QAAQ;QACvB,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,yBAAyB,EAAE,MAAM,EAAE,EAAE,0BAA0B,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;KACxG,CAAC,CAAC;IACH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,cAAc,EAAE,CAAC;IAC/C,MAAM,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC7B,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC;AAC3B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB;IACtC,8DAA8D;IAC9D,MAAM,GAAG,GAAG,MAAa,CAAC;IAC1B,IAAI,OAAO,GAAG,CAAC,kBAAkB,KAAK,UAAU,EAAE,CAAC;QAC/C,MAAM,IAAI,YAAY,CAClB,qBAAqB,EACrB,+EAA+E,CAClF,CAAC;IACN,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC;QAC1C,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,yBAAyB,EAAE,MAAM,EAAE,EAAE,0BAA0B,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;QACrG,QAAQ,EAAE,KAAK;KAClB,CAAC,CAAC;IACH,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IACpC,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;AAC9B,CAAC"}

package/dist/session/presets.d.ts

@@ -0,0 +1,31 @@
+/**
+ * presets.ts – Model size presets and config resolver for the MambaSession layer.
+ */
+import type { HybridMambaModelConfig, LayerSpec, LayerType } from '@seanhogg/builderforce-memory-engine';
+import type { MambaSessionOptions } from './session.js';
+/**
+ * Pre-defined model size presets.
+ * nHeads is used by Mamba-2/3 and Attention layers; ignored for Mamba-1.
+ */
+export declare const MODEL_PRESETS: Record<string, Partial<HybridMambaModelConfig>>;
+export type LayerSchedulePreset = 'jamba' | 'zamba';
+/**
+ * Resolve a layer schedule from a preset name, explicit array, or undefined.
+ *
+ *   'jamba' — Jamba-style: every 4th layer (index 3, 7, 11…) is attention, rest mamba2
+ *   'zamba' — Zamba-style: every 6th layer (index 5, 11…) is attention, rest mamba3
+ */
+export declare function resolveLayerSchedule(schedule: LayerSpec[] | LayerSchedulePreset | undefined, numLayers: number, defaultType: LayerType): LayerSpec[];
+/**
+ * Resolves a fully-populated HybridMambaModelConfig from session options and
+ * the actual tokenizer vocab size.
+ *
+ * Resolution order:
+ *  1. Preset fields (default: 'nano')
+ *  2. modelConfig overrides (only applied when modelSize === 'custom')
+ *  3. vocabSize from the tokenizer
+ *  4. mambaVersion → default layer type for schedule resolution
+ *  5. layerSchedule → per-layer type array (preset string or explicit array)
+ */
+export declare function resolveModelConfig(options: MambaSessionOptions, vocabSize: number): HybridMambaModelConfig;
+//# sourceMappingURL=presets.d.ts.map

package/dist/session/presets.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"presets.d.ts","sourceRoot":"","sources":["../../src/session/presets.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,sBAAsB,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,sCAAsC,CAAC;AACzG,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AAIxD;;;GAGG;AACH,eAAO,MAAM,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,sBAAsB,CAAC,CAKzE,CAAC;AAMF,MAAM,MAAM,mBAAmB,GAAG,OAAO,GAAG,OAAO,CAAC;AAEpD;;;;;GAKG;AACH,wBAAgB,oBAAoB,CAChC,QAAQ,EAAK,SAAS,EAAE,GAAG,mBAAmB,GAAG,SAAS,EAC1D,SAAS,EAAI,MAAM,EACnB,WAAW,EAAE,SAAS,GACvB,SAAS,EAAE,CAkBb;AAID;;;;;;;;;;GAUG;AACH,wBAAgB,kBAAkB,CAC9B,OAAO,EAAI,mBAAmB,EAC9B,SAAS,EAAE,MAAM,GAClB,sBAAsB,CAiDxB"}