@mlx-node/lm 0.0.5 → 0.0.7

package/dist/index.d.ts

@@ -12,9 +12,17 @@
  * const result = await model.generate([{ role: 'user', content: 'Hello!' }]);
  * ```
  */
-export { Qwen3Model, Qwen3Tokenizer } from '@mlx-node/core';
+export { Qwen3Tokenizer } from '@mlx-node/core';
+export { Qwen3Model } from './stream.js';
+export { Gemma4Model, Gemma4Model as Gemma4_Model } from './stream.js';
+export type { Gemma4Config } from '@mlx-node/core';
+export { HarrierModel } from '@mlx-node/core';
+export type { HarrierConfig } from '@mlx-node/core';
 export { Qwen35Model, Qwen35Model as Qwen3_5Model } from './stream.js';
 export type { Qwen35Config, Qwen35GenerationConfig, Qwen35GenerationResult } from '@mlx-node/core';
+export { Lfm2Model } from './stream.js';
+export type { Lfm2Config } from '@mlx-node/core';
+export { LFM2_CONFIGS, getLfm2Config } from './models/lfm2-configs.js';
 export { Qwen35MoeModel, Qwen35MoeModel as Qwen3_5MoeModel } from './stream.js';
 export type { Qwen35MoeConfig, Qwen35MoeGenerationConfig, Qwen35MoeGenerationResult } from '@mlx-node/core';
 export type { DType } from '@mlx-node/core';
@@ -22,10 +30,12 @@ export type { SamplingConfig, BatchGenerationResult } from '@mlx-node/core';
 export type { ChatConfig, ChatResult, ChatMessage, ToolCallResult, PerformanceMetrics } from '@mlx-node/core';
 export type { ChatStreamDelta, ChatStreamFinal, ChatStreamEvent } from './stream.js';
 export type { ChatStreamChunk, ChatStreamHandle } from '@mlx-node/core';
-export { _createChatStream } from './stream.js';
+export { _runChatStream } from './stream.js';
+export { ChatSession } from './chat-session.js';
+export type { ChatSessionOptions, SendOptions, SessionCapableModel } from './chat-session.js';
 export { type Qwen3Config, QWEN3_CONFIGS, type GenerationResult, type GenerationConfig, getQwen3Config, } from './models/qwen3-configs.js';
-export { loadModel, detectModelType, type ModelType } from './models/model-loader.js';
-export type { TrainableModel } from './interfaces.js';
+export { loadModel, loadSession, detectModelType, type ModelType } from './models/model-loader.js';
+export type { TrainableModel, LoadableModel, EmbeddingModel } from './interfaces.js';
 export { QWEN35_CONFIGS, getQwen35Config } from './models/qwen3_5-configs.js';
 export * from './tools/index.js';
 export { enableProfiling, disableProfiling } from './profiling.js';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAGH,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,WAAW,EAAE,WAAW,IAAI,YAAY,EAAE,MAAM,aAAa,CAAC;AACvE,YAAY,EAAE,YAAY,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AAGnG,OAAO,EAAE,cAAc,EAAE,cAAc,IAAI,eAAe,EAAE,MAAM,aAAa,CAAC;AAChF,YAAY,EAAE,eAAe,EAAE,yBAAyB,EAAE,yBAAyB,EAAE,MAAM,gBAAgB,CAAC;AAK5G,YAAY,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAC5C,YAAY,EAAE,cAAc,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AAG5E,YAAY,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAG9G,YAAY,EAAE,eAAe,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACrF,YAAY,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAGxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAGhD,OAAO,EACL,KAAK,WAAW,EAChB,aAAa,EACb,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EACrB,cAAc,GACf,MAAM,2BAA2B,CAAC;AAGnC,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,KAAK,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAGtF,YAAY,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEtD,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AAG9E,cAAc,kBAAkB,CAAC;AAGjC,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAGH,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAE,WAAW,EAAE,WAAW,IAAI,YAAY,EAAE,MAAM,aAAa,CAAC;AACvE,YAAY,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAGnD,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,YAAY,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,WAAW,IAAI,YAAY,EAAE,MAAM,aAAa,CAAC;AACvE,YAAY,EAAE,YAAY,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AAGnG,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,YAAY,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAGvE,OAAO,EAAE,cAAc,EAAE,cAAc,IAAI,eAAe,EAAE,MAAM,aAAa,CAAC;AAChF,YAAY,EAAE,eAAe,EAAE,yBAAyB,EAAE,yBAAyB,EAAE,MAAM,gBAAgB,CAAC;AAK5G,YAAY,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAC5C,YAAY,EAAE,cAAc,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AAG5E,YAAY,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAG9G,YAAY,EAAE,eAAe,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACrF,YAAY,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAMxE,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAM7C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,YAAY,EAAE,kBAAkB,EAAE,WAAW,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAG9F,OAAO,EACL,KAAK,WAAW,EAChB,aAAa,EACb,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EACrB,cAAc,GACf,MAAM,2BAA2B,CAAC;AAGnC,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,eAAe,EAAE,KAAK,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAGnG,YAAY,EAAE,cAAc,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAErF,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AAG9E,cAAc,kBAAkB,CAAC;AAGjC,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/index.js

@@ -13,17 +13,34 @@
  * ```
  */
 // Model classes (for inference)
-export { Qwen3Model, Qwen3Tokenizer } from '@mlx-node/core';
+export { Qwen3Tokenizer } from '@mlx-node/core';
+export { Qwen3Model } from './stream.js';
+// Gemma4 models
+export { Gemma4Model, Gemma4Model as Gemma4_Model } from './stream.js';
+// Embedding models
+export { HarrierModel } from '@mlx-node/core';
 export { Qwen35Model, Qwen35Model as Qwen3_5Model } from './stream.js';
+// LFM2 models
+export { Lfm2Model } from './stream.js';
+export { LFM2_CONFIGS, getLfm2Config } from './models/lfm2-configs.js';
 // MoE variant
 export { Qwen35MoeModel, Qwen35MoeModel as Qwen3_5MoeModel } from './stream.js';
 // Internal: exported for testing the callback-to-AsyncGenerator bridge
-// Not part of the public API — may change without notice
-export { _createChatStream } from './stream.js';
+// Not part of the public API — may change without notice.
+// `_runChatStream` is the generic adapter used by every model wrapper
+// (and the VLM package's QianfanOCR wrapper) to turn a callback-based
+// native stream into an `AsyncGenerator<ChatStreamEvent>`.
+export { _runChatStream } from './stream.js';
+// Cross-model chat session wrapper (see chat-session.ts for design notes).
+// `SessionCapableModel` is the structural interface matched by every
+// generative model wrapper and used as the upper-bound for
+// `ChatSession<M>`; exported so the VLM wrapper can pin a compile-time
+// conformance assertion.
+export { ChatSession } from './chat-session.js';
 // Model utilities (TypeScript-only)
 export { QWEN3_CONFIGS, getQwen3Config, } from './models/qwen3-configs.js';
 // Model loading
-export { loadModel, detectModelType } from './models/model-loader.js';
+export { loadModel, loadSession, detectModelType } from './models/model-loader.js';
 export { QWEN35_CONFIGS, getQwen35Config } from './models/qwen3_5-configs.js';
 // Tool calling utilities
 export * from './tools/index.js';

package/dist/interfaces.d.ts

@@ -1,8 +1,18 @@
-import type { Qwen3Model, Qwen35Model, Qwen35MoeModel } from '@mlx-node/core';
+import type { Gemma4Model, HarrierModel, Lfm2Model, Qwen3Model, Qwen35Model, Qwen35MoeModel, QianfanOCRModel } from '@mlx-node/core';
 /**
  * Union of all model classes that can be used with training engines.
  * Uses the native (core) types so trainers can pass instances directly
  * to Rust engine factory methods without type conflicts.
  */
 export type TrainableModel = Qwen3Model | Qwen35Model | Qwen35MoeModel;
+/**
+ * Union of all embedding model classes.
+ * These models encode text into fixed-size embedding vectors.
+ */
+export type EmbeddingModel = HarrierModel;
+/**
+ * Union of all model classes that loadModel can return.
+ * Includes trainable models, inference-only models, and embedding models.
+ */
+export type LoadableModel = TrainableModel | QianfanOCRModel | EmbeddingModel | Gemma4Model | Lfm2Model;
 //# sourceMappingURL=interfaces.d.ts.map

package/dist/interfaces.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"interfaces.d.ts","sourceRoot":"","sources":["../src/interfaces.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAE9E;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,UAAU,GAAG,WAAW,GAAG,cAAc,CAAC"}
+{"version":3,"file":"interfaces.d.ts","sourceRoot":"","sources":["../src/interfaces.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EACX,YAAY,EACZ,SAAS,EACT,UAAU,EACV,WAAW,EACX,cAAc,EACd,eAAe,EAChB,MAAM,gBAAgB,CAAC;AAExB;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,UAAU,GAAG,WAAW,GAAG,cAAc,CAAC;AAEvE;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG,YAAY,CAAC;AAE1C;;;GAGG;AACH,MAAM,MAAM,aAAa,GAAG,cAAc,GAAG,eAAe,GAAG,cAAc,GAAG,WAAW,GAAG,SAAS,CAAC"}

package/dist/models/lfm2-configs.d.ts

@@ -0,0 +1,6 @@
+import type { Lfm2Config } from '@mlx-node/core';
+export declare const LFM2_CONFIGS: {
+    [key: string]: Lfm2Config;
+};
+export declare function getLfm2Config(name: string): Lfm2Config;
+//# sourceMappingURL=lfm2-configs.d.ts.map

package/dist/models/lfm2-configs.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"lfm2-configs.d.ts","sourceRoot":"","sources":["../../src/models/lfm2-configs.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAEjD,eAAO,MAAM,YAAY,EAAE;IAAE,CAAC,GAAG,EAAE,MAAM,GAAG,UAAU,CAAA;CAwCrD,CAAC;AAEF,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,UAAU,CAMtD"}

package/dist/models/lfm2-configs.js

@@ -0,0 +1,48 @@
+export const LFM2_CONFIGS = {
+    'lfm2.5-1.2b-thinking': {
+        vocabSize: 65536,
+        hiddenSize: 2048,
+        numHiddenLayers: 16,
+        numAttentionHeads: 32,
+        numKeyValueHeads: 8,
+        maxPositionEmbeddings: 128000,
+        normEps: 1e-5,
+        convBias: false,
+        convLCache: 3,
+        blockDim: 2048,
+        blockFfDim: 12288,
+        blockMultipleOf: 256,
+        blockFfnDimMultiplier: 1.0,
+        blockAutoAdjustFfDim: true,
+        ropeTheta: 1000000.0,
+        layerTypes: [
+            'conv',
+            'conv',
+            'full_attention',
+            'conv',
+            'conv',
+            'full_attention',
+            'conv',
+            'conv',
+            'full_attention',
+            'conv',
+            'full_attention',
+            'conv',
+            'full_attention',
+            'conv',
+            'full_attention',
+            'conv',
+        ],
+        tieEmbedding: true,
+        eosTokenId: 7,
+        bosTokenId: 1,
+        padTokenId: 0,
+    },
+};
+export function getLfm2Config(name) {
+    const config = LFM2_CONFIGS[name];
+    if (!config) {
+        throw new Error(`Unknown LFM2 config: ${name}. Available: ${Object.keys(LFM2_CONFIGS).join(', ')}`);
+    }
+    return config;
+}

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

@@ -3,11 +3,32 @@
  *
  * Handles loading pretrained weights from MLX format or converting from HuggingFace.
  */
-import type { TrainableModel } from '../interfaces.js';
-export type ModelType = 'qwen3' | 'qwen3_5' | 'qwen3_5_moe';
+import { ChatSession, type SessionCapableModel } from '../chat-session.js';
+import type { LoadableModel } from '../interfaces.js';
+export type ModelType = 'qwen3' | 'qwen3_5' | 'qwen3_5_moe' | 'internvl_chat' | 'qianfan-ocr' | 'harrier' | 'gemma4' | 'lfm2';
 /**
- * Load a language model from disk, auto-detecting architecture from config.json.
+ * Load a model from disk, auto-detecting architecture from config.json.
+ *
+ * Supports both language models (Qwen3, Qwen3.5) and vision-language models
+ * (Qianfan-OCR / InternVL). Use `instanceof` to narrow the returned type.
+ */
+export declare function loadModel(modelPath: string): Promise<LoadableModel>;
+/**
+ * Load a model and wrap it in a {@link ChatSession} for multi-turn chat.
+ *
+ * Convenience around `loadModel()` + `new ChatSession(model)` for the
+ * common case where a caller just wants an ergonomic session handle.
+ *
+ * Rejects models that cannot be driven by a `ChatSession`:
+ *   - Embedding models (`HarrierModel`) have no chat surface.
+ *   - The native `QianfanOCRModel` exposes callback-based streaming
+ *     methods that do not structurally satisfy `SessionCapableModel`'s
+ *     `AsyncGenerator` overloads. The VLM AsyncGenerator wrapper lives
+ *     in `@mlx-node/vlm` (importing it here would create a circular
+ *     package dependency), so callers who want a Qianfan-OCR session
+ *     must import `QianfanOCRModel` from `@mlx-node/vlm` and construct
+ *     `new ChatSession(model)` directly.
  */
-export declare function loadModel(modelPath: string): Promise<TrainableModel>;
+export declare function loadSession(modelPath: string): Promise<ChatSession<SessionCapableModel>>;
 export declare function detectModelType(modelPath: string): Promise<ModelType>;
 //# sourceMappingURL=model-loader.d.ts.map

package/dist/models/model-loader.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"model-loader.d.ts","sourceRoot":"","sources":["../../src/models/model-loader.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAGvD,MAAM,MAAM,SAAS,GAAG,OAAO,GAAG,SAAS,GAAG,aAAa,CAAC;AAI5D;;GAEG;AACH,wBAAsB,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC,CAa1E;AAED,wBAAsB,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAa3E"}
+{"version":3,"file":"model-loader.d.ts","sourceRoot":"","sources":["../../src/models/model-loader.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,OAAO,EAAE,WAAW,EAAE,KAAK,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAC3E,OAAO,KAAK,EAAE,aAAa,EAAkB,MAAM,kBAAkB,CAAC;AAGtE,MAAM,MAAM,SAAS,GACjB,OAAO,GACP,SAAS,GACT,aAAa,GACb,eAAe,GACf,aAAa,GACb,SAAS,GACT,QAAQ,GACR,MAAM,CAAC;AAaX;;;;;GAKG;AACH,wBAAsB,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAoBzE;AAED;;;;;;;;;;;;;;;GAeG;AACH,wBAAsB,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC,CAW9F;AAED,wBAAsB,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAyB3E"}

package/dist/models/model-loader.js

@@ -5,30 +5,85 @@
  */
 import { readFile } from 'node:fs/promises';
 import { join } from 'node:path';
-import { Qwen3Model } from '@mlx-node/core';
-import { Qwen35Model, Qwen35MoeModel } from '../stream.js';
-const SUPPORTED_MODEL_TYPES = new Set(['qwen3', 'qwen3_5', 'qwen3_5_moe']);
+import { HarrierModel, QianfanOCRModel } from '@mlx-node/core';
+import { ChatSession } from '../chat-session.js';
+import { Gemma4Model, Lfm2Model, Qwen3Model, Qwen35Model, Qwen35MoeModel } from '../stream.js';
+const SUPPORTED_MODEL_TYPES = new Set([
+    'qwen3',
+    'qwen3_5',
+    'qwen3_5_moe',
+    'internvl_chat',
+    'qianfan-ocr',
+    'harrier',
+    'gemma4',
+    'lfm2',
+]);
 /**
- * Load a language model from disk, auto-detecting architecture from config.json.
+ * Load a model from disk, auto-detecting architecture from config.json.
+ *
+ * Supports both language models (Qwen3, Qwen3.5) and vision-language models
+ * (Qianfan-OCR / InternVL). Use `instanceof` to narrow the returned type.
  */
 export async function loadModel(modelPath) {
     const modelType = await detectModelType(modelPath);
     switch (modelType) {
         case 'qwen3_5_moe':
-            // Cast: stream wrapper extends native — safe for instanceof and engine factories
             return Qwen35MoeModel.load(modelPath);
         case 'qwen3_5':
-            // load() auto-detects vision weights and loads encoder if present
             return Qwen35Model.load(modelPath);
         case 'qwen3':
             return Qwen3Model.load(modelPath);
+        case 'harrier':
+            return HarrierModel.load(modelPath);
+        case 'internvl_chat':
+        case 'qianfan-ocr':
+            return QianfanOCRModel.load(modelPath);
+        case 'gemma4':
+            return Gemma4Model.load(modelPath);
+        case 'lfm2':
+            return Lfm2Model.load(modelPath);
     }
 }
+/**
+ * Load a model and wrap it in a {@link ChatSession} for multi-turn chat.
+ *
+ * Convenience around `loadModel()` + `new ChatSession(model)` for the
+ * common case where a caller just wants an ergonomic session handle.
+ *
+ * Rejects models that cannot be driven by a `ChatSession`:
+ *   - Embedding models (`HarrierModel`) have no chat surface.
+ *   - The native `QianfanOCRModel` exposes callback-based streaming
+ *     methods that do not structurally satisfy `SessionCapableModel`'s
+ *     `AsyncGenerator` overloads. The VLM AsyncGenerator wrapper lives
+ *     in `@mlx-node/vlm` (importing it here would create a circular
+ *     package dependency), so callers who want a Qianfan-OCR session
+ *     must import `QianfanOCRModel` from `@mlx-node/vlm` and construct
+ *     `new ChatSession(model)` directly.
+ */
+export async function loadSession(modelPath) {
+    const m = await loadModel(modelPath);
+    if (m instanceof HarrierModel) {
+        throw new Error('loadSession: embedding models (Harrier) cannot be wrapped in a ChatSession');
+    }
+    if (m instanceof QianfanOCRModel) {
+        throw new Error('loadSession: Qianfan-OCR / InternVL session support lives in @mlx-node/vlm. Import QianfanOCRModel from @mlx-node/vlm and construct ChatSession(model) directly.');
+    }
+    return new ChatSession(m);
+}
 export async function detectModelType(modelPath) {
     try {
         const raw = await readFile(join(modelPath, 'config.json'), 'utf-8');
         const config = JSON.parse(raw);
-        const modelType = config.model_type ?? 'qwen3';
+        const rawModelType = config.model_type ?? 'qwen3';
+        // Normalize model_type: gemma4_text → gemma4
+        let modelType = (rawModelType === 'gemma4_text' ? 'gemma4' : rawModelType);
+        // Detect embedding models: Qwen3 backbone with base architecture (no ForCausalLM)
+        if (modelType === 'qwen3') {
+            const architectures = config.architectures ?? [];
+            if (architectures.includes('Qwen3Model') && !architectures.includes('Qwen3ForCausalLM')) {
+                modelType = 'harrier';
+            }
+        }
         if (!SUPPORTED_MODEL_TYPES.has(modelType)) {
             throw new Error(`Unsupported model_type "${modelType}" in ${modelPath}/config.json`);
         }

package/dist/stream.d.ts

@@ -1,8 +1,9 @@
-import { Qwen35Model as Qwen35ModelNative, Qwen35MoeModel as Qwen35MoeModelNative } from '@mlx-node/core';
+import { Gemma4Model as Gemma4ModelNative, Lfm2Model as Lfm2ModelNative, Qwen3Model as Qwen3ModelNative, Qwen35Model as Qwen35ModelNative, Qwen35MoeModel as Qwen35MoeModelNative } from '@mlx-node/core';
 import type { ChatConfig, ChatMessage, ChatStreamChunk, ChatStreamHandle, PerformanceMetrics, ToolCallResult } from '@mlx-node/core';
 export interface ChatStreamDelta {
     text: string;
     done: false;
+    isReasoning?: boolean;
 }
 export interface ChatStreamFinal {
     text: string;
@@ -11,46 +12,173 @@ export interface ChatStreamFinal {
     toolCalls: ToolCallResult[];
     thinking: string | null;
     numTokens: number;
+    promptTokens: number;
+    reasoningTokens: number;
     rawText: string;
     performance?: PerformanceMetrics;
 }
 export type ChatStreamEvent = ChatStreamDelta | ChatStreamFinal;
 /**
- * Shared AsyncGenerator implementation that wraps a native callback-based
- * chatStream into a `for await...of`-compatible stream.
+ * Shared AsyncGenerator adapter for callback-based native streaming methods.
  *
- * Cancellation is automatic via the generator's `finally` block.
+ * Takes a `startCall` closure that, given the JS-side callback, dispatches
+ * the underlying native stream (whatever method signature that is — the
+ * closure captures `messages` / `config` / `userMessage` etc) and resolves
+ * with a `ChatStreamHandle`. The generator pumps the resulting chunk queue,
+ * transforms each chunk into a `ChatStreamEvent`, and calls `handle.cancel()`
+ * in a `finally` block so early termination (user `break`, exception) still
+ * cleans up native state.
+ *
+ * ## Signal-driven fast-abort
+ *
+ * When an optional `AbortSignal` is supplied and fires, the adapter:
+ *   1. Calls `handle.cancel()` immediately so the native decode stops
+ *      on the next safepoint rather than running to completion.
+ *   2. Wakes the pending `waitForItem()` await by pushing a synthetic
+ *      "aborted" marker into the queue and calling `notify()`. Without
+ *      this wake-up the generator would stay parked on the `await`
+ *      until the next native chunk arrived, which on a fast-abort
+ *      path (client disconnect before first token) never happens.
+ *   3. The generator sees the marker, breaks out of its loop, and the
+ *      finally block runs `cancelOnce()` — which is a no-op because
+ *      `triggerAbort` already flipped the `cancelled` flag. Some
+ *      backends throw on double-cancel, so routing every cancel site
+ *      through `cancelOnce` keeps abort behavior deterministic.
+ *
+ * The finally block is also the landing site for the consumer calling
+ * `.return()` on the outer generator — the existing `yield` cleanup
+ * covers that case. Signal-driven abort covers the window where the
+ * consumer cannot reach `.return()` because they are blocked waiting
+ * for the very `yield` that `waitForItem()` is gating.
+ *
+ * @internal Exported so the VLM wrapper (`@mlx-node/vlm`) can reuse the
+ * exact same bridge without duplicating the plumbing. Not part of the
+ * public API — may change without notice.
  */
-/** @internal Exported for testing only. */
-export declare function _createChatStream(nativeMethod: (messages: ChatMessage[], config: any, callback: (err: Error | null, chunk: ChatStreamChunk) => void) => Promise<ChatStreamHandle>, self: unknown, messages: ChatMessage[], config: unknown): AsyncGenerator<ChatStreamEvent>;
+export declare function _runChatStream(startCall: (callback: (err: Error | null, chunk: ChatStreamChunk) => void) => Promise<ChatStreamHandle>, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
 /**
- * Qwen3.5 dense model with AsyncGenerator-based `chatStream()`.
- *
- * @example
- * ```typescript
- * const model = await Qwen35Model.load('./models/qwen3.5-3b');
- * for await (const event of model.chatStream(messages)) {
- *   if (!event.done) process.stdout.write(event.text);
- * }
- * ```
+ * Qwen3.5 dense model with AsyncGenerator-based session streaming.
+ *
+ * Streaming is driven through the session API — `chatStreamSessionStart`,
+ * `chatStreamSessionContinue`, and `chatStreamSessionContinueTool` below —
+ * which adapt the callback-based native methods to
+ * `AsyncGenerator<ChatStreamEvent>` so the wrapper structurally satisfies
+ * `SessionCapableModel` and can be passed to `ChatSession<Qwen35Model>`.
  */
 export declare class Qwen35Model extends Qwen35ModelNative {
     static load(modelPath: string): Promise<Qwen35Model>;
-    chatStream(messages: ChatMessage[], config?: ChatConfig | null): AsyncGenerator<ChatStreamEvent>;
+    /**
+     * Streaming variant of {@link Qwen35Model#chatSessionStart}.
+     *
+     * Resets the KV caches, runs the jinja chat template, prefills on
+     * top of the fresh caches, and streams the decoded reply token-by-
+     * token. Stops on `<|im_end|>` so the cached history ends on a
+     * clean ChatML boundary that subsequent `chatStreamSessionContinue`
+     * deltas can append to. Text-only.
+     *
+     * The optional `signal` parameter wires an AbortSignal into the
+     * `_runChatStream` adapter's fast-abort path. Callers that need
+     * client-disconnect-aware cancellation (e.g. HTTP endpoints) pass
+     * one here and the native decode winds down at the next safepoint.
+     */
+    chatStreamSessionStart(messages: ChatMessage[], config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+    /**
+     * Streaming variant of {@link Qwen35Model#chatSessionContinue}.
+     *
+     * Builds a raw ChatML delta on top of the live session caches,
+     * tokenizes it, prefills the delta, and streams the decoded reply.
+     * Requires a live session started via `chatSessionStart` or
+     * `chatStreamSessionStart`. Stops on `<|im_end|>`.
+     *
+     * `images` is the native opt-in guard parameter — callers that
+     * attach a new image set must restart the session via
+     * `chatStreamSessionStart` with the full history. The high-level
+     * `ChatSession` wrapper handles that routing; callers that drive
+     * the wrapper directly should pass `null` for text-only continues.
+     */
+    chatStreamSessionContinue(userMessage: string, images: Uint8Array[] | null, config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+    /**
+     * Streaming variant of {@link Qwen35Model#chatSessionContinueTool}.
+     *
+     * Builds a ChatML `<tool_response>` delta on top of the live
+     * session caches and streams the decoded assistant reply. Requires
+     * a live session started via `chatSessionStart` /
+     * `chatStreamSessionStart`.
+     */
+    chatStreamSessionContinueTool(toolCallId: string, content: string, config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
 }
 /**
- * Qwen3.5 MoE model with AsyncGenerator-based `chatStream()`.
- *
- * @example
- * ```typescript
- * const model = await Qwen35MoeModel.load('./models/qwen3.5-moe');
- * for await (const event of model.chatStream(messages)) {
- *   if (!event.done) process.stdout.write(event.text);
- * }
- * ```
+ * Qwen3.5 MoE model wrapper.
+ *
+ * Streaming is driven through the `ChatSession` API — overrides below
+ * adapt the callback-based native methods to
+ * `AsyncGenerator<ChatStreamEvent>` so the wrapper structurally
+ * satisfies `SessionCapableModel`.
  */
 export declare class Qwen35MoeModel extends Qwen35MoeModelNative {
     static load(modelPath: string): Promise<Qwen35MoeModel>;
-    chatStream(messages: ChatMessage[], config?: ChatConfig | null): AsyncGenerator<ChatStreamEvent>;
+    /** Streaming variant of {@link Qwen35MoeModel#chatSessionStart}. */
+    chatStreamSessionStart(messages: ChatMessage[], config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+    /** Streaming variant of {@link Qwen35MoeModel#chatSessionContinue}. */
+    chatStreamSessionContinue(userMessage: string, images: Uint8Array[] | null, config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+    /** Streaming variant of {@link Qwen35MoeModel#chatSessionContinueTool}. */
+    chatStreamSessionContinueTool(toolCallId: string, content: string, config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+}
+/**
+ * LFM2 model wrapper.
+ *
+ * Streaming is driven through the `ChatSession` API — overrides below
+ * adapt the callback-based native methods to
+ * `AsyncGenerator<ChatStreamEvent>` so the wrapper structurally
+ * satisfies `SessionCapableModel`. LFM2 is text-only; the native
+ * `images` guard rejects non-empty image sets with an
+ * `IMAGE_CHANGE_REQUIRES_SESSION_RESTART:` prefix.
+ */
+export declare class Lfm2Model extends Lfm2ModelNative {
+    static load(modelPath: string): Promise<Lfm2Model>;
+    /** Streaming variant of {@link Lfm2Model#chatSessionStart}. */
+    chatStreamSessionStart(messages: ChatMessage[], config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+    /** Streaming variant of {@link Lfm2Model#chatSessionContinue}. */
+    chatStreamSessionContinue(userMessage: string, images: Uint8Array[] | null, config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+    /** Streaming variant of {@link Lfm2Model#chatSessionContinueTool}. */
+    chatStreamSessionContinueTool(toolCallId: string, content: string, config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+}
+/**
+ * Gemma4 model wrapper.
+ *
+ * Streaming is driven through the `ChatSession` API — overrides below
+ * adapt the callback-based native methods to
+ * `AsyncGenerator<ChatStreamEvent>` so the wrapper structurally
+ * satisfies `SessionCapableModel`. Gemma4 is text-only in the
+ * current refactor scope; the native `images` guard rejects non-empty
+ * image sets with an `IMAGE_CHANGE_REQUIRES_SESSION_RESTART:` prefix.
+ */
+export declare class Gemma4Model extends Gemma4ModelNative {
+    static load(modelPath: string): Promise<Gemma4Model>;
+    /** Streaming variant of {@link Gemma4Model#chatSessionStart}. */
+    chatStreamSessionStart(messages: ChatMessage[], config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+    /** Streaming variant of {@link Gemma4Model#chatSessionContinue}. */
+    chatStreamSessionContinue(userMessage: string, images: Uint8Array[] | null, config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+    /** Streaming variant of {@link Gemma4Model#chatSessionContinueTool}. */
+    chatStreamSessionContinueTool(toolCallId: string, content: string, config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+}
+/**
+ * Qwen3 (legacy) model wrapper.
+ *
+ * Streaming is driven through the `ChatSession` API — overrides below
+ * adapt the callback-based native methods to
+ * `AsyncGenerator<ChatStreamEvent>` so the wrapper structurally
+ * satisfies `SessionCapableModel`. Qwen3 legacy is text-only; the
+ * native `images` guard rejects non-empty image sets with an
+ * `IMAGE_CHANGE_REQUIRES_SESSION_RESTART:` prefix.
+ */
+export declare class Qwen3Model extends Qwen3ModelNative {
+    static load(modelPath: string): Promise<Qwen3Model>;
+    /** Streaming variant of {@link Qwen3Model#chatSessionStart}. */
+    chatStreamSessionStart(messages: ChatMessage[], config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+    /** Streaming variant of {@link Qwen3Model#chatSessionContinue}. */
+    chatStreamSessionContinue(userMessage: string, images: Uint8Array[] | null, config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
+    /** Streaming variant of {@link Qwen3Model#chatSessionContinueTool}. */
+    chatStreamSessionContinueTool(toolCallId: string, content: string, config?: ChatConfig | null, signal?: AbortSignal): AsyncGenerator<ChatStreamEvent>;
 }
 //# sourceMappingURL=stream.d.ts.map

package/dist/stream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../src/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,IAAI,iBAAiB,EAAE,cAAc,IAAI,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAC1G,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,kBAAkB,EAClB,cAAc,EACf,MAAM,gBAAgB,CAAC;AAExB,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,KAAK,CAAC;CACb;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,IAAI,CAAC;IACX,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,cAAc,EAAE,CAAC;IAC5B,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,kBAAkB,CAAC;CAClC;AAED,MAAM,MAAM,eAAe,GAAG,eAAe,GAAG,eAAe,CAAC;AAQhE;;;;;GAKG;AACH,2CAA2C;AAC3C,wBAAuB,iBAAiB,CAEtC,YAAY,EAAE,CACZ,QAAQ,EAAE,WAAW,EAAE,EACvB,MAAM,EAAE,GAAG,EACX,QAAQ,EAAE,CAAC,GAAG,EAAE,KAAK,GAAG,IAAI,EAAE,KAAK,EAAE,eAAe,KAAK,IAAI,KAC1D,OAAO,CAAC,gBAAgB,CAAC,EAC9B,IAAI,EAAE,OAAO,EACb,QAAQ,EAAE,WAAW,EAAE,EACvB,MAAM,EAAE,OAAO,GACd,cAAc,CAAC,eAAe,CAAC,CAoDjC;AAED;;;;;;;;;;GAUG;AACH,qBAAa,WAAY,SAAQ,iBAAiB;WAC1B,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAO5D,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,GAAG,cAAc,CAAC,eAAe,CAAC;CAGxG;AAED;;;;;;;;;;GAUG;AACH,qBAAa,cAAe,SAAQ,oBAAoB;WAChC,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC;IAO/D,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,GAAG,cAAc,CAAC,eAAe,CAAC;CAGxG"}
+{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../src/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,WAAW,IAAI,iBAAiB,EAChC,SAAS,IAAI,eAAe,EAC5B,UAAU,IAAI,gBAAgB,EAC9B,WAAW,IAAI,iBAAiB,EAChC,cAAc,IAAI,oBAAoB,EACvC,MAAM,gBAAgB,CAAC;AACxB,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,kBAAkB,EAClB,cAAc,EACf,MAAM,gBAAgB,CAAC;AAIxB,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,KAAK,CAAC;IACZ,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,IAAI,CAAC;IACX,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,cAAc,EAAE,CAAC;IAC5B,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,kBAAkB,CAAC;CAClC;AAED,MAAM,MAAM,eAAe,GAAG,eAAe,GAAG,eAAe,CAAC;AAoDhE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAoCG;AACH,wBAAuB,cAAc,CACnC,SAAS,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,KAAK,GAAG,IAAI,EAAE,KAAK,EAAE,eAAe,KAAK,IAAI,KAAK,OAAO,CAAC,gBAAgB,CAAC,EACvG,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC,CA2HjC;AAED;;;;;;;;GAQG;AACH,qBAAa,WAAY,SAAQ,iBAAiB;WAC1B,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAMnE;;;;;;;;;;;;;OAaG;IAEI,sBAAsB,CAC3B,QAAQ,EAAE,WAAW,EAAE,EACvB,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;IAOlC;;;;;;;;;;;;;OAaG;IAEI,yBAAyB,CAC9B,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,EAC3B,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;IAOlC;;;;;;;OAOG;IAEI,6BAA6B,CAClC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;CAMnC;AAED;;;;;;;GAOG;AACH,qBAAa,cAAe,SAAQ,oBAAoB;WAChC,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC;IAMtE,oEAAoE;IAE7D,sBAAsB,CAC3B,QAAQ,EAAE,WAAW,EAAE,EACvB,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;IAOlC,uEAAuE;IAEhE,yBAAyB,CAC9B,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,EAC3B,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;IAOlC,2EAA2E;IAEpE,6BAA6B,CAClC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;CAMnC;AAED;;;;;;;;;GASG;AACH,qBAAa,SAAU,SAAQ,eAAe;WACtB,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAMjE,+DAA+D;IAExD,sBAAsB,CAC3B,QAAQ,EAAE,WAAW,EAAE,EACvB,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;IAOlC,kEAAkE;IAE3D,yBAAyB,CAC9B,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,EAC3B,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;IAOlC,sEAAsE;IAE/D,6BAA6B,CAClC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;CAMnC;AAED;;;;;;;;;GASG;AACH,qBAAa,WAAY,SAAQ,iBAAiB;WAC1B,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAMnE,iEAAiE;IAE1D,sBAAsB,CAC3B,QAAQ,EAAE,WAAW,EAAE,EACvB,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;IAOlC,oEAAoE;IAE7D,yBAAyB,CAC9B,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,EAC3B,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;IAOlC,wEAAwE;IAEjE,6BAA6B,CAClC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;CAOnC;AAED;;;;;;;;;GASG;AACH,qBAAa,UAAW,SAAQ,gBAAgB;WACxB,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;IAMlE,gEAAgE;IAEzD,sBAAsB,CAC3B,QAAQ,EAAE,WAAW,EAAE,EACvB,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;IAOlC,mEAAmE;IAE5D,yBAAyB,CAC9B,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,EAC3B,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;IAOlC,uEAAuE;IAEhE,6BAA6B,CAClC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,UAAU,GAAG,IAAI,EAC1B,MAAM,CAAC,EAAE,WAAW,GACnB,cAAc,CAAC,eAAe,CAAC;CAMnC"}