@trymirai/uzu 0.2.20 → 0.3.0

package/interactors/chatModelsInteractor.d.ts

@@ -1,12 +1,12 @@
-import { ChatModel } from "../bridging/chatModel.js";
+import { Model } from "../bridging/model.js";
 import { ChatModelInteractor } from "./chatModelInteractor.js";
 import { EngineInteractor } from "./engineInteractor.js";
 import { Interactor, InteractorEntity } from "./interactor.js";
-export declare class ChatModelsInteractor implements Interactor<ChatModel[]> {
+export declare class ChatModelsInteractor implements Interactor<Model[]> {
     readonly engineInteractor: EngineInteractor;
-    readonly entity: InteractorEntity<ChatModel[]>;
-    constructor(engineInteractor: EngineInteractor, models: InteractorEntity<ChatModel[]>);
-    finalize(): Promise<ChatModel[]>;
+    readonly entity: InteractorEntity<Model[]>;
+    constructor(engineInteractor: EngineInteractor, models: InteractorEntity<Model[]>);
+    finalize(): Promise<Model[]>;
     get(repoId: string): ChatModelInteractor;
 }
 //# sourceMappingURL=chatModelsInteractor.d.ts.map

package/interactors/chatModelsInteractor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chatModelsInteractor.d.ts","sourceRoot":"","sources":["../src/interactors/chatModelsInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,SAAS,EAAE;OAGb,EAAE,mBAAmB,EAAE;OACvB,EAAE,gBAAgB,EAAE;OACpB,EAAE,UAAU,EAAE,gBAAgB,EAAE;AAEvC,qBAAa,oBAAqB,YAAW,UAAU,CAAC,SAAS,EAAE,CAAC;IAChE,QAAQ,CAAC,gBAAgB,EAAE,gBAAgB,CAAC;IAC5C,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,SAAS,EAAE,CAAC,CAAC;gBAEnC,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,EAAE,gBAAgB,CAAC,SAAS,EAAE,CAAC;IAK/E,QAAQ,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;IAItC,GAAG,CAAC,MAAM,EAAE,MAAM,GAAG,mBAAmB;CAQ3C"}
+{"version":3,"file":"chatModelsInteractor.d.ts","sourceRoot":"","sources":["../src/interactors/chatModelsInteractor.ts"],"names":[],"mappings":"OACO,EAAE,KAAK,EAAE;OAET,EAAE,mBAAmB,EAAE;OACvB,EAAE,gBAAgB,EAAE;OACpB,EAAE,UAAU,EAAE,gBAAgB,EAAE;AAEvC,qBAAa,oBAAqB,YAAW,UAAU,CAAC,KAAK,EAAE,CAAC;IAC5D,QAAQ,CAAC,gBAAgB,EAAE,gBAAgB,CAAC;IAC5C,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC;gBAE/B,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,EAAE,gBAAgB,CAAC,KAAK,EAAE,CAAC;IAK3E,QAAQ,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;IAIlC,GAAG,CAAC,MAAM,EAAE,MAAM,GAAG,mBAAmB;CAQ3C"}

package/interactors/chatModelsInteractor.js.map

@@ -1 +1 @@
-{"version":3,"file":"chatModelsInteractor.js","sourceRoot":"","sources":["../src/interactors/chatModelsInteractor.ts"],"names":[],"mappings":";;;AACA,kDAA4C;AAC5C,uCAA0D;AAC1D,kEAA4D;AAI5D,MAAa,oBAAoB;IAI7B,YAAY,gBAAkC,EAAE,MAAqC;QACjF,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,GAAG,CAAC,MAAc;QACd,MAAM,YAAY,GAAG,CAAC,KAAK,IAAI,EAAE;YAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;YAC9D,OAAO,IAAA,qBAAa,EAAC,KAAK,EAAE,uBAAe,CAAC,aAAa,CAAC,CAAC;QAC/D,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,yCAAmB,CAAC,IAAI,EAAE,YAAY,EAAE,eAAM,CAAC,OAAO,EAAE,CAAC,CAAC;IACzE,CAAC;CACJ;AArBD,oDAqBC"}
+{"version":3,"file":"chatModelsInteractor.js","sourceRoot":"","sources":["../src/interactors/chatModelsInteractor.ts"],"names":[],"mappings":";;;AAAA,kDAA4C;AAE5C,uCAA0D;AAC1D,kEAA4D;AAI5D,MAAa,oBAAoB;IAI7B,YAAY,gBAAkC,EAAE,MAAiC;QAC7E,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,GAAG,CAAC,MAAc;QACd,MAAM,YAAY,GAAG,CAAC,KAAK,IAAI,EAAE;YAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;YAC9D,OAAO,IAAA,qBAAa,EAAC,KAAK,EAAE,uBAAe,CAAC,aAAa,CAAC,CAAC;QAC/D,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,yCAAmB,CAAC,IAAI,EAAE,YAAY,EAAE,eAAM,CAAC,OAAO,EAAE,CAAC,CAAC;IACzE,CAAC;CACJ;AArBD,oDAqBC"}

package/interactors/chatModelsInteractor.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"chatModelsInteractor.mjs","sourceRoot":"","sources":["../src/interactors/chatModelsInteractor.ts"],"names":[],"mappings":"OACO,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE,aAAa,EAAE;OAClC,EAAE,mBAAmB,EAAE;AAI9B,MAAM,OAAO,oBAAoB;IAI7B,YAAY,gBAAkC,EAAE,MAAqC;QACjF,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,GAAG,CAAC,MAAc;QACd,MAAM,YAAY,GAAG,CAAC,KAAK,IAAI,EAAE;YAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;YAC9D,OAAO,aAAa,CAAC,KAAK,EAAE,eAAe,CAAC,aAAa,CAAC,CAAC;QAC/D,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,mBAAmB,CAAC,IAAI,EAAE,YAAY,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;IACzE,CAAC;CACJ"}
+{"version":3,"file":"chatModelsInteractor.mjs","sourceRoot":"","sources":["../src/interactors/chatModelsInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,EAAE;OAEV,EAAE,eAAe,EAAE,aAAa,EAAE;OAClC,EAAE,mBAAmB,EAAE;AAI9B,MAAM,OAAO,oBAAoB;IAI7B,YAAY,gBAAkC,EAAE,MAAiC;QAC7E,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,GAAG,CAAC,MAAc;QACd,MAAM,YAAY,GAAG,CAAC,KAAK,IAAI,EAAE;YAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;YAC9D,OAAO,aAAa,CAAC,KAAK,EAAE,eAAe,CAAC,aAAa,CAAC,CAAC;QAC/D,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,mBAAmB,CAAC,IAAI,EAAE,YAAY,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;IACzE,CAAC;CACJ"}

package/interactors/engineInteractor.d.mts

@@ -1,14 +1,14 @@
 import { Engine } from "../bridging/engine.mjs";
-import { ModelType } from "../bridging/modelType.mjs";
+import { ModelKind } from "../bridging/modelKind.mjs";
 import { ChatModelInteractor } from "./chatModelInteractor.mjs";
 import { ChatModelsInteractor } from "./chatModelsInteractor.mjs";
 import { Interactor, InteractorEntity } from "./interactor.mjs";
 export declare class EngineInteractor implements Interactor<Engine> {
     readonly entity: InteractorEntity<Engine>;
-    readonly modelTypeFilter: ModelType[] | null;
-    constructor(engine: InteractorEntity<Engine>, modelTypeFilter?: ModelType[] | null);
+    readonly modelKindFilter: ModelKind[] | null;
+    constructor(engine: InteractorEntity<Engine>, modelKindFilter?: ModelKind[] | null);
     finalize(): Promise<Engine>;
-    filterTypes(types: ModelType[] | null): EngineInteractor;
+    filterKinds(kinds: ModelKind[] | null): EngineInteractor;
     chatModels(): ChatModelsInteractor;
     chatModel(repoId: string): ChatModelInteractor;
 }

package/interactors/engineInteractor.d.ts

@@ -1,14 +1,14 @@
 import { Engine } from "../bridging/engine.js";
-import { ModelType } from "../bridging/modelType.js";
+import { ModelKind } from "../bridging/modelKind.js";
 import { ChatModelInteractor } from "./chatModelInteractor.js";
 import { ChatModelsInteractor } from "./chatModelsInteractor.js";
 import { Interactor, InteractorEntity } from "./interactor.js";
 export declare class EngineInteractor implements Interactor<Engine> {
     readonly entity: InteractorEntity<Engine>;
-    readonly modelTypeFilter: ModelType[] | null;
-    constructor(engine: InteractorEntity<Engine>, modelTypeFilter?: ModelType[] | null);
+    readonly modelKindFilter: ModelKind[] | null;
+    constructor(engine: InteractorEntity<Engine>, modelKindFilter?: ModelKind[] | null);
     finalize(): Promise<Engine>;
-    filterTypes(types: ModelType[] | null): EngineInteractor;
+    filterKinds(kinds: ModelKind[] | null): EngineInteractor;
     chatModels(): ChatModelsInteractor;
     chatModel(repoId: string): ChatModelInteractor;
 }

package/interactors/engineInteractor.js

@@ -3,21 +3,21 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.EngineInteractor = void 0;
 const chatModelsInteractor_1 = require("./chatModelsInteractor.js");
 class EngineInteractor {
-    constructor(engine, modelTypeFilter = null) {
+    constructor(engine, modelKindFilter = null) {
         this.entity = engine;
-        this.modelTypeFilter = modelTypeFilter;
+        this.modelKindFilter = modelKindFilter;
     }
     async finalize() {
         return await this.entity;
     }
     /* Models */
-    filterTypes(types) {
-        return new EngineInteractor(this.entity, types);
+    filterKinds(kinds) {
+        return new EngineInteractor(this.entity, kinds);
     }
     chatModels() {
         const modelsPromise = (async () => {
             const engine = await this.finalize();
-            return await engine.chatModels(this.modelTypeFilter);
+            return await engine.chatModels(this.modelKindFilter);
         })();
         return new chatModelsInteractor_1.ChatModelsInteractor(this, modelsPromise);
     }

package/interactors/engineInteractor.mjs

@@ -1,20 +1,20 @@
 import { ChatModelsInteractor } from "./chatModelsInteractor.mjs";
 export class EngineInteractor {
-    constructor(engine, modelTypeFilter = null) {
+    constructor(engine, modelKindFilter = null) {
         this.entity = engine;
-        this.modelTypeFilter = modelTypeFilter;
+        this.modelKindFilter = modelKindFilter;
     }
     async finalize() {
         return await this.entity;
     }
     /* Models */
-    filterTypes(types) {
-        return new EngineInteractor(this.entity, types);
+    filterKinds(kinds) {
+        return new EngineInteractor(this.entity, kinds);
     }
     chatModels() {
         const modelsPromise = (async () => {
             const engine = await this.finalize();
-            return await engine.chatModels(this.modelTypeFilter);
+            return await engine.chatModels(this.modelKindFilter);
         })();
         return new ChatModelsInteractor(this, modelsPromise);
     }

package/napi/uzu.d.ts

@@ -5,17 +5,25 @@ export declare class ChatSession {
   reset(): void
 }
 
+export declare class ClassificationSession {
+  run(input: Input): ClassificationOutput
+}
+
 export declare class Engine {
   activate(apiKey: string): Promise<LicenseStatus>
-  getChatModels(types: Array<ModelType>): Promise<Array<ChatModel>>
+  getChatModels(kinds: Array<ModelKind>): Promise<Array<Model>>
+  getTextToSpeechModels(): Promise<Array<Model>>
+  getClassificationModels(): Promise<Array<Model>>
   getModelDownloadState(repoId: string): ModelDownloadState
   createModelDownloadHandle(repoId: string): ModelDownloadHandle
   benchmark(task: BenchmarksTask, prefillStepSize?: number | undefined | null): Promise<Array<BenchmarksResult>>
   constructor()
-  createChatSession(model: ChatModel, config: Config): ChatSession
+  createChatSession(model: Model, config: Config): ChatSession
+  createTextToSpeechSession(model: Model): TextToSpeechSession
+  createClassificationSession(model: Model): ClassificationSession
   registerLicenseStatusHandler(callback?: ((arg: LicenseStatus) => void) | undefined | null): void
   registerModelDownloadStateHandler(callback?: ((arg0: string, arg1: ModelDownloadState) => void) | undefined | null): void
-  registerChatModelsHandler(callback?: ((arg: Array<ChatModel>) => void) | undefined | null): void
+  registerModelsHandler(callback?: ((arg: Array<Model>) => void) | undefined | null): void
 }
 
 export declare class ModelDownloadHandle {
@@ -40,6 +48,10 @@ export declare class ProgressUpdate {
   progress: number
 }
 
+export declare class TextToSpeechSession {
+  run(input: Input, outputPath: string): void
+}
+
 export type AsyncBatchSize =
   | { type: 'Default' }
   | { type: 'Custom', size: number }
@@ -74,20 +86,36 @@ export interface BenchmarksTask {
   greedy: boolean
 }
 
-export interface ChatModel {
-  readonly repoId: string
-  readonly type: ModelType
-  readonly name: string
-  readonly vendor: string
-  readonly quantization?: string
-  readonly outputParserRegex?: string
-}
-
 export interface ClassificationFeature {
   name: string
   values: Array<string>
 }
 
+export interface ClassificationLabel {
+  index: number
+  label: string
+  title: string
+  description: string
+  recommendedThreshold: number
+}
+
+export interface ClassificationOutput {
+  logits: Array<number>
+  probabilities: Record<string, number>
+  stats: ClassificationStats
+}
+
+export interface ClassificationStats {
+  preprocessingDuration: number
+  forwardPassDuration: number
+  postprocessingDuration: number
+  totalDuration: number
+  tokensCount: number
+  tokensPerSecond: number
+  predictedLabel: string
+  confidence: number
+}
+
 export interface Config {
   preset: Preset
   contextMode: ContextMode
@@ -148,6 +176,17 @@ export interface Message {
   reasoningContent?: string
 }
 
+export interface Model {
+  readonly repoId: string
+  readonly type: ModelType
+  readonly kind: ModelKind
+  readonly name: string
+  readonly vendor: string
+  readonly quantization?: string
+  readonly outputParserRegex?: string
+  readonly classificationLabels: Array<ClassificationLabel>
+}
+
 export declare const enum ModelDownloadPhase {
   NotDownloaded = 0,
   Downloading = 1,
@@ -168,6 +207,11 @@ export interface ModelDownloadState {
   readonly error?: string
 }
 
+export declare const enum ModelKind {
+  Local = 0,
+  Cloud = 1
+}
+
 export type ModelStorageError =
   | { type: 'Io', message: string }
   | { type: 'Network', message: string }
@@ -179,8 +223,9 @@ export type ModelStorageError =
   | { type: 'Paused' }
 
 export declare const enum ModelType {
-  Local = 0,
-  Cloud = 1
+  Chat = 0,
+  Classification = 1,
+  TTS = 2
 }
 
 export interface Output {
@@ -226,12 +271,17 @@ export interface RunStats {
 
 export type SamplingMethod =
   | { type: 'Greedy' }
-  | { type: 'Stochastic', temperature?: number, topK?: number, topP?: number, minP?: number }
+  | { type: 'Stochastic', temperature?: number, topK?: number, topP?: number, minP?: number, processingOrder: SamplingProcessingOrder }
 
 export type SamplingPolicy =
   | { type: 'Default' }
   | { type: 'Custom', value: SamplingMethod }
 
+export declare const enum SamplingProcessingOrder {
+  TemperatureThenFilters = 0,
+  FiltersThenTemperature = 1
+}
+
 export type SamplingSeed =
   | { type: 'Default' }
   | { type: 'Custom', seed: number }
@@ -250,6 +300,8 @@ export interface StepStats {
   processedTokensPerSecond: number
   modelRun: RunStats
   run?: RunStats
+  speculatorProposed: bigint
+  speculatorAccepted: bigint
 }
 
 export type StorageError =

package/napi/uzu.js

@@ -22,23 +22,30 @@ try {
 // Re-export all runtime exports
 const {
   ChatSession,
+  ClassificationSession,
   Engine,
   ModelDownloadHandle,
   ProgressStream,
   ProgressUpdate,
+  TextToSpeechSession,
   DataType,
   FinishReason,
   ModelDownloadPhase,
+  ModelKind,
   ModelType,
   Role,
+  SamplingProcessingOrder,
   version,
   BenchmarksDevice,
   BenchmarksResult,
   BenchmarksTask,
-  ChatModel,
   ClassificationFeature,
+  ClassificationLabel,
+  ClassificationOutput,
+  ClassificationStats,
   Config,
   Message,
+  Model,
   ModelDownloadState,
   Output,
   ParsedText,
@@ -65,23 +72,30 @@ const {
 
 module.exports = {
   ChatSession,
+  ClassificationSession,
   Engine,
   ModelDownloadHandle,
   ProgressStream,
   ProgressUpdate,
+  TextToSpeechSession,
   DataType,
   FinishReason,
   ModelDownloadPhase,
+  ModelKind,
   ModelType,
   Role,
+  SamplingProcessingOrder,
   version,
   BenchmarksDevice,
   BenchmarksResult,
   BenchmarksTask,
-  ChatModel,
   ClassificationFeature,
+  ClassificationLabel,
+  ClassificationOutput,
+  ClassificationStats,
   Config,
   Message,
+  Model,
   ModelDownloadState,
   Output,
   ParsedText,

package/napi/uzu.mjs

@@ -22,23 +22,30 @@ try {
 // Re-export all runtime exports as named exports  
 export const {
   ChatSession,
+  ClassificationSession,
   Engine,
   ModelDownloadHandle,
   ProgressStream,
   ProgressUpdate,
+  TextToSpeechSession,
   DataType,
   FinishReason,
   ModelDownloadPhase,
+  ModelKind,
   ModelType,
   Role,
+  SamplingProcessingOrder,
   version,
   BenchmarksDevice,
   BenchmarksResult,
   BenchmarksTask,
-  ChatModel,
   ClassificationFeature,
+  ClassificationLabel,
+  ClassificationOutput,
+  ClassificationStats,
   Config,
   Message,
+  Model,
   ModelDownloadState,
   Output,
   ParsedText,

package/package.json

@@ -101,5 +101,5 @@
   },
   "type": "commonjs",
   "types": "./index.d.ts",
-  "version": "0.2.20"
+  "version": "0.3.0"
 }

package/src/bridging/classificationLabel.ts

@@ -0,0 +1,43 @@
+import { ClassificationLabel as NapiClassificationLabel } from '../napi/uzu';
+
+export 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,
+    ) {
+        this.index = index;
+        this.label = label;
+        this.title = title;
+        this.description = description;
+        this.recommendedThreshold = recommendedThreshold;
+    }
+
+    static fromNapi(napiEntity: NapiClassificationLabel): ClassificationLabel {
+        return new ClassificationLabel(
+            napiEntity.index,
+            napiEntity.label,
+            napiEntity.title,
+            napiEntity.description,
+            napiEntity.recommendedThreshold,
+        );
+    }
+
+    toNapi(): NapiClassificationLabel {
+        return {
+            index: this.index,
+            label: this.label,
+            title: this.title,
+            description: this.description,
+            recommendedThreshold: this.recommendedThreshold,
+        };
+    }
+}

package/src/bridging/classificationOutput.ts

@@ -0,0 +1,22 @@
+import { ClassificationOutput as NapiClassificationOutput } from '../napi/uzu';
+import { ClassificationStats } from './classificationStats';
+
+export class ClassificationOutput {
+    readonly logits: number[];
+    readonly probabilities: Record<string, number>;
+    readonly stats: ClassificationStats;
+
+    constructor(logits: number[], probabilities: Record<string, number>, stats: ClassificationStats) {
+        this.logits = logits;
+        this.probabilities = probabilities;
+        this.stats = stats;
+    }
+
+    static fromNapi(napiEntity: NapiClassificationOutput): ClassificationOutput {
+        return new ClassificationOutput(
+            napiEntity.logits,
+            napiEntity.probabilities,
+            ClassificationStats.fromNapi(napiEntity.stats),
+        );
+    }
+}

package/src/bridging/classificationSession.ts

@@ -0,0 +1,17 @@
+import { ClassificationSession as NapiClassificationSession } from '../napi/uzu';
+import { ClassificationOutput } from './classificationOutput';
+import { Input } from './input';
+
+export class ClassificationSession {
+    private readonly napiClassificationSession: NapiClassificationSession;
+
+    constructor(napiClassificationSession: NapiClassificationSession) {
+        this.napiClassificationSession = napiClassificationSession;
+    }
+
+    run(input: Input): ClassificationOutput {
+        const napiInput = input.toNapi();
+        const napiOutput = this.napiClassificationSession.run(napiInput);
+        return ClassificationOutput.fromNapi(napiOutput);
+    }
+}

package/src/bridging/classificationStats.ts

@@ -0,0 +1,45 @@
+import { ClassificationStats as NapiClassificationStats } from '../napi/uzu';
+
+export 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,
+    ) {
+        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: NapiClassificationStats): ClassificationStats {
+        return new ClassificationStats(
+            napiEntity.preprocessingDuration,
+            napiEntity.forwardPassDuration,
+            napiEntity.postprocessingDuration,
+            napiEntity.totalDuration,
+            napiEntity.tokensCount,
+            napiEntity.tokensPerSecond,
+            napiEntity.predictedLabel,
+            napiEntity.confidence,
+        );
+    }
+}

package/src/bridging/engine.ts

@@ -2,14 +2,16 @@ import { EngineError, EngineErrorCode, unwrapOrThrow } from '../error';
 import { EngineInteractor } from '../interactors/engineInteractor';
 import { Engine as NapiEngine } from '../napi/uzu';
 import { readEnv } from '../utilities/env';
-import { ChatModel } from './chatModel';
 import { ChatSession } from './chatSession';
+import { ClassificationSession } from './classificationSession';
 import { Config } from './config';
 import { DownloadHandle } from './downloadHandle';
 import { DownloadProgressUpdate } from './downloadProgressUpdate';
 import { DownloadPhase, DownloadState } from './downloadState';
 import { LicenseStatus, licenseStatusFromNapiLicenseStatus } from './licenseStatus';
-import { ModelType, modelTypeToNapiModelType } from './modelType';
+import { Model } from './model';
+import { ModelKind, modelKindToNapiModelKind } from './modelKind';
+import { TextToSpeechSession } from './textToSpeechSession';
 
 export class Engine {
     private readonly napiEngine: NapiEngine;
@@ -55,32 +57,18 @@ export class Engine {
         return new DownloadHandle(this.napiEngine.createModelDownloadHandle(repoId));
     }
 
-    async chatModels(types: ModelType[] | null = null): Promise<ChatModel[]> {
-        let typesToFilter: ModelType[] = types ?? [ModelType.Local, ModelType.Cloud];
-        let napiModelTypes = typesToFilter.map(modelTypeToNapiModelType);
-        let napiChatModels = await this.napiEngine.getChatModels(napiModelTypes);
-        let results: ChatModel[] = napiChatModels.map(ChatModel.fromNapiChatModel);
-        return results;
-    }
-
-    async chatModel(repoId: string, types: ModelType[] | null = null): Promise<ChatModel> {
-        const chatModels = await this.chatModels(types);
-        const chatModel = chatModels.find((model) => model.repoId === repoId);
-        return unwrapOrThrow(chatModel, EngineErrorCode.ModelNotFound);
-    }
-
-    async downloadChatModel(
-        chatModel: ChatModel,
+    async downloadModel(
+        model: Model,
         progress: (progressUpdate: DownloadProgressUpdate) => void = () => {},
     ): Promise<DownloadState> {
-        switch (chatModel.type) {
-            case ModelType.Local:
+        switch (model.kind) {
+            case ModelKind.Local:
                 break;
-            case ModelType.Cloud:
-                throw new EngineError(EngineErrorCode.UnexpectedModelType);
+            case ModelKind.Cloud:
+                throw new EngineError(EngineErrorCode.UnexpectedModelKind);
         }
 
-        const downloadHandle = this.downloadHandle(chatModel.repoId);
+        const downloadHandle = this.downloadHandle(model.repoId);
         const state = await downloadHandle.state();
         switch (state.phase) {
             case DownloadPhase.Downloaded:
@@ -97,9 +85,57 @@ export class Engine {
         return finalState;
     }
 
-    chatSession(chatModel: ChatModel, config: Config = Config.default()): ChatSession {
-        const napiChatModel = chatModel.toNapi();
+    async chatModels(kinds: ModelKind[] | null = null): Promise<Model[]> {
+        let kindsToFilter: ModelKind[] = kinds ?? [ModelKind.Local, ModelKind.Cloud];
+        let napiModelKinds = kindsToFilter.map(modelKindToNapiModelKind);
+        let napiChatModels = await this.napiEngine.getChatModels(napiModelKinds);
+        let results: Model[] = napiChatModels.map(Model.fromNapiModel);
+        return results;
+    }
+
+    async chatModel(repoId: string, kinds: ModelKind[] | null = null): Promise<Model> {
+        const chatModels = await this.chatModels(kinds);
+        const chatModel = chatModels.find((model) => model.repoId === repoId);
+        return unwrapOrThrow(chatModel, EngineErrorCode.ModelNotFound);
+    }
+
+    chatSession(model: Model, config: Config = Config.default()): ChatSession {
+        const napiModel = model.toNapi();
         const napiConfig = config.toNapi();
-        return new ChatSession(this.napiEngine.createChatSession(napiChatModel, napiConfig));
+        return new ChatSession(this.napiEngine.createChatSession(napiModel, napiConfig));
+    }
+
+    async classificationModels(): Promise<Model[]> {
+        const napiClassificationModels = await this.napiEngine.getClassificationModels();
+        const results: Model[] = napiClassificationModels.map(Model.fromNapiModel);
+        return results;
+    }
+
+    async classificationModel(repoId: string): Promise<Model> {
+        const classificationModels = await this.classificationModels();
+        const classificationModel = classificationModels.find((model) => model.repoId === repoId);
+        return unwrapOrThrow(classificationModel, EngineErrorCode.ModelNotFound);
+    }
+
+    classificationSession(model: Model): ClassificationSession {
+        const napiModel = model.toNapi();
+        return new ClassificationSession(this.napiEngine.createClassificationSession(napiModel));
+    }
+
+    async textToSpeechModels(): Promise<Model[]> {
+        const napiTextToSpeechModels = await this.napiEngine.getTextToSpeechModels();
+        const results: Model[] = napiTextToSpeechModels.map(Model.fromNapiModel);
+        return results;
+    }
+
+    async textToSpeechModel(repoId: string): Promise<Model> {
+        const textToSpeechModels = await this.textToSpeechModels();
+        const textToSpeechModel = textToSpeechModels.find((model) => model.repoId === repoId);
+        return unwrapOrThrow(textToSpeechModel, EngineErrorCode.ModelNotFound);
+    }
+
+    textToSpeechSession(model: Model): TextToSpeechSession {
+        const napiModel = model.toNapi();
+        return new TextToSpeechSession(this.napiEngine.createTextToSpeechSession(napiModel));
     }
 }