@inbrowser/model 0.1.0

package/dist/engine.js

@@ -0,0 +1,353 @@
+/**
+ * `createEngine` — on-device LLM engine implementation.
+ *
+ * Wires `@huggingface/transformers` v4 to the narrow `Engine` surface
+ * declared in `./types.ts`:
+ *
+ *   - `ensureReady()` loads the `AutoProcessor` + `AutoModelForCausalLM`
+ *     pair from the configured HF Hub repo, mapping the runtime's
+ *     `ProgressInfo` stream into `LoadProgress` events.
+ *   - `generate()` applies the model's bundled chat template, sets up
+ *     a `TextStreamer` whose `callback_function` pushes tokens into
+ *     an async-iterator queue, and drives `model.generate()`. Yields
+ *     `{ kind: 'token' }` per decoded chunk, then a terminal `usage`
+ *     event with the engine's local accounting.
+ *
+ * Backend mapping: `Backend` ('auto' | 'webgpu' | 'wasm') passes
+ * straight through to Transformers.js's `DeviceType`. `dtype` is
+ * forwarded unchanged.
+ *
+ * Tool calling: the engine itself is toolless. The agent-side
+ * polyfill lives in `@inbrowser/agent` (see AGENTS.md).
+ *
+ * Stop sequences: `GenerateOpts.stop` is accepted but not yet
+ * honored — needs a `StoppingCriteria` adapter. Tracked as a
+ * follow-up.
+ */
+import { AutoModelForCausalLM, AutoTokenizer, TextStreamer, env as transformersEnv, } from '@huggingface/transformers';
+import { parseToolCalls } from './parse-tool-calls.js';
+export function createEngine(opts) {
+    const model = opts.model;
+    const capabilities = opts.capabilities;
+    let state = 'idle';
+    let loadPromise = null;
+    let tokenizer = null;
+    let llm = null;
+    const stateSubs = new Set();
+    const loadSubs = new Set();
+    if (opts.onLoadProgress)
+        loadSubs.add(opts.onLoadProgress);
+    function setState(next) {
+        if (state === next)
+            return;
+        state = next;
+        for (const sub of stateSubs)
+            sub(next);
+    }
+    function emitLoad(p) {
+        for (const sub of loadSubs)
+            sub(p);
+    }
+    function progressCallback(info) {
+        // Transformers.js emits five statuses across the load pipeline.
+        // Only 'progress' carries byte counts; the others are markers.
+        // `LoadProgress.fetch` covers the entire download phase end-to-end.
+        if (info.status === 'progress') {
+            emitLoad({
+                phase: 'fetch',
+                file: info.file,
+                loadedBytes: info.loaded,
+                totalBytes: info.total,
+            });
+        }
+        else if (info.status === 'initiate' || info.status === 'download') {
+            emitLoad({
+                phase: 'fetch',
+                file: info.file,
+                loadedBytes: 0,
+                totalBytes: 0,
+            });
+        }
+        // 'done' and 'ready' are deliberately ignored — phase transitions
+        // are driven from `ensureReady` so the order is deterministic.
+    }
+    async function ensureReady() {
+        if (state === 'ready')
+            return;
+        if (state === 'disposed')
+            throw new Error('engine disposed');
+        if (loadPromise)
+            return loadPromise;
+        setState('loading');
+        // `weightsBaseUrl` overrides the HF Hub origin for self-hosted
+        // mirrors. Transformers.js exposes this as the global
+        // `env.remoteHost`; we set it process-wide before load. Documented
+        // limitation: with multiple engines spanning different remotes,
+        // the last one to load wins. Realistic use case (one app, one
+        // mirror) is unaffected.
+        if (opts.weightsBaseUrl) {
+            transformersEnv.remoteHost = opts.weightsBaseUrl;
+        }
+        loadPromise = (async () => {
+            // AutoTokenizer (not AutoProcessor): text-only models like
+            // SmolLM2 ship no preprocessor_config.json and AutoProcessor
+            // 404s on them. Multimodal models (e.g., Gemma 4 audio) still
+            // resolve via AutoTokenizer because their text tokenizer is
+            // the same file — we just don't expose the audio path yet.
+            tokenizer = await AutoTokenizer.from_pretrained(model.modelId, {
+                ...(model.revision ? { revision: model.revision } : {}),
+                progress_callback: progressCallback,
+            });
+            emitLoad({ phase: 'init', backend: opts.backend });
+            llm = await AutoModelForCausalLM.from_pretrained(model.modelId, {
+                dtype: opts.dtype,
+                device: toDeviceOption(opts.backend),
+                ...(model.revision ? { revision: model.revision } : {}),
+                progress_callback: progressCallback,
+            });
+            setState('ready');
+            emitLoad({ phase: 'ready' });
+        })().catch((e) => {
+            setState('error');
+            loadPromise = null;
+            throw e;
+        });
+        return loadPromise;
+    }
+    function on(event, handler) {
+        if (event === 'state') {
+            const h = handler;
+            stateSubs.add(h);
+            return () => stateSubs.delete(h);
+        }
+        const h = handler;
+        loadSubs.add(h);
+        return () => loadSubs.delete(h);
+    }
+    async function* generate(messages, genOpts = {}) {
+        // Tool calling: pass `tools` to the chat template only when the
+        // preset declares support AND the caller provided tools. Then
+        // wrap the raw stream with `parseToolCalls` so we surface
+        // synthetic `tool_call` events when the model emits an envelope.
+        // Both gates matter: a preset claiming `supportsTools: false`
+        // (e.g., the base chat template doesn't know about tools) must
+        // refuse, and a tools-supporting preset called without tools
+        // pays no parsing cost.
+        const useTools = capabilities.supportsTools && genOpts.tools !== undefined && genOpts.tools.length > 0;
+        const raw = generateRaw(messages, genOpts, useTools);
+        if (useTools) {
+            yield* parseToolCalls(raw);
+        }
+        else {
+            yield* raw;
+        }
+    }
+    async function* generateRaw(messages, genOpts, useTools) {
+        try {
+            await ensureReady();
+        }
+        catch (e) {
+            yield {
+                kind: 'error',
+                message: e instanceof Error ? e.message : String(e),
+                recoverable: false,
+            };
+            return;
+        }
+        if (!tokenizer || !llm) {
+            yield { kind: 'error', message: 'engine not ready', recoverable: false };
+            return;
+        }
+        // ── Build prompt ──────────────────────────────────────────────
+        const tools = useTools ? genOpts.tools : undefined;
+        // Thinking mode: opt-in via GenerateOpts.enableThinking, gated on
+        // capabilities.supportsThinking. The chat template renders its
+        // thinking preamble; for models like Gemma 4 that use special
+        // tokens for the thinking channel, we also need to STOP stripping
+        // them from the TextStreamer output (see below).
+        const useThinking = capabilities.supportsThinking && genOpts.enableThinking === true;
+        const renderedPrompt = applyChatTemplate(tokenizer, messages, opts.chatTemplate, tools, useThinking);
+        // Calling the tokenizer as a function returns a BatchEncoding
+        // (`{ input_ids, attention_mask, ... }`) of Tensors.
+        const inputs = (await tokenizer(renderedPrompt));
+        const promptTokens = countTokens(inputs.input_ids);
+        // ── Set up streaming queue ────────────────────────────────────
+        const queue = [];
+        let resolver = null;
+        let producerDone = false;
+        let producerError = null;
+        let outputTokens = 0;
+        function wakeIterator() {
+            const r = resolver;
+            resolver = null;
+            if (r)
+                r();
+        }
+        const pushEvent = (evt) => {
+            queue.push(evt);
+            wakeIterator();
+        };
+        // Only expose special tokens in the output stream when the
+        // preset declares thinkingTags AND thinking is enabled. The
+        // signal that says "I expect a downstream parser to consume
+        // structural tokens" is `thinkingTags !== undefined` — without
+        // it, the default `skip_special_tokens: true` keeps the output
+        // clean (no `<|channel>`, no `<turn|>`, no BOS/EOS leak).
+        //
+        // Models with reliable channel formats (DeepSeek's `<think>` is
+        // literal text, not a special token, so it survives the default)
+        // need no special handling here. Models with structural-token
+        // channels but inconsistent emission (Gemma 4 family — see
+        // presets.ts) deliberately omit `thinkingTags` to take this path.
+        const preserveSpecialTokens = useThinking && capabilities.thinkingTags !== undefined;
+        const streamer = new TextStreamer(tokenizer, {
+            skip_prompt: true,
+            ...(preserveSpecialTokens ? { skip_special_tokens: false } : {}),
+            callback_function: (text) => {
+                if (text.length === 0)
+                    return;
+                pushEvent({ kind: 'token', text });
+            },
+            token_callback_function: (tokenIds) => {
+                outputTokens += tokenIds.length;
+            },
+        });
+        const startedAt = performance.now();
+        const generateArgs = {
+            ...inputs,
+            max_new_tokens: genOpts.maxNewTokens ?? 512,
+            streamer,
+        };
+        if (typeof genOpts.temperature === 'number') {
+            generateArgs.do_sample = true;
+            generateArgs.temperature = genOpts.temperature;
+        }
+        if (typeof genOpts.topP === 'number')
+            generateArgs.top_p = genOpts.topP;
+        if (typeof genOpts.topK === 'number')
+            generateArgs.top_k = genOpts.topK;
+        // Drive `model.generate` concurrently with the iterator drain.
+        const generatePromise = (async () => {
+            try {
+                await llm.generate(generateArgs);
+            }
+            catch (e) {
+                producerError = e;
+            }
+            finally {
+                producerDone = true;
+                wakeIterator();
+            }
+        })();
+        try {
+            while (!producerDone || queue.length > 0) {
+                if (queue.length === 0) {
+                    if (genOpts.signal?.aborted)
+                        break;
+                    await new Promise((r) => {
+                        resolver = r;
+                    });
+                }
+                const next = queue.shift();
+                if (next)
+                    yield next;
+            }
+        }
+        finally {
+            await generatePromise;
+        }
+        if (producerError) {
+            yield {
+                kind: 'error',
+                message: producerError instanceof Error ? producerError.message : String(producerError),
+                recoverable: false,
+            };
+            return;
+        }
+        yield {
+            kind: 'usage',
+            promptTokens,
+            outputTokens,
+            decodeMs: Math.round(performance.now() - startedAt),
+        };
+    }
+    async function dispose() {
+        setState('disposed');
+        stateSubs.clear();
+        loadSubs.clear();
+        // Transformers.js doesn't expose a public dispose; dropping
+        // references lets GC reclaim the wasm/webgpu sessions.
+        tokenizer = null;
+        llm = null;
+    }
+    return {
+        get model() {
+            return model;
+        },
+        get state() {
+            return state;
+        },
+        get capabilities() {
+            return capabilities;
+        },
+        ensureReady,
+        on,
+        generate,
+        dispose,
+    };
+}
+/**
+ * Type-safe preset authoring. Identity at runtime; the value of this
+ * helper is purely the compile-time completeness check it enforces
+ * on caller-defined presets.
+ */
+export function definePreset(p) {
+    return p;
+}
+// ── Helpers ──────────────────────────────────────────────────────────────────
+function toDeviceOption(backend) {
+    // Backend names map 1:1 to Transformers.js DeviceType strings.
+    return backend;
+}
+function applyChatTemplate(tokenizer, messages, override, tools, enableThinking) {
+    if (override)
+        return override(messages);
+    // EngineMessage's media field is dropped here — text-only path
+    // for POC. A multimodal preset will need a different code path.
+    const conversation = messages.map((m) => ({
+        role: m.role,
+        content: m.text,
+    }));
+    // When `tools` is provided the model's chat template formats them
+    // into its native tool-advertisement section (Qwen 3 wraps them in
+    // `<tools>`; DeepSeek R1 uses a different block). Transformers.js
+    // forwards the `tools` array to the underlying Jinja template
+    // unchanged — the OAI function-call shape is the canonical input.
+    //
+    // `enable_thinking` is a Gemma 3/4 / Qwen 3 convention — passing it
+    // true makes the model's template render its thinking-mode preamble
+    // (Gemma 4: "Inject Thinking token at the very top of the FIRST
+    // system turn"). Templates that don't know about it ignore the
+    // unknown key — Jinja is permissive about undeclared vars.
+    const rendered = tokenizer.apply_chat_template(conversation, {
+        add_generation_prompt: true,
+        tokenize: false,
+        ...(tools && tools.length > 0 ? { tools: tools } : {}),
+        ...(enableThinking ? { enable_thinking: true } : {}),
+    });
+    if (typeof rendered !== 'string') {
+        throw new Error('apply_chat_template returned non-string with tokenize:false');
+    }
+    return rendered;
+}
+function countTokens(inputIds) {
+    // input_ids is a Transformers.js Tensor with dims [batch, seq_len].
+    if (!inputIds || typeof inputIds !== 'object')
+        return 0;
+    const dims = inputIds.dims;
+    if (!Array.isArray(dims) || dims.length < 2)
+        return 0;
+    const seqLen = dims[dims.length - 1];
+    return typeof seqLen === 'number' ? seqLen : 0;
+}
+//# sourceMappingURL=engine.js.map

package/dist/engine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"engine.js","sourceRoot":"","sources":["../src/engine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AAEH,OAAO,EACL,oBAAoB,EACpB,aAAa,EAIb,YAAY,EACZ,GAAG,IAAI,eAAe,GACvB,MAAM,2BAA2B,CAAC;AAEnC,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAgBvD,MAAM,UAAU,YAAY,CAAC,IAAsB;IACjD,MAAM,KAAK,GAAa,IAAI,CAAC,KAAK,CAAC;IACnC,MAAM,YAAY,GAAuB,IAAI,CAAC,YAAY,CAAC;IAE3D,IAAI,KAAK,GAAgB,MAAM,CAAC;IAChC,IAAI,WAAW,GAAyB,IAAI,CAAC;IAC7C,IAAI,SAAS,GAA+B,IAAI,CAAC;IACjD,IAAI,GAAG,GAA2B,IAAI,CAAC;IAEvC,MAAM,SAAS,GAAG,IAAI,GAAG,EAA4B,CAAC;IACtD,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA6B,CAAC;IACtD,IAAI,IAAI,CAAC,cAAc;QAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAE3D,SAAS,QAAQ,CAAC,IAAiB;QACjC,IAAI,KAAK,KAAK,IAAI;YAAE,OAAO;QAC3B,KAAK,GAAG,IAAI,CAAC;QACb,KAAK,MAAM,GAAG,IAAI,SAAS;YAAE,GAAG,CAAC,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,SAAS,QAAQ,CAAC,CAAe;QAC/B,KAAK,MAAM,GAAG,IAAI,QAAQ;YAAE,GAAG,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC;IAED,SAAS,gBAAgB,CAAC,IAAkB;QAC1C,gEAAgE;QAChE,+DAA+D;QAC/D,oEAAoE;QACpE,IAAI,IAAI,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YAC/B,QAAQ,CAAC;gBACP,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,MAAM;gBACxB,UAAU,EAAE,IAAI,CAAC,KAAK;aACvB,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,KAAK,UAAU,IAAI,IAAI,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YACpE,QAAQ,CAAC;gBACP,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,CAAC;gBACd,UAAU,EAAE,CAAC;aACd,CAAC,CAAC;QACL,CAAC;QACD,kEAAkE;QAClE,+DAA+D;IACjE,CAAC;IAED,KAAK,UAAU,WAAW;QACxB,IAAI,KAAK,KAAK,OAAO;YAAE,OAAO;QAC9B,IAAI,KAAK,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;QAC7D,IAAI,WAAW;YAAE,OAAO,WAAW,CAAC;QAEpC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACpB,+DAA+D;QAC/D,sDAAsD;QACtD,mEAAmE;QACnE,gEAAgE;QAChE,8DAA8D;QAC9D,yBAAyB;QACzB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,eAAe,CAAC,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC;QACnD,CAAC;QACD,WAAW,GAAG,CAAC,KAAK,IAAI,EAAE;YACxB,2DAA2D;YAC3D,6DAA6D;YAC7D,8DAA8D;YAC9D,4DAA4D;YAC5D,2DAA2D;YAC3D,SAAS,GAAG,MAAM,aAAa,CAAC,eAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC7D,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvD,iBAAiB,EAAE,gBAAgB;aACpC,CAAC,CAAC;YAEH,QAAQ,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YAEnD,GAAG,GAAG,MAAM,oBAAoB,CAAC,eAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC9D,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC;gBACpC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvD,iBAAiB,EAAE,gBAAgB;aACpC,CAAC,CAAC;YAEH,QAAQ,CAAC,OAAO,CAAC,CAAC;YAClB,QAAQ,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/B,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;YACf,QAAQ,CAAC,OAAO,CAAC,CAAC;YAClB,WAAW,GAAG,IAAI,CAAC;YACnB,MAAM,CAAC,CAAC;QACV,CAAC,CAAC,CAAC;QACH,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,SAAS,EAAE,CACT,KAAQ,EACR,OAA2C;QAE3C,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;YACtB,MAAM,CAAC,GAAG,OAAmC,CAAC;YAC9C,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACjB,OAAO,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC;QACD,MAAM,CAAC,GAAG,OAAoC,CAAC;QAC/C,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAChB,OAAO,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,SAAS,CAAC,CAAC,QAAQ,CACtB,QAAsC,EACtC,UAAwB,EAAE;QAE1B,gEAAgE;QAChE,8DAA8D;QAC9D,0DAA0D;QAC1D,iEAAiE;QACjE,8DAA8D;QAC9D,+DAA+D;QAC/D,6DAA6D;QAC7D,wBAAwB;QACxB,MAAM,QAAQ,GACZ,YAAY,CAAC,aAAa,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;QACxF,MAAM,GAAG,GAAG,WAAW,CAAC,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;QACrD,IAAI,QAAQ,EAAE,CAAC;YACb,KAAK,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,CAAC,GAAG,CAAC;QACb,CAAC;IACH,CAAC;IAED,KAAK,SAAS,CAAC,CAAC,WAAW,CACzB,QAAsC,EACtC,OAAqB,EACrB,QAAiB;QAEjB,IAAI,CAAC;YACH,MAAM,WAAW,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM;gBACJ,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;gBACnD,WAAW,EAAE,KAAK;aACnB,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,CAAC,SAAS,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;YACzE,OAAO;QACT,CAAC;QAED,iEAAiE;QACjE,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAE,OAAO,CAAC,KAAiC,CAAC,CAAC,CAAC,SAAS,CAAC;QAChF,kEAAkE;QAClE,+DAA+D;QAC/D,8DAA8D;QAC9D,kEAAkE;QAClE,iDAAiD;QACjD,MAAM,WAAW,GAAG,YAAY,CAAC,gBAAgB,IAAI,OAAO,CAAC,cAAc,KAAK,IAAI,CAAC;QACrF,MAAM,cAAc,GAAG,iBAAiB,CACtC,SAAS,EACT,QAAQ,EACR,IAAI,CAAC,YAAY,EACjB,KAAK,EACL,WAAW,CACZ,CAAC;QACF,8DAA8D;QAC9D,qDAAqD;QACrD,MAAM,MAAM,GAAG,CAAC,MAAM,SAAS,CAAC,cAAc,CAAC,CAA4B,CAAC;QAC5E,MAAM,YAAY,GAAG,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAEnD,iEAAiE;QACjE,MAAM,KAAK,GAAkB,EAAE,CAAC;QAChC,IAAI,QAAQ,GAAwB,IAAI,CAAC;QACzC,IAAI,YAAY,GAAG,KAAK,CAAC;QACzB,IAAI,aAAa,GAAY,IAAI,CAAC;QAClC,IAAI,YAAY,GAAG,CAAC,CAAC;QAErB,SAAS,YAAY;YACnB,MAAM,CAAC,GAAG,QAAQ,CAAC;YACnB,QAAQ,GAAG,IAAI,CAAC;YAChB,IAAI,CAAC;gBAAE,CAAC,EAAE,CAAC;QACb,CAAC;QAED,MAAM,SAAS,GAAG,CAAC,GAAgB,EAAQ,EAAE;YAC3C,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAChB,YAAY,EAAE,CAAC;QACjB,CAAC,CAAC;QAEF,2DAA2D;QAC3D,4DAA4D;QAC5D,4DAA4D;QAC5D,+DAA+D;QAC/D,+DAA+D;QAC/D,0DAA0D;QAC1D,EAAE;QACF,gEAAgE;QAChE,iEAAiE;QACjE,8DAA8D;QAC9D,2DAA2D;QAC3D,kEAAkE;QAClE,MAAM,qBAAqB,GAAG,WAAW,IAAI,YAAY,CAAC,YAAY,KAAK,SAAS,CAAC;QACrF,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,SAAS,EAAE;YAC3C,WAAW,EAAE,IAAI;YACjB,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,EAAE,mBAAmB,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAChE,iBAAiB,EAAE,CAAC,IAAY,EAAE,EAAE;gBAClC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO;gBAC9B,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,CAAC;YACD,uBAAuB,EAAE,CAAC,QAAkB,EAAE,EAAE;gBAC9C,YAAY,IAAI,QAAQ,CAAC,MAAM,CAAC;YAClC,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,YAAY,GAA4B;YAC5C,GAAG,MAAM;YACT,cAAc,EAAE,OAAO,CAAC,YAAY,IAAI,GAAG;YAC3C,QAAQ;SACT,CAAC;QACF,IAAI,OAAO,OAAO,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;YAC5C,YAAY,CAAC,SAAS,GAAG,IAAI,CAAC;YAC9B,YAAY,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACjD,CAAC;QACD,IAAI,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ;YAAE,YAAY,CAAC,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC;QACxE,IAAI,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ;YAAE,YAAY,CAAC,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC;QAExE,+DAA+D;QAC/D,MAAM,eAAe,GAAG,CAAC,KAAK,IAAI,EAAE;YAClC,IAAI,CAAC;gBACH,MAAM,GAAI,CAAC,QAAQ,CAAC,YAAkD,CAAC,CAAC;YAC1E,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,aAAa,GAAG,CAAC,CAAC;YACpB,CAAC;oBAAS,CAAC;gBACT,YAAY,GAAG,IAAI,CAAC;gBACpB,YAAY,EAAE,CAAC;YACjB,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,IAAI,CAAC;YACH,OAAO,CAAC,YAAY,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACvB,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO;wBAAE,MAAM;oBACnC,MAAM,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE;wBAC5B,QAAQ,GAAG,CAAC,CAAC;oBACf,CAAC,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC3B,IAAI,IAAI;oBAAE,MAAM,IAAI,CAAC;YACvB,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,eAAe,CAAC;QACxB,CAAC;QAED,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM;gBACJ,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,aAAa,YAAY,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;gBACvF,WAAW,EAAE,KAAK;aACnB,CAAC;YACF,OAAO;QACT,CAAC;QAED,MAAM;YACJ,IAAI,EAAE,OAAO;YACb,YAAY;YACZ,YAAY;YACZ,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;SACpD,CAAC;IACJ,CAAC;IAED,KAAK,UAAU,OAAO;QACpB,QAAQ,CAAC,UAAU,CAAC,CAAC;QACrB,SAAS,CAAC,KAAK,EAAE,CAAC;QAClB,QAAQ,CAAC,KAAK,EAAE,CAAC;QACjB,4DAA4D;QAC5D,uDAAuD;QACvD,SAAS,GAAG,IAAI,CAAC;QACjB,GAAG,GAAG,IAAI,CAAC;IACb,CAAC;IAED,OAAO;QACL,IAAI,KAAK;YACP,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,KAAK;YACP,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,YAAY;YACd,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,WAAW;QACX,EAAE;QACF,QAAQ;QACR,OAAO;KACR,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,YAAY,CAA6C,CAAI;IAC3E,OAAO,CAAC,CAAC;AACX,CAAC;AAED,gFAAgF;AAEhF,SAAS,cAAc,CAAC,OAAgB;IACtC,+DAA+D;IAC/D,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,iBAAiB,CACxB,SAA8B,EAC9B,QAAsC,EACtC,QAAsD,EACtD,KAA+B,EAC/B,cAAwB;IAExB,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACxC,+DAA+D;IAC/D,gEAAgE;IAChE,MAAM,YAAY,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACxC,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,OAAO,EAAE,CAAC,CAAC,IAAI;KAChB,CAAC,CAAC,CAAC;IACJ,kEAAkE;IAClE,mEAAmE;IACnE,kEAAkE;IAClE,8DAA8D;IAC9D,kEAAkE;IAClE,EAAE;IACF,oEAAoE;IACpE,oEAAoE;IACpE,gEAAgE;IAChE,+DAA+D;IAC/D,2DAA2D;IAC3D,MAAM,QAAQ,GAAG,SAAS,CAAC,mBAAmB,CAAC,YAAY,EAAE;QAC3D,qBAAqB,EAAE,IAAI;QAC3B,QAAQ,EAAE,KAAK;QACf,GAAG,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnE,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAGrD,CAAC,CAAC;IACH,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,MAAM,IAAI,KAAK,CAAC,6DAA6D,CAAC,CAAC;IACjF,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,WAAW,CAAC,QAAiB;IACpC,oEAAoE;IACpE,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ;QAAE,OAAO,CAAC,CAAC;IACxD,MAAM,IAAI,GAAI,QAA+B,CAAC,IAAI,CAAC;IACnD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,CAAC,CAAC;IACtD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACrC,OAAO,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AACjD,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,21 @@
+/**
+ * `@inbrowser/model` — on-device LLM engine.
+ *
+ * Root export carries the engine factory + types + `definePreset`.
+ * Subpaths:
+ *   - `@inbrowser/model/presets` — bundled Gemma 4 presets.
+ *   - `@inbrowser/model/relay`   — adapter to `@inbrowser/relay`.
+ *   - `@inbrowser/model/agent`   — adapter to `@inbrowser/agent`.
+ *   - `@inbrowser/model/worker`  — host/connect helpers.
+ *
+ * Spread a preset into `createEngine` to get a running engine:
+ *
+ *   import { createEngine } from '@inbrowser/model';
+ *   import { gemma4_E2B } from '@inbrowser/model/presets';
+ *   const engine = createEngine(gemma4_E2B);
+ */
+export { createEngine, definePreset } from './engine.js';
+export { parseToolCalls, type ToolCallParseOpts } from './parse-tool-calls.js';
+export { splitThinking, type ThinkingSplitOpts } from './think.js';
+export type { Backend, CreateEngineOpts, Dtype, Engine, EngineCapabilities, EngineEvent, EngineEventMap, EngineHooks, EngineMessage, EngineState, GenerateOpts, LoadProgress, MediaPart, ModelPreset, ModelRef, ToolSpec, } from './types.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAEH,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,KAAK,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC/E,OAAO,EAAE,aAAa,EAAE,KAAK,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAEnE,YAAY,EACV,OAAO,EACP,gBAAgB,EAChB,KAAK,EACL,MAAM,EACN,kBAAkB,EAClB,WAAW,EACX,cAAc,EACd,WAAW,EACX,aAAa,EACb,WAAW,EACX,YAAY,EACZ,YAAY,EACZ,SAAS,EACT,WAAW,EACX,QAAQ,EACR,QAAQ,GACT,MAAM,YAAY,CAAC"}

package/dist/index.js

@@ -0,0 +1,20 @@
+/**
+ * `@inbrowser/model` — on-device LLM engine.
+ *
+ * Root export carries the engine factory + types + `definePreset`.
+ * Subpaths:
+ *   - `@inbrowser/model/presets` — bundled Gemma 4 presets.
+ *   - `@inbrowser/model/relay`   — adapter to `@inbrowser/relay`.
+ *   - `@inbrowser/model/agent`   — adapter to `@inbrowser/agent`.
+ *   - `@inbrowser/model/worker`  — host/connect helpers.
+ *
+ * Spread a preset into `createEngine` to get a running engine:
+ *
+ *   import { createEngine } from '@inbrowser/model';
+ *   import { gemma4_E2B } from '@inbrowser/model/presets';
+ *   const engine = createEngine(gemma4_E2B);
+ */
+export { createEngine, definePreset } from './engine.js';
+export { parseToolCalls } from './parse-tool-calls.js';
+export { splitThinking } from './think.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAEH,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EAAE,cAAc,EAA0B,MAAM,uBAAuB,CAAC;AAC/E,OAAO,EAAE,aAAa,EAA0B,MAAM,YAAY,CAAC"}

package/dist/parse-tool-calls.d.ts

@@ -0,0 +1,49 @@
+/**
+ * `parseToolCalls` — stream transformer that detects native
+ * tool-call envelopes the model emits and re-emits them as
+ * `kind: 'tool_call'` engine events.
+ *
+ * Different model families use different envelope formats:
+ *
+ *   Qwen 2/3, DeepSeek R1, Hermes-Pro:
+ *     <tool_call>
+ *     {"name": "func", "arguments": {...}}
+ *     </tool_call>
+ *
+ *   Llama 3.2+ (not yet supported — uses special tokens):
+ *     <|python_tag|>{"name": "func", "parameters": {...}}<|eom_id|>
+ *
+ *   Mistral v0.3+ (not yet supported):
+ *     [TOOL_CALLS]{"name": "func", "arguments": {...}}[/TOOL_CALLS]
+ *
+ * Default format is `'qwen'` (handles Qwen 2/3, DeepSeek R1, Hermes).
+ * Add new formats by extending the format union.
+ *
+ * The implementation is a buffer-aware state machine mirroring
+ * `splitThinking` — partial tags split across token boundaries
+ * resolve correctly. Inside-tag content is parsed as JSON to extract
+ * `name` and `arguments`; on parse failure the args field carries
+ * `{ _raw: string }` so the caller can surface or salvage.
+ *
+ * Pass-through behavior: `thinking`, `usage`, and `error` events
+ * forward unchanged. `token` events outside an envelope forward as
+ * `token`; inside an envelope they're buffered and converted to a
+ * single `tool_call` event on close.
+ */
+import type { EngineEvent } from './types.js';
+export interface ToolCallParseOpts {
+    /**
+     * Envelope format. Default `'qwen'` — `<tool_call>...</tool_call>`
+     * with a JSON body containing `name` and `arguments`. The body's
+     * JSON is parsed on close; malformed JSON falls through as
+     * `{ args: { _raw: string } }` so consumers can salvage or surface.
+     */
+    format?: 'qwen';
+    /**
+     * Override id generator. Default uses a short random suffix.
+     * Useful for tests that need deterministic ids.
+     */
+    generateId?: () => string;
+}
+export declare function parseToolCalls(source: AsyncIterable<EngineEvent>, opts?: ToolCallParseOpts): AsyncIterable<EngineEvent>;
+//# sourceMappingURL=parse-tool-calls.d.ts.map

package/dist/parse-tool-calls.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse-tool-calls.d.ts","sourceRoot":"","sources":["../src/parse-tool-calls.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAE9C,MAAM,WAAW,iBAAiB;IAChC;;;;;OAKG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,MAAM,CAAC;CAC3B;AAgCD,wBAAuB,cAAc,CACnC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,EAClC,IAAI,GAAE,iBAAsB,GAC3B,aAAa,CAAC,WAAW,CAAC,CA6D5B"}

package/dist/parse-tool-calls.js

@@ -0,0 +1,115 @@
+/**
+ * `parseToolCalls` — stream transformer that detects native
+ * tool-call envelopes the model emits and re-emits them as
+ * `kind: 'tool_call'` engine events.
+ *
+ * Different model families use different envelope formats:
+ *
+ *   Qwen 2/3, DeepSeek R1, Hermes-Pro:
+ *     <tool_call>
+ *     {"name": "func", "arguments": {...}}
+ *     </tool_call>
+ *
+ *   Llama 3.2+ (not yet supported — uses special tokens):
+ *     <|python_tag|>{"name": "func", "parameters": {...}}<|eom_id|>
+ *
+ *   Mistral v0.3+ (not yet supported):
+ *     [TOOL_CALLS]{"name": "func", "arguments": {...}}[/TOOL_CALLS]
+ *
+ * Default format is `'qwen'` (handles Qwen 2/3, DeepSeek R1, Hermes).
+ * Add new formats by extending the format union.
+ *
+ * The implementation is a buffer-aware state machine mirroring
+ * `splitThinking` — partial tags split across token boundaries
+ * resolve correctly. Inside-tag content is parsed as JSON to extract
+ * `name` and `arguments`; on parse failure the args field carries
+ * `{ _raw: string }` so the caller can surface or salvage.
+ *
+ * Pass-through behavior: `thinking`, `usage`, and `error` events
+ * forward unchanged. `token` events outside an envelope forward as
+ * `token`; inside an envelope they're buffered and converted to a
+ * single `tool_call` event on close.
+ */
+const DEFAULT_OPEN_TAG = '<tool_call>';
+const DEFAULT_CLOSE_TAG = '</tool_call>';
+function defaultId() {
+    return `tc_${Math.random().toString(36).slice(2, 10)}`;
+}
+function parseQwenBody(body) {
+    const trimmed = body.trim();
+    try {
+        const parsed = JSON.parse(trimmed);
+        const name = typeof parsed.name === 'string' ? parsed.name : '';
+        // Some templates use `arguments`, some use `parameters`. Accept
+        // both as a small charitable coercion — the model occasionally
+        // mis-keys.
+        const args = parsed.arguments !== undefined
+            ? parsed.arguments
+            : (parsed.parameters ?? {});
+        return { name, args };
+    }
+    catch {
+        return { name: '', args: { _raw: trimmed } };
+    }
+}
+export async function* parseToolCalls(source, opts = {}) {
+    const open = DEFAULT_OPEN_TAG;
+    const close = DEFAULT_CLOSE_TAG;
+    const genId = opts.generateId ?? defaultId;
+    let mode = 'normal';
+    let buffer = '';
+    for await (const evt of source) {
+        if (evt.kind !== 'token') {
+            yield evt;
+            continue;
+        }
+        buffer += evt.text;
+        while (buffer.length > 0) {
+            if (mode === 'normal') {
+                const idx = buffer.indexOf(open);
+                if (idx === -1) {
+                    // Hold back the tail in case it's a partial open-tag prefix.
+                    const safeLen = buffer.length - (open.length - 1);
+                    if (safeLen > 0) {
+                        yield { kind: 'token', text: buffer.slice(0, safeLen) };
+                        buffer = buffer.slice(safeLen);
+                    }
+                    break;
+                }
+                if (idx > 0) {
+                    yield { kind: 'token', text: buffer.slice(0, idx) };
+                }
+                buffer = buffer.slice(idx + open.length);
+                mode = 'inside';
+            }
+            else {
+                const idx = buffer.indexOf(close);
+                if (idx === -1) {
+                    // Don't dribble out partial tool-call body — wait for the
+                    // close tag so we emit one tool_call event per envelope.
+                    // Trade-off: we hold the entire body in `buffer` until close.
+                    // Tool-call bodies are small (~hundreds of bytes), fine.
+                    break;
+                }
+                const body = buffer.slice(0, idx);
+                const { name, args } = parseQwenBody(body);
+                yield { kind: 'tool_call', id: genId(), name, args };
+                buffer = buffer.slice(idx + close.length);
+                mode = 'normal';
+            }
+        }
+    }
+    // Flush. If we ended mid-envelope (model hit max_new_tokens before
+    // closing), parse whatever body we have and emit it — same
+    // best-effort posture as splitThinking.
+    if (buffer.length > 0) {
+        if (mode === 'inside') {
+            const { name, args } = parseQwenBody(buffer);
+            yield { kind: 'tool_call', id: genId(), name, args };
+        }
+        else {
+            yield { kind: 'token', text: buffer };
+        }
+    }
+}
+//# sourceMappingURL=parse-tool-calls.js.map

package/dist/parse-tool-calls.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse-tool-calls.js","sourceRoot":"","sources":["../src/parse-tool-calls.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AAmBH,MAAM,gBAAgB,GAAG,aAAa,CAAC;AACvC,MAAM,iBAAiB,GAAG,cAAc,CAAC;AAEzC,SAAS,SAAS;IAChB,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;AACzD,CAAC;AAOD,SAAS,aAAa,CAAC,IAAY;IACjC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAA4C,CAAC;QAC9E,MAAM,IAAI,GAAG,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;QAChE,gEAAgE;QAChE,+DAA+D;QAC/D,YAAY;QACZ,MAAM,IAAI,GACR,MAAM,CAAC,SAAS,KAAK,SAAS;YAC5B,CAAC,CAAC,MAAM,CAAC,SAAS;YAClB,CAAC,CAAC,CAAE,MAAmC,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;QAC9D,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IACxB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC;IAC/C,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,cAAc,CACnC,MAAkC,EAClC,OAA0B,EAAE;IAE5B,MAAM,IAAI,GAAG,gBAAgB,CAAC;IAC9B,MAAM,KAAK,GAAG,iBAAiB,CAAC;IAChC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,IAAI,SAAS,CAAC;IAE3C,IAAI,IAAI,GAAwB,QAAQ,CAAC;IACzC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC/B,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YACzB,MAAM,GAAG,CAAC;YACV,SAAS;QACX,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC;QAEnB,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBACjC,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;oBACf,6DAA6D;oBAC7D,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBAClD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;wBAChB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,CAAC;wBACxD,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACjC,CAAC;oBACD,MAAM;gBACR,CAAC;gBACD,IAAI,GAAG,GAAG,CAAC,EAAE,CAAC;oBACZ,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;gBACtD,CAAC;gBACD,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;gBACzC,IAAI,GAAG,QAAQ,CAAC;YAClB,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAClC,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;oBACf,0DAA0D;oBAC1D,yDAAyD;oBACzD,8DAA8D;oBAC9D,yDAAyD;oBACzD,MAAM;gBACR,CAAC;gBACD,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gBAClC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;gBAC3C,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBACrD,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC1C,IAAI,GAAG,QAAQ,CAAC;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED,mEAAmE;IACnE,2DAA2D;IAC3D,wCAAwC;IACxC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;YACtB,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACxC,CAAC;IACH,CAAC;AACH,CAAC"}