@trymirai/uzu 0.1.48 → 0.1.50
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 +1 -1
- package/bridging/asyncBatchSize.d.mts +10 -0
- package/bridging/asyncBatchSize.d.mts.map +1 -0
- package/bridging/asyncBatchSize.d.ts +10 -0
- package/bridging/asyncBatchSize.d.ts.map +1 -0
- package/bridging/asyncBatchSize.js +21 -0
- package/bridging/asyncBatchSize.js.map +1 -0
- package/bridging/asyncBatchSize.mjs +17 -0
- package/bridging/asyncBatchSize.mjs.map +1 -0
- package/bridging/config.d.mts +4 -1
- package/bridging/config.d.mts.map +1 -1
- package/bridging/config.d.ts +4 -1
- package/bridging/config.d.ts.map +1 -1
- package/bridging/config.js +13 -7
- package/bridging/config.js.map +1 -1
- package/bridging/config.mjs +13 -7
- package/bridging/config.mjs.map +1 -1
- package/index.d.mts +1 -0
- package/index.d.mts.map +1 -1
- package/index.d.ts +1 -0
- package/index.d.ts.map +1 -1
- package/index.js +3 -1
- package/index.js.map +1 -1
- package/index.mjs +1 -0
- package/index.mjs.map +1 -1
- package/interactors/chatModelInteractor.d.mts +2 -0
- package/interactors/chatModelInteractor.d.mts.map +1 -1
- package/interactors/chatModelInteractor.d.ts +2 -0
- package/interactors/chatModelInteractor.d.ts.map +1 -1
- package/interactors/chatModelInteractor.js +4 -0
- package/interactors/chatModelInteractor.js.map +1 -1
- package/interactors/chatModelInteractor.mjs +4 -0
- package/interactors/chatModelInteractor.mjs.map +1 -1
- package/napi/uzu.d.ts +5 -0
- package/napi/uzu.js +2 -0
- package/napi/uzu.mjs +1 -0
- package/napi/uzu.node +0 -0
- package/package.json +1 -1
- package/src/bridging/asyncBatchSize.ts +24 -0
- package/src/bridging/config.ts +22 -0
- package/src/index.ts +1 -0
- package/src/interactors/chatModelInteractor.ts +6 -0
- package/src/napi/uzu.d.ts +5 -0
- package/src/napi/uzu.js +2 -0
- package/src/napi/uzu.mjs +1 -0
- package/src/napi/uzu.node +0 -0
- package/src/version.ts +1 -1
- package/version.d.mts +1 -1
- package/version.d.ts +1 -1
- package/version.js +1 -1
- package/version.mjs +1 -1
package/README.md
CHANGED
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import { AsyncBatchSize as NapiAsyncBatchSize } from "../napi/uzu.mjs";
|
|
2
|
+
import { ToNapi } from "./napi.mjs";
|
|
3
|
+
export declare class AsyncBatchSize implements ToNapi<NapiAsyncBatchSize> {
|
|
4
|
+
private readonly napiAsyncBatchSize;
|
|
5
|
+
private constructor();
|
|
6
|
+
static default(): AsyncBatchSize;
|
|
7
|
+
static custom(size: number): AsyncBatchSize;
|
|
8
|
+
toNapi(): NapiAsyncBatchSize;
|
|
9
|
+
}
|
|
10
|
+
//# sourceMappingURL=asyncBatchSize.d.mts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"asyncBatchSize.d.mts","sourceRoot":"","sources":["../src/bridging/asyncBatchSize.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,IAAI,kBAAkB,EAAE;OACxC,EAAE,MAAM,EAAE;AAEjB,qBAAa,cAAe,YAAW,MAAM,CAAC,kBAAkB,CAAC;IAC7D,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAqB;IAExD,OAAO;IAIP,MAAM,CAAC,OAAO,IAAI,cAAc;IAKhC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAK3C,MAAM,IAAI,kBAAkB;CAG/B"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import { AsyncBatchSize as NapiAsyncBatchSize } from "../napi/uzu.js";
|
|
2
|
+
import { ToNapi } from "./napi.js";
|
|
3
|
+
export declare class AsyncBatchSize implements ToNapi<NapiAsyncBatchSize> {
|
|
4
|
+
private readonly napiAsyncBatchSize;
|
|
5
|
+
private constructor();
|
|
6
|
+
static default(): AsyncBatchSize;
|
|
7
|
+
static custom(size: number): AsyncBatchSize;
|
|
8
|
+
toNapi(): NapiAsyncBatchSize;
|
|
9
|
+
}
|
|
10
|
+
//# sourceMappingURL=asyncBatchSize.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"asyncBatchSize.d.ts","sourceRoot":"","sources":["../src/bridging/asyncBatchSize.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,IAAI,kBAAkB,EAAE;OACxC,EAAE,MAAM,EAAE;AAEjB,qBAAa,cAAe,YAAW,MAAM,CAAC,kBAAkB,CAAC;IAC7D,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAqB;IAExD,OAAO;IAIP,MAAM,CAAC,OAAO,IAAI,cAAc;IAKhC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAK3C,MAAM,IAAI,kBAAkB;CAG/B"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.AsyncBatchSize = void 0;
|
|
4
|
+
class AsyncBatchSize {
|
|
5
|
+
constructor(napiAsyncBatchSize) {
|
|
6
|
+
this.napiAsyncBatchSize = napiAsyncBatchSize;
|
|
7
|
+
}
|
|
8
|
+
static default() {
|
|
9
|
+
const napiAsyncBatchSize = { type: 'Default' };
|
|
10
|
+
return new AsyncBatchSize(napiAsyncBatchSize);
|
|
11
|
+
}
|
|
12
|
+
static custom(size) {
|
|
13
|
+
const napiAsyncBatchSize = { type: 'Custom', size };
|
|
14
|
+
return new AsyncBatchSize(napiAsyncBatchSize);
|
|
15
|
+
}
|
|
16
|
+
toNapi() {
|
|
17
|
+
return this.napiAsyncBatchSize;
|
|
18
|
+
}
|
|
19
|
+
}
|
|
20
|
+
exports.AsyncBatchSize = AsyncBatchSize;
|
|
21
|
+
//# sourceMappingURL=asyncBatchSize.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"asyncBatchSize.js","sourceRoot":"","sources":["../src/bridging/asyncBatchSize.ts"],"names":[],"mappings":";;;AAGA,MAAa,cAAc;IAGvB,YAAoB,kBAAsC;QACtD,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IACjD,CAAC;IAED,MAAM,CAAC,OAAO;QACV,MAAM,kBAAkB,GAAuB,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACnE,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,IAAY;QACtB,MAAM,kBAAkB,GAAuB,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;QACxE,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACnC,CAAC;CACJ;AApBD,wCAoBC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
export class AsyncBatchSize {
|
|
2
|
+
constructor(napiAsyncBatchSize) {
|
|
3
|
+
this.napiAsyncBatchSize = napiAsyncBatchSize;
|
|
4
|
+
}
|
|
5
|
+
static default() {
|
|
6
|
+
const napiAsyncBatchSize = { type: 'Default' };
|
|
7
|
+
return new AsyncBatchSize(napiAsyncBatchSize);
|
|
8
|
+
}
|
|
9
|
+
static custom(size) {
|
|
10
|
+
const napiAsyncBatchSize = { type: 'Custom', size };
|
|
11
|
+
return new AsyncBatchSize(napiAsyncBatchSize);
|
|
12
|
+
}
|
|
13
|
+
toNapi() {
|
|
14
|
+
return this.napiAsyncBatchSize;
|
|
15
|
+
}
|
|
16
|
+
}
|
|
17
|
+
//# sourceMappingURL=asyncBatchSize.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"asyncBatchSize.mjs","sourceRoot":"","sources":["../src/bridging/asyncBatchSize.ts"],"names":[],"mappings":"AAGA,MAAM,OAAO,cAAc;IAGvB,YAAoB,kBAAsC;QACtD,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IACjD,CAAC;IAED,MAAM,CAAC,OAAO;QACV,MAAM,kBAAkB,GAAuB,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACnE,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,IAAY;QACtB,MAAM,kBAAkB,GAAuB,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;QACxE,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACnC,CAAC;CACJ"}
|
package/bridging/config.d.mts
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import { Config as NapiConfig } from "../napi/uzu.mjs";
|
|
2
|
+
import { AsyncBatchSize } from "./asyncBatchSize.mjs";
|
|
2
3
|
import { ContextLength } from "./contextLength.mjs";
|
|
3
4
|
import { ContextMode } from "./contextMode.mjs";
|
|
4
5
|
import { ToNapi } from "./napi.mjs";
|
|
@@ -11,13 +12,15 @@ export declare class Config implements ToNapi<NapiConfig> {
|
|
|
11
12
|
readonly contextLength: ContextLength;
|
|
12
13
|
readonly prefillStepSize: PrefillStepSize;
|
|
13
14
|
readonly samplingSeed: SamplingSeed;
|
|
14
|
-
|
|
15
|
+
readonly asyncBatchSize: AsyncBatchSize;
|
|
16
|
+
constructor(preset: Preset, contextMode: ContextMode, contextLength: ContextLength, prefillStepSize: PrefillStepSize, samplingSeed: SamplingSeed, asyncBatchSize: AsyncBatchSize);
|
|
15
17
|
static default(): Config;
|
|
16
18
|
withPreset(preset: Preset): Config;
|
|
17
19
|
withContextMode(contextMode: ContextMode): Config;
|
|
18
20
|
withContextLength(contextLength: ContextLength): Config;
|
|
19
21
|
withPrefillStepSize(prefillStepSize: PrefillStepSize): Config;
|
|
20
22
|
withSamplingSeed(samplingSeed: SamplingSeed): Config;
|
|
23
|
+
withAsyncBatchSize(asyncBatchSize: AsyncBatchSize): Config;
|
|
21
24
|
toNapi(): NapiConfig;
|
|
22
25
|
}
|
|
23
26
|
//# sourceMappingURL=config.d.mts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.mts","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,YAAY,EAAE;AAEvB,qBAAa,MAAO,YAAW,MAAM,CAAC,UAAU,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC;IACtC,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;IAC1C,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;
|
|
1
|
+
{"version":3,"file":"config.d.mts","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,cAAc,EAAE;OAClB,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,YAAY,EAAE;AAEvB,qBAAa,MAAO,YAAW,MAAM,CAAC,UAAU,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC;IACtC,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;IAC1C,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;IACpC,QAAQ,CAAC,cAAc,EAAE,cAAc,CAAC;gBAGpC,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW,EACxB,aAAa,EAAE,aAAa,EAC5B,eAAe,EAAE,eAAe,EAChC,YAAY,EAAE,YAAY,EAC1B,cAAc,EAAE,cAAc;IAUlC,MAAM,CAAC,OAAO,IAAI,MAAM;IAWxB,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAWlC,eAAe,CAAC,WAAW,EAAE,WAAW,GAAG,MAAM;IAWjD,iBAAiB,CAAC,aAAa,EAAE,aAAa,GAAG,MAAM;IAWvD,mBAAmB,CAAC,eAAe,EAAE,eAAe,GAAG,MAAM;IAW7D,gBAAgB,CAAC,YAAY,EAAE,YAAY,GAAG,MAAM;IAWpD,kBAAkB,CAAC,cAAc,EAAE,cAAc,GAAG,MAAM;IAW1D,MAAM,IAAI,UAAU;CAWvB"}
|
package/bridging/config.d.ts
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import { Config as NapiConfig } from "../napi/uzu.js";
|
|
2
|
+
import { AsyncBatchSize } from "./asyncBatchSize.js";
|
|
2
3
|
import { ContextLength } from "./contextLength.js";
|
|
3
4
|
import { ContextMode } from "./contextMode.js";
|
|
4
5
|
import { ToNapi } from "./napi.js";
|
|
@@ -11,13 +12,15 @@ export declare class Config implements ToNapi<NapiConfig> {
|
|
|
11
12
|
readonly contextLength: ContextLength;
|
|
12
13
|
readonly prefillStepSize: PrefillStepSize;
|
|
13
14
|
readonly samplingSeed: SamplingSeed;
|
|
14
|
-
|
|
15
|
+
readonly asyncBatchSize: AsyncBatchSize;
|
|
16
|
+
constructor(preset: Preset, contextMode: ContextMode, contextLength: ContextLength, prefillStepSize: PrefillStepSize, samplingSeed: SamplingSeed, asyncBatchSize: AsyncBatchSize);
|
|
15
17
|
static default(): Config;
|
|
16
18
|
withPreset(preset: Preset): Config;
|
|
17
19
|
withContextMode(contextMode: ContextMode): Config;
|
|
18
20
|
withContextLength(contextLength: ContextLength): Config;
|
|
19
21
|
withPrefillStepSize(prefillStepSize: PrefillStepSize): Config;
|
|
20
22
|
withSamplingSeed(samplingSeed: SamplingSeed): Config;
|
|
23
|
+
withAsyncBatchSize(asyncBatchSize: AsyncBatchSize): Config;
|
|
21
24
|
toNapi(): NapiConfig;
|
|
22
25
|
}
|
|
23
26
|
//# sourceMappingURL=config.d.ts.map
|
package/bridging/config.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,YAAY,EAAE;AAEvB,qBAAa,MAAO,YAAW,MAAM,CAAC,UAAU,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC;IACtC,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;IAC1C,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":"OAAO,EAAE,MAAM,IAAI,UAAU,EAAE;OACxB,EAAE,cAAc,EAAE;OAClB,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,YAAY,EAAE;AAEvB,qBAAa,MAAO,YAAW,MAAM,CAAC,UAAU,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC;IACtC,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;IAC1C,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;IACpC,QAAQ,CAAC,cAAc,EAAE,cAAc,CAAC;gBAGpC,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW,EACxB,aAAa,EAAE,aAAa,EAC5B,eAAe,EAAE,eAAe,EAChC,YAAY,EAAE,YAAY,EAC1B,cAAc,EAAE,cAAc;IAUlC,MAAM,CAAC,OAAO,IAAI,MAAM;IAWxB,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAWlC,eAAe,CAAC,WAAW,EAAE,WAAW,GAAG,MAAM;IAWjD,iBAAiB,CAAC,aAAa,EAAE,aAAa,GAAG,MAAM;IAWvD,mBAAmB,CAAC,eAAe,EAAE,eAAe,GAAG,MAAM;IAW7D,gBAAgB,CAAC,YAAY,EAAE,YAAY,GAAG,MAAM;IAWpD,kBAAkB,CAAC,cAAc,EAAE,cAAc,GAAG,MAAM;IAW1D,MAAM,IAAI,UAAU;CAWvB"}
|
package/bridging/config.js
CHANGED
|
@@ -1,36 +1,41 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.Config = void 0;
|
|
4
|
+
const asyncBatchSize_1 = require("./asyncBatchSize.js");
|
|
4
5
|
const contextLength_1 = require("./contextLength.js");
|
|
5
6
|
const contextMode_1 = require("./contextMode.js");
|
|
6
7
|
const prefillStepSize_1 = require("./prefillStepSize.js");
|
|
7
8
|
const preset_1 = require("./preset.js");
|
|
8
9
|
const samplingSeed_1 = require("./samplingSeed.js");
|
|
9
10
|
class Config {
|
|
10
|
-
constructor(preset, contextMode, contextLength, prefillStepSize, samplingSeed) {
|
|
11
|
+
constructor(preset, contextMode, contextLength, prefillStepSize, samplingSeed, asyncBatchSize) {
|
|
11
12
|
this.preset = preset;
|
|
12
13
|
this.contextMode = contextMode;
|
|
13
14
|
this.contextLength = contextLength;
|
|
14
15
|
this.prefillStepSize = prefillStepSize;
|
|
15
16
|
this.samplingSeed = samplingSeed;
|
|
17
|
+
this.asyncBatchSize = asyncBatchSize;
|
|
16
18
|
}
|
|
17
19
|
static default() {
|
|
18
|
-
return new Config(preset_1.Preset.general(), contextMode_1.ContextMode.none(), contextLength_1.ContextLength.default(), prefillStepSize_1.PrefillStepSize.default(), samplingSeed_1.SamplingSeed.default());
|
|
20
|
+
return new Config(preset_1.Preset.general(), contextMode_1.ContextMode.none(), contextLength_1.ContextLength.default(), prefillStepSize_1.PrefillStepSize.default(), samplingSeed_1.SamplingSeed.default(), asyncBatchSize_1.AsyncBatchSize.default());
|
|
19
21
|
}
|
|
20
22
|
withPreset(preset) {
|
|
21
|
-
return new Config(preset, this.contextMode, this.contextLength, this.prefillStepSize, this.samplingSeed);
|
|
23
|
+
return new Config(preset, this.contextMode, this.contextLength, this.prefillStepSize, this.samplingSeed, this.asyncBatchSize);
|
|
22
24
|
}
|
|
23
25
|
withContextMode(contextMode) {
|
|
24
|
-
return new Config(this.preset, contextMode, this.contextLength, this.prefillStepSize, this.samplingSeed);
|
|
26
|
+
return new Config(this.preset, contextMode, this.contextLength, this.prefillStepSize, this.samplingSeed, this.asyncBatchSize);
|
|
25
27
|
}
|
|
26
28
|
withContextLength(contextLength) {
|
|
27
|
-
return new Config(this.preset, this.contextMode, contextLength, this.prefillStepSize, this.samplingSeed);
|
|
29
|
+
return new Config(this.preset, this.contextMode, contextLength, this.prefillStepSize, this.samplingSeed, this.asyncBatchSize);
|
|
28
30
|
}
|
|
29
31
|
withPrefillStepSize(prefillStepSize) {
|
|
30
|
-
return new Config(this.preset, this.contextMode, this.contextLength, prefillStepSize, this.samplingSeed);
|
|
32
|
+
return new Config(this.preset, this.contextMode, this.contextLength, prefillStepSize, this.samplingSeed, this.asyncBatchSize);
|
|
31
33
|
}
|
|
32
34
|
withSamplingSeed(samplingSeed) {
|
|
33
|
-
return new Config(this.preset, this.contextMode, this.contextLength, this.prefillStepSize, samplingSeed);
|
|
35
|
+
return new Config(this.preset, this.contextMode, this.contextLength, this.prefillStepSize, samplingSeed, this.asyncBatchSize);
|
|
36
|
+
}
|
|
37
|
+
withAsyncBatchSize(asyncBatchSize) {
|
|
38
|
+
return new Config(this.preset, this.contextMode, this.contextLength, this.prefillStepSize, this.samplingSeed, asyncBatchSize);
|
|
34
39
|
}
|
|
35
40
|
toNapi() {
|
|
36
41
|
const napiConfig = {
|
|
@@ -39,6 +44,7 @@ class Config {
|
|
|
39
44
|
contextLength: this.contextLength.toNapi(),
|
|
40
45
|
prefillStepSize: this.prefillStepSize.toNapi(),
|
|
41
46
|
samplingSeed: this.samplingSeed.toNapi(),
|
|
47
|
+
asyncBatchSize: this.asyncBatchSize.toNapi(),
|
|
42
48
|
};
|
|
43
49
|
return napiConfig;
|
|
44
50
|
}
|
package/bridging/config.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":";;;AACA,sDAAgD;AAChD,kDAA4C;AAE5C,0DAAoD;AACpD,wCAAkC;AAClC,oDAA8C;AAE9C,MAAa,MAAM;
|
|
1
|
+
{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":";;;AACA,wDAAkD;AAClD,sDAAgD;AAChD,kDAA4C;AAE5C,0DAAoD;AACpD,wCAAkC;AAClC,oDAA8C;AAE9C,MAAa,MAAM;IAQf,YACI,MAAc,EACd,WAAwB,EACxB,aAA4B,EAC5B,eAAgC,EAChC,YAA0B,EAC1B,cAA8B;QAE9B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;IACzC,CAAC;IAED,MAAM,CAAC,OAAO;QACV,OAAO,IAAI,MAAM,CACb,eAAM,CAAC,OAAO,EAAE,EAChB,yBAAW,CAAC,IAAI,EAAE,EAClB,6BAAa,CAAC,OAAO,EAAE,EACvB,iCAAe,CAAC,OAAO,EAAE,EACzB,2BAAY,CAAC,OAAO,EAAE,EACtB,+BAAc,CAAC,OAAO,EAAE,CAC3B,CAAC;IACN,CAAC;IAED,UAAU,CAAC,MAAc;QACrB,OAAO,IAAI,MAAM,CACb,MAAM,EACN,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,cAAc,CACtB,CAAC;IACN,CAAC;IAED,eAAe,CAAC,WAAwB;QACpC,OAAO,IAAI,MAAM,CACb,IAAI,CAAC,MAAM,EACX,WAAW,EACX,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,cAAc,CACtB,CAAC;IACN,CAAC;IAED,iBAAiB,CAAC,aAA4B;QAC1C,OAAO,IAAI,MAAM,CACb,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,WAAW,EAChB,aAAa,EACb,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,cAAc,CACtB,CAAC;IACN,CAAC;IAED,mBAAmB,CAAC,eAAgC;QAChD,OAAO,IAAI,MAAM,CACb,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,aAAa,EAClB,eAAe,EACf,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,cAAc,CACtB,CAAC;IACN,CAAC;IAED,gBAAgB,CAAC,YAA0B;QACvC,OAAO,IAAI,MAAM,CACb,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,eAAe,EACpB,YAAY,EACZ,IAAI,CAAC,cAAc,CACtB,CAAC;IACN,CAAC;IAED,kBAAkB,CAAC,cAA8B;QAC7C,OAAO,IAAI,MAAM,CACb,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,YAAY,EACjB,cAAc,CACjB,CAAC;IACN,CAAC;IAED,MAAM;QACF,MAAM,UAAU,GAAe;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;YAC5B,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE;YACtC,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE;YAC1C,eAAe,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;YAC9C,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;YACxC,cAAc,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE;SAC/C,CAAC;QACF,OAAO,UAAU,CAAC;IACtB,CAAC;CACJ;AAhHD,wBAgHC"}
|
package/bridging/config.mjs
CHANGED
|
@@ -1,33 +1,38 @@
|
|
|
1
|
+
import { AsyncBatchSize } from "./asyncBatchSize.mjs";
|
|
1
2
|
import { ContextLength } from "./contextLength.mjs";
|
|
2
3
|
import { ContextMode } from "./contextMode.mjs";
|
|
3
4
|
import { PrefillStepSize } from "./prefillStepSize.mjs";
|
|
4
5
|
import { Preset } from "./preset.mjs";
|
|
5
6
|
import { SamplingSeed } from "./samplingSeed.mjs";
|
|
6
7
|
export class Config {
|
|
7
|
-
constructor(preset, contextMode, contextLength, prefillStepSize, samplingSeed) {
|
|
8
|
+
constructor(preset, contextMode, contextLength, prefillStepSize, samplingSeed, asyncBatchSize) {
|
|
8
9
|
this.preset = preset;
|
|
9
10
|
this.contextMode = contextMode;
|
|
10
11
|
this.contextLength = contextLength;
|
|
11
12
|
this.prefillStepSize = prefillStepSize;
|
|
12
13
|
this.samplingSeed = samplingSeed;
|
|
14
|
+
this.asyncBatchSize = asyncBatchSize;
|
|
13
15
|
}
|
|
14
16
|
static default() {
|
|
15
|
-
return new Config(Preset.general(), ContextMode.none(), ContextLength.default(), PrefillStepSize.default(), SamplingSeed.default());
|
|
17
|
+
return new Config(Preset.general(), ContextMode.none(), ContextLength.default(), PrefillStepSize.default(), SamplingSeed.default(), AsyncBatchSize.default());
|
|
16
18
|
}
|
|
17
19
|
withPreset(preset) {
|
|
18
|
-
return new Config(preset, this.contextMode, this.contextLength, this.prefillStepSize, this.samplingSeed);
|
|
20
|
+
return new Config(preset, this.contextMode, this.contextLength, this.prefillStepSize, this.samplingSeed, this.asyncBatchSize);
|
|
19
21
|
}
|
|
20
22
|
withContextMode(contextMode) {
|
|
21
|
-
return new Config(this.preset, contextMode, this.contextLength, this.prefillStepSize, this.samplingSeed);
|
|
23
|
+
return new Config(this.preset, contextMode, this.contextLength, this.prefillStepSize, this.samplingSeed, this.asyncBatchSize);
|
|
22
24
|
}
|
|
23
25
|
withContextLength(contextLength) {
|
|
24
|
-
return new Config(this.preset, this.contextMode, contextLength, this.prefillStepSize, this.samplingSeed);
|
|
26
|
+
return new Config(this.preset, this.contextMode, contextLength, this.prefillStepSize, this.samplingSeed, this.asyncBatchSize);
|
|
25
27
|
}
|
|
26
28
|
withPrefillStepSize(prefillStepSize) {
|
|
27
|
-
return new Config(this.preset, this.contextMode, this.contextLength, prefillStepSize, this.samplingSeed);
|
|
29
|
+
return new Config(this.preset, this.contextMode, this.contextLength, prefillStepSize, this.samplingSeed, this.asyncBatchSize);
|
|
28
30
|
}
|
|
29
31
|
withSamplingSeed(samplingSeed) {
|
|
30
|
-
return new Config(this.preset, this.contextMode, this.contextLength, this.prefillStepSize, samplingSeed);
|
|
32
|
+
return new Config(this.preset, this.contextMode, this.contextLength, this.prefillStepSize, samplingSeed, this.asyncBatchSize);
|
|
33
|
+
}
|
|
34
|
+
withAsyncBatchSize(asyncBatchSize) {
|
|
35
|
+
return new Config(this.preset, this.contextMode, this.contextLength, this.prefillStepSize, this.samplingSeed, asyncBatchSize);
|
|
31
36
|
}
|
|
32
37
|
toNapi() {
|
|
33
38
|
const napiConfig = {
|
|
@@ -36,6 +41,7 @@ export class Config {
|
|
|
36
41
|
contextLength: this.contextLength.toNapi(),
|
|
37
42
|
prefillStepSize: this.prefillStepSize.toNapi(),
|
|
38
43
|
samplingSeed: this.samplingSeed.toNapi(),
|
|
44
|
+
asyncBatchSize: this.asyncBatchSize.toNapi(),
|
|
39
45
|
};
|
|
40
46
|
return napiConfig;
|
|
41
47
|
}
|
package/bridging/config.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.mjs","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":"OACO,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OAEf,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,YAAY,EAAE;AAEvB,MAAM,OAAO,MAAM;
|
|
1
|
+
{"version":3,"file":"config.mjs","sourceRoot":"","sources":["../src/bridging/config.ts"],"names":[],"mappings":"OACO,EAAE,cAAc,EAAE;OAClB,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OAEf,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,YAAY,EAAE;AAEvB,MAAM,OAAO,MAAM;IAQf,YACI,MAAc,EACd,WAAwB,EACxB,aAA4B,EAC5B,eAAgC,EAChC,YAA0B,EAC1B,cAA8B;QAE9B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;IACzC,CAAC;IAED,MAAM,CAAC,OAAO;QACV,OAAO,IAAI,MAAM,CACb,MAAM,CAAC,OAAO,EAAE,EAChB,WAAW,CAAC,IAAI,EAAE,EAClB,aAAa,CAAC,OAAO,EAAE,EACvB,eAAe,CAAC,OAAO,EAAE,EACzB,YAAY,CAAC,OAAO,EAAE,EACtB,cAAc,CAAC,OAAO,EAAE,CAC3B,CAAC;IACN,CAAC;IAED,UAAU,CAAC,MAAc;QACrB,OAAO,IAAI,MAAM,CACb,MAAM,EACN,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,cAAc,CACtB,CAAC;IACN,CAAC;IAED,eAAe,CAAC,WAAwB;QACpC,OAAO,IAAI,MAAM,CACb,IAAI,CAAC,MAAM,EACX,WAAW,EACX,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,cAAc,CACtB,CAAC;IACN,CAAC;IAED,iBAAiB,CAAC,aAA4B;QAC1C,OAAO,IAAI,MAAM,CACb,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,WAAW,EAChB,aAAa,EACb,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,cAAc,CACtB,CAAC;IACN,CAAC;IAED,mBAAmB,CAAC,eAAgC;QAChD,OAAO,IAAI,MAAM,CACb,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,aAAa,EAClB,eAAe,EACf,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,cAAc,CACtB,CAAC;IACN,CAAC;IAED,gBAAgB,CAAC,YAA0B;QACvC,OAAO,IAAI,MAAM,CACb,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,eAAe,EACpB,YAAY,EACZ,IAAI,CAAC,cAAc,CACtB,CAAC;IACN,CAAC;IAED,kBAAkB,CAAC,cAA8B;QAC7C,OAAO,IAAI,MAAM,CACb,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,YAAY,EACjB,cAAc,CACjB,CAAC;IACN,CAAC;IAED,MAAM;QACF,MAAM,UAAU,GAAe;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;YAC5B,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE;YACtC,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE;YAC1C,eAAe,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;YAC9C,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;YACxC,cAAc,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE;SAC/C,CAAC;QACF,OAAO,UAAU,CAAC;IACtB,CAAC;CACJ"}
|
package/index.d.mts
CHANGED
package/index.d.mts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.mts","sourceRoot":"","sources":["src/index.ts"],"names":[],"mappings":"OAAO,EAAE,SAAS,EAAE;OACb,EAAE,WAAW,EAAE;OACf,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;OACV,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,cAAc,EAAE;OAClB,EAAE,sBAAsB,EAAE;OAC1B,EAAE,aAAa,EAAE,aAAa,EAAE;OAChC,EAAE,MAAM,IAAI,OAAO,EAAE;OACrB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,OAAO,EAAE;OACX,EAAE,SAAS,EAAE;OACb,EAAE,MAAM,EAAE;OACV,EAAE,UAAU,EAAE;OACd,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,IAAI,EAAE;OACR,EAAE,SAAS,EAAE;OACb,EAAE,QAAQ,EAAE;OACZ,EAAE,cAAc,EAAE;OAClB,EAAE,cAAc,EAAE;OAClB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,SAAS,EAAE;OACb,EAAE,IAAI,EAAE;OACR,EAAE,UAAU,EAAE;OACd,EAAE,WAAW,EAAE;OACf,EAAE,mBAAmB,EAAE;OACvB,EAAE,oBAAoB,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;OACtB,EAAE,gBAAgB,EAAE"}
|
|
1
|
+
{"version":3,"file":"index.d.mts","sourceRoot":"","sources":["src/index.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,EAAE;OAClB,EAAE,SAAS,EAAE;OACb,EAAE,WAAW,EAAE;OACf,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;OACV,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,cAAc,EAAE;OAClB,EAAE,sBAAsB,EAAE;OAC1B,EAAE,aAAa,EAAE,aAAa,EAAE;OAChC,EAAE,MAAM,IAAI,OAAO,EAAE;OACrB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,OAAO,EAAE;OACX,EAAE,SAAS,EAAE;OACb,EAAE,MAAM,EAAE;OACV,EAAE,UAAU,EAAE;OACd,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,IAAI,EAAE;OACR,EAAE,SAAS,EAAE;OACb,EAAE,QAAQ,EAAE;OACZ,EAAE,cAAc,EAAE;OAClB,EAAE,cAAc,EAAE;OAClB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,SAAS,EAAE;OACb,EAAE,IAAI,EAAE;OACR,EAAE,UAAU,EAAE;OACd,EAAE,WAAW,EAAE;OACf,EAAE,mBAAmB,EAAE;OACvB,EAAE,oBAAoB,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;OACtB,EAAE,gBAAgB,EAAE"}
|
package/index.d.ts
CHANGED
package/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["src/index.ts"],"names":[],"mappings":"OAAO,EAAE,SAAS,EAAE;OACb,EAAE,WAAW,EAAE;OACf,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;OACV,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,cAAc,EAAE;OAClB,EAAE,sBAAsB,EAAE;OAC1B,EAAE,aAAa,EAAE,aAAa,EAAE;OAChC,EAAE,MAAM,IAAI,OAAO,EAAE;OACrB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,OAAO,EAAE;OACX,EAAE,SAAS,EAAE;OACb,EAAE,MAAM,EAAE;OACV,EAAE,UAAU,EAAE;OACd,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,IAAI,EAAE;OACR,EAAE,SAAS,EAAE;OACb,EAAE,QAAQ,EAAE;OACZ,EAAE,cAAc,EAAE;OAClB,EAAE,cAAc,EAAE;OAClB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,SAAS,EAAE;OACb,EAAE,IAAI,EAAE;OACR,EAAE,UAAU,EAAE;OACd,EAAE,WAAW,EAAE;OACf,EAAE,mBAAmB,EAAE;OACvB,EAAE,oBAAoB,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;OACtB,EAAE,gBAAgB,EAAE"}
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["src/index.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,EAAE;OAClB,EAAE,SAAS,EAAE;OACb,EAAE,WAAW,EAAE;OACf,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;OACV,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,cAAc,EAAE;OAClB,EAAE,sBAAsB,EAAE;OAC1B,EAAE,aAAa,EAAE,aAAa,EAAE;OAChC,EAAE,MAAM,IAAI,OAAO,EAAE;OACrB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,OAAO,EAAE;OACX,EAAE,SAAS,EAAE;OACb,EAAE,MAAM,EAAE;OACV,EAAE,UAAU,EAAE;OACd,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,IAAI,EAAE;OACR,EAAE,SAAS,EAAE;OACb,EAAE,QAAQ,EAAE;OACZ,EAAE,cAAc,EAAE;OAClB,EAAE,cAAc,EAAE;OAClB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,SAAS,EAAE;OACb,EAAE,IAAI,EAAE;OACR,EAAE,UAAU,EAAE;OACd,EAAE,WAAW,EAAE;OACf,EAAE,mBAAmB,EAAE;OACvB,EAAE,oBAAoB,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;OACtB,EAAE,gBAAgB,EAAE"}
|
package/index.js
CHANGED
|
@@ -3,7 +3,9 @@ exports = module.exports = function (...args) {
|
|
|
3
3
|
return new exports.default(...args)
|
|
4
4
|
}
|
|
5
5
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
-
exports.EngineInteractor = exports.DownloadInteractor = exports.ChatSessionInteractor = exports.ChatModelsInteractor = exports.ChatModelInteractor = exports.EngineError = exports.TotalStats = exports.Text = exports.StepStats = exports.Stats = exports.SamplingSeed = exports.SamplingPolicy = exports.SamplingMethod = exports.RunStats = exports.RunConfig = exports.Role = exports.Preset = exports.PrefillStepSize = exports.ParsedText = exports.Output = exports.ModelType = exports.Message = exports.Input = exports.FinishReason = exports.default = exports.DownloadState = exports.DownloadPhase = exports.DownloadProgressUpdate = exports.DownloadHandle = exports.ContextMode = exports.ContextLength = exports.Config = exports.ClassificationFeature = exports.ChatSession = exports.ChatModel = void 0;
|
|
6
|
+
exports.EngineInteractor = exports.DownloadInteractor = exports.ChatSessionInteractor = exports.ChatModelsInteractor = exports.ChatModelInteractor = exports.EngineError = exports.TotalStats = exports.Text = exports.StepStats = exports.Stats = exports.SamplingSeed = exports.SamplingPolicy = exports.SamplingMethod = exports.RunStats = exports.RunConfig = exports.Role = exports.Preset = exports.PrefillStepSize = exports.ParsedText = exports.Output = exports.ModelType = exports.Message = exports.Input = exports.FinishReason = exports.default = exports.DownloadState = exports.DownloadPhase = exports.DownloadProgressUpdate = exports.DownloadHandle = exports.ContextMode = exports.ContextLength = exports.Config = exports.ClassificationFeature = exports.ChatSession = exports.ChatModel = exports.AsyncBatchSize = void 0;
|
|
7
|
+
var asyncBatchSize_1 = require("./bridging/asyncBatchSize.js");
|
|
8
|
+
Object.defineProperty(exports, "AsyncBatchSize", { enumerable: true, get: function () { return asyncBatchSize_1.AsyncBatchSize; } });
|
|
7
9
|
var chatModel_1 = require("./bridging/chatModel.js");
|
|
8
10
|
Object.defineProperty(exports, "ChatModel", { enumerable: true, get: function () { return chatModel_1.ChatModel; } });
|
|
9
11
|
var chatSession_1 = require("./bridging/chatSession.js");
|
package/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["src/index.ts"],"names":[],"mappings":";;;AAAA,qDAAiD;AAAxC,sGAAA,SAAS,OAAA;AAClB,yDAAqD;AAA5C,0GAAA,WAAW,OAAA;AACpB,6EAAyE;AAAhE,8HAAA,qBAAqB,OAAA;AAC9B,+CAA2C;AAAlC,gGAAA,MAAM,OAAA;AACf,6DAAyD;AAAhD,8GAAA,aAAa,OAAA;AACtB,yDAAqD;AAA5C,0GAAA,WAAW,OAAA;AACpB,+DAA2D;AAAlD,gHAAA,cAAc,OAAA;AACvB,+EAA2E;AAAlE,gIAAA,sBAAsB,OAAA;AAC/B,6DAAwE;AAA/D,8GAAA,aAAa,OAAA;AAAE,8GAAA,aAAa,OAAA;AACrC,+CAAsD;AAA7C,iGAAA,MAAM,OAAW;AAC1B,2DAAuD;AAA9C,4GAAA,YAAY,OAAA;AACrB,6CAAyC;AAAhC,8FAAA,KAAK,OAAA;AACd,iDAA6C;AAApC,kGAAA,OAAO,OAAA;AAChB,qDAAiD;AAAxC,sGAAA,SAAS,OAAA;AAClB,+CAA2C;AAAlC,gGAAA,MAAM,OAAA;AACf,uDAAmD;AAA1C,wGAAA,UAAU,OAAA;AACnB,iEAA6D;AAApD,kHAAA,eAAe,OAAA;AACxB,+CAA2C;AAAlC,gGAAA,MAAM,OAAA;AACf,2CAAuC;AAA9B,4FAAA,IAAI,OAAA;AACb,qDAAiD;AAAxC,sGAAA,SAAS,OAAA;AAClB,mDAA+C;AAAtC,oGAAA,QAAQ,OAAA;AACjB,+DAA2D;AAAlD,gHAAA,cAAc,OAAA;AACvB,+DAA2D;AAAlD,gHAAA,cAAc,OAAA;AACvB,2DAAuD;AAA9C,4GAAA,YAAY,OAAA;AACrB,6CAAyC;AAAhC,8FAAA,KAAK,OAAA;AACd,qDAAiD;AAAxC,sGAAA,SAAS,OAAA;AAClB,2CAAuC;AAA9B,4FAAA,IAAI,OAAA;AACb,uDAAmD;AAA1C,wGAAA,UAAU,OAAA;AACnB,oCAAsC;AAA7B,oGAAA,WAAW,OAAA;AACpB,4EAAwE;AAA/D,0HAAA,mBAAmB,OAAA;AAC5B,8EAA0E;AAAjE,4HAAA,oBAAoB,OAAA;AAC7B,gFAA4E;AAAnE,8HAAA,qBAAqB,OAAA;AAC9B,0EAAsE;AAA7D,wHAAA,kBAAkB,OAAA;AAC3B,sEAAkE;AAAzD,oHAAA,gBAAgB,OAAA"}
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["src/index.ts"],"names":[],"mappings":";;;AAAA,+DAA2D;AAAlD,gHAAA,cAAc,OAAA;AACvB,qDAAiD;AAAxC,sGAAA,SAAS,OAAA;AAClB,yDAAqD;AAA5C,0GAAA,WAAW,OAAA;AACpB,6EAAyE;AAAhE,8HAAA,qBAAqB,OAAA;AAC9B,+CAA2C;AAAlC,gGAAA,MAAM,OAAA;AACf,6DAAyD;AAAhD,8GAAA,aAAa,OAAA;AACtB,yDAAqD;AAA5C,0GAAA,WAAW,OAAA;AACpB,+DAA2D;AAAlD,gHAAA,cAAc,OAAA;AACvB,+EAA2E;AAAlE,gIAAA,sBAAsB,OAAA;AAC/B,6DAAwE;AAA/D,8GAAA,aAAa,OAAA;AAAE,8GAAA,aAAa,OAAA;AACrC,+CAAsD;AAA7C,iGAAA,MAAM,OAAW;AAC1B,2DAAuD;AAA9C,4GAAA,YAAY,OAAA;AACrB,6CAAyC;AAAhC,8FAAA,KAAK,OAAA;AACd,iDAA6C;AAApC,kGAAA,OAAO,OAAA;AAChB,qDAAiD;AAAxC,sGAAA,SAAS,OAAA;AAClB,+CAA2C;AAAlC,gGAAA,MAAM,OAAA;AACf,uDAAmD;AAA1C,wGAAA,UAAU,OAAA;AACnB,iEAA6D;AAApD,kHAAA,eAAe,OAAA;AACxB,+CAA2C;AAAlC,gGAAA,MAAM,OAAA;AACf,2CAAuC;AAA9B,4FAAA,IAAI,OAAA;AACb,qDAAiD;AAAxC,sGAAA,SAAS,OAAA;AAClB,mDAA+C;AAAtC,oGAAA,QAAQ,OAAA;AACjB,+DAA2D;AAAlD,gHAAA,cAAc,OAAA;AACvB,+DAA2D;AAAlD,gHAAA,cAAc,OAAA;AACvB,2DAAuD;AAA9C,4GAAA,YAAY,OAAA;AACrB,6CAAyC;AAAhC,8FAAA,KAAK,OAAA;AACd,qDAAiD;AAAxC,sGAAA,SAAS,OAAA;AAClB,2CAAuC;AAA9B,4FAAA,IAAI,OAAA;AACb,uDAAmD;AAA1C,wGAAA,UAAU,OAAA;AACnB,oCAAsC;AAA7B,oGAAA,WAAW,OAAA;AACpB,4EAAwE;AAA/D,0HAAA,mBAAmB,OAAA;AAC5B,8EAA0E;AAAjE,4HAAA,oBAAoB,OAAA;AAC7B,gFAA4E;AAAnE,8HAAA,qBAAqB,OAAA;AAC9B,0EAAsE;AAA7D,wHAAA,kBAAkB,OAAA;AAC3B,sEAAkE;AAAzD,oHAAA,gBAAgB,OAAA"}
|
package/index.mjs
CHANGED
package/index.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.mjs","sourceRoot":"","sources":["src/index.ts"],"names":[],"mappings":"OAAO,EAAE,SAAS,EAAE;OACb,EAAE,WAAW,EAAE;OACf,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;OACV,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,cAAc,EAAE;OAClB,EAAE,sBAAsB,EAAE;OAC1B,EAAE,aAAa,EAAE,aAAa,EAAE;OAChC,EAAE,MAAM,IAAI,OAAO,EAAE;OACrB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,OAAO,EAAE;OACX,EAAE,SAAS,EAAE;OACb,EAAE,MAAM,EAAE;OACV,EAAE,UAAU,EAAE;OACd,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,IAAI,EAAE;OACR,EAAE,SAAS,EAAE;OACb,EAAE,QAAQ,EAAE;OACZ,EAAE,cAAc,EAAE;OAClB,EAAE,cAAc,EAAE;OAClB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,SAAS,EAAE;OACb,EAAE,IAAI,EAAE;OACR,EAAE,UAAU,EAAE;OACd,EAAE,WAAW,EAAE;OACf,EAAE,mBAAmB,EAAE;OACvB,EAAE,oBAAoB,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;OACtB,EAAE,gBAAgB,EAAE"}
|
|
1
|
+
{"version":3,"file":"index.mjs","sourceRoot":"","sources":["src/index.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,EAAE;OAClB,EAAE,SAAS,EAAE;OACb,EAAE,WAAW,EAAE;OACf,EAAE,qBAAqB,EAAE;OACzB,EAAE,MAAM,EAAE;OACV,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,cAAc,EAAE;OAClB,EAAE,sBAAsB,EAAE;OAC1B,EAAE,aAAa,EAAE,aAAa,EAAE;OAChC,EAAE,MAAM,IAAI,OAAO,EAAE;OACrB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,OAAO,EAAE;OACX,EAAE,SAAS,EAAE;OACb,EAAE,MAAM,EAAE;OACV,EAAE,UAAU,EAAE;OACd,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OACV,EAAE,IAAI,EAAE;OACR,EAAE,SAAS,EAAE;OACb,EAAE,QAAQ,EAAE;OACZ,EAAE,cAAc,EAAE;OAClB,EAAE,cAAc,EAAE;OAClB,EAAE,YAAY,EAAE;OAChB,EAAE,KAAK,EAAE;OACT,EAAE,SAAS,EAAE;OACb,EAAE,IAAI,EAAE;OACR,EAAE,UAAU,EAAE;OACd,EAAE,WAAW,EAAE;OACf,EAAE,mBAAmB,EAAE;OACvB,EAAE,oBAAoB,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;OACtB,EAAE,gBAAgB,EAAE"}
|
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
import { AsyncBatchSize } from "../bridging/asyncBatchSize.mjs";
|
|
1
2
|
import { ChatModel } from "../bridging/chatModel.mjs";
|
|
2
3
|
import { Config } from "../bridging/config.mjs";
|
|
3
4
|
import { ContextLength } from "../bridging/contextLength.mjs";
|
|
@@ -25,6 +26,7 @@ export declare class ChatModelInteractor implements Interactor<ChatModel> {
|
|
|
25
26
|
contextLength(contextLength: ContextLength): ChatModelInteractor;
|
|
26
27
|
prefillStepSize(prefillStepSize: PrefillStepSize): ChatModelInteractor;
|
|
27
28
|
samplingSeed(samplingSeed: SamplingSeed): ChatModelInteractor;
|
|
29
|
+
asyncBatchSize(asyncBatchSize: AsyncBatchSize): ChatModelInteractor;
|
|
28
30
|
session(): ChatSessionInteractor;
|
|
29
31
|
reply(text: string, progress?: (partialOutput: Output) => boolean): Promise<Output>;
|
|
30
32
|
replyToMessages(messages: Message[], progress?: (partialOutput: Output) => boolean): Promise<Output>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"chatModelInteractor.d.mts","sourceRoot":"","sources":["../src/interactors/chatModelInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,SAAS,EAAE;OACb,EAAE,MAAM,EAAE;OACV,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,sBAAsB,EAAE;OAE1B,EAAE,OAAO,EAAE;OAEX,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OAEV,EAAE,YAAY,EAAE;OAChB,EAAE,oBAAoB,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;OACtB,EAAE,UAAU,EAAE,gBAAgB,EAAE;AAEvC,qBAAa,mBAAoB,YAAW,UAAU,CAAC,SAAS,CAAC;IAC7D,QAAQ,CAAC,gBAAgB,EAAE,oBAAoB,CAAC;IAChD,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,SAAS,CAAC,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;gBAEZ,gBAAgB,EAAE,oBAAoB,EAAE,KAAK,EAAE,gBAAgB,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,MAAM;IAMhG,QAAQ,IAAI,OAAO,CAAC,SAAS,CAAC;IAMpC,WAAW,IAAI,kBAAkB;IAUjC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,cAAc,EAAE,sBAAsB,KAAK,IAAI,GAAG,mBAAmB;IAsB1F,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,mBAAmB;IAK3C,WAAW,CAAC,WAAW,EAAE,WAAW,GAAG,mBAAmB;IAK1D,aAAa,CAAC,aAAa,EAAE,aAAa,GAAG,mBAAmB;IAKhE,eAAe,CAAC,eAAe,EAAE,eAAe,GAAG,mBAAmB;IAKtE,YAAY,CAAC,YAAY,EAAE,YAAY,GAAG,mBAAmB;
|
|
1
|
+
{"version":3,"file":"chatModelInteractor.d.mts","sourceRoot":"","sources":["../src/interactors/chatModelInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,EAAE;OAClB,EAAE,SAAS,EAAE;OACb,EAAE,MAAM,EAAE;OACV,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,sBAAsB,EAAE;OAE1B,EAAE,OAAO,EAAE;OAEX,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OAEV,EAAE,YAAY,EAAE;OAChB,EAAE,oBAAoB,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;OACtB,EAAE,UAAU,EAAE,gBAAgB,EAAE;AAEvC,qBAAa,mBAAoB,YAAW,UAAU,CAAC,SAAS,CAAC;IAC7D,QAAQ,CAAC,gBAAgB,EAAE,oBAAoB,CAAC;IAChD,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,SAAS,CAAC,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;gBAEZ,gBAAgB,EAAE,oBAAoB,EAAE,KAAK,EAAE,gBAAgB,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,MAAM;IAMhG,QAAQ,IAAI,OAAO,CAAC,SAAS,CAAC;IAMpC,WAAW,IAAI,kBAAkB;IAUjC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,cAAc,EAAE,sBAAsB,KAAK,IAAI,GAAG,mBAAmB;IAsB1F,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,mBAAmB;IAK3C,WAAW,CAAC,WAAW,EAAE,WAAW,GAAG,mBAAmB;IAK1D,aAAa,CAAC,aAAa,EAAE,aAAa,GAAG,mBAAmB;IAKhE,eAAe,CAAC,eAAe,EAAE,eAAe,GAAG,mBAAmB;IAKtE,YAAY,CAAC,YAAY,EAAE,YAAY,GAAG,mBAAmB;IAK7D,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,mBAAmB;IAOnE,OAAO,IAAI,qBAAqB;IAe1B,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"}
|
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
import { AsyncBatchSize } from "../bridging/asyncBatchSize.js";
|
|
1
2
|
import { ChatModel } from "../bridging/chatModel.js";
|
|
2
3
|
import { Config } from "../bridging/config.js";
|
|
3
4
|
import { ContextLength } from "../bridging/contextLength.js";
|
|
@@ -25,6 +26,7 @@ export declare class ChatModelInteractor implements Interactor<ChatModel> {
|
|
|
25
26
|
contextLength(contextLength: ContextLength): ChatModelInteractor;
|
|
26
27
|
prefillStepSize(prefillStepSize: PrefillStepSize): ChatModelInteractor;
|
|
27
28
|
samplingSeed(samplingSeed: SamplingSeed): ChatModelInteractor;
|
|
29
|
+
asyncBatchSize(asyncBatchSize: AsyncBatchSize): ChatModelInteractor;
|
|
28
30
|
session(): ChatSessionInteractor;
|
|
29
31
|
reply(text: string, progress?: (partialOutput: Output) => boolean): Promise<Output>;
|
|
30
32
|
replyToMessages(messages: Message[], progress?: (partialOutput: Output) => boolean): Promise<Output>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"chatModelInteractor.d.ts","sourceRoot":"","sources":["../src/interactors/chatModelInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,SAAS,EAAE;OACb,EAAE,MAAM,EAAE;OACV,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,sBAAsB,EAAE;OAE1B,EAAE,OAAO,EAAE;OAEX,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OAEV,EAAE,YAAY,EAAE;OAChB,EAAE,oBAAoB,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;OACtB,EAAE,UAAU,EAAE,gBAAgB,EAAE;AAEvC,qBAAa,mBAAoB,YAAW,UAAU,CAAC,SAAS,CAAC;IAC7D,QAAQ,CAAC,gBAAgB,EAAE,oBAAoB,CAAC;IAChD,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,SAAS,CAAC,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;gBAEZ,gBAAgB,EAAE,oBAAoB,EAAE,KAAK,EAAE,gBAAgB,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,MAAM;IAMhG,QAAQ,IAAI,OAAO,CAAC,SAAS,CAAC;IAMpC,WAAW,IAAI,kBAAkB;IAUjC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,cAAc,EAAE,sBAAsB,KAAK,IAAI,GAAG,mBAAmB;IAsB1F,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,mBAAmB;IAK3C,WAAW,CAAC,WAAW,EAAE,WAAW,GAAG,mBAAmB;IAK1D,aAAa,CAAC,aAAa,EAAE,aAAa,GAAG,mBAAmB;IAKhE,eAAe,CAAC,eAAe,EAAE,eAAe,GAAG,mBAAmB;IAKtE,YAAY,CAAC,YAAY,EAAE,YAAY,GAAG,mBAAmB;
|
|
1
|
+
{"version":3,"file":"chatModelInteractor.d.ts","sourceRoot":"","sources":["../src/interactors/chatModelInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,EAAE;OAClB,EAAE,SAAS,EAAE;OACb,EAAE,MAAM,EAAE;OACV,EAAE,aAAa,EAAE;OACjB,EAAE,WAAW,EAAE;OACf,EAAE,sBAAsB,EAAE;OAE1B,EAAE,OAAO,EAAE;OAEX,EAAE,MAAM,EAAE;OACV,EAAE,eAAe,EAAE;OACnB,EAAE,MAAM,EAAE;OAEV,EAAE,YAAY,EAAE;OAChB,EAAE,oBAAoB,EAAE;OACxB,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;OACtB,EAAE,UAAU,EAAE,gBAAgB,EAAE;AAEvC,qBAAa,mBAAoB,YAAW,UAAU,CAAC,SAAS,CAAC;IAC7D,QAAQ,CAAC,gBAAgB,EAAE,oBAAoB,CAAC;IAChD,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,SAAS,CAAC,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;gBAEZ,gBAAgB,EAAE,oBAAoB,EAAE,KAAK,EAAE,gBAAgB,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,MAAM;IAMhG,QAAQ,IAAI,OAAO,CAAC,SAAS,CAAC;IAMpC,WAAW,IAAI,kBAAkB;IAUjC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,cAAc,EAAE,sBAAsB,KAAK,IAAI,GAAG,mBAAmB;IAsB1F,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,mBAAmB;IAK3C,WAAW,CAAC,WAAW,EAAE,WAAW,GAAG,mBAAmB;IAK1D,aAAa,CAAC,aAAa,EAAE,aAAa,GAAG,mBAAmB;IAKhE,eAAe,CAAC,eAAe,EAAE,eAAe,GAAG,mBAAmB;IAKtE,YAAY,CAAC,YAAY,EAAE,YAAY,GAAG,mBAAmB;IAK7D,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,mBAAmB;IAOnE,OAAO,IAAI,qBAAqB;IAe1B,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"}
|
|
@@ -65,6 +65,10 @@ class ChatModelInteractor {
|
|
|
65
65
|
const config = this.config.withSamplingSeed(samplingSeed);
|
|
66
66
|
return new ChatModelInteractor(this.modelsInteractor, this.entity, config);
|
|
67
67
|
}
|
|
68
|
+
asyncBatchSize(asyncBatchSize) {
|
|
69
|
+
const config = this.config.withAsyncBatchSize(asyncBatchSize);
|
|
70
|
+
return new ChatModelInteractor(this.modelsInteractor, this.entity, config);
|
|
71
|
+
}
|
|
68
72
|
/* Session */
|
|
69
73
|
session() {
|
|
70
74
|
const sessionPromise = (async () => {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"chatModelInteractor.js","sourceRoot":"","sources":["../src/interactors/chatModelInteractor.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"chatModelInteractor.js","sourceRoot":"","sources":["../src/interactors/chatModelInteractor.ts"],"names":[],"mappings":";;;AAMA,gEAA0D;AAE1D,wDAAkD;AAIlD,wDAAkD;AAGlD,sEAAgE;AAChE,gEAA0D;AAG1D,MAAa,mBAAmB;IAK5B,YAAY,gBAAsC,EAAE,KAAkC,EAAE,MAAc;QAClG,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,MAAM,CAAC,CAAC;YACnD,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,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACrF,CAAC;IAED,YAAY;IAEZ,MAAM,CAAC,MAAc;QACjB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC9C,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,WAAW,CAAC,WAAwB;QAChC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACxD,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,aAAa,CAAC,aAA4B;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAC5D,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,eAAe,CAAC,eAAgC;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC;QAChE,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,YAAY,CAAC,YAA0B;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;QAC1D,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,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,qBAAS,CAAC,KAAK,EAAE,CAAC;gBACjC,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;YAC7C,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACvD,OAAO,OAAO,CAAC;QACnB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,6CAAqB,CAAC,IAAI,EAAE,cAAc,EAAE,qBAAS,CAAC,OAAO,EAAE,CAAC,CAAC;IAChF,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;AA1GD,kDA0GC"}
|
|
@@ -62,6 +62,10 @@ export class ChatModelInteractor {
|
|
|
62
62
|
const config = this.config.withSamplingSeed(samplingSeed);
|
|
63
63
|
return new ChatModelInteractor(this.modelsInteractor, this.entity, config);
|
|
64
64
|
}
|
|
65
|
+
asyncBatchSize(asyncBatchSize) {
|
|
66
|
+
const config = this.config.withAsyncBatchSize(asyncBatchSize);
|
|
67
|
+
return new ChatModelInteractor(this.modelsInteractor, this.entity, config);
|
|
68
|
+
}
|
|
65
69
|
/* Session */
|
|
66
70
|
session() {
|
|
67
71
|
const sessionPromise = (async () => {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"chatModelInteractor.mjs","sourceRoot":"","sources":["../src/interactors/chatModelInteractor.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"chatModelInteractor.mjs","sourceRoot":"","sources":["../src/interactors/chatModelInteractor.ts"],"names":[],"mappings":"OAMO,EAAE,aAAa,EAAE;OAEjB,EAAE,SAAS,EAAE;OAIb,EAAE,SAAS,EAAE;OAGb,EAAE,qBAAqB,EAAE;OACzB,EAAE,kBAAkB,EAAE;AAG7B,MAAM,OAAO,mBAAmB;IAK5B,YAAY,gBAAsC,EAAE,KAAkC,EAAE,MAAc;QAClG,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,MAAM,CAAC,CAAC;YACnD,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,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACrF,CAAC;IAED,YAAY;IAEZ,MAAM,CAAC,MAAc;QACjB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC9C,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,WAAW,CAAC,WAAwB;QAChC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACxD,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,aAAa,CAAC,aAA4B;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAC5D,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,eAAe,CAAC,eAAgC;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC;QAChE,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,YAAY,CAAC,YAA0B;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;QAC1D,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC/E,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,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACvD,OAAO,OAAO,CAAC;QACnB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,qBAAqB,CAAC,IAAI,EAAE,cAAc,EAAE,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC;IAChF,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/napi/uzu.d.ts
CHANGED
|
@@ -39,6 +39,10 @@ export declare class ProgressUpdate {
|
|
|
39
39
|
progress: number
|
|
40
40
|
}
|
|
41
41
|
|
|
42
|
+
export type AsyncBatchSize =
|
|
43
|
+
| { type: 'Default' }
|
|
44
|
+
| { type: 'Custom', size: number }
|
|
45
|
+
|
|
42
46
|
export interface ChatModel {
|
|
43
47
|
readonly repoId: string
|
|
44
48
|
readonly type: ModelType
|
|
@@ -59,6 +63,7 @@ export interface Config {
|
|
|
59
63
|
contextLength: ContextLength
|
|
60
64
|
prefillStepSize: PrefillStepSize
|
|
61
65
|
samplingSeed: SamplingSeed
|
|
66
|
+
asyncBatchSize: AsyncBatchSize
|
|
62
67
|
}
|
|
63
68
|
|
|
64
69
|
export type ContextLength =
|
package/napi/uzu.js
CHANGED
|
@@ -44,6 +44,7 @@ const {
|
|
|
44
44
|
StepStats,
|
|
45
45
|
Text,
|
|
46
46
|
TotalStats,
|
|
47
|
+
AsyncBatchSize,
|
|
47
48
|
ContextLength,
|
|
48
49
|
ContextMode,
|
|
49
50
|
Input,
|
|
@@ -81,6 +82,7 @@ module.exports = {
|
|
|
81
82
|
StepStats,
|
|
82
83
|
Text,
|
|
83
84
|
TotalStats,
|
|
85
|
+
AsyncBatchSize,
|
|
84
86
|
ContextLength,
|
|
85
87
|
ContextMode,
|
|
86
88
|
Input,
|
package/napi/uzu.mjs
CHANGED
package/napi/uzu.node
CHANGED
|
Binary file
|
package/package.json
CHANGED
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
import { AsyncBatchSize as NapiAsyncBatchSize } from '../napi/uzu';
|
|
2
|
+
import { ToNapi } from './napi';
|
|
3
|
+
|
|
4
|
+
export class AsyncBatchSize implements ToNapi<NapiAsyncBatchSize> {
|
|
5
|
+
private readonly napiAsyncBatchSize: NapiAsyncBatchSize;
|
|
6
|
+
|
|
7
|
+
private constructor(napiAsyncBatchSize: NapiAsyncBatchSize) {
|
|
8
|
+
this.napiAsyncBatchSize = napiAsyncBatchSize;
|
|
9
|
+
}
|
|
10
|
+
|
|
11
|
+
static default(): AsyncBatchSize {
|
|
12
|
+
const napiAsyncBatchSize: NapiAsyncBatchSize = { type: 'Default' };
|
|
13
|
+
return new AsyncBatchSize(napiAsyncBatchSize);
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
static custom(size: number): AsyncBatchSize {
|
|
17
|
+
const napiAsyncBatchSize: NapiAsyncBatchSize = { type: 'Custom', size };
|
|
18
|
+
return new AsyncBatchSize(napiAsyncBatchSize);
|
|
19
|
+
}
|
|
20
|
+
|
|
21
|
+
toNapi(): NapiAsyncBatchSize {
|
|
22
|
+
return this.napiAsyncBatchSize;
|
|
23
|
+
}
|
|
24
|
+
}
|
package/src/bridging/config.ts
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import { Config as NapiConfig } from '../napi/uzu';
|
|
2
|
+
import { AsyncBatchSize } from './asyncBatchSize';
|
|
2
3
|
import { ContextLength } from './contextLength';
|
|
3
4
|
import { ContextMode } from './contextMode';
|
|
4
5
|
import { ToNapi } from './napi';
|
|
@@ -12,6 +13,7 @@ export class Config implements ToNapi<NapiConfig> {
|
|
|
12
13
|
readonly contextLength: ContextLength;
|
|
13
14
|
readonly prefillStepSize: PrefillStepSize;
|
|
14
15
|
readonly samplingSeed: SamplingSeed;
|
|
16
|
+
readonly asyncBatchSize: AsyncBatchSize;
|
|
15
17
|
|
|
16
18
|
constructor(
|
|
17
19
|
preset: Preset,
|
|
@@ -19,12 +21,14 @@ export class Config implements ToNapi<NapiConfig> {
|
|
|
19
21
|
contextLength: ContextLength,
|
|
20
22
|
prefillStepSize: PrefillStepSize,
|
|
21
23
|
samplingSeed: SamplingSeed,
|
|
24
|
+
asyncBatchSize: AsyncBatchSize,
|
|
22
25
|
) {
|
|
23
26
|
this.preset = preset;
|
|
24
27
|
this.contextMode = contextMode;
|
|
25
28
|
this.contextLength = contextLength;
|
|
26
29
|
this.prefillStepSize = prefillStepSize;
|
|
27
30
|
this.samplingSeed = samplingSeed;
|
|
31
|
+
this.asyncBatchSize = asyncBatchSize;
|
|
28
32
|
}
|
|
29
33
|
|
|
30
34
|
static default(): Config {
|
|
@@ -34,6 +38,7 @@ export class Config implements ToNapi<NapiConfig> {
|
|
|
34
38
|
ContextLength.default(),
|
|
35
39
|
PrefillStepSize.default(),
|
|
36
40
|
SamplingSeed.default(),
|
|
41
|
+
AsyncBatchSize.default(),
|
|
37
42
|
);
|
|
38
43
|
}
|
|
39
44
|
|
|
@@ -44,6 +49,7 @@ export class Config implements ToNapi<NapiConfig> {
|
|
|
44
49
|
this.contextLength,
|
|
45
50
|
this.prefillStepSize,
|
|
46
51
|
this.samplingSeed,
|
|
52
|
+
this.asyncBatchSize,
|
|
47
53
|
);
|
|
48
54
|
}
|
|
49
55
|
|
|
@@ -54,6 +60,7 @@ export class Config implements ToNapi<NapiConfig> {
|
|
|
54
60
|
this.contextLength,
|
|
55
61
|
this.prefillStepSize,
|
|
56
62
|
this.samplingSeed,
|
|
63
|
+
this.asyncBatchSize,
|
|
57
64
|
);
|
|
58
65
|
}
|
|
59
66
|
|
|
@@ -64,6 +71,7 @@ export class Config implements ToNapi<NapiConfig> {
|
|
|
64
71
|
contextLength,
|
|
65
72
|
this.prefillStepSize,
|
|
66
73
|
this.samplingSeed,
|
|
74
|
+
this.asyncBatchSize,
|
|
67
75
|
);
|
|
68
76
|
}
|
|
69
77
|
|
|
@@ -74,6 +82,7 @@ export class Config implements ToNapi<NapiConfig> {
|
|
|
74
82
|
this.contextLength,
|
|
75
83
|
prefillStepSize,
|
|
76
84
|
this.samplingSeed,
|
|
85
|
+
this.asyncBatchSize,
|
|
77
86
|
);
|
|
78
87
|
}
|
|
79
88
|
|
|
@@ -84,6 +93,18 @@ export class Config implements ToNapi<NapiConfig> {
|
|
|
84
93
|
this.contextLength,
|
|
85
94
|
this.prefillStepSize,
|
|
86
95
|
samplingSeed,
|
|
96
|
+
this.asyncBatchSize,
|
|
97
|
+
);
|
|
98
|
+
}
|
|
99
|
+
|
|
100
|
+
withAsyncBatchSize(asyncBatchSize: AsyncBatchSize): Config {
|
|
101
|
+
return new Config(
|
|
102
|
+
this.preset,
|
|
103
|
+
this.contextMode,
|
|
104
|
+
this.contextLength,
|
|
105
|
+
this.prefillStepSize,
|
|
106
|
+
this.samplingSeed,
|
|
107
|
+
asyncBatchSize,
|
|
87
108
|
);
|
|
88
109
|
}
|
|
89
110
|
|
|
@@ -94,6 +115,7 @@ export class Config implements ToNapi<NapiConfig> {
|
|
|
94
115
|
contextLength: this.contextLength.toNapi(),
|
|
95
116
|
prefillStepSize: this.prefillStepSize.toNapi(),
|
|
96
117
|
samplingSeed: this.samplingSeed.toNapi(),
|
|
118
|
+
asyncBatchSize: this.asyncBatchSize.toNapi(),
|
|
97
119
|
};
|
|
98
120
|
return napiConfig;
|
|
99
121
|
}
|
package/src/index.ts
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
import { AsyncBatchSize } from '../bridging/asyncBatchSize';
|
|
1
2
|
import { ChatModel } from '../bridging/chatModel';
|
|
2
3
|
import { Config } from '../bridging/config';
|
|
3
4
|
import { ContextLength } from '../bridging/contextLength';
|
|
@@ -90,6 +91,11 @@ export class ChatModelInteractor implements Interactor<ChatModel> {
|
|
|
90
91
|
return new ChatModelInteractor(this.modelsInteractor, this.entity, config);
|
|
91
92
|
}
|
|
92
93
|
|
|
94
|
+
asyncBatchSize(asyncBatchSize: AsyncBatchSize): ChatModelInteractor {
|
|
95
|
+
const config = this.config.withAsyncBatchSize(asyncBatchSize);
|
|
96
|
+
return new ChatModelInteractor(this.modelsInteractor, this.entity, config);
|
|
97
|
+
}
|
|
98
|
+
|
|
93
99
|
/* Session */
|
|
94
100
|
|
|
95
101
|
session(): ChatSessionInteractor {
|
package/src/napi/uzu.d.ts
CHANGED
|
@@ -39,6 +39,10 @@ export declare class ProgressUpdate {
|
|
|
39
39
|
progress: number
|
|
40
40
|
}
|
|
41
41
|
|
|
42
|
+
export type AsyncBatchSize =
|
|
43
|
+
| { type: 'Default' }
|
|
44
|
+
| { type: 'Custom', size: number }
|
|
45
|
+
|
|
42
46
|
export interface ChatModel {
|
|
43
47
|
readonly repoId: string
|
|
44
48
|
readonly type: ModelType
|
|
@@ -59,6 +63,7 @@ export interface Config {
|
|
|
59
63
|
contextLength: ContextLength
|
|
60
64
|
prefillStepSize: PrefillStepSize
|
|
61
65
|
samplingSeed: SamplingSeed
|
|
66
|
+
asyncBatchSize: AsyncBatchSize
|
|
62
67
|
}
|
|
63
68
|
|
|
64
69
|
export type ContextLength =
|
package/src/napi/uzu.js
CHANGED
|
@@ -44,6 +44,7 @@ const {
|
|
|
44
44
|
StepStats,
|
|
45
45
|
Text,
|
|
46
46
|
TotalStats,
|
|
47
|
+
AsyncBatchSize,
|
|
47
48
|
ContextLength,
|
|
48
49
|
ContextMode,
|
|
49
50
|
Input,
|
|
@@ -81,6 +82,7 @@ module.exports = {
|
|
|
81
82
|
StepStats,
|
|
82
83
|
Text,
|
|
83
84
|
TotalStats,
|
|
85
|
+
AsyncBatchSize,
|
|
84
86
|
ContextLength,
|
|
85
87
|
ContextMode,
|
|
86
88
|
Input,
|
package/src/napi/uzu.mjs
CHANGED
package/src/napi/uzu.node
CHANGED
|
Binary file
|
package/src/version.ts
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
export const VERSION = '0.1.
|
|
1
|
+
export const VERSION = '0.1.50';
|
package/version.d.mts
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
export declare const VERSION = "0.1.
|
|
1
|
+
export declare const VERSION = "0.1.50";
|
|
2
2
|
//# sourceMappingURL=version.d.mts.map
|
package/version.d.ts
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
export declare const VERSION = "0.1.
|
|
1
|
+
export declare const VERSION = "0.1.50";
|
|
2
2
|
//# sourceMappingURL=version.d.ts.map
|
package/version.js
CHANGED
package/version.mjs
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
export const VERSION = '0.1.
|
|
1
|
+
export const VERSION = '0.1.50';
|
|
2
2
|
//# sourceMappingURL=version.mjs.map
|