@trymirai/uzu 0.1.21 → 0.1.22

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Mirai Tech Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -18,185 +18,152 @@ Node package for [uzu](https://github.com/trymirai/uzu), a **high-performance**
 - Simple, high-level API
 - Specialized configurations with significant performance boosts for common use cases like classification and summarization
 - [Broad model support](https://trymirai.com/models)
-- Observable model manager
 
-## Setup
+## Quick Start
 
 Add the `uzu` dependency to your project's `package.json`:
 
 ```json
 "dependencies": {
-    "@trymirai/uzu": "0.1.21"
+    "@trymirai/uzu": "0.1.22"
 }
 ```
 
-Create and activate engine:
+Set up your project through [Platform](https://platform.trymirai.com) and obtain an `API_KEY`. Then, choose the model you want from the [library](https://platform.trymirai.com/models) and run it with the following snippet using the corresponding identifier:
 
 ```ts
-const engine = new Engine()
-const licenseStatus: LicenseStatus = await engine.activate(apiKey)
+const output = await Engine
+    .create('API_KEY')
+    .model('Alibaba-Qwen3-0.6B')
+    .reply('Tell me a short, funny story about a robot');
 ```
 
-### Refresh models registry / list cloud models:
+Everything from model downloading to inference configuration is handled automatically. Refer to the [documentation](https://docs.trymirai.com) for details on how to customize each step of the process.
 
-```ts
-const localModels: LocalModel[] = await engine.updateRegistry()
-const localModelId = 'Meta-Llama-3.2-1B-Instruct'
-```
-
-### Download with progress handle
-
-```ts
-const donwloadHandle = engine.downloadHandle(localModelId)
-donwloadHandle.start()
-
-for await (const donwloadProgress of donwloadHandle.progress()) {
-    // Implement a custom download progress handler
-    handleProgress(donwloadProgress)
-}
-```
-
-Alternatively, you may use engine to control and observe model download:
+## Examples
 
-```ts
-const modelState: ModelDownloadState = engine.getState(localModelId)
+Place the `API_KEY` you obtained earlier in the corresponding example file, and then run it using one of the following commands:
 
-// engine.download(localModelId)
-// engine.pause(localModelId)
-// engine.resume(localModelId)
-// engine.stop(localModelId)
-// engine.delete(localModelId)
-```
-
-### Session
-
-`Session` is the core entity used to communicate with the model:
-
-```ts
-const modelId: ModelID = { type: 'Local', id: localModelId }
-const config: Config = {
-    preset: { type: 'General' },
-    prefillStepSize: { type: 'Default' },
-    contextLength: { type: 'Default' },
-    samplingSeed: { type: 'Default' },
-}
-const session: Session = engine.createSession(modelId, config)
+```bash
+pnpm run tsn examples/chat.ts
+pnpm run tsn examples/summarization.ts
+pnpm run tsn examples/classification.ts
 ```
 
 ### Chat
 
-After creating it, you can run the `Session` with a specific prompt or a list of messages:
-
-```ts
-const input: Input = {
-    type: 'Messages',
-    messages: [
-        {
-            role: Role.System,
-            content: 'You are a helpful assistant.'
-        },
-        {
-            role: Role.User,
-            content: 'Tell me a short, funny story about a robot.'
-        },
-    ],
-}
-```
+In this example, we will download a model and get a reply to a specific list of messages:
 
 ```ts
-const runConfig: RunConfig = {
-    tokensLimit: 128,
-    enableThinking: true,
-    samplingPolicy: { type: 'Default' },
+import Engine, { Message } from '@trymirai/uzu';
+
+async function main() {
+    const output = await Engine.create('API_KEY')
+        .model('Alibaba-Qwen3-0.6B')
+        .download((update) => {
+            console.log('Progress:', update.progress);
+        })
+        .replyToMessages(
+            [
+                Message.system('You are a helpful assistant'),
+                Message.user('Tell me a short, funny story about a robot')
+            ],
+            (partialOutput) => {
+                return true;
+            },
+        );
+    console.log(output.text.original);
 }
 
-const output = session.run(input, runConfig, (partialOutput) => {
-    // Implement a custom partial output handler
-    return handlePartialOutput(partialOutput)
-})
+main().catch((error) => {
+    console.error(error);
+});
 ```
 
 ### Summarization
 
-In this example, we will extract a summary of the input text:
+In this example, we will use the `summarization` preset to generate a summary of the input text:
 
 ```ts
-const modelId: ModelID = { type: 'Local', id: localModelId }
-const config: Config = {
-    preset: { type: 'Summarization' },
-    prefillStepSize: { type: 'Default' },
-    contextLength: { type: 'Default' },
-    samplingSeed: { type: 'Default' },
+import Engine, { Preset, SamplingMethod } from '@trymirai/uzu';
+
+async function main() {
+    const textToSummarize =
+        "A Large Language Model (LLM) is a type of artificial intelligence that processes and generates human-like text. It is trained on vast datasets containing books, articles, and web content, allowing it to understand and predict language patterns. LLMs use deep learning, particularly transformer-based architectures, to analyze text, recognize context, and generate coherent responses. These models have a wide range of applications, including chatbots, content creation, translation, and code generation. One of the key strengths of LLMs is their ability to generate contextually relevant text based on prompts. They utilize self-attention mechanisms to weigh the importance of words within a sentence, improving accuracy and fluency. Examples of popular LLMs include OpenAI's GPT series, Google's BERT, and Meta's LLaMA. As these models grow in size and sophistication, they continue to enhance human-computer interactions, making AI-powered communication more natural and effective.";
+    const prompt = `Text is: "${textToSummarize}". Write only summary itself.`;
+
+    const output = await Engine.create('API_KEY')
+        .model('Alibaba-Qwen3-0.6B')
+        .download((update) => {
+            console.log('Progress:', update.progress);
+        })
+        .preset(Preset.summarization())
+        .session()
+        .tokensLimit(256)
+        .enableThinking(false)
+        .samplingMethod(SamplingMethod.greedy())
+        .reply(prompt);
+
+    console.log('Summary:', output.text.original);
+    console.log(
+        'Model runs:',
+        output.stats.prefillStats.modelRun.count + (output.stats.generateStats?.modelRun.count ?? 0),
+    );
+    console.log('Tokens count:', output.stats.totalStats.tokensCountOutput);
 }
-const session = engine.createSession(modelId, config)
-```
 
-```ts
-const textToSummarize =
-    'A Large Language Model (LLM) is a type of AI that processes and generates text using transformer-based architectures trained on vast datasets. They power chatbots, translation, code assistants, and more.'
-const input: Input = {
-    type: 'Text',
-    text: `Text is: "${textToSummarize}". Write only summary itself.`,
-}
+main().catch((error) => {
+    console.error(error);
+});
 ```
 
-```ts
-const runConfig: RunConfig = {
-    tokensLimit: 256,
-    enableThinking: true,
-    samplingPolicy: { type: 'Custom', value: { type: 'Greedy' } },
-}
-
-const output = session.run(input, runConfig, (partialOutput) => {
-    // Implement a custom partial output handler
-    return handlePartialOutput(partialOutput)
-})
-```
+You will notice that the model’s run count is lower than the actual number of generated tokens due to speculative decoding, which significantly improves generation speed.
 
 ### Classification
 
-Let’s look at a case where you need to classify input text based on a specific feature, such as `sentiment`:
-
-```ts
-const feature: ClassificationFeature = {
-    name: 'sentiment',
-    values: ['Happy', 'Sad', 'Angry', 'Fearful', 'Surprised', 'Disgusted'],
-}
-const config: Config = {
-    preset: { type: 'Classification', feature },
-    prefillStepSize: { type: 'Default' },
-    contextLength: { type: 'Default' },
-    samplingSeed: { type: 'Default' },
-}
-
-const modelId: ModelID = { type: 'Local', id: localModelId }
-const session = engine.createSession(modelId, config)
-```
-
-```ts
-const textToDetectFeature =
-    "Today's been awesome! Everything just feels right, and I can't stop smiling."
-const classificationPrompt =
-    `Text is: "${textToDetectFeature}". Choose ${feature.name} from the list: ${feature.values.join(', ')}. ` +
-    "Answer with one word. Don't add a dot at the end."
-const input: Input = { type: 'Text', text: classificationPrompt }
-```
+In this example, we will use the `classification` preset to determine the sentiment of the user's input:
 
 ```ts
-const runConfig: RunConfig = {
-    tokensLimit: 32,
-    enableThinking: true,
-    samplingPolicy: { type: 'Custom', value: { type: 'Greedy' } },
+import Engine, { ClassificationFeature, Preset, SamplingMethod } from '@trymirai/uzu';
+
+async function main() {
+    const feature = new ClassificationFeature('sentiment', [
+        'Happy',
+        'Sad',
+        'Angry',
+        'Fearful',
+        'Surprised',
+        'Disgusted',
+    ]);
+    const textToDetectFeature =
+        "Today's been awesome! Everything just feels right, and I can't stop smiling.";
+    const prompt =
+        `Text is: "${textToDetectFeature}". Choose ${feature.name} from the list: ${feature.values.join(', ')}. ` +
+        "Answer with one word. Don't add a dot at the end.";
+
+    const output = await Engine.create('API_KEY')
+        .model('Alibaba-Qwen3-0.6B')
+        .download((update) => {
+            console.log('Progress:', update.progress);
+        })
+        .preset(Preset.classification(feature))
+        .session()
+        .tokensLimit(32)
+        .enableThinking(false)
+        .samplingMethod(SamplingMethod.greedy())
+        .reply(prompt);
+
+    console.log('Prediction:', output.text.original);
+    console.log('Stats:', output.stats);
 }
 
-const output = session.run(input, runConfig, (partialOutput) => {
-    // Implement a custom partial output handler
-    return handlePartialOutput(partialOutput)
-})
+main().catch((error) => {
+    console.error(error);
+});
 ```
 
-In this example, you will get the answer `Happy` immediately after the prefill step, and the actual generation won't even start.
+You can view the stats to see that the answer will be ready immediately after the prefill step, and actual generation won’t even start due to speculative decoding, which significantly improves generation speed.
 
 ## License
 
-This project is licensed under the MIT License. See the [LICENSE](LICENSE) file for details.
+This project is licensed under the MIT License. See the [LICENSE](LICENSE) file for details.

package/bridging/classificationFeature.d.mts

@@ -0,0 +1,9 @@
+import { ClassificationFeature as NapiClassificationFeature } from "../napi/uzu.mjs";
+import { ToNapi } from "./napi.mjs";
+export declare class ClassificationFeature implements ToNapi<NapiClassificationFeature> {
+    readonly name: string;
+    readonly values: string[];
+    constructor(name: string, values: string[]);
+    toNapi(): NapiClassificationFeature;
+}
+//# sourceMappingURL=classificationFeature.d.mts.map

package/bridging/classificationFeature.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationFeature.d.mts","sourceRoot":"","sources":["../src/bridging/classificationFeature.ts"],"names":[],"mappings":"OAAO,EAAE,qBAAqB,IAAI,yBAAyB,EAAE;OACtD,EAAE,MAAM,EAAE;AAEjB,qBAAa,qBAAsB,YAAW,MAAM,CAAC,yBAAyB,CAAC;IAC3E,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;gBAEd,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE;IAK1C,MAAM,IAAI,yBAAyB;CAOtC"}

package/bridging/classificationFeature.d.ts

@@ -0,0 +1,9 @@
+import { ClassificationFeature as NapiClassificationFeature } from "../napi/uzu.js";
+import { ToNapi } from "./napi.js";
+export declare class ClassificationFeature implements ToNapi<NapiClassificationFeature> {
+    readonly name: string;
+    readonly values: string[];
+    constructor(name: string, values: string[]);
+    toNapi(): NapiClassificationFeature;
+}
+//# sourceMappingURL=classificationFeature.d.ts.map

package/bridging/classificationFeature.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationFeature.d.ts","sourceRoot":"","sources":["../src/bridging/classificationFeature.ts"],"names":[],"mappings":"OAAO,EAAE,qBAAqB,IAAI,yBAAyB,EAAE;OACtD,EAAE,MAAM,EAAE;AAEjB,qBAAa,qBAAsB,YAAW,MAAM,CAAC,yBAAyB,CAAC;IAC3E,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;gBAEd,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE;IAK1C,MAAM,IAAI,yBAAyB;CAOtC"}

package/bridging/classificationFeature.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClassificationFeature = void 0;
+class ClassificationFeature {
+    constructor(name, values) {
+        this.name = name;
+        this.values = values;
+    }
+    toNapi() {
+        const result = {
+            name: this.name,
+            values: this.values,
+        };
+        return result;
+    }
+}
+exports.ClassificationFeature = ClassificationFeature;
+//# sourceMappingURL=classificationFeature.js.map

package/bridging/classificationFeature.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationFeature.js","sourceRoot":"","sources":["../src/bridging/classificationFeature.ts"],"names":[],"mappings":";;;AAGA,MAAa,qBAAqB;IAI9B,YAAY,IAAY,EAAE,MAAgB;QACtC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,MAAM;QACF,MAAM,MAAM,GAA8B;YACtC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,MAAM;SACtB,CAAC;QACF,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ;AAhBD,sDAgBC"}

package/bridging/classificationFeature.mjs

@@ -0,0 +1,14 @@
+export class ClassificationFeature {
+    constructor(name, values) {
+        this.name = name;
+        this.values = values;
+    }
+    toNapi() {
+        const result = {
+            name: this.name,
+            values: this.values,
+        };
+        return result;
+    }
+}
+//# sourceMappingURL=classificationFeature.mjs.map

package/bridging/classificationFeature.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationFeature.mjs","sourceRoot":"","sources":["../src/bridging/classificationFeature.ts"],"names":[],"mappings":"AAGA,MAAM,OAAO,qBAAqB;IAI9B,YAAY,IAAY,EAAE,MAAgB;QACtC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,MAAM;QACF,MAAM,MAAM,GAA8B;YACtC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,MAAM;SACtB,CAAC;QACF,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ"}

package/bridging/config.d.mts

@@ -0,0 +1,20 @@
+import { Config as NapiConfig } from "../napi/uzu.mjs";
+import { ContextLength } from "./contextLength.mjs";
+import { ToNapi } from "./napi.mjs";
+import { PrefillStepSize } from "./prefillStepSize.mjs";
+import { Preset } from "./preset.mjs";
+import { SamplingSeed } from "./samplingSeed.mjs";
+export declare class Config implements ToNapi<NapiConfig> {
+    readonly preset: Preset;
+    readonly prefillStepSize: PrefillStepSize;
+    readonly contextLength: ContextLength;
+    readonly samplingSeed: SamplingSeed;
+    constructor(preset: Preset, prefillStepSize: PrefillStepSize, contextLength: ContextLength, samplingSeed: SamplingSeed);
+    static default(): Config;
+    withPreset(preset: Preset): Config;
+    withPrefillStepSize(prefillStepSize: PrefillStepSize): Config;
+    withContextLength(contextLength: ContextLength): Config;
+    withSamplingSeed(samplingSeed: SamplingSeed): Config;
+    toNapi(): NapiConfig;
+}
+//# sourceMappingURL=config.d.mts.map

package/bridging/config.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.mts","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,aAAa,EAAE;OACjB,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,YAAY,EAAE;AAEvB,qBAAa,MAAO,YAAW,MAAM,CAAC,UAAU,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;IAC1C,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC;IACtC,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;gBAGhC,MAAM,EAAE,MAAM,EACd,eAAe,EAAE,eAAe,EAChC,aAAa,EAAE,aAAa,EAC5B,YAAY,EAAE,YAAY;IAQ9B,MAAM,CAAC,OAAO,IAAI,MAAM;IASxB,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAIlC,mBAAmB,CAAC,eAAe,EAAE,eAAe,GAAG,MAAM;IAI7D,iBAAiB,CAAC,aAAa,EAAE,aAAa,GAAG,MAAM;IAIvD,gBAAgB,CAAC,YAAY,EAAE,YAAY,GAAG,MAAM;IAIpD,MAAM,IAAI,UAAU;CASvB"}

package/bridging/config.d.ts

@@ -0,0 +1,20 @@
+import { Config as NapiConfig } from "../napi/uzu.js";
+import { ContextLength } from "./contextLength.js";
+import { ToNapi } from "./napi.js";
+import { PrefillStepSize } from "./prefillStepSize.js";
+import { Preset } from "./preset.js";
+import { SamplingSeed } from "./samplingSeed.js";
+export declare class Config implements ToNapi<NapiConfig> {
+    readonly preset: Preset;
+    readonly prefillStepSize: PrefillStepSize;
+    readonly contextLength: ContextLength;
+    readonly samplingSeed: SamplingSeed;
+    constructor(preset: Preset, prefillStepSize: PrefillStepSize, contextLength: ContextLength, samplingSeed: SamplingSeed);
+    static default(): Config;
+    withPreset(preset: Preset): Config;
+    withPrefillStepSize(prefillStepSize: PrefillStepSize): Config;
+    withContextLength(contextLength: ContextLength): Config;
+    withSamplingSeed(samplingSeed: SamplingSeed): Config;
+    toNapi(): NapiConfig;
+}
+//# sourceMappingURL=config.d.ts.map

package/bridging/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,aAAa,EAAE;OACjB,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,YAAY,EAAE;AAEvB,qBAAa,MAAO,YAAW,MAAM,CAAC,UAAU,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;IAC1C,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC;IACtC,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;gBAGhC,MAAM,EAAE,MAAM,EACd,eAAe,EAAE,eAAe,EAChC,aAAa,EAAE,aAAa,EAC5B,YAAY,EAAE,YAAY;IAQ9B,MAAM,CAAC,OAAO,IAAI,MAAM;IASxB,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAIlC,mBAAmB,CAAC,eAAe,EAAE,eAAe,GAAG,MAAM;IAI7D,iBAAiB,CAAC,aAAa,EAAE,aAAa,GAAG,MAAM;IAIvD,gBAAgB,CAAC,YAAY,EAAE,YAAY,GAAG,MAAM;IAIpD,MAAM,IAAI,UAAU;CASvB"}

package/bridging/config.js

@@ -0,0 +1,41 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Config = void 0;
+const contextLength_1 = require("./contextLength.js");
+const prefillStepSize_1 = require("./prefillStepSize.js");
+const preset_1 = require("./preset.js");
+const samplingSeed_1 = require("./samplingSeed.js");
+class Config {
+    constructor(preset, prefillStepSize, contextLength, samplingSeed) {
+        this.preset = preset;
+        this.prefillStepSize = prefillStepSize;
+        this.contextLength = contextLength;
+        this.samplingSeed = samplingSeed;
+    }
+    static default() {
+        return new Config(preset_1.Preset.general(), prefillStepSize_1.PrefillStepSize.default(), contextLength_1.ContextLength.default(), samplingSeed_1.SamplingSeed.default());
+    }
+    withPreset(preset) {
+        return new Config(preset, this.prefillStepSize, this.contextLength, this.samplingSeed);
+    }
+    withPrefillStepSize(prefillStepSize) {
+        return new Config(this.preset, prefillStepSize, this.contextLength, this.samplingSeed);
+    }
+    withContextLength(contextLength) {
+        return new Config(this.preset, this.prefillStepSize, contextLength, this.samplingSeed);
+    }
+    withSamplingSeed(samplingSeed) {
+        return new Config(this.preset, this.prefillStepSize, this.contextLength, samplingSeed);
+    }
+    toNapi() {
+        const napiConfig = {
+            preset: this.preset.toNapi(),
+            prefillStepSize: this.prefillStepSize.toNapi(),
+            contextLength: this.contextLength.toNapi(),
+            samplingSeed: this.samplingSeed.toNapi(),
+        };
+        return napiConfig;
+    }
+}
+exports.Config = Config;
+//# sourceMappingURL=config.js.map

package/bridging/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":";;;AACA,sDAAgD;AAEhD,0DAAoD;AACpD,wCAAkC;AAClC,oDAA8C;AAE9C,MAAa,MAAM;IAMf,YACI,MAAc,EACd,eAAgC,EAChC,aAA4B,EAC5B,YAA0B;QAE1B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACrC,CAAC;IAED,MAAM,CAAC,OAAO;QACV,OAAO,IAAI,MAAM,CACb,eAAM,CAAC,OAAO,EAAE,EAChB,iCAAe,CAAC,OAAO,EAAE,EACzB,6BAAa,CAAC,OAAO,EAAE,EACvB,2BAAY,CAAC,OAAO,EAAE,CACzB,CAAC;IACN,CAAC;IAED,UAAU,CAAC,MAAc;QACrB,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IAC3F,CAAC;IAED,mBAAmB,CAAC,eAAgC;QAChD,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,eAAe,EAAE,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IAC3F,CAAC;IAED,iBAAiB,CAAC,aAA4B;QAC1C,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IAC3F,CAAC;IAED,gBAAgB,CAAC,YAA0B;QACvC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;IAC3F,CAAC;IAED,MAAM;QACF,MAAM,UAAU,GAAe;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;YAC5B,eAAe,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;YAC9C,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE;YAC1C,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;SAC3C,CAAC;QACF,OAAO,UAAU,CAAC;IACtB,CAAC;CACJ;AApDD,wBAoDC"}

package/bridging/config.mjs

@@ -0,0 +1,37 @@
+import { ContextLength } from "./contextLength.mjs";
+import { PrefillStepSize } from "./prefillStepSize.mjs";
+import { Preset } from "./preset.mjs";
+import { SamplingSeed } from "./samplingSeed.mjs";
+export class Config {
+    constructor(preset, prefillStepSize, contextLength, samplingSeed) {
+        this.preset = preset;
+        this.prefillStepSize = prefillStepSize;
+        this.contextLength = contextLength;
+        this.samplingSeed = samplingSeed;
+    }
+    static default() {
+        return new Config(Preset.general(), PrefillStepSize.default(), ContextLength.default(), SamplingSeed.default());
+    }
+    withPreset(preset) {
+        return new Config(preset, this.prefillStepSize, this.contextLength, this.samplingSeed);
+    }
+    withPrefillStepSize(prefillStepSize) {
+        return new Config(this.preset, prefillStepSize, this.contextLength, this.samplingSeed);
+    }
+    withContextLength(contextLength) {
+        return new Config(this.preset, this.prefillStepSize, contextLength, this.samplingSeed);
+    }
+    withSamplingSeed(samplingSeed) {
+        return new Config(this.preset, this.prefillStepSize, this.contextLength, samplingSeed);
+    }
+    toNapi() {
+        const napiConfig = {
+            preset: this.preset.toNapi(),
+            prefillStepSize: this.prefillStepSize.toNapi(),
+            contextLength: this.contextLength.toNapi(),
+            samplingSeed: this.samplingSeed.toNapi(),
+        };
+        return napiConfig;
+    }
+}
+//# sourceMappingURL=config.mjs.map

package/bridging/config.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.mjs","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":"OACO,EAAE,aAAa,EAAE;OAEjB,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,YAAY,EAAE;AAEvB,MAAM,OAAO,MAAM;IAMf,YACI,MAAc,EACd,eAAgC,EAChC,aAA4B,EAC5B,YAA0B;QAE1B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACrC,CAAC;IAED,MAAM,CAAC,OAAO;QACV,OAAO,IAAI,MAAM,CACb,MAAM,CAAC,OAAO,EAAE,EAChB,eAAe,CAAC,OAAO,EAAE,EACzB,aAAa,CAAC,OAAO,EAAE,EACvB,YAAY,CAAC,OAAO,EAAE,CACzB,CAAC;IACN,CAAC;IAED,UAAU,CAAC,MAAc;QACrB,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IAC3F,CAAC;IAED,mBAAmB,CAAC,eAAgC;QAChD,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,eAAe,EAAE,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IAC3F,CAAC;IAED,iBAAiB,CAAC,aAA4B;QAC1C,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IAC3F,CAAC;IAED,gBAAgB,CAAC,YAA0B;QACvC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;IAC3F,CAAC;IAED,MAAM;QACF,MAAM,UAAU,GAAe;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;YAC5B,eAAe,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;YAC9C,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE;YAC1C,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;SAC3C,CAAC;QACF,OAAO,UAAU,CAAC;IACtB,CAAC;CACJ"}

package/bridging/contextLength.d.mts

@@ -0,0 +1,11 @@
+import { ContextLength as NapiContextLength } from "../napi/uzu.mjs";
+import { ToNapi } from "./napi.mjs";
+export declare class ContextLength implements ToNapi<NapiContextLength> {
+    private readonly napiContextLength;
+    private constructor();
+    static default(): ContextLength;
+    static maximal(): ContextLength;
+    static custom(length: number): ContextLength;
+    toNapi(): NapiContextLength;
+}
+//# sourceMappingURL=contextLength.d.mts.map

package/bridging/contextLength.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextLength.d.mts","sourceRoot":"","sources":["../src/bridging/contextLength.ts"],"names":[],"mappings":"OAAO,EAAE,aAAa,IAAI,iBAAiB,EAAE;OACtC,EAAE,MAAM,EAAE;AAEjB,qBAAa,aAAc,YAAW,MAAM,CAAC,iBAAiB,CAAC;IAC3D,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAoB;IAEtD,OAAO;IAIP,MAAM,CAAC,OAAO,IAAI,aAAa;IAK/B,MAAM,CAAC,OAAO,IAAI,aAAa;IAK/B,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa;IAK5C,MAAM,IAAI,iBAAiB;CAG9B"}

package/bridging/contextLength.d.ts

@@ -0,0 +1,11 @@
+import { ContextLength as NapiContextLength } from "../napi/uzu.js";
+import { ToNapi } from "./napi.js";
+export declare class ContextLength implements ToNapi<NapiContextLength> {
+    private readonly napiContextLength;
+    private constructor();
+    static default(): ContextLength;
+    static maximal(): ContextLength;
+    static custom(length: number): ContextLength;
+    toNapi(): NapiContextLength;
+}
+//# sourceMappingURL=contextLength.d.ts.map

package/bridging/contextLength.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextLength.d.ts","sourceRoot":"","sources":["../src/bridging/contextLength.ts"],"names":[],"mappings":"OAAO,EAAE,aAAa,IAAI,iBAAiB,EAAE;OACtC,EAAE,MAAM,EAAE;AAEjB,qBAAa,aAAc,YAAW,MAAM,CAAC,iBAAiB,CAAC;IAC3D,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAoB;IAEtD,OAAO;IAIP,MAAM,CAAC,OAAO,IAAI,aAAa;IAK/B,MAAM,CAAC,OAAO,IAAI,aAAa;IAK/B,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa;IAK5C,MAAM,IAAI,iBAAiB;CAG9B"}

package/bridging/contextLength.js

@@ -0,0 +1,25 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContextLength = void 0;
+class ContextLength {
+    constructor(napiContextLength) {
+        this.napiContextLength = napiContextLength;
+    }
+    static default() {
+        const napiContextLength = { type: 'Default' };
+        return new ContextLength(napiContextLength);
+    }
+    static maximal() {
+        const napiContextLength = { type: 'Maximal' };
+        return new ContextLength(napiContextLength);
+    }
+    static custom(length) {
+        const napiContextLength = { type: 'Custom', length };
+        return new ContextLength(napiContextLength);
+    }
+    toNapi() {
+        return this.napiContextLength;
+    }
+}
+exports.ContextLength = ContextLength;
+//# sourceMappingURL=contextLength.js.map

package/bridging/contextLength.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextLength.js","sourceRoot":"","sources":["../src/bridging/contextLength.ts"],"names":[],"mappings":";;;AAGA,MAAa,aAAa;IAGtB,YAAoB,iBAAoC;QACpD,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,OAAO;QACV,MAAM,iBAAiB,GAAsB,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACjE,OAAO,IAAI,aAAa,CAAC,iBAAiB,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,CAAC,OAAO;QACV,MAAM,iBAAiB,GAAsB,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACjE,OAAO,IAAI,aAAa,CAAC,iBAAiB,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,MAAc;QACxB,MAAM,iBAAiB,GAAsB,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;QACxE,OAAO,IAAI,aAAa,CAAC,iBAAiB,CAAC,CAAC;IAChD,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAClC,CAAC;CACJ;AAzBD,sCAyBC"}

package/bridging/contextLength.mjs

@@ -0,0 +1,21 @@
+export class ContextLength {
+    constructor(napiContextLength) {
+        this.napiContextLength = napiContextLength;
+    }
+    static default() {
+        const napiContextLength = { type: 'Default' };
+        return new ContextLength(napiContextLength);
+    }
+    static maximal() {
+        const napiContextLength = { type: 'Maximal' };
+        return new ContextLength(napiContextLength);
+    }
+    static custom(length) {
+        const napiContextLength = { type: 'Custom', length };
+        return new ContextLength(napiContextLength);
+    }
+    toNapi() {
+        return this.napiContextLength;
+    }
+}
+//# sourceMappingURL=contextLength.mjs.map

package/bridging/contextLength.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextLength.mjs","sourceRoot":"","sources":["../src/bridging/contextLength.ts"],"names":[],"mappings":"AAGA,MAAM,OAAO,aAAa;IAGtB,YAAoB,iBAAoC;QACpD,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,OAAO;QACV,MAAM,iBAAiB,GAAsB,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACjE,OAAO,IAAI,aAAa,CAAC,iBAAiB,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,CAAC,OAAO;QACV,MAAM,iBAAiB,GAAsB,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACjE,OAAO,IAAI,aAAa,CAAC,iBAAiB,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,MAAc;QACxB,MAAM,iBAAiB,GAAsB,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;QACxE,OAAO,IAAI,aAAa,CAAC,iBAAiB,CAAC,CAAC;IAChD,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAClC,CAAC;CACJ"}

package/bridging/downloadHandle.d.mts

@@ -0,0 +1,14 @@
+import { ModelDownloadHandle as NapiModelDownloadHandle } from "../napi/uzu.mjs";
+import { DownloadProgressUpdate } from "./downloadProgressUpdate.mjs";
+import { DownloadState } from "./downloadState.mjs";
+export declare class DownloadHandle {
+    private readonly napiDownloadHandle;
+    constructor(napiDownloadHandle: NapiModelDownloadHandle);
+    identifier(): string;
+    state(): Promise<DownloadState>;
+    download(): Promise<void>;
+    pause(): Promise<void>;
+    delete(): Promise<void>;
+    progressUpdate(): AsyncIterable<DownloadProgressUpdate>;
+}
+//# sourceMappingURL=downloadHandle.d.mts.map

package/bridging/downloadHandle.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"downloadHandle.d.mts","sourceRoot":"","sources":["../src/bridging/downloadHandle.ts"],"names":[],"mappings":"OAAO,EAAE,mBAAmB,IAAI,uBAAuB,EAAE;OAClD,EAAE,sBAAsB,EAAE;OAC1B,EAAE,aAAa,EAAE;AAExB,qBAAa,cAAc;IACvB,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAA0B;gBAEjD,kBAAkB,EAAE,uBAAuB;IAIvD,UAAU,IAAI,MAAM;IAId,KAAK,IAAI,OAAO,CAAC,aAAa,CAAC;IAK/B,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAIzB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAItB,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC;IAI7B,cAAc,IAAI,aAAa,CAAC,sBAAsB,CAAC;CAQ1D"}