@trymirai/uzu 0.1.39 → 0.1.42

package/interactors/engineInteractor.js

@@ -1,32 +1,28 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.EngineInteractor = void 0;
-const modelsInteractor_1 = require("./modelsInteractor.js");
+const chatModelsInteractor_1 = require("./chatModelsInteractor.js");
 class EngineInteractor {
-    constructor(engine, modelTypeFilter = null, modelKindFilter = null) {
+    constructor(engine, modelTypeFilter = 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, this.modelKindFilter);
+        return new EngineInteractor(this.entity, types);
     }
-    filterKinds(kinds) {
-        return new EngineInteractor(this.entity, this.modelTypeFilter, kinds);
-    }
-    models() {
+    chatModels() {
         const modelsPromise = (async () => {
             const engine = await this.finalize();
-            return await engine.models(this.modelTypeFilter, this.modelKindFilter);
+            return await engine.chatModels(this.modelTypeFilter);
         })();
-        return new modelsInteractor_1.ModelsInteractor(this, modelsPromise);
+        return new chatModelsInteractor_1.ChatModelsInteractor(this, modelsPromise);
     }
-    model(repoId) {
-        const models = this.models();
+    chatModel(repoId) {
+        const models = this.chatModels();
         return models.get(repoId);
     }
 }

package/interactors/engineInteractor.js.map

@@ -1 +1 @@
-{"version":3,"file":"engineInteractor.js","sourceRoot":"","sources":["../src/interactors/engineInteractor.ts"],"names":[],"mappings":";;;AAKA,4DAAsD;AAEtD,MAAa,gBAAgB;IAKzB,YACI,MAAgC,EAChC,kBAAsC,IAAI,EAC1C,kBAAsC,IAAI;QAE1C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY;IAEZ,WAAW,CAAC,KAAyB;QACjC,OAAO,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;IAC1E,CAAC;IAED,WAAW,CAAC,KAAyB;QACjC,OAAO,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;IAC1E,CAAC;IAED,MAAM;QACF,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACrC,OAAO,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAC3E,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,mCAAgB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,MAAc;QAChB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAC7B,OAAO,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;CACJ;AAzCD,4CAyCC"}
+{"version":3,"file":"engineInteractor.js","sourceRoot":"","sources":["../src/interactors/engineInteractor.ts"],"names":[],"mappings":";;;AAGA,oEAA8D;AAG9D,MAAa,gBAAgB;IAIzB,YAAY,MAAgC,EAAE,kBAAsC,IAAI;QACpF,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY;IAEZ,WAAW,CAAC,KAAyB;QACjC,OAAO,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACpD,CAAC;IAED,UAAU;QACN,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACrC,OAAO,MAAM,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACzD,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,2CAAoB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACzD,CAAC;IAED,SAAS,CAAC,MAAc;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,OAAO,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;CACJ;AA/BD,4CA+BC"}

package/interactors/engineInteractor.mjs

@@ -1,29 +1,25 @@
-import { ModelsInteractor } from "./modelsInteractor.mjs";
+import { ChatModelsInteractor } from "./chatModelsInteractor.mjs";
 export class EngineInteractor {
-    constructor(engine, modelTypeFilter = null, modelKindFilter = null) {
+    constructor(engine, modelTypeFilter = 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, this.modelKindFilter);
+        return new EngineInteractor(this.entity, types);
     }
-    filterKinds(kinds) {
-        return new EngineInteractor(this.entity, this.modelTypeFilter, kinds);
-    }
-    models() {
+    chatModels() {
         const modelsPromise = (async () => {
             const engine = await this.finalize();
-            return await engine.models(this.modelTypeFilter, this.modelKindFilter);
+            return await engine.chatModels(this.modelTypeFilter);
         })();
-        return new ModelsInteractor(this, modelsPromise);
+        return new ChatModelsInteractor(this, modelsPromise);
     }
-    model(repoId) {
-        const models = this.models();
+    chatModel(repoId) {
+        const models = this.chatModels();
         return models.get(repoId);
     }
 }

package/interactors/engineInteractor.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"engineInteractor.mjs","sourceRoot":"","sources":["../src/interactors/engineInteractor.ts"],"names":[],"mappings":"OAKO,EAAE,gBAAgB,EAAE;AAE3B,MAAM,OAAO,gBAAgB;IAKzB,YACI,MAAgC,EAChC,kBAAsC,IAAI,EAC1C,kBAAsC,IAAI;QAE1C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY;IAEZ,WAAW,CAAC,KAAyB;QACjC,OAAO,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;IAC1E,CAAC;IAED,WAAW,CAAC,KAAyB;QACjC,OAAO,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;IAC1E,CAAC;IAED,MAAM;QACF,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACrC,OAAO,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAC3E,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,gBAAgB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,MAAc;QAChB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAC7B,OAAO,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;CACJ"}
+{"version":3,"file":"engineInteractor.mjs","sourceRoot":"","sources":["../src/interactors/engineInteractor.ts"],"names":[],"mappings":"OAGO,EAAE,oBAAoB,EAAE;AAG/B,MAAM,OAAO,gBAAgB;IAIzB,YAAY,MAAgC,EAAE,kBAAsC,IAAI;QACpF,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY;IAEZ,WAAW,CAAC,KAAyB;QACjC,OAAO,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACpD,CAAC;IAED,UAAU;QACN,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACrC,OAAO,MAAM,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACzD,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,oBAAoB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACzD,CAAC;IAED,SAAS,CAAC,MAAc;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,OAAO,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;CACJ"}

package/napi/uzu.d.ts

@@ -1,19 +1,19 @@
 /* auto-generated by NAPI-RS */
 /* eslint-disable */
-/** Thin FFI wrapper around `ModelStorage`. */
+export declare class ChatSession {
+  run(input: Input, config: RunConfig, progress?: (arg: Output) => boolean | undefined | null): Output
+}
+
 export declare class Engine {
-  getCloudModels(): Promise<Array<CloudModel>>
-  fetchCloudModels(): Promise<Array<CloudModel>>
-  /** Returns a `DownloadHandle` for the given model identifier. */
-  downloadHandle(repoId: string): ModelDownloadHandle
-  getLocalModels(): Array<LocalModel>
-  getState(repoId: string): ModelDownloadState
   activate(apiKey: string): Promise<LicenseStatus>
-  downloadModel(repoId: string): Promise<void>
-  pauseModel(repoId: string): Promise<void>
-  deleteModel(repoId: string): Promise<void>
+  getChatModels(types: Array<ModelType>): Promise<Array<ChatModel>>
+  getModelDownloadState(repoId: string): ModelDownloadState
+  createModelDownloadHandle(repoId: string): ModelDownloadHandle
   constructor()
-  createSession(repoId: string, modelType: ModelType, config: Config): Session
+  createChatSession(model: ChatModel, config: Config): ChatSession
+  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
 }
 
 export declare class ModelDownloadHandle {
@@ -38,8 +38,13 @@ export declare class ProgressUpdate {
   progress: number
 }
 
-export declare class Session {
-  run(input: Input, config: RunConfig, progress?: (arg: Output) => boolean | undefined | null): Output
+export interface ChatModel {
+  readonly repoId: string
+  readonly type: ModelType
+  readonly name: string
+  readonly vendor: string
+  readonly quantization?: string
+  readonly outputParserRegex?: string
 }
 
 export interface ClassificationFeature {
@@ -47,13 +52,6 @@ export interface ClassificationFeature {
   values: Array<string>
 }
 
-export interface CloudModel {
-  readonly repoId: string
-  readonly name: string
-  readonly vendor: string
-  readonly outputParserRegex?: string
-}
-
 export interface Config {
   preset: Preset
   prefillStepSize: PrefillStepSize
@@ -88,18 +86,6 @@ export type LicenseStatus =
   | { type: 'Timeout' }
   | { type: 'HttpError', code: number }
 
-export interface LocalModel {
-  readonly identifier: string
-  readonly repoId: string
-  readonly family: string
-  readonly vendor: string
-  readonly name: string
-  readonly size: string
-  readonly quantization?: string
-  readonly outputParserRegex?: string
-  readonly state: ModelDownloadState
-}
-
 export interface Message {
   role: Role
   content: string

package/napi/uzu.js

@@ -21,20 +21,19 @@ try {
 
 // Re-export all runtime exports
 const {
+  ChatSession,
   Engine,
   ModelDownloadHandle,
   ProgressStream,
   ProgressUpdate,
-  Session,
   FinishReason,
   ModelDownloadPhase,
   ModelType,
   Role,
   version,
+  ChatModel,
   ClassificationFeature,
-  CloudModel,
   Config,
-  LocalModel,
   Message,
   ModelDownloadState,
   Output,
@@ -58,20 +57,19 @@ const {
 } = nativeBinding;
 
 module.exports = {
+  ChatSession,
   Engine,
   ModelDownloadHandle,
   ProgressStream,
   ProgressUpdate,
-  Session,
   FinishReason,
   ModelDownloadPhase,
   ModelType,
   Role,
   version,
+  ChatModel,
   ClassificationFeature,
-  CloudModel,
   Config,
-  LocalModel,
   Message,
   ModelDownloadState,
   Output,

package/napi/uzu.mjs

@@ -21,20 +21,19 @@ try {
 
 // Re-export all runtime exports as named exports  
 export const {
+  ChatSession,
   Engine,
   ModelDownloadHandle,
   ProgressStream,
   ProgressUpdate,
-  Session,
   FinishReason,
   ModelDownloadPhase,
   ModelType,
   Role,
   version,
+  ChatModel,
   ClassificationFeature,
-  CloudModel,
   Config,
-  LocalModel,
   Message,
   ModelDownloadState,
   Output,

package/package.json

@@ -98,5 +98,5 @@
   },
   "type": "commonjs",
   "types": "./index.d.ts",
-  "version": "0.1.39"
+  "version": "0.1.42"
 }

package/src/bridging/chatModel.ts

@@ -0,0 +1,54 @@
+import { ChatModel as NapiChatModel } from '../napi/uzu';
+import { ModelType, modelTypeFromNapiModelType, modelTypeToNapiModelType } from './modelType';
+
+export class ChatModel {
+    readonly repoId: string;
+    readonly type: ModelType;
+    readonly name: string;
+    readonly vendor: string;
+    readonly quantization: string | null;
+    readonly outputParserRegex: string | null;
+
+    get isThinking(): boolean {
+        return this.outputParserRegex !== null;
+    }
+
+    private constructor(
+        repoId: string,
+        type: ModelType,
+        name: string,
+        vendor: string,
+        quantization: string | null,
+        outputParserRegex: string | null,
+    ) {
+        this.repoId = repoId;
+        this.type = type;
+        this.name = name;
+        this.vendor = vendor;
+        this.quantization = quantization;
+        this.outputParserRegex = outputParserRegex;
+    }
+
+    static fromNapiChatModel(napiChatModel: NapiChatModel): ChatModel {
+        return new ChatModel(
+            napiChatModel.repoId,
+            modelTypeFromNapiModelType(napiChatModel.type),
+            napiChatModel.name,
+            napiChatModel.vendor,
+            napiChatModel.quantization ?? null,
+            napiChatModel.outputParserRegex ?? null,
+        );
+    }
+
+    toNapi(): NapiChatModel {
+        const napiChatModel: NapiChatModel = {
+            repoId: this.repoId,
+            type: modelTypeToNapiModelType(this.type),
+            name: this.name,
+            vendor: this.vendor,
+            ...(this.quantization !== null && { quantization: this.quantization }),
+            ...(this.outputParserRegex !== null && { outputParserRegex: this.outputParserRegex }),
+        };
+        return napiChatModel;
+    }
+}

package/src/bridging/{session.ts → chatSession.ts}

@@ -1,13 +1,13 @@
-import { Output as NapiOutput, Session as NapiSession } from '../napi/uzu';
+import { ChatSession as NapiChatSession, Output as NapiOutput } from '../napi/uzu';
 import { Input } from './input';
 import { Output } from './output';
 import { RunConfig } from './runConfig';
 
-export class Session {
-    private readonly napiSession: NapiSession;
+export class ChatSession {
+    private readonly napiChatSession: NapiChatSession;
 
-    constructor(napiSession: NapiSession) {
-        this.napiSession = napiSession;
+    constructor(napiChatSession: NapiChatSession) {
+        this.napiChatSession = napiChatSession;
     }
 
     run(
@@ -18,7 +18,7 @@ export class Session {
         const napiInput = input.toNapi();
         const napiConfig = config.toNapi();
         const napiProgress = (partialOutput: NapiOutput) => progress(Output.fromNapi(partialOutput));
-        const napiOutput = this.napiSession.run(napiInput, napiConfig, napiProgress);
+        const napiOutput = this.napiChatSession.run(napiInput, napiConfig, napiProgress);
         return Output.fromNapi(napiOutput);
     }
 }

package/src/bridging/engine.ts

@@ -2,12 +2,14 @@ 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 { Config } from './config';
 import { DownloadHandle } from './downloadHandle';
+import { DownloadProgressUpdate } from './downloadProgressUpdate';
+import { DownloadPhase, DownloadState } from './downloadState';
 import { LicenseStatus, licenseStatusFromNapiLicenseStatus } from './licenseStatus';
-import { Model, ModelKind } from './model';
-import { ModelType, modelTypeToNapi } from './modelType';
-import { Session } from './session';
+import { ModelType, modelTypeToNapiModelType } from './modelType';
 
 export class Engine {
     private readonly napiEngine: NapiEngine;
@@ -44,62 +46,60 @@ export class Engine {
         return new EngineInteractor(enginePromise);
     }
 
-    async models(types: ModelType[] | null = null, kinds: ModelKind[] | null = null): Promise<Model[]> {
-        function isTypeAllowed(type: ModelType): boolean {
-            return types === null || types.includes(type);
-        }
+    getDownloadState(repoId: string): DownloadState {
+        let downloadState = this.napiEngine.getModelDownloadState(repoId);
+        return DownloadState.fromNapi(downloadState);
+    }
 
-        function isKindAllowed(kind: ModelKind): boolean {
-            return kinds === null || kinds.includes(kind);
-        }
+    downloadHandle(repoId: string): DownloadHandle {
+        return new DownloadHandle(this.napiEngine.createModelDownloadHandle(repoId));
+    }
 
-        let results: Model[] = [];
-        if (isTypeAllowed(ModelType.Local) && isKindAllowed(ModelKind.Text)) {
-            const napiLocalModels = this.napiEngine.getLocalModels();
-            results.push(...napiLocalModels.map(Model.fromNapiLocalModel));
-        }
-        if (isTypeAllowed(ModelType.Cloud) && isKindAllowed(ModelKind.Text)) {
-            const napiCloudModels = await this.napiEngine.getCloudModels();
-            results.push(...napiCloudModels.map(Model.fromNapiCloudModel));
-        }
+    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 model(repoId: string): Promise<Model> {
-        const models = await this.models();
-        const model = models.find((model) => model.repoId === repoId);
-        return unwrapOrThrow(model, EngineErrorCode.ModelNotFound);
+    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);
     }
 
-    downloadHandle(model: Model): DownloadHandle {
-        if (model.type !== ModelType.Local) {
-            throw new EngineError(EngineErrorCode.ExpectedLocalModel);
+    async downloadChatModel(
+        chatModel: ChatModel,
+        progress: (progressUpdate: DownloadProgressUpdate) => void = () => {},
+    ): Promise<DownloadState> {
+        switch (chatModel.type) {
+            case ModelType.Local:
+                break;
+            case ModelType.Cloud:
+                throw new EngineError(EngineErrorCode.UnexpectedModelType);
         }
-        return new DownloadHandle(this.napiEngine.downloadHandle(model.repoId));
-    }
 
-    session(model: Model, config: Config = Config.default()): Session {
-        if (model.kind !== ModelKind.Text) {
-            throw new EngineError(EngineErrorCode.ExpectedTextModel);
+        const downloadHandle = this.downloadHandle(chatModel.repoId);
+        const state = await downloadHandle.state();
+        switch (state.phase) {
+            case DownloadPhase.Downloaded:
+                break;
+            default:
+                await downloadHandle.download();
+                for await (const progressUpdate of downloadHandle.progressUpdate()) {
+                    progress(progressUpdate);
+                }
+                break;
         }
-        const napiModelType = modelTypeToNapi(model.type);
-        const napiConfig = config.toNapi();
-        return new Session(this.napiEngine.createSession(model.repoId, napiModelType, napiConfig));
-    }
 
-    async downloadModel(repoId: string): Promise<void> {
-        await this.napiEngine.downloadModel(repoId);
+        const finalState = await downloadHandle.state();
+        return finalState;
     }
 
-    async pauseModel(repoId: string): Promise<void> {
-        await this.napiEngine.pauseModel(repoId);
-    }
-
-    async deleteModel(repoId: string): Promise<void> {
-        await this.napiEngine.deleteModel(repoId);
-    }
-
-    getState(repoId: string) {
-        return this.napiEngine.getState(repoId);
+    chatSession(chatModel: ChatModel, config: Config = Config.default()): ChatSession {
+        const napiChatModel = chatModel.toNapi();
+        const napiConfig = config.toNapi();
+        return new ChatSession(this.napiEngine.createChatSession(napiChatModel, napiConfig));
     }
 }

package/src/bridging/modelType.ts

@@ -1,3 +1,4 @@
+import { EngineError, EngineErrorCode } from '../error';
 import { ModelType as NapiModelType } from '../napi/uzu';
 
 export enum ModelType {
@@ -5,7 +6,18 @@ export enum ModelType {
     Cloud = 'Cloud',
 }
 
-export function modelTypeToNapi(modelType: ModelType): NapiModelType {
+export function modelTypeFromNapiModelType(napiModelType: NapiModelType): ModelType {
+    switch (napiModelType) {
+        case 0:
+            return ModelType.Local;
+        case 1:
+            return ModelType.Cloud;
+        default:
+            throw new EngineError(EngineErrorCode.UnexpectedModelType);
+    }
+}
+
+export function modelTypeToNapiModelType(modelType: ModelType): NapiModelType {
     switch (modelType) {
         case ModelType.Local:
             return 0;

package/src/error.ts

@@ -2,10 +2,10 @@ export enum EngineErrorCode {
     APIKeyNotFound = 'apiKeyNotFound',
     ModelNotFound = 'modelNotFound',
     ExpectedLocalModel = 'expectedLocalModel',
-    ExpectedTextModel = 'expectedTextModel',
     UnexpectedDownloadPhase = 'unexpectedDownloadPhase',
     UnexpectedFinishReason = 'unexpectedFinishReason',
     UnexpectedLicenseStatus = 'unexpectedLicenseStatus',
+    UnexpectedModelType = 'unexpectedModelType',
     LicenseNotActivated = 'licenseNotActivated',
     Unknown = 'unknown',
 }
@@ -14,10 +14,10 @@ const errorCodeToMessage = {
     [EngineErrorCode.APIKeyNotFound]: 'API key not found',
     [EngineErrorCode.ModelNotFound]: 'Model not found',
     [EngineErrorCode.ExpectedLocalModel]: 'Expected local model',
-    [EngineErrorCode.ExpectedTextModel]: 'Expected text model',
     [EngineErrorCode.UnexpectedDownloadPhase]: 'Unexpected download phase',
     [EngineErrorCode.UnexpectedFinishReason]: 'Unexpected finish reason',
     [EngineErrorCode.UnexpectedLicenseStatus]: 'Unexpected license status',
+    [EngineErrorCode.UnexpectedModelType]: 'Unexpected model type',
     [EngineErrorCode.LicenseNotActivated]: 'License not activated',
     [EngineErrorCode.Unknown]: 'Unknown error',
 };

package/src/index.ts

@@ -1,3 +1,5 @@
+export { ChatModel } from './bridging/chatModel';
+export { ChatSession } from './bridging/chatSession';
 export { ClassificationFeature } from './bridging/classificationFeature';
 export { Config } from './bridging/config';
 export { ContextLength } from './bridging/contextLength';
@@ -8,7 +10,6 @@ export { Engine as default } from './bridging/engine';
 export { FinishReason } from './bridging/finishReason';
 export { Input } from './bridging/input';
 export { Message } from './bridging/message';
-export { Model, ModelKind } from './bridging/model';
 export { ModelType } from './bridging/modelType';
 export { Output } from './bridging/output';
 export { ParsedText } from './bridging/parsedText';
@@ -20,14 +21,13 @@ export { RunStats } from './bridging/runStats';
 export { SamplingMethod } from './bridging/samplingMethod';
 export { SamplingPolicy } from './bridging/samplingPolicy';
 export { SamplingSeed } from './bridging/samplingSeed';
-export { Session } from './bridging/session';
 export { Stats } from './bridging/stats';
 export { StepStats } from './bridging/stepStats';
 export { Text } from './bridging/text';
 export { TotalStats } from './bridging/totalStats';
 export { EngineError } from './error';
+export { ChatModelInteractor } from './interactors/chatModelInteractor';
+export { ChatModelsInteractor } from './interactors/chatModelsInteractor';
+export { ChatSessionInteractor } from './interactors/chatSessionInteractor';
 export { DownloadInteractor } from './interactors/downloadInteractor';
 export { EngineInteractor } from './interactors/engineInteractor';
-export { ModelInteractor } from './interactors/modelInteractor';
-export { ModelsInteractor } from './interactors/modelsInteractor';
-export { SessionInteractor } from './interactors/sessionInteractor';