@trymirai/uzu 0.2.10 → 0.3.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +52 -6
- package/bridging/classificationLabel.d.mts +12 -0
- package/bridging/classificationLabel.d.mts.map +1 -0
- package/bridging/classificationLabel.d.ts +12 -0
- package/bridging/classificationLabel.d.ts.map +1 -0
- package/bridging/classificationLabel.js +26 -0
- package/bridging/classificationLabel.js.map +1 -0
- package/bridging/classificationLabel.mjs +22 -0
- package/bridging/classificationLabel.mjs.map +1 -0
- package/bridging/classificationOutput.d.mts +10 -0
- package/bridging/classificationOutput.d.mts.map +1 -0
- package/bridging/classificationOutput.d.ts +10 -0
- package/bridging/classificationOutput.d.ts.map +1 -0
- package/bridging/classificationOutput.js +16 -0
- package/bridging/classificationOutput.js.map +1 -0
- package/bridging/classificationOutput.mjs +12 -0
- package/bridging/classificationOutput.mjs.map +1 -0
- package/bridging/classificationSession.d.mts +9 -0
- package/bridging/classificationSession.d.mts.map +1 -0
- package/bridging/classificationSession.d.ts +9 -0
- package/bridging/classificationSession.d.ts.map +1 -0
- package/bridging/classificationSession.js +16 -0
- package/bridging/classificationSession.js.map +1 -0
- package/bridging/classificationSession.mjs +12 -0
- package/bridging/classificationSession.mjs.map +1 -0
- package/bridging/classificationStats.d.mts +14 -0
- package/bridging/classificationStats.d.mts.map +1 -0
- package/bridging/classificationStats.d.ts +14 -0
- package/bridging/classificationStats.d.ts.map +1 -0
- package/bridging/classificationStats.js +20 -0
- package/bridging/classificationStats.js.map +1 -0
- package/bridging/classificationStats.mjs +16 -0
- package/bridging/classificationStats.mjs.map +1 -0
- package/bridging/engine.d.mts +14 -6
- package/bridging/engine.d.mts.map +1 -1
- package/bridging/engine.d.ts +14 -6
- package/bridging/engine.d.ts.map +1 -1
- package/bridging/engine.js +53 -23
- package/bridging/engine.js.map +1 -1
- package/bridging/engine.mjs +53 -23
- package/bridging/engine.mjs.map +1 -1
- package/bridging/model.d.mts +19 -0
- package/bridging/model.d.mts.map +1 -0
- package/bridging/model.d.ts +19 -0
- package/bridging/model.d.ts.map +1 -0
- package/bridging/model.js +39 -0
- package/bridging/model.js.map +1 -0
- package/bridging/model.mjs +35 -0
- package/bridging/model.mjs.map +1 -0
- package/bridging/modelKind.d.mts +8 -0
- package/bridging/modelKind.d.mts.map +1 -0
- package/bridging/modelKind.d.ts +8 -0
- package/bridging/modelKind.d.ts.map +1 -0
- package/bridging/modelKind.js +30 -0
- package/bridging/modelKind.js.map +1 -0
- package/bridging/modelKind.mjs +25 -0
- package/bridging/modelKind.mjs.map +1 -0
- package/bridging/modelType.d.mts +3 -2
- package/bridging/modelType.d.mts.map +1 -1
- package/bridging/modelType.d.ts +3 -2
- package/bridging/modelType.d.ts.map +1 -1
- package/bridging/modelType.js +11 -6
- package/bridging/modelType.js.map +1 -1
- package/bridging/modelType.mjs +11 -6
- package/bridging/modelType.mjs.map +1 -1
- package/bridging/preset.d.mts +2 -0
- package/bridging/preset.d.mts.map +1 -1
- package/bridging/preset.d.ts +2 -0
- package/bridging/preset.d.ts.map +1 -1
- package/bridging/preset.js +12 -0
- package/bridging/preset.js.map +1 -1
- package/bridging/preset.mjs +12 -0
- package/bridging/preset.mjs.map +1 -1
- package/bridging/samplingMethod.d.mts +2 -1
- package/bridging/samplingMethod.d.mts.map +1 -1
- package/bridging/samplingMethod.d.ts +2 -1
- package/bridging/samplingMethod.d.ts.map +1 -1
- package/bridging/samplingMethod.js +3 -1
- package/bridging/samplingMethod.js.map +1 -1
- package/bridging/samplingMethod.mjs +3 -1
- package/bridging/samplingMethod.mjs.map +1 -1
- package/bridging/samplingProcessingOrder.d.mts +8 -0
- package/bridging/samplingProcessingOrder.d.mts.map +1 -0
- package/bridging/samplingProcessingOrder.d.ts +8 -0
- package/bridging/samplingProcessingOrder.d.ts.map +1 -0
- package/bridging/samplingProcessingOrder.js +30 -0
- package/bridging/samplingProcessingOrder.js.map +1 -0
- package/bridging/samplingProcessingOrder.mjs +25 -0
- package/bridging/samplingProcessingOrder.mjs.map +1 -0
- package/bridging/textToSpeechSession.d.mts +8 -0
- package/bridging/textToSpeechSession.d.mts.map +1 -0
- package/bridging/textToSpeechSession.d.ts +8 -0
- package/bridging/textToSpeechSession.d.ts.map +1 -0
- package/bridging/textToSpeechSession.js +14 -0
- package/bridging/textToSpeechSession.js.map +1 -0
- package/bridging/textToSpeechSession.mjs +10 -0
- package/bridging/textToSpeechSession.mjs.map +1 -0
- package/error.d.mts +2 -0
- package/error.d.mts.map +1 -1
- package/error.d.ts +2 -0
- package/error.d.ts.map +1 -1
- package/error.js +4 -0
- package/error.js.map +1 -1
- package/error.mjs +4 -0
- package/error.mjs.map +1 -1
- package/index.d.mts +8 -1
- package/index.d.mts.map +1 -1
- package/index.d.ts +8 -1
- package/index.d.ts.map +1 -1
- package/index.js +17 -3
- package/index.js.map +1 -1
- package/index.mjs +8 -1
- package/index.mjs.map +1 -1
- package/interactors/chatModelInteractor.d.mts +5 -5
- package/interactors/chatModelInteractor.d.mts.map +1 -1
- package/interactors/chatModelInteractor.d.ts +5 -5
- package/interactors/chatModelInteractor.d.ts.map +1 -1
- package/interactors/chatModelInteractor.js +2 -2
- package/interactors/chatModelInteractor.js.map +1 -1
- package/interactors/chatModelInteractor.mjs +2 -2
- package/interactors/chatModelInteractor.mjs.map +1 -1
- package/interactors/chatModelsInteractor.d.mts +5 -5
- package/interactors/chatModelsInteractor.d.mts.map +1 -1
- package/interactors/chatModelsInteractor.d.ts +5 -5
- package/interactors/chatModelsInteractor.d.ts.map +1 -1
- package/interactors/chatModelsInteractor.js.map +1 -1
- package/interactors/chatModelsInteractor.mjs.map +1 -1
- package/interactors/engineInteractor.d.mts +4 -4
- package/interactors/engineInteractor.d.ts +4 -4
- package/interactors/engineInteractor.js +5 -5
- package/interactors/engineInteractor.mjs +5 -5
- package/napi/uzu.d.ts +69 -15
- package/napi/uzu.js +16 -2
- package/napi/uzu.mjs +8 -1
- package/napi/uzu.node +0 -0
- package/package.json +1 -1
- package/src/bridging/classificationLabel.ts +43 -0
- package/src/bridging/classificationOutput.ts +22 -0
- package/src/bridging/classificationSession.ts +17 -0
- package/src/bridging/classificationStats.ts +45 -0
- package/src/bridging/engine.ts +62 -26
- package/src/bridging/model.ts +68 -0
- package/src/bridging/modelKind.ts +27 -0
- package/src/bridging/modelType.ts +11 -6
- package/src/bridging/preset.ts +14 -0
- package/src/bridging/samplingMethod.ts +6 -0
- package/src/bridging/samplingProcessingOrder.ts +31 -0
- package/src/bridging/textToSpeechSession.ts +15 -0
- package/src/error.ts +4 -0
- package/src/index.ts +8 -1
- package/src/interactors/chatModelInteractor.ts +7 -7
- package/src/interactors/chatModelsInteractor.ts +5 -5
- package/src/interactors/engineInteractor.ts +7 -7
- package/src/napi/uzu.d.ts +69 -15
- package/src/napi/uzu.js +16 -2
- package/src/napi/uzu.mjs +8 -1
- package/src/napi/uzu.node +0 -0
- package/src/version.ts +1 -1
- package/version.d.mts +1 -1
- package/version.d.mts.map +1 -1
- package/version.d.ts +1 -1
- package/version.d.ts.map +1 -1
- package/version.js +1 -1
- package/version.js.map +1 -1
- package/version.mjs +1 -1
- package/version.mjs.map +1 -1
- package/bridging/chatModel.d.mts +0 -15
- package/bridging/chatModel.d.mts.map +0 -1
- package/bridging/chatModel.d.ts +0 -15
- package/bridging/chatModel.d.ts.map +0 -1
- package/bridging/chatModel.js +0 -33
- package/bridging/chatModel.js.map +0 -1
- package/bridging/chatModel.mjs +0 -29
- package/bridging/chatModel.mjs.map +0 -1
- package/src/bridging/chatModel.ts +0 -54
package/bridging/engine.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"engine.js","sourceRoot":"","sources":["../src/bridging/engine.ts"],"names":[],"mappings":";;;AAAA,uCAAuE;AACvE,yEAAmE;AACnE,wCAAmD;AACnD,6CAA2C;AAC3C,
|
|
1
|
+
{"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"}
|
package/bridging/engine.mjs
CHANGED
|
@@ -2,13 +2,15 @@ import { EngineError, EngineErrorCode, unwrapOrThrow } from "../error.mjs";
|
|
|
2
2
|
import { EngineInteractor } from "../interactors/engineInteractor.mjs";
|
|
3
3
|
import { Engine as NapiEngine } from "../napi/uzu.mjs";
|
|
4
4
|
import { readEnv } from "../utilities/env.mjs";
|
|
5
|
-
import { ChatModel } from "./chatModel.mjs";
|
|
6
5
|
import { ChatSession } from "./chatSession.mjs";
|
|
6
|
+
import { ClassificationSession } from "./classificationSession.mjs";
|
|
7
7
|
import { Config } from "./config.mjs";
|
|
8
8
|
import { DownloadHandle } from "./downloadHandle.mjs";
|
|
9
9
|
import { DownloadPhase, DownloadState } from "./downloadState.mjs";
|
|
10
10
|
import { LicenseStatus, licenseStatusFromNapiLicenseStatus } from "./licenseStatus.mjs";
|
|
11
|
-
import {
|
|
11
|
+
import { Model } from "./model.mjs";
|
|
12
|
+
import { ModelKind, modelKindToNapiModelKind } from "./modelKind.mjs";
|
|
13
|
+
import { TextToSpeechSession } from "./textToSpeechSession.mjs";
|
|
12
14
|
export class Engine {
|
|
13
15
|
constructor(napiEngine) {
|
|
14
16
|
this.napiEngine = napiEngine;
|
|
@@ -40,26 +42,14 @@ export class Engine {
|
|
|
40
42
|
downloadHandle(repoId) {
|
|
41
43
|
return new DownloadHandle(this.napiEngine.createModelDownloadHandle(repoId));
|
|
42
44
|
}
|
|
43
|
-
async
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
let napiChatModels = await this.napiEngine.getChatModels(napiModelTypes);
|
|
47
|
-
let results = napiChatModels.map(ChatModel.fromNapiChatModel);
|
|
48
|
-
return results;
|
|
49
|
-
}
|
|
50
|
-
async chatModel(repoId, types = null) {
|
|
51
|
-
const chatModels = await this.chatModels(types);
|
|
52
|
-
const chatModel = chatModels.find((model) => model.repoId === repoId);
|
|
53
|
-
return unwrapOrThrow(chatModel, EngineErrorCode.ModelNotFound);
|
|
54
|
-
}
|
|
55
|
-
async downloadChatModel(chatModel, progress = () => { }) {
|
|
56
|
-
switch (chatModel.type) {
|
|
57
|
-
case ModelType.Local:
|
|
45
|
+
async downloadModel(model, progress = () => { }) {
|
|
46
|
+
switch (model.kind) {
|
|
47
|
+
case ModelKind.Local:
|
|
58
48
|
break;
|
|
59
|
-
case
|
|
60
|
-
throw new EngineError(EngineErrorCode.
|
|
49
|
+
case ModelKind.Cloud:
|
|
50
|
+
throw new EngineError(EngineErrorCode.UnexpectedModelKind);
|
|
61
51
|
}
|
|
62
|
-
const downloadHandle = this.downloadHandle(
|
|
52
|
+
const downloadHandle = this.downloadHandle(model.repoId);
|
|
63
53
|
const state = await downloadHandle.state();
|
|
64
54
|
switch (state.phase) {
|
|
65
55
|
case DownloadPhase.Downloaded:
|
|
@@ -74,10 +64,50 @@ export class Engine {
|
|
|
74
64
|
const finalState = await downloadHandle.state();
|
|
75
65
|
return finalState;
|
|
76
66
|
}
|
|
77
|
-
|
|
78
|
-
|
|
67
|
+
async chatModels(kinds = null) {
|
|
68
|
+
let kindsToFilter = kinds ?? [ModelKind.Local, ModelKind.Cloud];
|
|
69
|
+
let napiModelKinds = kindsToFilter.map(modelKindToNapiModelKind);
|
|
70
|
+
let napiChatModels = await this.napiEngine.getChatModels(napiModelKinds);
|
|
71
|
+
let results = napiChatModels.map(Model.fromNapiModel);
|
|
72
|
+
return results;
|
|
73
|
+
}
|
|
74
|
+
async chatModel(repoId, kinds = null) {
|
|
75
|
+
const chatModels = await this.chatModels(kinds);
|
|
76
|
+
const chatModel = chatModels.find((model) => model.repoId === repoId);
|
|
77
|
+
return unwrapOrThrow(chatModel, EngineErrorCode.ModelNotFound);
|
|
78
|
+
}
|
|
79
|
+
chatSession(model, config = Config.default()) {
|
|
80
|
+
const napiModel = model.toNapi();
|
|
79
81
|
const napiConfig = config.toNapi();
|
|
80
|
-
return new ChatSession(this.napiEngine.createChatSession(
|
|
82
|
+
return new ChatSession(this.napiEngine.createChatSession(napiModel, napiConfig));
|
|
83
|
+
}
|
|
84
|
+
async classificationModels() {
|
|
85
|
+
const napiClassificationModels = await this.napiEngine.getClassificationModels();
|
|
86
|
+
const results = napiClassificationModels.map(Model.fromNapiModel);
|
|
87
|
+
return results;
|
|
88
|
+
}
|
|
89
|
+
async classificationModel(repoId) {
|
|
90
|
+
const classificationModels = await this.classificationModels();
|
|
91
|
+
const classificationModel = classificationModels.find((model) => model.repoId === repoId);
|
|
92
|
+
return unwrapOrThrow(classificationModel, EngineErrorCode.ModelNotFound);
|
|
93
|
+
}
|
|
94
|
+
classificationSession(model) {
|
|
95
|
+
const napiModel = model.toNapi();
|
|
96
|
+
return new ClassificationSession(this.napiEngine.createClassificationSession(napiModel));
|
|
97
|
+
}
|
|
98
|
+
async textToSpeechModels() {
|
|
99
|
+
const napiTextToSpeechModels = await this.napiEngine.getTextToSpeechModels();
|
|
100
|
+
const results = napiTextToSpeechModels.map(Model.fromNapiModel);
|
|
101
|
+
return results;
|
|
102
|
+
}
|
|
103
|
+
async textToSpeechModel(repoId) {
|
|
104
|
+
const textToSpeechModels = await this.textToSpeechModels();
|
|
105
|
+
const textToSpeechModel = textToSpeechModels.find((model) => model.repoId === repoId);
|
|
106
|
+
return unwrapOrThrow(textToSpeechModel, EngineErrorCode.ModelNotFound);
|
|
107
|
+
}
|
|
108
|
+
textToSpeechSession(model) {
|
|
109
|
+
const napiModel = model.toNapi();
|
|
110
|
+
return new TextToSpeechSession(this.napiEngine.createTextToSpeechSession(napiModel));
|
|
81
111
|
}
|
|
82
112
|
}
|
|
83
113
|
//# sourceMappingURL=engine.mjs.map
|
package/bridging/engine.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"engine.mjs","sourceRoot":"","sources":["../src/bridging/engine.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,aAAa,EAAE;OAC/C,EAAE,gBAAgB,EAAE;OACpB,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,OAAO,EAAE;OACX,EAAE,
|
|
1
|
+
{"version":3,"file":"engine.mjs","sourceRoot":"","sources":["../src/bridging/engine.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,aAAa,EAAE;OAC/C,EAAE,gBAAgB,EAAE;OACpB,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,OAAO,EAAE;OACX,EAAE,WAAW,EAAE;OACf,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;OACV,EAAE,cAAc,EAAE;OAElB,EAAE,aAAa,EAAE,aAAa,EAAE;OAChC,EAAE,aAAa,EAAE,kCAAkC,EAAE;OACrD,EAAE,KAAK,EAAE;OACT,EAAE,SAAS,EAAE,wBAAwB,EAAE;OACvC,EAAE,mBAAmB,EAAE;AAE9B,MAAM,OAAO,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,aAAa,CAChC,MAAM,IAAI,OAAO,CAAC,eAAe,CAAC,EAClC,eAAe,CAAC,cAAc,CACjC,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;QAEpC,MAAM,iBAAiB,GAAG,MAAM,UAAU,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QACpE,MAAM,aAAa,GAAG,kCAAkC,CAAC,iBAAiB,CAAC,CAAC;QAC5E,QAAQ,aAAa,EAAE,CAAC;YACpB,KAAK,aAAa,CAAC,SAAS,CAAC;YAC7B,KAAK,aAAa,CAAC,iBAAiB;gBAChC,MAAM;YACV;gBACI,MAAM,IAAI,WAAW,CAAC,eAAe,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,gBAAgB,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,aAAa,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IACjD,CAAC;IAED,cAAc,CAAC,MAAc;QACzB,OAAO,IAAI,cAAc,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,SAAS,CAAC,KAAK;gBAChB,MAAM;YACV,KAAK,SAAS,CAAC,KAAK;gBAChB,MAAM,IAAI,WAAW,CAAC,eAAe,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,aAAa,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,SAAS,CAAC,KAAK,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC;QAC7E,IAAI,cAAc,GAAG,aAAa,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QACjE,IAAI,cAAc,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC;QACzE,IAAI,OAAO,GAAY,cAAc,CAAC,GAAG,CAAC,KAAK,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,aAAa,CAAC,SAAS,EAAE,eAAe,CAAC,aAAa,CAAC,CAAC;IACnE,CAAC;IAED,WAAW,CAAC,KAAY,EAAE,SAAiB,MAAM,CAAC,OAAO,EAAE;QACvD,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QACnC,OAAO,IAAI,WAAW,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,KAAK,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,aAAa,CAAC,mBAAmB,EAAE,eAAe,CAAC,aAAa,CAAC,CAAC;IAC7E,CAAC;IAED,qBAAqB,CAAC,KAAY;QAC9B,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,OAAO,IAAI,qBAAqB,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,KAAK,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,aAAa,CAAC,iBAAiB,EAAE,eAAe,CAAC,aAAa,CAAC,CAAC;IAC3E,CAAC;IAED,mBAAmB,CAAC,KAAY;QAC5B,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,UAAU,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,CAAC;IACzF,CAAC;CACJ"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import { Model as NapiModel } from "../napi/uzu.mjs";
|
|
2
|
+
import { ClassificationLabel } from "./classificationLabel.mjs";
|
|
3
|
+
import { ModelKind } from "./modelKind.mjs";
|
|
4
|
+
import { ModelType } from "./modelType.mjs";
|
|
5
|
+
export declare class Model {
|
|
6
|
+
readonly repoId: string;
|
|
7
|
+
readonly type: ModelType;
|
|
8
|
+
readonly kind: ModelKind;
|
|
9
|
+
readonly name: string;
|
|
10
|
+
readonly vendor: string;
|
|
11
|
+
readonly quantization: string | null;
|
|
12
|
+
readonly outputParserRegex: string | null;
|
|
13
|
+
readonly classificationLabels: ClassificationLabel[];
|
|
14
|
+
get isThinking(): boolean;
|
|
15
|
+
private constructor();
|
|
16
|
+
static fromNapiModel(napiModel: NapiModel): Model;
|
|
17
|
+
toNapi(): NapiModel;
|
|
18
|
+
}
|
|
19
|
+
//# sourceMappingURL=model.d.mts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model.d.mts","sourceRoot":"","sources":["../src/bridging/model.ts"],"names":[],"mappings":"OAAO,EAAE,KAAK,IAAI,SAAS,EAAE;OACtB,EAAE,mBAAmB,EAAE;OACvB,EAAE,SAAS,EAAwD;OACnE,EAAE,SAAS,EAAwD;AAE1E,qBAAa,KAAK;IACd,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;IACzB,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;IACzB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IACrC,QAAQ,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1C,QAAQ,CAAC,oBAAoB,EAAE,mBAAmB,EAAE,CAAC;IAErD,IAAI,UAAU,IAAI,OAAO,CAExB;IAED,OAAO;IAoBP,MAAM,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,GAAG,KAAK;IAajD,MAAM,IAAI,SAAS;CAetB"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import { Model as NapiModel } from "../napi/uzu.js";
|
|
2
|
+
import { ClassificationLabel } from "./classificationLabel.js";
|
|
3
|
+
import { ModelKind } from "./modelKind.js";
|
|
4
|
+
import { ModelType } from "./modelType.js";
|
|
5
|
+
export declare class Model {
|
|
6
|
+
readonly repoId: string;
|
|
7
|
+
readonly type: ModelType;
|
|
8
|
+
readonly kind: ModelKind;
|
|
9
|
+
readonly name: string;
|
|
10
|
+
readonly vendor: string;
|
|
11
|
+
readonly quantization: string | null;
|
|
12
|
+
readonly outputParserRegex: string | null;
|
|
13
|
+
readonly classificationLabels: ClassificationLabel[];
|
|
14
|
+
get isThinking(): boolean;
|
|
15
|
+
private constructor();
|
|
16
|
+
static fromNapiModel(napiModel: NapiModel): Model;
|
|
17
|
+
toNapi(): NapiModel;
|
|
18
|
+
}
|
|
19
|
+
//# sourceMappingURL=model.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../src/bridging/model.ts"],"names":[],"mappings":"OAAO,EAAE,KAAK,IAAI,SAAS,EAAE;OACtB,EAAE,mBAAmB,EAAE;OACvB,EAAE,SAAS,EAAwD;OACnE,EAAE,SAAS,EAAwD;AAE1E,qBAAa,KAAK;IACd,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;IACzB,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;IACzB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IACrC,QAAQ,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1C,QAAQ,CAAC,oBAAoB,EAAE,mBAAmB,EAAE,CAAC;IAErD,IAAI,UAAU,IAAI,OAAO,CAExB;IAED,OAAO;IAoBP,MAAM,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,GAAG,KAAK;IAajD,MAAM,IAAI,SAAS;CAetB"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.Model = void 0;
|
|
4
|
+
const classificationLabel_1 = require("./classificationLabel.js");
|
|
5
|
+
const modelKind_1 = require("./modelKind.js");
|
|
6
|
+
const modelType_1 = require("./modelType.js");
|
|
7
|
+
class Model {
|
|
8
|
+
get isThinking() {
|
|
9
|
+
return this.outputParserRegex !== null;
|
|
10
|
+
}
|
|
11
|
+
constructor(repoId, type, kind, name, vendor, quantization, outputParserRegex, classificationLabels) {
|
|
12
|
+
this.repoId = repoId;
|
|
13
|
+
this.type = type;
|
|
14
|
+
this.kind = kind;
|
|
15
|
+
this.name = name;
|
|
16
|
+
this.vendor = vendor;
|
|
17
|
+
this.quantization = quantization;
|
|
18
|
+
this.outputParserRegex = outputParserRegex;
|
|
19
|
+
this.classificationLabels = classificationLabels;
|
|
20
|
+
}
|
|
21
|
+
static fromNapiModel(napiModel) {
|
|
22
|
+
return new Model(napiModel.repoId, (0, modelType_1.modelTypeFromNapiModelType)(napiModel.type), (0, modelKind_1.modelKindFromNapiModelKind)(napiModel.kind), napiModel.name, napiModel.vendor, napiModel.quantization ?? null, napiModel.outputParserRegex ?? null, napiModel.classificationLabels.map(classificationLabel_1.ClassificationLabel.fromNapi));
|
|
23
|
+
}
|
|
24
|
+
toNapi() {
|
|
25
|
+
const napiModel = {
|
|
26
|
+
repoId: this.repoId,
|
|
27
|
+
type: (0, modelType_1.modelTypeToNapiModelType)(this.type),
|
|
28
|
+
kind: (0, modelKind_1.modelKindToNapiModelKind)(this.kind),
|
|
29
|
+
name: this.name,
|
|
30
|
+
vendor: this.vendor,
|
|
31
|
+
...(this.quantization !== null && { quantization: this.quantization }),
|
|
32
|
+
...(this.outputParserRegex !== null && { outputParserRegex: this.outputParserRegex }),
|
|
33
|
+
classificationLabels: this.classificationLabels.map((classificationLabel) => classificationLabel.toNapi()),
|
|
34
|
+
};
|
|
35
|
+
return napiModel;
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
exports.Model = Model;
|
|
39
|
+
//# sourceMappingURL=model.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model.js","sourceRoot":"","sources":["../src/bridging/model.ts"],"names":[],"mappings":";;;AACA,kEAA4D;AAC5D,8CAA8F;AAC9F,8CAA8F;AAE9F,MAAa,KAAK;IAUd,IAAI,UAAU;QACV,OAAO,IAAI,CAAC,iBAAiB,KAAK,IAAI,CAAC;IAC3C,CAAC;IAED,YACI,MAAc,EACd,IAAe,EACf,IAAe,EACf,IAAY,EACZ,MAAc,EACd,YAA2B,EAC3B,iBAAgC,EAChC,oBAA2C;QAE3C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;IACrD,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,SAAoB;QACrC,OAAO,IAAI,KAAK,CACZ,SAAS,CAAC,MAAM,EAChB,IAAA,sCAA0B,EAAC,SAAS,CAAC,IAAI,CAAC,EAC1C,IAAA,sCAA0B,EAAC,SAAS,CAAC,IAAI,CAAC,EAC1C,SAAS,CAAC,IAAI,EACd,SAAS,CAAC,MAAM,EAChB,SAAS,CAAC,YAAY,IAAI,IAAI,EAC9B,SAAS,CAAC,iBAAiB,IAAI,IAAI,EACnC,SAAS,CAAC,oBAAoB,CAAC,GAAG,CAAC,yCAAmB,CAAC,QAAQ,CAAC,CACnE,CAAC;IACN,CAAC;IAED,MAAM;QACF,MAAM,SAAS,GAAc;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,IAAA,oCAAwB,EAAC,IAAI,CAAC,IAAI,CAAC;YACzC,IAAI,EAAE,IAAA,oCAAwB,EAAC,IAAI,CAAC,IAAI,CAAC;YACzC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,IAAI,IAAI,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC;YACtE,GAAG,CAAC,IAAI,CAAC,iBAAiB,KAAK,IAAI,IAAI,EAAE,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACrF,oBAAoB,EAAE,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,mBAAmB,EAAE,EAAE,CACxE,mBAAmB,CAAC,MAAM,EAAE,CAC/B;SACJ,CAAC;QACF,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ;AA9DD,sBA8DC"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
import { ClassificationLabel } from "./classificationLabel.mjs";
|
|
2
|
+
import { modelKindFromNapiModelKind, modelKindToNapiModelKind } from "./modelKind.mjs";
|
|
3
|
+
import { modelTypeFromNapiModelType, modelTypeToNapiModelType } from "./modelType.mjs";
|
|
4
|
+
export class Model {
|
|
5
|
+
get isThinking() {
|
|
6
|
+
return this.outputParserRegex !== null;
|
|
7
|
+
}
|
|
8
|
+
constructor(repoId, type, kind, name, vendor, quantization, outputParserRegex, classificationLabels) {
|
|
9
|
+
this.repoId = repoId;
|
|
10
|
+
this.type = type;
|
|
11
|
+
this.kind = kind;
|
|
12
|
+
this.name = name;
|
|
13
|
+
this.vendor = vendor;
|
|
14
|
+
this.quantization = quantization;
|
|
15
|
+
this.outputParserRegex = outputParserRegex;
|
|
16
|
+
this.classificationLabels = classificationLabels;
|
|
17
|
+
}
|
|
18
|
+
static fromNapiModel(napiModel) {
|
|
19
|
+
return new Model(napiModel.repoId, modelTypeFromNapiModelType(napiModel.type), modelKindFromNapiModelKind(napiModel.kind), napiModel.name, napiModel.vendor, napiModel.quantization ?? null, napiModel.outputParserRegex ?? null, napiModel.classificationLabels.map(ClassificationLabel.fromNapi));
|
|
20
|
+
}
|
|
21
|
+
toNapi() {
|
|
22
|
+
const napiModel = {
|
|
23
|
+
repoId: this.repoId,
|
|
24
|
+
type: modelTypeToNapiModelType(this.type),
|
|
25
|
+
kind: modelKindToNapiModelKind(this.kind),
|
|
26
|
+
name: this.name,
|
|
27
|
+
vendor: this.vendor,
|
|
28
|
+
...(this.quantization !== null && { quantization: this.quantization }),
|
|
29
|
+
...(this.outputParserRegex !== null && { outputParserRegex: this.outputParserRegex }),
|
|
30
|
+
classificationLabels: this.classificationLabels.map((classificationLabel) => classificationLabel.toNapi()),
|
|
31
|
+
};
|
|
32
|
+
return napiModel;
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
//# sourceMappingURL=model.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model.mjs","sourceRoot":"","sources":["../src/bridging/model.ts"],"names":[],"mappings":"OACO,EAAE,mBAAmB,EAAE;OACvB,EAAa,0BAA0B,EAAE,wBAAwB,EAAE;OACnE,EAAa,0BAA0B,EAAE,wBAAwB,EAAE;AAE1E,MAAM,OAAO,KAAK;IAUd,IAAI,UAAU;QACV,OAAO,IAAI,CAAC,iBAAiB,KAAK,IAAI,CAAC;IAC3C,CAAC;IAED,YACI,MAAc,EACd,IAAe,EACf,IAAe,EACf,IAAY,EACZ,MAAc,EACd,YAA2B,EAC3B,iBAAgC,EAChC,oBAA2C;QAE3C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;IACrD,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,SAAoB;QACrC,OAAO,IAAI,KAAK,CACZ,SAAS,CAAC,MAAM,EAChB,0BAA0B,CAAC,SAAS,CAAC,IAAI,CAAC,EAC1C,0BAA0B,CAAC,SAAS,CAAC,IAAI,CAAC,EAC1C,SAAS,CAAC,IAAI,EACd,SAAS,CAAC,MAAM,EAChB,SAAS,CAAC,YAAY,IAAI,IAAI,EAC9B,SAAS,CAAC,iBAAiB,IAAI,IAAI,EACnC,SAAS,CAAC,oBAAoB,CAAC,GAAG,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CACnE,CAAC;IACN,CAAC;IAED,MAAM;QACF,MAAM,SAAS,GAAc;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,wBAAwB,CAAC,IAAI,CAAC,IAAI,CAAC;YACzC,IAAI,EAAE,wBAAwB,CAAC,IAAI,CAAC,IAAI,CAAC;YACzC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,IAAI,IAAI,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC;YACtE,GAAG,CAAC,IAAI,CAAC,iBAAiB,KAAK,IAAI,IAAI,EAAE,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACrF,oBAAoB,EAAE,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,mBAAmB,EAAE,EAAE,CACxE,mBAAmB,CAAC,MAAM,EAAE,CAC/B;SACJ,CAAC;QACF,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import { ModelKind as NapiModelKind } from "../napi/uzu.mjs";
|
|
2
|
+
export declare enum ModelKind {
|
|
3
|
+
Local = "Local",
|
|
4
|
+
Cloud = "Cloud"
|
|
5
|
+
}
|
|
6
|
+
export declare function modelKindFromNapiModelKind(napiModelKind: NapiModelKind): ModelKind;
|
|
7
|
+
export declare function modelKindToNapiModelKind(modelKind: ModelKind): NapiModelKind;
|
|
8
|
+
//# sourceMappingURL=modelKind.d.mts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"modelKind.d.mts","sourceRoot":"","sources":["../src/bridging/modelKind.ts"],"names":[],"mappings":"OACO,EAAE,SAAS,IAAI,aAAa,EAAE;AAErC,oBAAY,SAAS;IACjB,KAAK,UAAU;IACf,KAAK,UAAU;CAClB;AAED,wBAAgB,0BAA0B,CAAC,aAAa,EAAE,aAAa,GAAG,SAAS,CASlF;AAED,wBAAgB,wBAAwB,CAAC,SAAS,EAAE,SAAS,GAAG,aAAa,CAO5E"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import { ModelKind as NapiModelKind } from "../napi/uzu.js";
|
|
2
|
+
export declare enum ModelKind {
|
|
3
|
+
Local = "Local",
|
|
4
|
+
Cloud = "Cloud"
|
|
5
|
+
}
|
|
6
|
+
export declare function modelKindFromNapiModelKind(napiModelKind: NapiModelKind): ModelKind;
|
|
7
|
+
export declare function modelKindToNapiModelKind(modelKind: ModelKind): NapiModelKind;
|
|
8
|
+
//# sourceMappingURL=modelKind.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"modelKind.d.ts","sourceRoot":"","sources":["../src/bridging/modelKind.ts"],"names":[],"mappings":"OACO,EAAE,SAAS,IAAI,aAAa,EAAE;AAErC,oBAAY,SAAS;IACjB,KAAK,UAAU;IACf,KAAK,UAAU;CAClB;AAED,wBAAgB,0BAA0B,CAAC,aAAa,EAAE,aAAa,GAAG,SAAS,CASlF;AAED,wBAAgB,wBAAwB,CAAC,SAAS,EAAE,SAAS,GAAG,aAAa,CAO5E"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ModelKind = void 0;
|
|
4
|
+
exports.modelKindFromNapiModelKind = modelKindFromNapiModelKind;
|
|
5
|
+
exports.modelKindToNapiModelKind = modelKindToNapiModelKind;
|
|
6
|
+
const error_1 = require("../error.js");
|
|
7
|
+
var ModelKind;
|
|
8
|
+
(function (ModelKind) {
|
|
9
|
+
ModelKind["Local"] = "Local";
|
|
10
|
+
ModelKind["Cloud"] = "Cloud";
|
|
11
|
+
})(ModelKind || (exports.ModelKind = ModelKind = {}));
|
|
12
|
+
function modelKindFromNapiModelKind(napiModelKind) {
|
|
13
|
+
switch (napiModelKind) {
|
|
14
|
+
case 0:
|
|
15
|
+
return ModelKind.Local;
|
|
16
|
+
case 1:
|
|
17
|
+
return ModelKind.Cloud;
|
|
18
|
+
default:
|
|
19
|
+
throw new error_1.EngineError(error_1.EngineErrorCode.UnexpectedModelKind);
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
function modelKindToNapiModelKind(modelKind) {
|
|
23
|
+
switch (modelKind) {
|
|
24
|
+
case ModelKind.Local:
|
|
25
|
+
return 0;
|
|
26
|
+
case ModelKind.Cloud:
|
|
27
|
+
return 1;
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=modelKind.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"modelKind.js","sourceRoot":"","sources":["../src/bridging/modelKind.ts"],"names":[],"mappings":";;;AAQA,gEASC;AAED,4DAOC;AA1BD,uCAAwD;AAGxD,IAAY,SAGX;AAHD,WAAY,SAAS;IACjB,4BAAe,CAAA;IACf,4BAAe,CAAA;AACnB,CAAC,EAHW,SAAS,yBAAT,SAAS,QAGpB;AAED,SAAgB,0BAA0B,CAAC,aAA4B;IACnE,QAAQ,aAAa,EAAE,CAAC;QACpB,KAAK,CAAC;YACF,OAAO,SAAS,CAAC,KAAK,CAAC;QAC3B,KAAK,CAAC;YACF,OAAO,SAAS,CAAC,KAAK,CAAC;QAC3B;YACI,MAAM,IAAI,mBAAW,CAAC,uBAAe,CAAC,mBAAmB,CAAC,CAAC;IACnE,CAAC;AACL,CAAC;AAED,SAAgB,wBAAwB,CAAC,SAAoB;IACzD,QAAQ,SAAS,EAAE,CAAC;QAChB,KAAK,SAAS,CAAC,KAAK;YAChB,OAAO,CAAC,CAAC;QACb,KAAK,SAAS,CAAC,KAAK;YAChB,OAAO,CAAC,CAAC;IACjB,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import { EngineError, EngineErrorCode } from "../error.mjs";
|
|
2
|
+
export var ModelKind;
|
|
3
|
+
(function (ModelKind) {
|
|
4
|
+
ModelKind["Local"] = "Local";
|
|
5
|
+
ModelKind["Cloud"] = "Cloud";
|
|
6
|
+
})(ModelKind || (ModelKind = {}));
|
|
7
|
+
export function modelKindFromNapiModelKind(napiModelKind) {
|
|
8
|
+
switch (napiModelKind) {
|
|
9
|
+
case 0:
|
|
10
|
+
return ModelKind.Local;
|
|
11
|
+
case 1:
|
|
12
|
+
return ModelKind.Cloud;
|
|
13
|
+
default:
|
|
14
|
+
throw new EngineError(EngineErrorCode.UnexpectedModelKind);
|
|
15
|
+
}
|
|
16
|
+
}
|
|
17
|
+
export function modelKindToNapiModelKind(modelKind) {
|
|
18
|
+
switch (modelKind) {
|
|
19
|
+
case ModelKind.Local:
|
|
20
|
+
return 0;
|
|
21
|
+
case ModelKind.Cloud:
|
|
22
|
+
return 1;
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
//# sourceMappingURL=modelKind.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"modelKind.mjs","sourceRoot":"","sources":["../src/bridging/modelKind.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,EAAE,eAAe,EAAE;AAGvC,MAAM,CAAN,IAAY,SAGX;AAHD,WAAY,SAAS;IACjB,4BAAe,CAAA;IACf,4BAAe,CAAA;AACnB,CAAC,EAHW,SAAS,KAAT,SAAS,QAGpB;AAED,MAAM,UAAU,0BAA0B,CAAC,aAA4B;IACnE,QAAQ,aAAa,EAAE,CAAC;QACpB,KAAK,CAAC;YACF,OAAO,SAAS,CAAC,KAAK,CAAC;QAC3B,KAAK,CAAC;YACF,OAAO,SAAS,CAAC,KAAK,CAAC;QAC3B;YACI,MAAM,IAAI,WAAW,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;IACnE,CAAC;AACL,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,SAAoB;IACzD,QAAQ,SAAS,EAAE,CAAC;QAChB,KAAK,SAAS,CAAC,KAAK;YAChB,OAAO,CAAC,CAAC;QACb,KAAK,SAAS,CAAC,KAAK;YAChB,OAAO,CAAC,CAAC;IACjB,CAAC;AACL,CAAC"}
|
package/bridging/modelType.d.mts
CHANGED
|
@@ -1,7 +1,8 @@
|
|
|
1
1
|
import { ModelType as NapiModelType } from "../napi/uzu.mjs";
|
|
2
2
|
export declare enum ModelType {
|
|
3
|
-
|
|
4
|
-
|
|
3
|
+
Chat = "Chat",
|
|
4
|
+
Classification = "Classification",
|
|
5
|
+
TTS = "TTS"
|
|
5
6
|
}
|
|
6
7
|
export declare function modelTypeFromNapiModelType(napiModelType: NapiModelType): ModelType;
|
|
7
8
|
export declare function modelTypeToNapiModelType(modelType: ModelType): NapiModelType;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"modelType.d.mts","sourceRoot":"","sources":["../src/bridging/modelType.ts"],"names":[],"mappings":"OACO,EAAE,SAAS,IAAI,aAAa,EAAE;AAErC,oBAAY,SAAS;IACjB,
|
|
1
|
+
{"version":3,"file":"modelType.d.mts","sourceRoot":"","sources":["../src/bridging/modelType.ts"],"names":[],"mappings":"OACO,EAAE,SAAS,IAAI,aAAa,EAAE;AAErC,oBAAY,SAAS;IACjB,IAAI,SAAS;IACb,cAAc,mBAAmB;IACjC,GAAG,QAAQ;CACd;AAED,wBAAgB,0BAA0B,CAAC,aAAa,EAAE,aAAa,GAAG,SAAS,CAWlF;AAED,wBAAgB,wBAAwB,CAAC,SAAS,EAAE,SAAS,GAAG,aAAa,CAS5E"}
|
package/bridging/modelType.d.ts
CHANGED
|
@@ -1,7 +1,8 @@
|
|
|
1
1
|
import { ModelType as NapiModelType } from "../napi/uzu.js";
|
|
2
2
|
export declare enum ModelType {
|
|
3
|
-
|
|
4
|
-
|
|
3
|
+
Chat = "Chat",
|
|
4
|
+
Classification = "Classification",
|
|
5
|
+
TTS = "TTS"
|
|
5
6
|
}
|
|
6
7
|
export declare function modelTypeFromNapiModelType(napiModelType: NapiModelType): ModelType;
|
|
7
8
|
export declare function modelTypeToNapiModelType(modelType: ModelType): NapiModelType;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"modelType.d.ts","sourceRoot":"","sources":["../src/bridging/modelType.ts"],"names":[],"mappings":"OACO,EAAE,SAAS,IAAI,aAAa,EAAE;AAErC,oBAAY,SAAS;IACjB,
|
|
1
|
+
{"version":3,"file":"modelType.d.ts","sourceRoot":"","sources":["../src/bridging/modelType.ts"],"names":[],"mappings":"OACO,EAAE,SAAS,IAAI,aAAa,EAAE;AAErC,oBAAY,SAAS;IACjB,IAAI,SAAS;IACb,cAAc,mBAAmB;IACjC,GAAG,QAAQ;CACd;AAED,wBAAgB,0BAA0B,CAAC,aAAa,EAAE,aAAa,GAAG,SAAS,CAWlF;AAED,wBAAgB,wBAAwB,CAAC,SAAS,EAAE,SAAS,GAAG,aAAa,CAS5E"}
|
package/bridging/modelType.js
CHANGED
|
@@ -6,25 +6,30 @@ exports.modelTypeToNapiModelType = modelTypeToNapiModelType;
|
|
|
6
6
|
const error_1 = require("../error.js");
|
|
7
7
|
var ModelType;
|
|
8
8
|
(function (ModelType) {
|
|
9
|
-
ModelType["
|
|
10
|
-
ModelType["
|
|
9
|
+
ModelType["Chat"] = "Chat";
|
|
10
|
+
ModelType["Classification"] = "Classification";
|
|
11
|
+
ModelType["TTS"] = "TTS";
|
|
11
12
|
})(ModelType || (exports.ModelType = ModelType = {}));
|
|
12
13
|
function modelTypeFromNapiModelType(napiModelType) {
|
|
13
14
|
switch (napiModelType) {
|
|
14
15
|
case 0:
|
|
15
|
-
return ModelType.
|
|
16
|
+
return ModelType.Chat;
|
|
16
17
|
case 1:
|
|
17
|
-
return ModelType.
|
|
18
|
+
return ModelType.Classification;
|
|
19
|
+
case 2:
|
|
20
|
+
return ModelType.TTS;
|
|
18
21
|
default:
|
|
19
22
|
throw new error_1.EngineError(error_1.EngineErrorCode.UnexpectedModelType);
|
|
20
23
|
}
|
|
21
24
|
}
|
|
22
25
|
function modelTypeToNapiModelType(modelType) {
|
|
23
26
|
switch (modelType) {
|
|
24
|
-
case ModelType.
|
|
27
|
+
case ModelType.Chat:
|
|
25
28
|
return 0;
|
|
26
|
-
case ModelType.
|
|
29
|
+
case ModelType.Classification:
|
|
27
30
|
return 1;
|
|
31
|
+
case ModelType.TTS:
|
|
32
|
+
return 2;
|
|
28
33
|
}
|
|
29
34
|
}
|
|
30
35
|
//# sourceMappingURL=modelType.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"modelType.js","sourceRoot":"","sources":["../src/bridging/modelType.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"modelType.js","sourceRoot":"","sources":["../src/bridging/modelType.ts"],"names":[],"mappings":";;;AASA,gEAWC;AAED,4DASC;AA/BD,uCAAwD;AAGxD,IAAY,SAIX;AAJD,WAAY,SAAS;IACjB,0BAAa,CAAA;IACb,8CAAiC,CAAA;IACjC,wBAAW,CAAA;AACf,CAAC,EAJW,SAAS,yBAAT,SAAS,QAIpB;AAED,SAAgB,0BAA0B,CAAC,aAA4B;IACnE,QAAQ,aAAa,EAAE,CAAC;QACpB,KAAK,CAAC;YACF,OAAO,SAAS,CAAC,IAAI,CAAC;QAC1B,KAAK,CAAC;YACF,OAAO,SAAS,CAAC,cAAc,CAAC;QACpC,KAAK,CAAC;YACF,OAAO,SAAS,CAAC,GAAG,CAAC;QACzB;YACI,MAAM,IAAI,mBAAW,CAAC,uBAAe,CAAC,mBAAmB,CAAC,CAAC;IACnE,CAAC;AACL,CAAC;AAED,SAAgB,wBAAwB,CAAC,SAAoB;IACzD,QAAQ,SAAS,EAAE,CAAC;QAChB,KAAK,SAAS,CAAC,IAAI;YACf,OAAO,CAAC,CAAC;QACb,KAAK,SAAS,CAAC,cAAc;YACzB,OAAO,CAAC,CAAC;QACb,KAAK,SAAS,CAAC,GAAG;YACd,OAAO,CAAC,CAAC;IACjB,CAAC;AACL,CAAC"}
|
package/bridging/modelType.mjs
CHANGED
|
@@ -1,25 +1,30 @@
|
|
|
1
1
|
import { EngineError, EngineErrorCode } from "../error.mjs";
|
|
2
2
|
export var ModelType;
|
|
3
3
|
(function (ModelType) {
|
|
4
|
-
ModelType["
|
|
5
|
-
ModelType["
|
|
4
|
+
ModelType["Chat"] = "Chat";
|
|
5
|
+
ModelType["Classification"] = "Classification";
|
|
6
|
+
ModelType["TTS"] = "TTS";
|
|
6
7
|
})(ModelType || (ModelType = {}));
|
|
7
8
|
export function modelTypeFromNapiModelType(napiModelType) {
|
|
8
9
|
switch (napiModelType) {
|
|
9
10
|
case 0:
|
|
10
|
-
return ModelType.
|
|
11
|
+
return ModelType.Chat;
|
|
11
12
|
case 1:
|
|
12
|
-
return ModelType.
|
|
13
|
+
return ModelType.Classification;
|
|
14
|
+
case 2:
|
|
15
|
+
return ModelType.TTS;
|
|
13
16
|
default:
|
|
14
17
|
throw new EngineError(EngineErrorCode.UnexpectedModelType);
|
|
15
18
|
}
|
|
16
19
|
}
|
|
17
20
|
export function modelTypeToNapiModelType(modelType) {
|
|
18
21
|
switch (modelType) {
|
|
19
|
-
case ModelType.
|
|
22
|
+
case ModelType.Chat:
|
|
20
23
|
return 0;
|
|
21
|
-
case ModelType.
|
|
24
|
+
case ModelType.Classification:
|
|
22
25
|
return 1;
|
|
26
|
+
case ModelType.TTS:
|
|
27
|
+
return 2;
|
|
23
28
|
}
|
|
24
29
|
}
|
|
25
30
|
//# sourceMappingURL=modelType.mjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"modelType.mjs","sourceRoot":"","sources":["../src/bridging/modelType.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,EAAE,eAAe,EAAE;AAGvC,MAAM,CAAN,IAAY,
|
|
1
|
+
{"version":3,"file":"modelType.mjs","sourceRoot":"","sources":["../src/bridging/modelType.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,EAAE,eAAe,EAAE;AAGvC,MAAM,CAAN,IAAY,SAIX;AAJD,WAAY,SAAS;IACjB,0BAAa,CAAA;IACb,8CAAiC,CAAA;IACjC,wBAAW,CAAA;AACf,CAAC,EAJW,SAAS,KAAT,SAAS,QAIpB;AAED,MAAM,UAAU,0BAA0B,CAAC,aAA4B;IACnE,QAAQ,aAAa,EAAE,CAAC;QACpB,KAAK,CAAC;YACF,OAAO,SAAS,CAAC,IAAI,CAAC;QAC1B,KAAK,CAAC;YACF,OAAO,SAAS,CAAC,cAAc,CAAC;QACpC,KAAK,CAAC;YACF,OAAO,SAAS,CAAC,GAAG,CAAC;QACzB;YACI,MAAM,IAAI,WAAW,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;IACnE,CAAC;AACL,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,SAAoB;IACzD,QAAQ,SAAS,EAAE,CAAC;QAChB,KAAK,SAAS,CAAC,IAAI;YACf,OAAO,CAAC,CAAC;QACb,KAAK,SAAS,CAAC,cAAc;YACzB,OAAO,CAAC,CAAC;QACb,KAAK,SAAS,CAAC,GAAG;YACd,OAAO,CAAC,CAAC;IACjB,CAAC;AACL,CAAC"}
|
package/bridging/preset.d.mts
CHANGED
|
@@ -7,6 +7,8 @@ export declare class Preset implements ToNapi<NapiPreset> {
|
|
|
7
7
|
static general(): Preset;
|
|
8
8
|
static classification(feature: ClassificationFeature): Preset;
|
|
9
9
|
static summarization(): Preset;
|
|
10
|
+
static nGramSpeculator(useCase: string, numberOfSpeculatedTokens: number): Preset;
|
|
11
|
+
static chat(): Preset;
|
|
10
12
|
toNapi(): NapiPreset;
|
|
11
13
|
}
|
|
12
14
|
//# sourceMappingURL=preset.d.mts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"preset.d.mts","sourceRoot":"","sources":["../src/bridging/preset.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;AAEjB,qBAAa,MAAO,YAAW,MAAM,CAAC,UAAU,CAAC;IAC7C,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IAExC,OAAO;IAIP,MAAM,CAAC,OAAO,IAAI,MAAM;IAKxB,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,qBAAqB,GAAG,MAAM;IAS7D,MAAM,CAAC,aAAa,IAAI,MAAM;IAK9B,MAAM,IAAI,UAAU;CAGvB"}
|
|
1
|
+
{"version":3,"file":"preset.d.mts","sourceRoot":"","sources":["../src/bridging/preset.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;AAEjB,qBAAa,MAAO,YAAW,MAAM,CAAC,UAAU,CAAC;IAC7C,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IAExC,OAAO;IAIP,MAAM,CAAC,OAAO,IAAI,MAAM;IAKxB,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,qBAAqB,GAAG,MAAM;IAS7D,MAAM,CAAC,aAAa,IAAI,MAAM;IAK9B,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,wBAAwB,EAAE,MAAM,GAAG,MAAM;IASjF,MAAM,CAAC,IAAI,IAAI,MAAM;IAKrB,MAAM,IAAI,UAAU;CAGvB"}
|
package/bridging/preset.d.ts
CHANGED
|
@@ -7,6 +7,8 @@ export declare class Preset implements ToNapi<NapiPreset> {
|
|
|
7
7
|
static general(): Preset;
|
|
8
8
|
static classification(feature: ClassificationFeature): Preset;
|
|
9
9
|
static summarization(): Preset;
|
|
10
|
+
static nGramSpeculator(useCase: string, numberOfSpeculatedTokens: number): Preset;
|
|
11
|
+
static chat(): Preset;
|
|
10
12
|
toNapi(): NapiPreset;
|
|
11
13
|
}
|
|
12
14
|
//# sourceMappingURL=preset.d.ts.map
|
package/bridging/preset.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"preset.d.ts","sourceRoot":"","sources":["../src/bridging/preset.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;AAEjB,qBAAa,MAAO,YAAW,MAAM,CAAC,UAAU,CAAC;IAC7C,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IAExC,OAAO;IAIP,MAAM,CAAC,OAAO,IAAI,MAAM;IAKxB,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,qBAAqB,GAAG,MAAM;IAS7D,MAAM,CAAC,aAAa,IAAI,MAAM;IAK9B,MAAM,IAAI,UAAU;CAGvB"}
|
|
1
|
+
{"version":3,"file":"preset.d.ts","sourceRoot":"","sources":["../src/bridging/preset.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;AAEjB,qBAAa,MAAO,YAAW,MAAM,CAAC,UAAU,CAAC;IAC7C,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IAExC,OAAO;IAIP,MAAM,CAAC,OAAO,IAAI,MAAM;IAKxB,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,qBAAqB,GAAG,MAAM;IAS7D,MAAM,CAAC,aAAa,IAAI,MAAM;IAK9B,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,wBAAwB,EAAE,MAAM,GAAG,MAAM;IASjF,MAAM,CAAC,IAAI,IAAI,MAAM;IAKrB,MAAM,IAAI,UAAU;CAGvB"}
|
package/bridging/preset.js
CHANGED
|
@@ -21,6 +21,18 @@ class Preset {
|
|
|
21
21
|
const napiPreset = { type: 'Summarization' };
|
|
22
22
|
return new Preset(napiPreset);
|
|
23
23
|
}
|
|
24
|
+
static nGramSpeculator(useCase, numberOfSpeculatedTokens) {
|
|
25
|
+
const napiPreset = {
|
|
26
|
+
type: 'NGramSpeculator',
|
|
27
|
+
useCase,
|
|
28
|
+
numberOfSpeculatedTokens,
|
|
29
|
+
};
|
|
30
|
+
return new Preset(napiPreset);
|
|
31
|
+
}
|
|
32
|
+
static chat() {
|
|
33
|
+
const napiPreset = { type: 'Chat' };
|
|
34
|
+
return new Preset(napiPreset);
|
|
35
|
+
}
|
|
24
36
|
toNapi() {
|
|
25
37
|
return this.napiPreset;
|
|
26
38
|
}
|
package/bridging/preset.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"preset.js","sourceRoot":"","sources":["../src/bridging/preset.ts"],"names":[],"mappings":";;;AAIA,MAAa,MAAM;IAGf,YAAoB,UAAsB;QACtC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,OAAO;QACV,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACnD,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,OAA8B;QAChD,MAAM,yBAAyB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACnD,MAAM,UAAU,GAAe;YAC3B,IAAI,EAAE,gBAAgB;YACtB,OAAO,EAAE,yBAAyB;SACrC,CAAC;QACF,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,aAAa;QAChB,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;QACzD,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;CACJ;
|
|
1
|
+
{"version":3,"file":"preset.js","sourceRoot":"","sources":["../src/bridging/preset.ts"],"names":[],"mappings":";;;AAIA,MAAa,MAAM;IAGf,YAAoB,UAAsB;QACtC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,OAAO;QACV,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACnD,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,OAA8B;QAChD,MAAM,yBAAyB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACnD,MAAM,UAAU,GAAe;YAC3B,IAAI,EAAE,gBAAgB;YACtB,OAAO,EAAE,yBAAyB;SACrC,CAAC;QACF,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,aAAa;QAChB,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;QACzD,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,eAAe,CAAC,OAAe,EAAE,wBAAgC;QACpE,MAAM,UAAU,GAAe;YAC3B,IAAI,EAAE,iBAAiB;YACvB,OAAO;YACP,wBAAwB;SAC3B,CAAC;QACF,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,IAAI;QACP,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QAChD,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;CACJ;AA3CD,wBA2CC"}
|
package/bridging/preset.mjs
CHANGED
|
@@ -18,6 +18,18 @@ export class Preset {
|
|
|
18
18
|
const napiPreset = { type: 'Summarization' };
|
|
19
19
|
return new Preset(napiPreset);
|
|
20
20
|
}
|
|
21
|
+
static nGramSpeculator(useCase, numberOfSpeculatedTokens) {
|
|
22
|
+
const napiPreset = {
|
|
23
|
+
type: 'NGramSpeculator',
|
|
24
|
+
useCase,
|
|
25
|
+
numberOfSpeculatedTokens,
|
|
26
|
+
};
|
|
27
|
+
return new Preset(napiPreset);
|
|
28
|
+
}
|
|
29
|
+
static chat() {
|
|
30
|
+
const napiPreset = { type: 'Chat' };
|
|
31
|
+
return new Preset(napiPreset);
|
|
32
|
+
}
|
|
21
33
|
toNapi() {
|
|
22
34
|
return this.napiPreset;
|
|
23
35
|
}
|
package/bridging/preset.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"preset.mjs","sourceRoot":"","sources":["../src/bridging/preset.ts"],"names":[],"mappings":"AAIA,MAAM,OAAO,MAAM;IAGf,YAAoB,UAAsB;QACtC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,OAAO;QACV,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACnD,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,OAA8B;QAChD,MAAM,yBAAyB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACnD,MAAM,UAAU,GAAe;YAC3B,IAAI,EAAE,gBAAgB;YACtB,OAAO,EAAE,yBAAyB;SACrC,CAAC;QACF,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,aAAa;QAChB,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;QACzD,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;CACJ"}
|
|
1
|
+
{"version":3,"file":"preset.mjs","sourceRoot":"","sources":["../src/bridging/preset.ts"],"names":[],"mappings":"AAIA,MAAM,OAAO,MAAM;IAGf,YAAoB,UAAsB;QACtC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,OAAO;QACV,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACnD,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,OAA8B;QAChD,MAAM,yBAAyB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACnD,MAAM,UAAU,GAAe;YAC3B,IAAI,EAAE,gBAAgB;YACtB,OAAO,EAAE,yBAAyB;SACrC,CAAC;QACF,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,aAAa;QAChB,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;QACzD,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,eAAe,CAAC,OAAe,EAAE,wBAAgC;QACpE,MAAM,UAAU,GAAe;YAC3B,IAAI,EAAE,iBAAiB;YACvB,OAAO;YACP,wBAAwB;SAC3B,CAAC;QACF,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,IAAI;QACP,MAAM,UAAU,GAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QAChD,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;CACJ"}
|
|
@@ -1,10 +1,11 @@
|
|
|
1
1
|
import { SamplingMethod as NapiSamplingMethod } from "../napi/uzu.mjs";
|
|
2
2
|
import { ToNapi } from "./napi.mjs";
|
|
3
|
+
import { SamplingProcessingOrder } from "./samplingProcessingOrder.mjs";
|
|
3
4
|
export declare class SamplingMethod implements ToNapi<NapiSamplingMethod> {
|
|
4
5
|
private readonly napiSamplingMethod;
|
|
5
6
|
private constructor();
|
|
6
7
|
static greedy(): SamplingMethod;
|
|
7
|
-
static stochastic(temperature: number | null, topK: number | null, topP: number | null, minP: number | null): SamplingMethod;
|
|
8
|
+
static stochastic(temperature: number | null, topK: number | null, topP: number | null, minP: number | null, processingOrder: SamplingProcessingOrder): SamplingMethod;
|
|
8
9
|
toNapi(): NapiSamplingMethod;
|
|
9
10
|
}
|
|
10
11
|
//# sourceMappingURL=samplingMethod.d.mts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"samplingMethod.d.mts","sourceRoot":"","sources":["../src/bridging/samplingMethod.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,IAAI,kBAAkB,EAAE;OACxC,EAAE,MAAM,EAAE;
|
|
1
|
+
{"version":3,"file":"samplingMethod.d.mts","sourceRoot":"","sources":["../src/bridging/samplingMethod.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,IAAI,kBAAkB,EAAE;OACxC,EAAE,MAAM,EAAE;OACV,EACH,uBAAuB,EAE1B;AAED,qBAAa,cAAe,YAAW,MAAM,CAAC,kBAAkB,CAAC;IAC7D,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAqB;IAExD,OAAO;IAIP,MAAM,CAAC,MAAM,IAAI,cAAc;IAK/B,MAAM,CAAC,UAAU,CACb,WAAW,EAAE,MAAM,GAAG,IAAI,EAC1B,IAAI,EAAE,MAAM,GAAG,IAAI,EACnB,IAAI,EAAE,MAAM,GAAG,IAAI,EACnB,IAAI,EAAE,MAAM,GAAG,IAAI,EACnB,eAAe,EAAE,uBAAuB,GACzC,cAAc;IAoBjB,MAAM,IAAI,kBAAkB;CAG/B"}
|