@trymirai/uzu 0.1.21 → 0.1.22

package/interactors/modelInteractor.js

@@ -0,0 +1,85 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModelInteractor = void 0;
+const downloadState_1 = require("../bridging/downloadState.js");
+const model_1 = require("../bridging/model.js");
+const runConfig_1 = require("../bridging/runConfig.js");
+const downloadInteractor_1 = require("./downloadInteractor.js");
+const sessionInteractor_1 = require("./sessionInteractor.js");
+class ModelInteractor {
+    constructor(modelsInteractor, model, config) {
+        this.modelsInteractor = modelsInteractor;
+        this.entity = model;
+        this.config = config;
+    }
+    async finalize() {
+        return await this.entity;
+    }
+    /* Downloading */
+    downloading() {
+        const handlePromise = (async () => {
+            const model = await this.finalize();
+            const engine = await this.modelsInteractor.engineInteractor.finalize();
+            const handle = engine.downloadHandle(model);
+            return handle;
+        })();
+        return new downloadInteractor_1.DownloadInteractor(this, handlePromise);
+    }
+    download(callback) {
+        const modelPromise = (async () => {
+            const downloadInteractor = this.downloading();
+            const handle = await downloadInteractor.finalize();
+            const state = await handle.state();
+            switch (state.phase) {
+                case downloadState_1.DownloadPhase.Downloaded:
+                    break;
+                default:
+                    await downloadInteractor
+                        .resume()
+                        .onProgressUpdate(callback ?? (() => { }))
+                        .finalize();
+                    break;
+            }
+            return this.finalize();
+        })();
+        return new ModelInteractor(this.modelsInteractor, modelPromise, this.config);
+    }
+    /* Config */
+    preset(preset) {
+        const config = this.config.withPreset(preset);
+        return new ModelInteractor(this.modelsInteractor, this.entity, config);
+    }
+    prefillStepSize(prefillStepSize) {
+        const config = this.config.withPrefillStepSize(prefillStepSize);
+        return new ModelInteractor(this.modelsInteractor, this.entity, config);
+    }
+    contextLength(contextLength) {
+        const config = this.config.withContextLength(contextLength);
+        return new ModelInteractor(this.modelsInteractor, this.entity, config);
+    }
+    samplingSeed(samplingSeed) {
+        const config = this.config.withSamplingSeed(samplingSeed);
+        return new ModelInteractor(this.modelsInteractor, this.entity, config);
+    }
+    /* Session */
+    session() {
+        const sessionPromise = (async () => {
+            const engine = await this.modelsInteractor.engineInteractor.finalize();
+            let model = await this.finalize();
+            if (model.type === model_1.ModelType.Local) {
+                model = await this.download().finalize();
+            }
+            const session = engine.session(model, this.config);
+            return session;
+        })();
+        return new sessionInteractor_1.SessionInteractor(this, sessionPromise, runConfig_1.RunConfig.default());
+    }
+    async reply(text, progress = () => true) {
+        return await this.session().reply(text, progress);
+    }
+    async replyToMessages(messages, progress = () => true) {
+        return await this.session().replyToMessages(messages, progress);
+    }
+}
+exports.ModelInteractor = ModelInteractor;
+//# sourceMappingURL=modelInteractor.js.map

package/interactors/modelInteractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelInteractor.js","sourceRoot":"","sources":["../src/interactors/modelInteractor.ts"],"names":[],"mappings":";;;AAGA,gEAA0D;AAE1D,gDAAqD;AAIrD,wDAAkD;AAElD,gEAA0D;AAG1D,8DAAwD;AAExD,MAAa,eAAe;IAKxB,YAAY,gBAAkC,EAAE,KAA8B,EAAE,MAAc;QAC1F,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,iBAAiB;IAEjB,WAAW;QACP,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC;YACvE,MAAM,MAAM,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAC5C,OAAO,MAAM,CAAC;QAClB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,uCAAkB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACvD,CAAC;IAED,QAAQ,CAAC,QAA2D;QAChE,MAAM,YAAY,GAAG,CAAC,KAAK,IAAI,EAAE;YAC7B,MAAM,kBAAkB,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YAC9C,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,EAAE,CAAC;YACnD,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACnC,QAAQ,KAAK,CAAC,KAAK,EAAE,CAAC;gBAClB,KAAK,6BAAa,CAAC,UAAU;oBACzB,MAAM;gBACV;oBACI,MAAM,kBAAkB;yBACnB,MAAM,EAAE;yBACR,gBAAgB,CAAC,QAAQ,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;yBACxC,QAAQ,EAAE,CAAC;oBAChB,MAAM;YACd,CAAC;YACD,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC3B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,gBAAgB,EAAE,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACjF,CAAC;IAED,YAAY;IAEZ,MAAM,CAAC,MAAc;QACjB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC9C,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAED,eAAe,CAAC,eAAgC;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC;QAChE,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAED,aAAa,CAAC,aAA4B;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAC5D,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAED,YAAY,CAAC,YAA0B;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;QAC1D,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAED,aAAa;IAEb,OAAO;QACH,MAAM,cAAc,GAAG,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC;YAEvE,IAAI,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClC,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAS,CAAC,KAAK,EAAE,CAAC;gBACjC,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;YAC7C,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACnD,OAAO,OAAO,CAAC;QACnB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,qCAAiB,CAAC,IAAI,EAAE,cAAc,EAAE,qBAAS,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5E,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC/E,OAAO,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,eAAe,CACjB,QAAmB,EACnB,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,OAAO,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACpE,CAAC;CACJ;AAhGD,0CAgGC"}

package/interactors/modelInteractor.mjs

@@ -0,0 +1,81 @@
+import { DownloadPhase } from "../bridging/downloadState.mjs";
+import { ModelType } from "../bridging/model.mjs";
+import { RunConfig } from "../bridging/runConfig.mjs";
+import { DownloadInteractor } from "./downloadInteractor.mjs";
+import { SessionInteractor } from "./sessionInteractor.mjs";
+export class ModelInteractor {
+    constructor(modelsInteractor, model, config) {
+        this.modelsInteractor = modelsInteractor;
+        this.entity = model;
+        this.config = config;
+    }
+    async finalize() {
+        return await this.entity;
+    }
+    /* Downloading */
+    downloading() {
+        const handlePromise = (async () => {
+            const model = await this.finalize();
+            const engine = await this.modelsInteractor.engineInteractor.finalize();
+            const handle = engine.downloadHandle(model);
+            return handle;
+        })();
+        return new DownloadInteractor(this, handlePromise);
+    }
+    download(callback) {
+        const modelPromise = (async () => {
+            const downloadInteractor = this.downloading();
+            const handle = await downloadInteractor.finalize();
+            const state = await handle.state();
+            switch (state.phase) {
+                case DownloadPhase.Downloaded:
+                    break;
+                default:
+                    await downloadInteractor
+                        .resume()
+                        .onProgressUpdate(callback ?? (() => { }))
+                        .finalize();
+                    break;
+            }
+            return this.finalize();
+        })();
+        return new ModelInteractor(this.modelsInteractor, modelPromise, this.config);
+    }
+    /* Config */
+    preset(preset) {
+        const config = this.config.withPreset(preset);
+        return new ModelInteractor(this.modelsInteractor, this.entity, config);
+    }
+    prefillStepSize(prefillStepSize) {
+        const config = this.config.withPrefillStepSize(prefillStepSize);
+        return new ModelInteractor(this.modelsInteractor, this.entity, config);
+    }
+    contextLength(contextLength) {
+        const config = this.config.withContextLength(contextLength);
+        return new ModelInteractor(this.modelsInteractor, this.entity, config);
+    }
+    samplingSeed(samplingSeed) {
+        const config = this.config.withSamplingSeed(samplingSeed);
+        return new ModelInteractor(this.modelsInteractor, this.entity, config);
+    }
+    /* Session */
+    session() {
+        const sessionPromise = (async () => {
+            const engine = await this.modelsInteractor.engineInteractor.finalize();
+            let model = await this.finalize();
+            if (model.type === ModelType.Local) {
+                model = await this.download().finalize();
+            }
+            const session = engine.session(model, this.config);
+            return session;
+        })();
+        return new SessionInteractor(this, sessionPromise, RunConfig.default());
+    }
+    async reply(text, progress = () => true) {
+        return await this.session().reply(text, progress);
+    }
+    async replyToMessages(messages, progress = () => true) {
+        return await this.session().replyToMessages(messages, progress);
+    }
+}
+//# sourceMappingURL=modelInteractor.mjs.map

package/interactors/modelInteractor.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelInteractor.mjs","sourceRoot":"","sources":["../src/interactors/modelInteractor.ts"],"names":[],"mappings":"OAGO,EAAE,aAAa,EAAE;OAEjB,EAAS,SAAS,EAAE;OAIpB,EAAE,SAAS,EAAE;OAEb,EAAE,kBAAkB,EAAE;OAGtB,EAAE,iBAAiB,EAAE;AAE5B,MAAM,OAAO,eAAe;IAKxB,YAAY,gBAAkC,EAAE,KAA8B,EAAE,MAAc;QAC1F,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,iBAAiB;IAEjB,WAAW;QACP,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC;YACvE,MAAM,MAAM,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAC5C,OAAO,MAAM,CAAC;QAClB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,kBAAkB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACvD,CAAC;IAED,QAAQ,CAAC,QAA2D;QAChE,MAAM,YAAY,GAAG,CAAC,KAAK,IAAI,EAAE;YAC7B,MAAM,kBAAkB,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YAC9C,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,EAAE,CAAC;YACnD,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACnC,QAAQ,KAAK,CAAC,KAAK,EAAE,CAAC;gBAClB,KAAK,aAAa,CAAC,UAAU;oBACzB,MAAM;gBACV;oBACI,MAAM,kBAAkB;yBACnB,MAAM,EAAE;yBACR,gBAAgB,CAAC,QAAQ,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;yBACxC,QAAQ,EAAE,CAAC;oBAChB,MAAM;YACd,CAAC;YACD,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC3B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,gBAAgB,EAAE,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACjF,CAAC;IAED,YAAY;IAEZ,MAAM,CAAC,MAAc;QACjB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC9C,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAED,eAAe,CAAC,eAAgC;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC;QAChE,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAED,aAAa,CAAC,aAA4B;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAC5D,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAED,YAAY,CAAC,YAA0B;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;QAC1D,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAED,aAAa;IAEb,OAAO;QACH,MAAM,cAAc,GAAG,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC;YAEvE,IAAI,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClC,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,KAAK,EAAE,CAAC;gBACjC,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;YAC7C,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACnD,OAAO,OAAO,CAAC;QACnB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,iBAAiB,CAAC,IAAI,EAAE,cAAc,EAAE,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5E,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC/E,OAAO,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,eAAe,CACjB,QAAmB,EACnB,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,OAAO,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACpE,CAAC;CACJ"}

package/interactors/modelsInteractor.d.mts

@@ -0,0 +1,12 @@
+import { Model } from "../bridging/model.mjs";
+import { EngineInteractor } from "./engineInteractor.mjs";
+import { Interactor, InteractorEntity } from "./interactor.mjs";
+import { ModelInteractor } from "./modelInteractor.mjs";
+export declare class ModelsInteractor implements Interactor<Model[]> {
+    readonly engineInteractor: EngineInteractor;
+    readonly entity: InteractorEntity<Model[]>;
+    constructor(engineInteractor: EngineInteractor, models: InteractorEntity<Model[]>);
+    finalize(): Promise<Model[]>;
+    get(repoId: string): ModelInteractor;
+}
+//# sourceMappingURL=modelsInteractor.d.mts.map

package/interactors/modelsInteractor.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelsInteractor.d.mts","sourceRoot":"","sources":["../src/interactors/modelsInteractor.ts"],"names":[],"mappings":"OACO,EAAE,KAAK,EAAE;OAET,EAAE,gBAAgB,EAAE;OACpB,EAAE,UAAU,EAAE,gBAAgB,EAAE;OAChC,EAAE,eAAe,EAAE;AAE1B,qBAAa,gBAAiB,YAAW,UAAU,CAAC,KAAK,EAAE,CAAC;IACxD,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,eAAe;CAQvC"}

package/interactors/modelsInteractor.d.ts

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

package/interactors/modelsInteractor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelsInteractor.d.ts","sourceRoot":"","sources":["../src/interactors/modelsInteractor.ts"],"names":[],"mappings":"OACO,EAAE,KAAK,EAAE;OAET,EAAE,gBAAgB,EAAE;OACpB,EAAE,UAAU,EAAE,gBAAgB,EAAE;OAChC,EAAE,eAAe,EAAE;AAE1B,qBAAa,gBAAiB,YAAW,UAAU,CAAC,KAAK,EAAE,CAAC;IACxD,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,eAAe;CAQvC"}

package/interactors/modelsInteractor.js

@@ -0,0 +1,25 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModelsInteractor = void 0;
+const config_1 = require("../bridging/config.js");
+const error_1 = require("../error.js");
+const modelInteractor_1 = require("./modelInteractor.js");
+class ModelsInteractor {
+    constructor(engineInteractor, models) {
+        this.engineInteractor = engineInteractor;
+        this.entity = models;
+    }
+    async finalize() {
+        return await this.entity;
+    }
+    get(repoId) {
+        const modelPromise = (async () => {
+            const models = await this.finalize();
+            const model = models.find((model) => model.repoId === repoId);
+            return (0, error_1.unwrapOrThrow)(model, error_1.EngineErrorCode.ModelNotFound);
+        })();
+        return new modelInteractor_1.ModelInteractor(this, modelPromise, config_1.Config.default());
+    }
+}
+exports.ModelsInteractor = ModelsInteractor;
+//# sourceMappingURL=modelsInteractor.js.map

package/interactors/modelsInteractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelsInteractor.js","sourceRoot":"","sources":["../src/interactors/modelsInteractor.ts"],"names":[],"mappings":";;;AAAA,kDAA4C;AAE5C,uCAA0D;AAG1D,0DAAoD;AAEpD,MAAa,gBAAgB;IAIzB,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,iCAAe,CAAC,IAAI,EAAE,YAAY,EAAE,eAAM,CAAC,OAAO,EAAE,CAAC,CAAC;IACrE,CAAC;CACJ;AArBD,4CAqBC"}

package/interactors/modelsInteractor.mjs

@@ -0,0 +1,21 @@
+import { Config } from "../bridging/config.mjs";
+import { EngineErrorCode, unwrapOrThrow } from "../error.mjs";
+import { ModelInteractor } from "./modelInteractor.mjs";
+export class ModelsInteractor {
+    constructor(engineInteractor, models) {
+        this.engineInteractor = engineInteractor;
+        this.entity = models;
+    }
+    async finalize() {
+        return await this.entity;
+    }
+    get(repoId) {
+        const modelPromise = (async () => {
+            const models = await this.finalize();
+            const model = models.find((model) => model.repoId === repoId);
+            return unwrapOrThrow(model, EngineErrorCode.ModelNotFound);
+        })();
+        return new ModelInteractor(this, modelPromise, Config.default());
+    }
+}
+//# sourceMappingURL=modelsInteractor.mjs.map

package/interactors/modelsInteractor.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelsInteractor.mjs","sourceRoot":"","sources":["../src/interactors/modelsInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,EAAE;OAEV,EAAE,eAAe,EAAE,aAAa,EAAE;OAGlC,EAAE,eAAe,EAAE;AAE1B,MAAM,OAAO,gBAAgB;IAIzB,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,eAAe,CAAC,IAAI,EAAE,YAAY,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;IACrE,CAAC;CACJ"}

package/interactors/sessionInteractor.d.mts

@@ -0,0 +1,24 @@
+import { Input } from "../bridging/input.mjs";
+import { Message } from "../bridging/message.mjs";
+import { Output } from "../bridging/output.mjs";
+import { RunConfig } from "../bridging/runConfig.mjs";
+import { SamplingMethod } from "../bridging/samplingMethod.mjs";
+import { SamplingPolicy } from "../bridging/samplingPolicy.mjs";
+import { Session } from "../bridging/session.mjs";
+import { Interactor, InteractorEntity } from "./interactor.mjs";
+import { ModelInteractor } from "./modelInteractor.mjs";
+export declare class SessionInteractor implements Interactor<Session> {
+    readonly modelInteractor: ModelInteractor;
+    readonly entity: InteractorEntity<Session>;
+    readonly config: RunConfig;
+    constructor(modelInteractor: ModelInteractor, session: InteractorEntity<Session>, config: RunConfig);
+    finalize(): Promise<Session>;
+    tokensLimit(tokensLimit: number): SessionInteractor;
+    enableThinking(enableThinking: boolean): SessionInteractor;
+    samplingPolicy(samplingPolicy: SamplingPolicy): SessionInteractor;
+    samplingMethod(samplingMethod: SamplingMethod): SessionInteractor;
+    run(input: Input, progress?: (partialOutput: Output) => boolean): Promise<Output>;
+    reply(text: string, progress?: (partialOutput: Output) => boolean): Promise<Output>;
+    replyToMessages(messages: Message[], progress?: (partialOutput: Output) => boolean): Promise<Output>;
+}
+//# sourceMappingURL=sessionInteractor.d.mts.map

package/interactors/sessionInteractor.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionInteractor.d.mts","sourceRoot":"","sources":["../src/interactors/sessionInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,KAAK,EAAE;OACT,EAAE,OAAO,EAAE;OACX,EAAE,MAAM,EAAE;OACV,EAAE,SAAS,EAAE;OACb,EAAE,cAAc,EAAE;OAClB,EAAE,cAAc,EAAE;OAClB,EAAE,OAAO,EAAE;OACX,EAAE,UAAU,EAAE,gBAAgB,EAAE;OAChC,EAAE,eAAe,EAAE;AAE1B,qBAAa,iBAAkB,YAAW,UAAU,CAAC,OAAO,CAAC;IACzD,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;IAC1C,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC3C,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC;gBAEf,eAAe,EAAE,eAAe,EAAE,OAAO,EAAE,gBAAgB,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS;IAM7F,QAAQ,IAAI,OAAO,CAAC,OAAO,CAAC;IAMlC,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,iBAAiB;IAKnD,cAAc,CAAC,cAAc,EAAE,OAAO,GAAG,iBAAiB;IAK1D,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,iBAAiB;IAKjE,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,iBAAiB;IAO3D,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAAG,OAAO,CAAC,MAAM,CAAC;IAU7F,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAAG,OAAO,CAAC,MAAM,CAAC;IAI/F,eAAe,CACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAC1D,OAAO,CAAC,MAAM,CAAC;CAGrB"}

package/interactors/sessionInteractor.d.ts

@@ -0,0 +1,24 @@
+import { Input } from "../bridging/input.js";
+import { Message } from "../bridging/message.js";
+import { Output } from "../bridging/output.js";
+import { RunConfig } from "../bridging/runConfig.js";
+import { SamplingMethod } from "../bridging/samplingMethod.js";
+import { SamplingPolicy } from "../bridging/samplingPolicy.js";
+import { Session } from "../bridging/session.js";
+import { Interactor, InteractorEntity } from "./interactor.js";
+import { ModelInteractor } from "./modelInteractor.js";
+export declare class SessionInteractor implements Interactor<Session> {
+    readonly modelInteractor: ModelInteractor;
+    readonly entity: InteractorEntity<Session>;
+    readonly config: RunConfig;
+    constructor(modelInteractor: ModelInteractor, session: InteractorEntity<Session>, config: RunConfig);
+    finalize(): Promise<Session>;
+    tokensLimit(tokensLimit: number): SessionInteractor;
+    enableThinking(enableThinking: boolean): SessionInteractor;
+    samplingPolicy(samplingPolicy: SamplingPolicy): SessionInteractor;
+    samplingMethod(samplingMethod: SamplingMethod): SessionInteractor;
+    run(input: Input, progress?: (partialOutput: Output) => boolean): Promise<Output>;
+    reply(text: string, progress?: (partialOutput: Output) => boolean): Promise<Output>;
+    replyToMessages(messages: Message[], progress?: (partialOutput: Output) => boolean): Promise<Output>;
+}
+//# sourceMappingURL=sessionInteractor.d.ts.map

package/interactors/sessionInteractor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionInteractor.d.ts","sourceRoot":"","sources":["../src/interactors/sessionInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,KAAK,EAAE;OACT,EAAE,OAAO,EAAE;OACX,EAAE,MAAM,EAAE;OACV,EAAE,SAAS,EAAE;OACb,EAAE,cAAc,EAAE;OAClB,EAAE,cAAc,EAAE;OAClB,EAAE,OAAO,EAAE;OACX,EAAE,UAAU,EAAE,gBAAgB,EAAE;OAChC,EAAE,eAAe,EAAE;AAE1B,qBAAa,iBAAkB,YAAW,UAAU,CAAC,OAAO,CAAC;IACzD,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;IAC1C,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC3C,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC;gBAEf,eAAe,EAAE,eAAe,EAAE,OAAO,EAAE,gBAAgB,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS;IAM7F,QAAQ,IAAI,OAAO,CAAC,OAAO,CAAC;IAMlC,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,iBAAiB;IAKnD,cAAc,CAAC,cAAc,EAAE,OAAO,GAAG,iBAAiB;IAK1D,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,iBAAiB;IAKjE,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,iBAAiB;IAO3D,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAAG,OAAO,CAAC,MAAM,CAAC;IAU7F,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAAG,OAAO,CAAC,MAAM,CAAC;IAI/F,eAAe,CACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAC1D,OAAO,CAAC,MAAM,CAAC;CAGrB"}

package/interactors/sessionInteractor.js

@@ -0,0 +1,50 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SessionInteractor = void 0;
+const input_1 = require("../bridging/input.js");
+const samplingPolicy_1 = require("../bridging/samplingPolicy.js");
+class SessionInteractor {
+    constructor(modelInteractor, session, config) {
+        this.modelInteractor = modelInteractor;
+        this.entity = session;
+        this.config = config;
+    }
+    async finalize() {
+        return await this.entity;
+    }
+    /* Config */
+    tokensLimit(tokensLimit) {
+        const config = this.config.withTokensLimit(tokensLimit);
+        return new SessionInteractor(this.modelInteractor, this.entity, config);
+    }
+    enableThinking(enableThinking) {
+        const config = this.config.withEnableThinking(enableThinking);
+        return new SessionInteractor(this.modelInteractor, this.entity, config);
+    }
+    samplingPolicy(samplingPolicy) {
+        const config = this.config.withSamplingPolicy(samplingPolicy);
+        return new SessionInteractor(this.modelInteractor, this.entity, config);
+    }
+    samplingMethod(samplingMethod) {
+        const config = this.config.withSamplingPolicy(samplingPolicy_1.SamplingPolicy.custom(samplingMethod));
+        return new SessionInteractor(this.modelInteractor, this.entity, config);
+    }
+    /* Run */
+    async run(input, progress = () => true) {
+        const outputPromise = (async () => {
+            const session = await this.finalize();
+            const output = session.run(input, this.config, progress);
+            return output;
+        })();
+        const output = await outputPromise;
+        return output;
+    }
+    async reply(text, progress = () => true) {
+        return await this.run(input_1.Input.text(text), progress);
+    }
+    async replyToMessages(messages, progress = () => true) {
+        return await this.run(input_1.Input.messages(messages), progress);
+    }
+}
+exports.SessionInteractor = SessionInteractor;
+//# sourceMappingURL=sessionInteractor.js.map

package/interactors/sessionInteractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionInteractor.js","sourceRoot":"","sources":["../src/interactors/sessionInteractor.ts"],"names":[],"mappings":";;;AAAA,gDAA0C;AAK1C,kEAA4D;AAK5D,MAAa,iBAAiB;IAK1B,YAAY,eAAgC,EAAE,OAAkC,EAAE,MAAiB;QAC/F,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY;IAEZ,WAAW,CAAC,WAAmB;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACxD,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,cAAc,CAAC,cAAuB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,+BAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;QACrF,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,SAAS;IAET,KAAK,CAAC,GAAG,CAAC,KAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC7E,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YACzD,OAAO,MAAM,CAAC;QAClB,CAAC,CAAC,EAAE,CAAC;QACL,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;QACnC,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC/E,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,aAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,eAAe,CACjB,QAAmB,EACnB,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,aAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC;IAC9D,CAAC;CACJ;AA3DD,8CA2DC"}

package/interactors/sessionInteractor.mjs

@@ -0,0 +1,46 @@
+import { Input } from "../bridging/input.mjs";
+import { SamplingPolicy } from "../bridging/samplingPolicy.mjs";
+export class SessionInteractor {
+    constructor(modelInteractor, session, config) {
+        this.modelInteractor = modelInteractor;
+        this.entity = session;
+        this.config = config;
+    }
+    async finalize() {
+        return await this.entity;
+    }
+    /* Config */
+    tokensLimit(tokensLimit) {
+        const config = this.config.withTokensLimit(tokensLimit);
+        return new SessionInteractor(this.modelInteractor, this.entity, config);
+    }
+    enableThinking(enableThinking) {
+        const config = this.config.withEnableThinking(enableThinking);
+        return new SessionInteractor(this.modelInteractor, this.entity, config);
+    }
+    samplingPolicy(samplingPolicy) {
+        const config = this.config.withSamplingPolicy(samplingPolicy);
+        return new SessionInteractor(this.modelInteractor, this.entity, config);
+    }
+    samplingMethod(samplingMethod) {
+        const config = this.config.withSamplingPolicy(SamplingPolicy.custom(samplingMethod));
+        return new SessionInteractor(this.modelInteractor, this.entity, config);
+    }
+    /* Run */
+    async run(input, progress = () => true) {
+        const outputPromise = (async () => {
+            const session = await this.finalize();
+            const output = session.run(input, this.config, progress);
+            return output;
+        })();
+        const output = await outputPromise;
+        return output;
+    }
+    async reply(text, progress = () => true) {
+        return await this.run(Input.text(text), progress);
+    }
+    async replyToMessages(messages, progress = () => true) {
+        return await this.run(Input.messages(messages), progress);
+    }
+}
+//# sourceMappingURL=sessionInteractor.mjs.map

package/interactors/sessionInteractor.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionInteractor.mjs","sourceRoot":"","sources":["../src/interactors/sessionInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,KAAK,EAAE;OAKT,EAAE,cAAc,EAAE;AAKzB,MAAM,OAAO,iBAAiB;IAK1B,YAAY,eAAgC,EAAE,OAAkC,EAAE,MAAiB;QAC/F,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY;IAEZ,WAAW,CAAC,WAAmB;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACxD,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,cAAc,CAAC,cAAuB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;QACrF,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAED,SAAS;IAET,KAAK,CAAC,GAAG,CAAC,KAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC7E,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YACzD,OAAO,MAAM,CAAC;QAClB,CAAC,CAAC,EAAE,CAAC;QACL,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;QACnC,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC/E,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,eAAe,CACjB,QAAmB,EACnB,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC;IAC9D,CAAC;CACJ"}

package/internal/tslib.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/internal/tslib.mjs

@@ -0,0 +1 @@
+export {};

package/{uzu.d.ts → napi/uzu.d.ts}

@@ -1,35 +1,30 @@
 /* auto-generated by NAPI-RS */
 /* eslint-disable */
-export declare class DownloadHandle {
-  start(): void
-  pause(): void
-  resume(): void
-  stop(): void
-  delete(): void
-  state(): ModelDownloadState
-  identifier(): string
-  progress(): AsyncIterable<ProgressUpdate>
-}
-
 /** Thin FFI wrapper around `ModelStorage`. */
 export declare class Engine {
   getCloudModels(): Promise<Array<CloudModel>>
   fetchCloudModels(): Promise<Array<CloudModel>>
   /** Returns a `DownloadHandle` for the given model identifier. */
-  downloadHandle(identifier: string): DownloadHandle
+  downloadHandle(identifier: string): ModelDownloadHandle
   getLocalModels(): Array<LocalModel>
   getState(identifier: string): ModelDownloadState
-  download(identifier: string): void
-  pause(identifier: string): void
-  resume(identifier: string): void
-  stop(identifier: string): void
-  delete(identifier: string): void
-  updateRegistry(): Promise<Array<LocalModel>>
   activate(apiKey: string): Promise<LicenseStatus>
+  downloadModel(identifier: string): Promise<void>
+  pauseModel(identifier: string): Promise<void>
+  deleteModel(identifier: string): Promise<void>
   constructor()
   createSession(modelId: ModelID, config: Config): Session
 }
 
+export declare class ModelDownloadHandle {
+  identifier(): string
+  download(): Promise<void>
+  pause(): Promise<void>
+  delete(): Promise<void>
+  state(): Promise<ModelDownloadState>
+  progress(): AsyncIterable<ProgressUpdate>
+}
+
 export declare class ProgressStream {
 
 }
@@ -70,12 +65,6 @@ export type ContextLength =
   | { type: 'Maximal' }
   | { type: 'Custom', length: number }
 
-export type DownloaderError =
-  | { type: 'Http', message: string }
-  | { type: 'Io', message: string }
-  | { type: 'AlreadyExists', path: string }
-  | { type: 'Generic', message: string }
-
 export declare const enum FinishReason {
   Stop = 0,
   Length = 1,
@@ -99,12 +88,14 @@ export type LicenseStatus =
   | { type: 'HttpError', code: number }
 
 export interface LocalModel {
-  /** Unique identifier of the model in the form `<vendor>-<name>`. */
+  /** Unique identifier of the model in the form `<vendor>-<name>-<precision>`. */
   readonly identifier: string
-  /** Vendor/author of the model (e.g. "Llama"). */
+  /** Vendor/author of the model (e.g. "Alibaba", "Meta"). */
   readonly vendor: string
-  /** Human-readable model name without vendor (e.g. "3B-Instruct"). */
+  /** Human-readable model name without vendor (e.g. "Qwen2.5-0.5B-Instruct"). */
   readonly name: string
+  /** Precision of the model (e.g. "bfloat16", "float16"). */
+  readonly precision: string
   /** Quantization type if the model is quantized (e.g. "uint4"). */
   readonly quantization?: string
   /** Optional regex to parse model output provided by the backend. */
@@ -119,14 +110,22 @@ export interface Message {
   reasoningContent?: string
 }
 
+export declare const enum ModelDownloadPhase {
+  NotDownloaded = 0,
+  Downloading = 1,
+  Paused = 2,
+  Downloaded = 3,
+  Error = 4
+}
+
 export interface ModelDownloadState {
   /** Total size of all model files in kilobytes. */
   readonly totalKbytes: number
   /** Bytes already downloaded (meaningful in Downloading / Paused). */
   readonly downloadedKbytes: number
   /** Current phase of the download. */
-  readonly phase: Phase
-  /** Stringified error when `phase == Phase::Error`. */
+  readonly phase: ModelDownloadPhase
+  /** Stringified error when `phase == ModelDownloadPhase::Error`. */
   readonly error?: string
 }
 
@@ -134,6 +133,16 @@ export type ModelID =
   | { type: 'Local', id: string }
   | { type: 'Cloud', id: string }
 
+export type ModelStorageError =
+  | { type: 'Io', message: string }
+  | { type: 'Network', message: string }
+  | { type: 'DownloadFailed', message: string }
+  | { type: 'Download', message: string }
+  | { type: 'InvalidPath', message: string }
+  | { type: 'ModelNotFound', identifier: string }
+  | { type: 'FileVerificationFailed', message: string }
+  | { type: 'Paused' }
+
 export interface Output {
   text: Text
   stats: Stats
@@ -145,14 +154,6 @@ export interface ParsedText {
   response?: string
 }
 
-export declare const enum Phase {
-  NotDownloaded = 0,
-  Downloading = 1,
-  Paused = 2,
-  Downloaded = 3,
-  Error = 4
-}
-
 export type PrefillStepSize =
   | { type: 'Default' }
   | { type: 'Maximal' }