@trymirai/uzu 0.2.20 → 0.3.1

package/README.md

@@ -24,7 +24,7 @@ Add the `uzu` dependency to your project's `package.json`:
 
 ```json
 "dependencies": {
-    "@trymirai/uzu": "0.2.20"
+    "@trymirai/uzu": "0.3.1"
 }
 ```
 
@@ -51,7 +51,9 @@ pnpm run tsn examples/chatStaticContext.ts
 pnpm run tsn examples/summarization.ts
 pnpm run tsn examples/classification.ts
 pnpm run tsn examples/cloud.ts
-pnpm run tsn examples/structutedOutput.ts
+pnpm run tsn examples/structuredOutput.ts
+pnpm run tsn examples/classifier.ts
+pnpm run tsn examples/textToSpeech.ts
 ```
 
 ### Chat
@@ -140,7 +142,7 @@ async function main() {
     const engine = await Engine.load('API_KEY');
 
     const model = await engine.chatModel('Qwen/Qwen3-0.6B');
-    await engine.downloadChatModel(model, (update) => {
+    await engine.downloadModel(model, (update) => {
         console.log('Progress:', update.progress);
     });
 
@@ -190,7 +192,7 @@ async function main() {
     const engine = await Engine.load('API_KEY');
 
     const model = await engine.chatModel('Qwen/Qwen3-0.6B');
-    await engine.downloadChatModel(model, (update) => {
+    await engine.downloadModel(model, (update) => {
         console.log('Progress:', update.progress);
     });
 

package/bridging/classificationLabel.d.mts

@@ -0,0 +1,12 @@
+import { ClassificationLabel as NapiClassificationLabel } from "../napi/uzu.mjs";
+export declare class ClassificationLabel {
+    readonly index: number;
+    readonly label: string;
+    readonly title: string;
+    readonly description: string;
+    readonly recommendedThreshold: number;
+    constructor(index: number, label: string, title: string, description: string, recommendedThreshold: number);
+    static fromNapi(napiEntity: NapiClassificationLabel): ClassificationLabel;
+    toNapi(): NapiClassificationLabel;
+}
+//# sourceMappingURL=classificationLabel.d.mts.map

package/bridging/classificationLabel.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationLabel.d.mts","sourceRoot":"","sources":["../src/bridging/classificationLabel.ts"],"names":[],"mappings":"OAAO,EAAE,mBAAmB,IAAI,uBAAuB,EAAE;AAEzD,qBAAa,mBAAmB;IAC5B,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,oBAAoB,EAAE,MAAM,CAAC;gBAGlC,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,EACnB,oBAAoB,EAAE,MAAM;IAShC,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,uBAAuB,GAAG,mBAAmB;IAUzE,MAAM,IAAI,uBAAuB;CASpC"}

package/bridging/classificationLabel.d.ts

@@ -0,0 +1,12 @@
+import { ClassificationLabel as NapiClassificationLabel } from "../napi/uzu.js";
+export declare class ClassificationLabel {
+    readonly index: number;
+    readonly label: string;
+    readonly title: string;
+    readonly description: string;
+    readonly recommendedThreshold: number;
+    constructor(index: number, label: string, title: string, description: string, recommendedThreshold: number);
+    static fromNapi(napiEntity: NapiClassificationLabel): ClassificationLabel;
+    toNapi(): NapiClassificationLabel;
+}
+//# sourceMappingURL=classificationLabel.d.ts.map

package/bridging/classificationLabel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationLabel.d.ts","sourceRoot":"","sources":["../src/bridging/classificationLabel.ts"],"names":[],"mappings":"OAAO,EAAE,mBAAmB,IAAI,uBAAuB,EAAE;AAEzD,qBAAa,mBAAmB;IAC5B,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,oBAAoB,EAAE,MAAM,CAAC;gBAGlC,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,EACnB,oBAAoB,EAAE,MAAM;IAShC,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,uBAAuB,GAAG,mBAAmB;IAUzE,MAAM,IAAI,uBAAuB;CASpC"}

package/bridging/classificationLabel.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClassificationLabel = void 0;
+class ClassificationLabel {
+    constructor(index, label, title, description, recommendedThreshold) {
+        this.index = index;
+        this.label = label;
+        this.title = title;
+        this.description = description;
+        this.recommendedThreshold = recommendedThreshold;
+    }
+    static fromNapi(napiEntity) {
+        return new ClassificationLabel(napiEntity.index, napiEntity.label, napiEntity.title, napiEntity.description, napiEntity.recommendedThreshold);
+    }
+    toNapi() {
+        return {
+            index: this.index,
+            label: this.label,
+            title: this.title,
+            description: this.description,
+            recommendedThreshold: this.recommendedThreshold,
+        };
+    }
+}
+exports.ClassificationLabel = ClassificationLabel;
+//# sourceMappingURL=classificationLabel.js.map

package/bridging/classificationLabel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationLabel.js","sourceRoot":"","sources":["../src/bridging/classificationLabel.ts"],"names":[],"mappings":";;;AAEA,MAAa,mBAAmB;IAO5B,YACI,KAAa,EACb,KAAa,EACb,KAAa,EACb,WAAmB,EACnB,oBAA4B;QAE5B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;IACrD,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,UAAmC;QAC/C,OAAO,IAAI,mBAAmB,CAC1B,UAAU,CAAC,KAAK,EAChB,UAAU,CAAC,KAAK,EAChB,UAAU,CAAC,KAAK,EAChB,UAAU,CAAC,WAAW,EACtB,UAAU,CAAC,oBAAoB,CAClC,CAAC;IACN,CAAC;IAED,MAAM;QACF,OAAO;YACH,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,oBAAoB,EAAE,IAAI,CAAC,oBAAoB;SAClD,CAAC;IACN,CAAC;CACJ;AAxCD,kDAwCC"}

package/bridging/classificationLabel.mjs

@@ -0,0 +1,22 @@
+export class ClassificationLabel {
+    constructor(index, label, title, description, recommendedThreshold) {
+        this.index = index;
+        this.label = label;
+        this.title = title;
+        this.description = description;
+        this.recommendedThreshold = recommendedThreshold;
+    }
+    static fromNapi(napiEntity) {
+        return new ClassificationLabel(napiEntity.index, napiEntity.label, napiEntity.title, napiEntity.description, napiEntity.recommendedThreshold);
+    }
+    toNapi() {
+        return {
+            index: this.index,
+            label: this.label,
+            title: this.title,
+            description: this.description,
+            recommendedThreshold: this.recommendedThreshold,
+        };
+    }
+}
+//# sourceMappingURL=classificationLabel.mjs.map

package/bridging/classificationLabel.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationLabel.mjs","sourceRoot":"","sources":["../src/bridging/classificationLabel.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,mBAAmB;IAO5B,YACI,KAAa,EACb,KAAa,EACb,KAAa,EACb,WAAmB,EACnB,oBAA4B;QAE5B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;IACrD,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,UAAmC;QAC/C,OAAO,IAAI,mBAAmB,CAC1B,UAAU,CAAC,KAAK,EAChB,UAAU,CAAC,KAAK,EAChB,UAAU,CAAC,KAAK,EAChB,UAAU,CAAC,WAAW,EACtB,UAAU,CAAC,oBAAoB,CAClC,CAAC;IACN,CAAC;IAED,MAAM;QACF,OAAO;YACH,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,oBAAoB,EAAE,IAAI,CAAC,oBAAoB;SAClD,CAAC;IACN,CAAC;CACJ"}

package/bridging/classificationOutput.d.mts

@@ -0,0 +1,10 @@
+import { ClassificationOutput as NapiClassificationOutput } from "../napi/uzu.mjs";
+import { ClassificationStats } from "./classificationStats.mjs";
+export declare class ClassificationOutput {
+    readonly logits: number[];
+    readonly probabilities: Record<string, number>;
+    readonly stats: ClassificationStats;
+    constructor(logits: number[], probabilities: Record<string, number>, stats: ClassificationStats);
+    static fromNapi(napiEntity: NapiClassificationOutput): ClassificationOutput;
+}
+//# sourceMappingURL=classificationOutput.d.mts.map

package/bridging/classificationOutput.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationOutput.d.mts","sourceRoot":"","sources":["../src/bridging/classificationOutput.ts"],"names":[],"mappings":"OAAO,EAAE,oBAAoB,IAAI,wBAAwB,EAAE;OACpD,EAAE,mBAAmB,EAAE;AAE9B,qBAAa,oBAAoB;IAC7B,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;IAC1B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/C,QAAQ,CAAC,KAAK,EAAE,mBAAmB,CAAC;gBAExB,MAAM,EAAE,MAAM,EAAE,EAAE,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,KAAK,EAAE,mBAAmB;IAM/F,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,wBAAwB,GAAG,oBAAoB;CAO9E"}

package/bridging/classificationOutput.d.ts

@@ -0,0 +1,10 @@
+import { ClassificationOutput as NapiClassificationOutput } from "../napi/uzu.js";
+import { ClassificationStats } from "./classificationStats.js";
+export declare class ClassificationOutput {
+    readonly logits: number[];
+    readonly probabilities: Record<string, number>;
+    readonly stats: ClassificationStats;
+    constructor(logits: number[], probabilities: Record<string, number>, stats: ClassificationStats);
+    static fromNapi(napiEntity: NapiClassificationOutput): ClassificationOutput;
+}
+//# sourceMappingURL=classificationOutput.d.ts.map

package/bridging/classificationOutput.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationOutput.d.ts","sourceRoot":"","sources":["../src/bridging/classificationOutput.ts"],"names":[],"mappings":"OAAO,EAAE,oBAAoB,IAAI,wBAAwB,EAAE;OACpD,EAAE,mBAAmB,EAAE;AAE9B,qBAAa,oBAAoB;IAC7B,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;IAC1B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/C,QAAQ,CAAC,KAAK,EAAE,mBAAmB,CAAC;gBAExB,MAAM,EAAE,MAAM,EAAE,EAAE,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,KAAK,EAAE,mBAAmB;IAM/F,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,wBAAwB,GAAG,oBAAoB;CAO9E"}

package/bridging/classificationOutput.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClassificationOutput = void 0;
+const classificationStats_1 = require("./classificationStats.js");
+class ClassificationOutput {
+    constructor(logits, probabilities, stats) {
+        this.logits = logits;
+        this.probabilities = probabilities;
+        this.stats = stats;
+    }
+    static fromNapi(napiEntity) {
+        return new ClassificationOutput(napiEntity.logits, napiEntity.probabilities, classificationStats_1.ClassificationStats.fromNapi(napiEntity.stats));
+    }
+}
+exports.ClassificationOutput = ClassificationOutput;
+//# sourceMappingURL=classificationOutput.js.map

package/bridging/classificationOutput.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationOutput.js","sourceRoot":"","sources":["../src/bridging/classificationOutput.ts"],"names":[],"mappings":";;;AACA,kEAA4D;AAE5D,MAAa,oBAAoB;IAK7B,YAAY,MAAgB,EAAE,aAAqC,EAAE,KAA0B;QAC3F,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACvB,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,UAAoC;QAChD,OAAO,IAAI,oBAAoB,CAC3B,UAAU,CAAC,MAAM,EACjB,UAAU,CAAC,aAAa,EACxB,yCAAmB,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,CACjD,CAAC;IACN,CAAC;CACJ;AAlBD,oDAkBC"}

package/bridging/classificationOutput.mjs

@@ -0,0 +1,12 @@
+import { ClassificationStats } from "./classificationStats.mjs";
+export class ClassificationOutput {
+    constructor(logits, probabilities, stats) {
+        this.logits = logits;
+        this.probabilities = probabilities;
+        this.stats = stats;
+    }
+    static fromNapi(napiEntity) {
+        return new ClassificationOutput(napiEntity.logits, napiEntity.probabilities, ClassificationStats.fromNapi(napiEntity.stats));
+    }
+}
+//# sourceMappingURL=classificationOutput.mjs.map

package/bridging/classificationOutput.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationOutput.mjs","sourceRoot":"","sources":["../src/bridging/classificationOutput.ts"],"names":[],"mappings":"OACO,EAAE,mBAAmB,EAAE;AAE9B,MAAM,OAAO,oBAAoB;IAK7B,YAAY,MAAgB,EAAE,aAAqC,EAAE,KAA0B;QAC3F,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACvB,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,UAAoC;QAChD,OAAO,IAAI,oBAAoB,CAC3B,UAAU,CAAC,MAAM,EACjB,UAAU,CAAC,aAAa,EACxB,mBAAmB,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,CACjD,CAAC;IACN,CAAC;CACJ"}

package/bridging/classificationSession.d.mts

@@ -0,0 +1,9 @@
+import { ClassificationSession as NapiClassificationSession } from "../napi/uzu.mjs";
+import { ClassificationOutput } from "./classificationOutput.mjs";
+import { Input } from "./input.mjs";
+export declare class ClassificationSession {
+    private readonly napiClassificationSession;
+    constructor(napiClassificationSession: NapiClassificationSession);
+    run(input: Input): ClassificationOutput;
+}
+//# sourceMappingURL=classificationSession.d.mts.map

package/bridging/classificationSession.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationSession.d.mts","sourceRoot":"","sources":["../src/bridging/classificationSession.ts"],"names":[],"mappings":"OAAO,EAAE,qBAAqB,IAAI,yBAAyB,EAAE;OACtD,EAAE,oBAAoB,EAAE;OACxB,EAAE,KAAK,EAAE;AAEhB,qBAAa,qBAAqB;IAC9B,OAAO,CAAC,QAAQ,CAAC,yBAAyB,CAA4B;gBAE1D,yBAAyB,EAAE,yBAAyB;IAIhE,GAAG,CAAC,KAAK,EAAE,KAAK,GAAG,oBAAoB;CAK1C"}

package/bridging/classificationSession.d.ts

@@ -0,0 +1,9 @@
+import { ClassificationSession as NapiClassificationSession } from "../napi/uzu.js";
+import { ClassificationOutput } from "./classificationOutput.js";
+import { Input } from "./input.js";
+export declare class ClassificationSession {
+    private readonly napiClassificationSession;
+    constructor(napiClassificationSession: NapiClassificationSession);
+    run(input: Input): ClassificationOutput;
+}
+//# sourceMappingURL=classificationSession.d.ts.map

package/bridging/classificationSession.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationSession.d.ts","sourceRoot":"","sources":["../src/bridging/classificationSession.ts"],"names":[],"mappings":"OAAO,EAAE,qBAAqB,IAAI,yBAAyB,EAAE;OACtD,EAAE,oBAAoB,EAAE;OACxB,EAAE,KAAK,EAAE;AAEhB,qBAAa,qBAAqB;IAC9B,OAAO,CAAC,QAAQ,CAAC,yBAAyB,CAA4B;gBAE1D,yBAAyB,EAAE,yBAAyB;IAIhE,GAAG,CAAC,KAAK,EAAE,KAAK,GAAG,oBAAoB;CAK1C"}

package/bridging/classificationSession.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClassificationSession = void 0;
+const classificationOutput_1 = require("./classificationOutput.js");
+class ClassificationSession {
+    constructor(napiClassificationSession) {
+        this.napiClassificationSession = napiClassificationSession;
+    }
+    run(input) {
+        const napiInput = input.toNapi();
+        const napiOutput = this.napiClassificationSession.run(napiInput);
+        return classificationOutput_1.ClassificationOutput.fromNapi(napiOutput);
+    }
+}
+exports.ClassificationSession = ClassificationSession;
+//# sourceMappingURL=classificationSession.js.map

package/bridging/classificationSession.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationSession.js","sourceRoot":"","sources":["../src/bridging/classificationSession.ts"],"names":[],"mappings":";;;AACA,oEAA8D;AAG9D,MAAa,qBAAqB;IAG9B,YAAY,yBAAoD;QAC5D,IAAI,CAAC,yBAAyB,GAAG,yBAAyB,CAAC;IAC/D,CAAC;IAED,GAAG,CAAC,KAAY;QACZ,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,IAAI,CAAC,yBAAyB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjE,OAAO,2CAAoB,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACrD,CAAC;CACJ;AAZD,sDAYC"}

package/bridging/classificationSession.mjs

@@ -0,0 +1,12 @@
+import { ClassificationOutput } from "./classificationOutput.mjs";
+export class ClassificationSession {
+    constructor(napiClassificationSession) {
+        this.napiClassificationSession = napiClassificationSession;
+    }
+    run(input) {
+        const napiInput = input.toNapi();
+        const napiOutput = this.napiClassificationSession.run(napiInput);
+        return ClassificationOutput.fromNapi(napiOutput);
+    }
+}
+//# sourceMappingURL=classificationSession.mjs.map

package/bridging/classificationSession.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationSession.mjs","sourceRoot":"","sources":["../src/bridging/classificationSession.ts"],"names":[],"mappings":"OACO,EAAE,oBAAoB,EAAE;AAG/B,MAAM,OAAO,qBAAqB;IAG9B,YAAY,yBAAoD;QAC5D,IAAI,CAAC,yBAAyB,GAAG,yBAAyB,CAAC;IAC/D,CAAC;IAED,GAAG,CAAC,KAAY;QACZ,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,IAAI,CAAC,yBAAyB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjE,OAAO,oBAAoB,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACrD,CAAC;CACJ"}

package/bridging/classificationStats.d.mts

@@ -0,0 +1,14 @@
+import { ClassificationStats as NapiClassificationStats } from "../napi/uzu.mjs";
+export declare class ClassificationStats {
+    readonly preprocessingDuration: number;
+    readonly forwardPassDuration: number;
+    readonly postprocessingDuration: number;
+    readonly totalDuration: number;
+    readonly tokensCount: number;
+    readonly tokensPerSecond: number;
+    readonly predictedLabel: string;
+    readonly confidence: number;
+    constructor(preprocessingDuration: number, forwardPassDuration: number, postprocessingDuration: number, totalDuration: number, tokensCount: number, tokensPerSecond: number, predictedLabel: string, confidence: number);
+    static fromNapi(napiEntity: NapiClassificationStats): ClassificationStats;
+}
+//# sourceMappingURL=classificationStats.d.mts.map

package/bridging/classificationStats.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationStats.d.mts","sourceRoot":"","sources":["../src/bridging/classificationStats.ts"],"names":[],"mappings":"OAAO,EAAE,mBAAmB,IAAI,uBAAuB,EAAE;AAEzD,qBAAa,mBAAmB;IAC5B,QAAQ,CAAC,qBAAqB,EAAE,MAAM,CAAC;IACvC,QAAQ,CAAC,mBAAmB,EAAE,MAAM,CAAC;IACrC,QAAQ,CAAC,sBAAsB,EAAE,MAAM,CAAC;IACxC,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC;IACjC,QAAQ,CAAC,cAAc,EAAE,MAAM,CAAC;IAChC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;gBAGxB,qBAAqB,EAAE,MAAM,EAC7B,mBAAmB,EAAE,MAAM,EAC3B,sBAAsB,EAAE,MAAM,EAC9B,aAAa,EAAE,MAAM,EACrB,WAAW,EAAE,MAAM,EACnB,eAAe,EAAE,MAAM,EACvB,cAAc,EAAE,MAAM,EACtB,UAAU,EAAE,MAAM;IAYtB,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,uBAAuB,GAAG,mBAAmB;CAY5E"}

package/bridging/classificationStats.d.ts

@@ -0,0 +1,14 @@
+import { ClassificationStats as NapiClassificationStats } from "../napi/uzu.js";
+export declare class ClassificationStats {
+    readonly preprocessingDuration: number;
+    readonly forwardPassDuration: number;
+    readonly postprocessingDuration: number;
+    readonly totalDuration: number;
+    readonly tokensCount: number;
+    readonly tokensPerSecond: number;
+    readonly predictedLabel: string;
+    readonly confidence: number;
+    constructor(preprocessingDuration: number, forwardPassDuration: number, postprocessingDuration: number, totalDuration: number, tokensCount: number, tokensPerSecond: number, predictedLabel: string, confidence: number);
+    static fromNapi(napiEntity: NapiClassificationStats): ClassificationStats;
+}
+//# sourceMappingURL=classificationStats.d.ts.map

package/bridging/classificationStats.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationStats.d.ts","sourceRoot":"","sources":["../src/bridging/classificationStats.ts"],"names":[],"mappings":"OAAO,EAAE,mBAAmB,IAAI,uBAAuB,EAAE;AAEzD,qBAAa,mBAAmB;IAC5B,QAAQ,CAAC,qBAAqB,EAAE,MAAM,CAAC;IACvC,QAAQ,CAAC,mBAAmB,EAAE,MAAM,CAAC;IACrC,QAAQ,CAAC,sBAAsB,EAAE,MAAM,CAAC;IACxC,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC;IACjC,QAAQ,CAAC,cAAc,EAAE,MAAM,CAAC;IAChC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;gBAGxB,qBAAqB,EAAE,MAAM,EAC7B,mBAAmB,EAAE,MAAM,EAC3B,sBAAsB,EAAE,MAAM,EAC9B,aAAa,EAAE,MAAM,EACrB,WAAW,EAAE,MAAM,EACnB,eAAe,EAAE,MAAM,EACvB,cAAc,EAAE,MAAM,EACtB,UAAU,EAAE,MAAM;IAYtB,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,uBAAuB,GAAG,mBAAmB;CAY5E"}

package/bridging/classificationStats.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClassificationStats = void 0;
+class ClassificationStats {
+    constructor(preprocessingDuration, forwardPassDuration, postprocessingDuration, totalDuration, tokensCount, tokensPerSecond, predictedLabel, confidence) {
+        this.preprocessingDuration = preprocessingDuration;
+        this.forwardPassDuration = forwardPassDuration;
+        this.postprocessingDuration = postprocessingDuration;
+        this.totalDuration = totalDuration;
+        this.tokensCount = tokensCount;
+        this.tokensPerSecond = tokensPerSecond;
+        this.predictedLabel = predictedLabel;
+        this.confidence = confidence;
+    }
+    static fromNapi(napiEntity) {
+        return new ClassificationStats(napiEntity.preprocessingDuration, napiEntity.forwardPassDuration, napiEntity.postprocessingDuration, napiEntity.totalDuration, napiEntity.tokensCount, napiEntity.tokensPerSecond, napiEntity.predictedLabel, napiEntity.confidence);
+    }
+}
+exports.ClassificationStats = ClassificationStats;
+//# sourceMappingURL=classificationStats.js.map

package/bridging/classificationStats.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationStats.js","sourceRoot":"","sources":["../src/bridging/classificationStats.ts"],"names":[],"mappings":";;;AAEA,MAAa,mBAAmB;IAU5B,YACI,qBAA6B,EAC7B,mBAA2B,EAC3B,sBAA8B,EAC9B,aAAqB,EACrB,WAAmB,EACnB,eAAuB,EACvB,cAAsB,EACtB,UAAkB;QAElB,IAAI,CAAC,qBAAqB,GAAG,qBAAqB,CAAC;QACnD,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,UAAmC;QAC/C,OAAO,IAAI,mBAAmB,CAC1B,UAAU,CAAC,qBAAqB,EAChC,UAAU,CAAC,mBAAmB,EAC9B,UAAU,CAAC,sBAAsB,EACjC,UAAU,CAAC,aAAa,EACxB,UAAU,CAAC,WAAW,EACtB,UAAU,CAAC,eAAe,EAC1B,UAAU,CAAC,cAAc,EACzB,UAAU,CAAC,UAAU,CACxB,CAAC;IACN,CAAC;CACJ;AA1CD,kDA0CC"}

package/bridging/classificationStats.mjs

@@ -0,0 +1,16 @@
+export class ClassificationStats {
+    constructor(preprocessingDuration, forwardPassDuration, postprocessingDuration, totalDuration, tokensCount, tokensPerSecond, predictedLabel, confidence) {
+        this.preprocessingDuration = preprocessingDuration;
+        this.forwardPassDuration = forwardPassDuration;
+        this.postprocessingDuration = postprocessingDuration;
+        this.totalDuration = totalDuration;
+        this.tokensCount = tokensCount;
+        this.tokensPerSecond = tokensPerSecond;
+        this.predictedLabel = predictedLabel;
+        this.confidence = confidence;
+    }
+    static fromNapi(napiEntity) {
+        return new ClassificationStats(napiEntity.preprocessingDuration, napiEntity.forwardPassDuration, napiEntity.postprocessingDuration, napiEntity.totalDuration, napiEntity.tokensCount, napiEntity.tokensPerSecond, napiEntity.predictedLabel, napiEntity.confidence);
+    }
+}
+//# sourceMappingURL=classificationStats.mjs.map

package/bridging/classificationStats.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"classificationStats.mjs","sourceRoot":"","sources":["../src/bridging/classificationStats.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,mBAAmB;IAU5B,YACI,qBAA6B,EAC7B,mBAA2B,EAC3B,sBAA8B,EAC9B,aAAqB,EACrB,WAAmB,EACnB,eAAuB,EACvB,cAAsB,EACtB,UAAkB;QAElB,IAAI,CAAC,qBAAqB,GAAG,qBAAqB,CAAC;QACnD,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,UAAmC;QAC/C,OAAO,IAAI,mBAAmB,CAC1B,UAAU,CAAC,qBAAqB,EAChC,UAAU,CAAC,mBAAmB,EAC9B,UAAU,CAAC,sBAAsB,EACjC,UAAU,CAAC,aAAa,EACxB,UAAU,CAAC,WAAW,EACtB,UAAU,CAAC,eAAe,EAC1B,UAAU,CAAC,cAAc,EACzB,UAAU,CAAC,UAAU,CACxB,CAAC;IACN,CAAC;CACJ"}

package/bridging/engine.d.mts

@@ -1,11 +1,13 @@
 import { EngineInteractor } from "../interactors/engineInteractor.mjs";
-import { ChatModel } from "./chatModel.mjs";
 import { ChatSession } from "./chatSession.mjs";
+import { ClassificationSession } from "./classificationSession.mjs";
 import { Config } from "./config.mjs";
 import { DownloadHandle } from "./downloadHandle.mjs";
 import { DownloadProgressUpdate } from "./downloadProgressUpdate.mjs";
 import { DownloadState } from "./downloadState.mjs";
-import { ModelType } from "./modelType.mjs";
+import { Model } from "./model.mjs";
+import { ModelKind } from "./modelKind.mjs";
+import { TextToSpeechSession } from "./textToSpeechSession.mjs";
 export declare class Engine {
     private readonly napiEngine;
     private constructor();
@@ -13,9 +15,15 @@ export declare class Engine {
     static create(apiKey?: string | undefined): EngineInteractor;
     getDownloadState(repoId: string): DownloadState;
     downloadHandle(repoId: string): DownloadHandle;
-    chatModels(types?: ModelType[] | null): Promise<ChatModel[]>;
-    chatModel(repoId: string, types?: ModelType[] | null): Promise<ChatModel>;
-    downloadChatModel(chatModel: ChatModel, progress?: (progressUpdate: DownloadProgressUpdate) => void): Promise<DownloadState>;
-    chatSession(chatModel: ChatModel, config?: Config): ChatSession;
+    downloadModel(model: Model, progress?: (progressUpdate: DownloadProgressUpdate) => void): Promise<DownloadState>;
+    chatModels(kinds?: ModelKind[] | null): Promise<Model[]>;
+    chatModel(repoId: string, kinds?: ModelKind[] | null): Promise<Model>;
+    chatSession(model: Model, config?: Config): ChatSession;
+    classificationModels(): Promise<Model[]>;
+    classificationModel(repoId: string): Promise<Model>;
+    classificationSession(model: Model): ClassificationSession;
+    textToSpeechModels(): Promise<Model[]>;
+    textToSpeechModel(repoId: string): Promise<Model>;
+    textToSpeechSession(model: Model): TextToSpeechSession;
 }
 //# sourceMappingURL=engine.d.mts.map

package/bridging/engine.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"engine.d.mts","sourceRoot":"","sources":["../src/bridging/engine.ts"],"names":[],"mappings":"OACO,EAAE,gBAAgB,EAAE;OAGpB,EAAE,SAAS,EAAE;OACb,EAAE,WAAW,EAAE;OACf,EAAE,MAAM,EAAE;OACV,EAAE,cAAc,EAAE;OAClB,EAAE,sBAAsB,EAAE;OAC1B,EAAiB,aAAa,EAAE;OAEhC,EAAE,SAAS,EAA4B;AAE9C,qBAAa,MAAM;IACf,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IAExC,OAAO;WAIM,IAAI,CAAC,MAAM,GAAE,MAAM,GAAG,SAAqB,GAAG,OAAO,CAAC,MAAM,CAAC;IAqB1E,MAAM,CAAC,MAAM,CAAC,MAAM,GAAE,MAAM,GAAG,SAAqB,GAAG,gBAAgB;IAOvE,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa;IAK/C,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,cAAc;IAIxC,UAAU,CAAC,KAAK,GAAE,SAAS,EAAE,GAAG,IAAW,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAQlE,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,SAAS,EAAE,GAAG,IAAW,GAAG,OAAO,CAAC,SAAS,CAAC;IAM/E,iBAAiB,CACnB,SAAS,EAAE,SAAS,EACpB,QAAQ,GAAE,CAAC,cAAc,EAAE,sBAAsB,KAAK,IAAe,GACtE,OAAO,CAAC,aAAa,CAAC;IAyBzB,WAAW,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,GAAE,MAAyB,GAAG,WAAW;CAKpF"}
+{"version":3,"file":"engine.d.mts","sourceRoot":"","sources":["../src/bridging/engine.ts"],"names":[],"mappings":"OACO,EAAE,gBAAgB,EAAE;OAGpB,EAAE,WAAW,EAAE;OACf,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;OACV,EAAE,cAAc,EAAE;OAClB,EAAE,sBAAsB,EAAE;OAC1B,EAAiB,aAAa,EAAE;OAEhC,EAAE,KAAK,EAAE;OACT,EAAE,SAAS,EAA4B;OACvC,EAAE,mBAAmB,EAAE;AAE9B,qBAAa,MAAM;IACf,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IAExC,OAAO;WAIM,IAAI,CAAC,MAAM,GAAE,MAAM,GAAG,SAAqB,GAAG,OAAO,CAAC,MAAM,CAAC;IAqB1E,MAAM,CAAC,MAAM,CAAC,MAAM,GAAE,MAAM,GAAG,SAAqB,GAAG,gBAAgB;IAOvE,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa;IAK/C,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,cAAc;IAIxC,aAAa,CACf,KAAK,EAAE,KAAK,EACZ,QAAQ,GAAE,CAAC,cAAc,EAAE,sBAAsB,KAAK,IAAe,GACtE,OAAO,CAAC,aAAa,CAAC;IAyBnB,UAAU,CAAC,KAAK,GAAE,SAAS,EAAE,GAAG,IAAW,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC;IAQ9D,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,SAAS,EAAE,GAAG,IAAW,GAAG,OAAO,CAAC,KAAK,CAAC;IAMjF,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,GAAE,MAAyB,GAAG,WAAW;IAMnE,oBAAoB,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;IAMxC,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;IAMzD,qBAAqB,CAAC,KAAK,EAAE,KAAK,GAAG,qBAAqB;IAKpD,kBAAkB,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;IAMtC,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;IAMvD,mBAAmB,CAAC,KAAK,EAAE,KAAK,GAAG,mBAAmB;CAIzD"}

package/bridging/engine.d.ts

@@ -1,11 +1,13 @@
 import { EngineInteractor } from "../interactors/engineInteractor.js";
-import { ChatModel } from "./chatModel.js";
 import { ChatSession } from "./chatSession.js";
+import { ClassificationSession } from "./classificationSession.js";
 import { Config } from "./config.js";
 import { DownloadHandle } from "./downloadHandle.js";
 import { DownloadProgressUpdate } from "./downloadProgressUpdate.js";
 import { DownloadState } from "./downloadState.js";
-import { ModelType } from "./modelType.js";
+import { Model } from "./model.js";
+import { ModelKind } from "./modelKind.js";
+import { TextToSpeechSession } from "./textToSpeechSession.js";
 export declare class Engine {
     private readonly napiEngine;
     private constructor();
@@ -13,9 +15,15 @@ export declare class Engine {
     static create(apiKey?: string | undefined): EngineInteractor;
     getDownloadState(repoId: string): DownloadState;
     downloadHandle(repoId: string): DownloadHandle;
-    chatModels(types?: ModelType[] | null): Promise<ChatModel[]>;
-    chatModel(repoId: string, types?: ModelType[] | null): Promise<ChatModel>;
-    downloadChatModel(chatModel: ChatModel, progress?: (progressUpdate: DownloadProgressUpdate) => void): Promise<DownloadState>;
-    chatSession(chatModel: ChatModel, config?: Config): ChatSession;
+    downloadModel(model: Model, progress?: (progressUpdate: DownloadProgressUpdate) => void): Promise<DownloadState>;
+    chatModels(kinds?: ModelKind[] | null): Promise<Model[]>;
+    chatModel(repoId: string, kinds?: ModelKind[] | null): Promise<Model>;
+    chatSession(model: Model, config?: Config): ChatSession;
+    classificationModels(): Promise<Model[]>;
+    classificationModel(repoId: string): Promise<Model>;
+    classificationSession(model: Model): ClassificationSession;
+    textToSpeechModels(): Promise<Model[]>;
+    textToSpeechModel(repoId: string): Promise<Model>;
+    textToSpeechSession(model: Model): TextToSpeechSession;
 }
 //# sourceMappingURL=engine.d.ts.map

package/bridging/engine.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"engine.d.ts","sourceRoot":"","sources":["../src/bridging/engine.ts"],"names":[],"mappings":"OACO,EAAE,gBAAgB,EAAE;OAGpB,EAAE,SAAS,EAAE;OACb,EAAE,WAAW,EAAE;OACf,EAAE,MAAM,EAAE;OACV,EAAE,cAAc,EAAE;OAClB,EAAE,sBAAsB,EAAE;OAC1B,EAAiB,aAAa,EAAE;OAEhC,EAAE,SAAS,EAA4B;AAE9C,qBAAa,MAAM;IACf,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IAExC,OAAO;WAIM,IAAI,CAAC,MAAM,GAAE,MAAM,GAAG,SAAqB,GAAG,OAAO,CAAC,MAAM,CAAC;IAqB1E,MAAM,CAAC,MAAM,CAAC,MAAM,GAAE,MAAM,GAAG,SAAqB,GAAG,gBAAgB;IAOvE,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa;IAK/C,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,cAAc;IAIxC,UAAU,CAAC,KAAK,GAAE,SAAS,EAAE,GAAG,IAAW,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAQlE,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,SAAS,EAAE,GAAG,IAAW,GAAG,OAAO,CAAC,SAAS,CAAC;IAM/E,iBAAiB,CACnB,SAAS,EAAE,SAAS,EACpB,QAAQ,GAAE,CAAC,cAAc,EAAE,sBAAsB,KAAK,IAAe,GACtE,OAAO,CAAC,aAAa,CAAC;IAyBzB,WAAW,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,GAAE,MAAyB,GAAG,WAAW;CAKpF"}
+{"version":3,"file":"engine.d.ts","sourceRoot":"","sources":["../src/bridging/engine.ts"],"names":[],"mappings":"OACO,EAAE,gBAAgB,EAAE;OAGpB,EAAE,WAAW,EAAE;OACf,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;OACV,EAAE,cAAc,EAAE;OAClB,EAAE,sBAAsB,EAAE;OAC1B,EAAiB,aAAa,EAAE;OAEhC,EAAE,KAAK,EAAE;OACT,EAAE,SAAS,EAA4B;OACvC,EAAE,mBAAmB,EAAE;AAE9B,qBAAa,MAAM;IACf,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IAExC,OAAO;WAIM,IAAI,CAAC,MAAM,GAAE,MAAM,GAAG,SAAqB,GAAG,OAAO,CAAC,MAAM,CAAC;IAqB1E,MAAM,CAAC,MAAM,CAAC,MAAM,GAAE,MAAM,GAAG,SAAqB,GAAG,gBAAgB;IAOvE,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa;IAK/C,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,cAAc;IAIxC,aAAa,CACf,KAAK,EAAE,KAAK,EACZ,QAAQ,GAAE,CAAC,cAAc,EAAE,sBAAsB,KAAK,IAAe,GACtE,OAAO,CAAC,aAAa,CAAC;IAyBnB,UAAU,CAAC,KAAK,GAAE,SAAS,EAAE,GAAG,IAAW,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC;IAQ9D,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,SAAS,EAAE,GAAG,IAAW,GAAG,OAAO,CAAC,KAAK,CAAC;IAMjF,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,GAAE,MAAyB,GAAG,WAAW;IAMnE,oBAAoB,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;IAMxC,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;IAMzD,qBAAqB,CAAC,KAAK,EAAE,KAAK,GAAG,qBAAqB;IAKpD,kBAAkB,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;IAMtC,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;IAMvD,mBAAmB,CAAC,KAAK,EAAE,KAAK,GAAG,mBAAmB;CAIzD"}

package/bridging/engine.js

@@ -5,13 +5,15 @@ const error_1 = require("../error.js");
 const engineInteractor_1 = require("../interactors/engineInteractor.js");
 const uzu_1 = require("../napi/uzu.js");
 const env_1 = require("../utilities/env.js");
-const chatModel_1 = require("./chatModel.js");
 const chatSession_1 = require("./chatSession.js");
+const classificationSession_1 = require("./classificationSession.js");
 const config_1 = require("./config.js");
 const downloadHandle_1 = require("./downloadHandle.js");
 const downloadState_1 = require("./downloadState.js");
 const licenseStatus_1 = require("./licenseStatus.js");
-const modelType_1 = require("./modelType.js");
+const model_1 = require("./model.js");
+const modelKind_1 = require("./modelKind.js");
+const textToSpeechSession_1 = require("./textToSpeechSession.js");
 class Engine {
     constructor(napiEngine) {
         this.napiEngine = napiEngine;
@@ -43,26 +45,14 @@ class Engine {
     downloadHandle(repoId) {
         return new downloadHandle_1.DownloadHandle(this.napiEngine.createModelDownloadHandle(repoId));
     }
-    async chatModels(types = null) {
-        let typesToFilter = types ?? [modelType_1.ModelType.Local, modelType_1.ModelType.Cloud];
-        let napiModelTypes = typesToFilter.map(modelType_1.modelTypeToNapiModelType);
-        let napiChatModels = await this.napiEngine.getChatModels(napiModelTypes);
-        let results = napiChatModels.map(chatModel_1.ChatModel.fromNapiChatModel);
-        return results;
-    }
-    async chatModel(repoId, types = null) {
-        const chatModels = await this.chatModels(types);
-        const chatModel = chatModels.find((model) => model.repoId === repoId);
-        return (0, error_1.unwrapOrThrow)(chatModel, error_1.EngineErrorCode.ModelNotFound);
-    }
-    async downloadChatModel(chatModel, progress = () => { }) {
-        switch (chatModel.type) {
-            case modelType_1.ModelType.Local:
+    async downloadModel(model, progress = () => { }) {
+        switch (model.kind) {
+            case modelKind_1.ModelKind.Local:
                 break;
-            case modelType_1.ModelType.Cloud:
-                throw new error_1.EngineError(error_1.EngineErrorCode.UnexpectedModelType);
+            case modelKind_1.ModelKind.Cloud:
+                throw new error_1.EngineError(error_1.EngineErrorCode.UnexpectedModelKind);
         }
-        const downloadHandle = this.downloadHandle(chatModel.repoId);
+        const downloadHandle = this.downloadHandle(model.repoId);
         const state = await downloadHandle.state();
         switch (state.phase) {
             case downloadState_1.DownloadPhase.Downloaded:
@@ -77,10 +67,50 @@ class Engine {
         const finalState = await downloadHandle.state();
         return finalState;
     }
-    chatSession(chatModel, config = config_1.Config.default()) {
-        const napiChatModel = chatModel.toNapi();
+    async chatModels(kinds = null) {
+        let kindsToFilter = kinds ?? [modelKind_1.ModelKind.Local, modelKind_1.ModelKind.Cloud];
+        let napiModelKinds = kindsToFilter.map(modelKind_1.modelKindToNapiModelKind);
+        let napiChatModels = await this.napiEngine.getChatModels(napiModelKinds);
+        let results = napiChatModels.map(model_1.Model.fromNapiModel);
+        return results;
+    }
+    async chatModel(repoId, kinds = null) {
+        const chatModels = await this.chatModels(kinds);
+        const chatModel = chatModels.find((model) => model.repoId === repoId);
+        return (0, error_1.unwrapOrThrow)(chatModel, error_1.EngineErrorCode.ModelNotFound);
+    }
+    chatSession(model, config = config_1.Config.default()) {
+        const napiModel = model.toNapi();
         const napiConfig = config.toNapi();
-        return new chatSession_1.ChatSession(this.napiEngine.createChatSession(napiChatModel, napiConfig));
+        return new chatSession_1.ChatSession(this.napiEngine.createChatSession(napiModel, napiConfig));
+    }
+    async classificationModels() {
+        const napiClassificationModels = await this.napiEngine.getClassificationModels();
+        const results = napiClassificationModels.map(model_1.Model.fromNapiModel);
+        return results;
+    }
+    async classificationModel(repoId) {
+        const classificationModels = await this.classificationModels();
+        const classificationModel = classificationModels.find((model) => model.repoId === repoId);
+        return (0, error_1.unwrapOrThrow)(classificationModel, error_1.EngineErrorCode.ModelNotFound);
+    }
+    classificationSession(model) {
+        const napiModel = model.toNapi();
+        return new classificationSession_1.ClassificationSession(this.napiEngine.createClassificationSession(napiModel));
+    }
+    async textToSpeechModels() {
+        const napiTextToSpeechModels = await this.napiEngine.getTextToSpeechModels();
+        const results = napiTextToSpeechModels.map(model_1.Model.fromNapiModel);
+        return results;
+    }
+    async textToSpeechModel(repoId) {
+        const textToSpeechModels = await this.textToSpeechModels();
+        const textToSpeechModel = textToSpeechModels.find((model) => model.repoId === repoId);
+        return (0, error_1.unwrapOrThrow)(textToSpeechModel, error_1.EngineErrorCode.ModelNotFound);
+    }
+    textToSpeechSession(model) {
+        const napiModel = model.toNapi();
+        return new textToSpeechSession_1.TextToSpeechSession(this.napiEngine.createTextToSpeechSession(napiModel));
     }
 }
 exports.Engine = Engine;

package/bridging/engine.js.map

@@ -1 +1 @@
-{"version":3,"file":"engine.js","sourceRoot":"","sources":["../src/bridging/engine.ts"],"names":[],"mappings":";;;AAAA,uCAAuE;AACvE,yEAAmE;AACnE,wCAAmD;AACnD,6CAA2C;AAC3C,8CAAwC;AACxC,kDAA4C;AAC5C,wCAAkC;AAClC,wDAAkD;AAElD,sDAA+D;AAC/D,sDAAoF;AACpF,8CAAkE;AAElE,MAAa,MAAM;IAGf,YAAoB,UAAsB;QACtC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,SAA6B,SAAS;QACpD,MAAM,cAAc,GAAG,IAAA,qBAAa,EAChC,MAAM,IAAI,IAAA,aAAO,EAAC,eAAe,CAAC,EAClC,uBAAe,CAAC,cAAc,CACjC,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,YAAU,EAAE,CAAC;QAEpC,MAAM,iBAAiB,GAAG,MAAM,UAAU,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QACpE,MAAM,aAAa,GAAG,IAAA,kDAAkC,EAAC,iBAAiB,CAAC,CAAC;QAC5E,QAAQ,aAAa,EAAE,CAAC;YACpB,KAAK,6BAAa,CAAC,SAAS,CAAC;YAC7B,KAAK,6BAAa,CAAC,iBAAiB;gBAChC,MAAM;YACV;gBACI,MAAM,IAAI,mBAAW,CAAC,uBAAe,CAAC,mBAAmB,CAAC,CAAC;QACnE,CAAC;QAED,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,SAA6B,SAAS;QAChD,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,OAAO,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,mCAAgB,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC;IAED,gBAAgB,CAAC,MAAc;QAC3B,IAAI,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAClE,OAAO,6BAAa,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IACjD,CAAC;IAED,cAAc,CAAC,MAAc;QACzB,OAAO,IAAI,+BAAc,CAAC,IAAI,CAAC,UAAU,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC,CAAC;IACjF,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAA4B,IAAI;QAC7C,IAAI,aAAa,GAAgB,KAAK,IAAI,CAAC,qBAAS,CAAC,KAAK,EAAE,qBAAS,CAAC,KAAK,CAAC,CAAC;QAC7E,IAAI,cAAc,GAAG,aAAa,CAAC,GAAG,CAAC,oCAAwB,CAAC,CAAC;QACjE,IAAI,cAAc,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC;QACzE,IAAI,OAAO,GAAgB,cAAc,CAAC,GAAG,CAAC,qBAAS,CAAC,iBAAiB,CAAC,CAAC;QAC3E,OAAO,OAAO,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,QAA4B,IAAI;QAC5D,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;QACtE,OAAO,IAAA,qBAAa,EAAC,SAAS,EAAE,uBAAe,CAAC,aAAa,CAAC,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,iBAAiB,CACnB,SAAoB,EACpB,WAA6D,GAAG,EAAE,GAAE,CAAC;QAErE,QAAQ,SAAS,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,qBAAS,CAAC,KAAK;gBAChB,MAAM;YACV,KAAK,qBAAS,CAAC,KAAK;gBAChB,MAAM,IAAI,mBAAW,CAAC,uBAAe,CAAC,mBAAmB,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC7D,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,CAAC;QAC3C,QAAQ,KAAK,CAAC,KAAK,EAAE,CAAC;YAClB,KAAK,6BAAa,CAAC,UAAU;gBACzB,MAAM;YACV;gBACI,MAAM,cAAc,CAAC,QAAQ,EAAE,CAAC;gBAChC,IAAI,KAAK,EAAE,MAAM,cAAc,IAAI,cAAc,CAAC,cAAc,EAAE,EAAE,CAAC;oBACjE,QAAQ,CAAC,cAAc,CAAC,CAAC;gBAC7B,CAAC;gBACD,MAAM;QACd,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,CAAC;QAChD,OAAO,UAAU,CAAC;IACtB,CAAC;IAED,WAAW,CAAC,SAAoB,EAAE,SAAiB,eAAM,CAAC,OAAO,EAAE;QAC/D,MAAM,aAAa,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC;QACzC,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QACnC,OAAO,IAAI,yBAAW,CAAC,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC,CAAC;IACzF,CAAC;CACJ;AA3FD,wBA2FC"}
+{"version":3,"file":"engine.js","sourceRoot":"","sources":["../src/bridging/engine.ts"],"names":[],"mappings":";;;AAAA,uCAAuE;AACvE,yEAAmE;AACnE,wCAAmD;AACnD,6CAA2C;AAC3C,kDAA4C;AAC5C,sEAAgE;AAChE,wCAAkC;AAClC,wDAAkD;AAElD,sDAA+D;AAC/D,sDAAoF;AACpF,sCAAgC;AAChC,8CAAkE;AAClE,kEAA4D;AAE5D,MAAa,MAAM;IAGf,YAAoB,UAAsB;QACtC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,SAA6B,SAAS;QACpD,MAAM,cAAc,GAAG,IAAA,qBAAa,EAChC,MAAM,IAAI,IAAA,aAAO,EAAC,eAAe,CAAC,EAClC,uBAAe,CAAC,cAAc,CACjC,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,YAAU,EAAE,CAAC;QAEpC,MAAM,iBAAiB,GAAG,MAAM,UAAU,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QACpE,MAAM,aAAa,GAAG,IAAA,kDAAkC,EAAC,iBAAiB,CAAC,CAAC;QAC5E,QAAQ,aAAa,EAAE,CAAC;YACpB,KAAK,6BAAa,CAAC,SAAS,CAAC;YAC7B,KAAK,6BAAa,CAAC,iBAAiB;gBAChC,MAAM;YACV;gBACI,MAAM,IAAI,mBAAW,CAAC,uBAAe,CAAC,mBAAmB,CAAC,CAAC;QACnE,CAAC;QAED,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,SAA6B,SAAS;QAChD,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,OAAO,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,mCAAgB,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC;IAED,gBAAgB,CAAC,MAAc;QAC3B,IAAI,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAClE,OAAO,6BAAa,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IACjD,CAAC;IAED,cAAc,CAAC,MAAc;QACzB,OAAO,IAAI,+BAAc,CAAC,IAAI,CAAC,UAAU,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC,CAAC;IACjF,CAAC;IAED,KAAK,CAAC,aAAa,CACf,KAAY,EACZ,WAA6D,GAAG,EAAE,GAAE,CAAC;QAErE,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACjB,KAAK,qBAAS,CAAC,KAAK;gBAChB,MAAM;YACV,KAAK,qBAAS,CAAC,KAAK;gBAChB,MAAM,IAAI,mBAAW,CAAC,uBAAe,CAAC,mBAAmB,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,CAAC;QAC3C,QAAQ,KAAK,CAAC,KAAK,EAAE,CAAC;YAClB,KAAK,6BAAa,CAAC,UAAU;gBACzB,MAAM;YACV;gBACI,MAAM,cAAc,CAAC,QAAQ,EAAE,CAAC;gBAChC,IAAI,KAAK,EAAE,MAAM,cAAc,IAAI,cAAc,CAAC,cAAc,EAAE,EAAE,CAAC;oBACjE,QAAQ,CAAC,cAAc,CAAC,CAAC;gBAC7B,CAAC;gBACD,MAAM;QACd,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,CAAC;QAChD,OAAO,UAAU,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAA4B,IAAI;QAC7C,IAAI,aAAa,GAAgB,KAAK,IAAI,CAAC,qBAAS,CAAC,KAAK,EAAE,qBAAS,CAAC,KAAK,CAAC,CAAC;QAC7E,IAAI,cAAc,GAAG,aAAa,CAAC,GAAG,CAAC,oCAAwB,CAAC,CAAC;QACjE,IAAI,cAAc,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC;QACzE,IAAI,OAAO,GAAY,cAAc,CAAC,GAAG,CAAC,aAAK,CAAC,aAAa,CAAC,CAAC;QAC/D,OAAO,OAAO,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,QAA4B,IAAI;QAC5D,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;QACtE,OAAO,IAAA,qBAAa,EAAC,SAAS,EAAE,uBAAe,CAAC,aAAa,CAAC,CAAC;IACnE,CAAC;IAED,WAAW,CAAC,KAAY,EAAE,SAAiB,eAAM,CAAC,OAAO,EAAE;QACvD,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QACnC,OAAO,IAAI,yBAAW,CAAC,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC;IACrF,CAAC;IAED,KAAK,CAAC,oBAAoB;QACtB,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,uBAAuB,EAAE,CAAC;QACjF,MAAM,OAAO,GAAY,wBAAwB,CAAC,GAAG,CAAC,aAAK,CAAC,aAAa,CAAC,CAAC;QAC3E,OAAO,OAAO,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,MAAc;QACpC,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC/D,MAAM,mBAAmB,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;QAC1F,OAAO,IAAA,qBAAa,EAAC,mBAAmB,EAAE,uBAAe,CAAC,aAAa,CAAC,CAAC;IAC7E,CAAC;IAED,qBAAqB,CAAC,KAAY;QAC9B,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,OAAO,IAAI,6CAAqB,CAAC,IAAI,CAAC,UAAU,CAAC,2BAA2B,CAAC,SAAS,CAAC,CAAC,CAAC;IAC7F,CAAC;IAED,KAAK,CAAC,kBAAkB;QACpB,MAAM,sBAAsB,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,qBAAqB,EAAE,CAAC;QAC7E,MAAM,OAAO,GAAY,sBAAsB,CAAC,GAAG,CAAC,aAAK,CAAC,aAAa,CAAC,CAAC;QACzE,OAAO,OAAO,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,MAAc;QAClC,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC3D,MAAM,iBAAiB,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;QACtF,OAAO,IAAA,qBAAa,EAAC,iBAAiB,EAAE,uBAAe,CAAC,aAAa,CAAC,CAAC;IAC3E,CAAC;IAED,mBAAmB,CAAC,KAAY;QAC5B,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,OAAO,IAAI,yCAAmB,CAAC,IAAI,CAAC,UAAU,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC;IACzF,CAAC;CACJ;AA7HD,wBA6HC"}