@litert-lm/core 0.13.1 → 0.14.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 +2 -2
- package/dist/conversation.d.ts +6 -3
- package/dist/conversation.d.ts.map +1 -1
- package/dist/conversation.js +52 -53
- package/dist/conversation.js.map +1 -1
- package/dist/conversation_config.d.ts +68 -22
- package/dist/conversation_config.d.ts.map +1 -1
- package/dist/engine.js +5 -2
- package/dist/engine.js.map +1 -1
- package/dist/engine_settings.d.ts +2 -1
- package/dist/engine_settings.d.ts.map +1 -1
- package/dist/engine_settings.js +4 -0
- package/dist/engine_settings.js.map +1 -1
- package/dist/index.d.ts +4 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +4 -0
- package/dist/index.js.map +1 -1
- package/dist/litertlm_web.js +1 -1
- package/dist/load.d.ts.map +1 -1
- package/dist/load.js +9 -2
- package/dist/load.js.map +1 -1
- package/dist/load_litertlm.d.ts +1 -1
- package/dist/load_litertlm.d.ts.map +1 -1
- package/dist/load_litertlm.js +14 -2
- package/dist/load_litertlm.js.map +1 -1
- package/dist/orchestration/auto_tool_chat.d.ts +82 -0
- package/dist/orchestration/auto_tool_chat.d.ts.map +1 -0
- package/dist/orchestration/auto_tool_chat.js +304 -0
- package/dist/orchestration/auto_tool_chat.js.map +1 -0
- package/dist/orchestration/chat_interface.d.ts +30 -0
- package/dist/orchestration/chat_interface.d.ts.map +1 -0
- package/dist/orchestration/chat_interface.js +17 -0
- package/dist/orchestration/chat_interface.js.map +1 -0
- package/dist/orchestration/webmcp_tool.d.ts +41 -0
- package/dist/orchestration/webmcp_tool.d.ts.map +1 -0
- package/dist/orchestration/webmcp_tool.js +32 -0
- package/dist/orchestration/webmcp_tool.js.map +1 -0
- package/dist/testing/fakes.d.ts +65 -0
- package/dist/testing/fakes.d.ts.map +1 -0
- package/dist/testing/fakes.js +115 -0
- package/dist/testing/fakes.js.map +1 -0
- package/dist/testing/index.d.ts +17 -0
- package/dist/testing/index.d.ts.map +1 -0
- package/dist/testing/index.js +17 -0
- package/dist/testing/index.js.map +1 -0
- package/dist/wasm_binding_types.d.ts +13 -2
- package/dist/wasm_binding_types.d.ts.map +1 -1
- package/dist/wasm_binding_types.js.map +1 -1
- package/dist/wasm_feature_detect.d.ts +3 -0
- package/dist/wasm_feature_detect.d.ts.map +1 -1
- package/dist/wasm_feature_detect.js +22 -0
- package/dist/wasm_feature_detect.js.map +1 -1
- package/package.json +15 -1
- package/wasm/litertlm_wasm_asyncify_internal.js +8615 -0
- package/wasm/litertlm_wasm_asyncify_internal.wasm +0 -0
- package/wasm/litertlm_wasm_compat_asyncify_internal.js +8624 -0
- package/wasm/litertlm_wasm_compat_asyncify_internal.wasm +0 -0
- package/wasm/litertlm_wasm_compat_internal.js +13 -4
- package/wasm/litertlm_wasm_compat_internal.wasm +0 -0
- package/wasm/litertlm_wasm_internal.js +13 -4
- package/wasm/litertlm_wasm_internal.wasm +0 -0
package/README.md
CHANGED
|
@@ -3,8 +3,8 @@
|
|
|
3
3
|
The JavaScript API of LiteRT-LM for running LLMs in the browser. This is an
|
|
4
4
|
early preview that supports text-in / text-out running in WebGPU.
|
|
5
5
|
[LiteRT-LM](https://www.google.com/url?sa=D&q=https%3A%2F%2Fai.google.dev%2Fedge%2Flitert-lm%2F)
|
|
6
|
-
is
|
|
7
|
-
high-performance
|
|
6
|
+
is the **production-ready** orchestration layer to run LLMs with LiteRT,
|
|
7
|
+
engineered for **high-performance**, **cross-platform** execution.
|
|
8
8
|
|
|
9
9
|
- **Documentation**: [https://ai.google.dev/edge/litert-lm/js](https://ai.google.dev/edge/litert-lm/js)
|
|
10
10
|
- **GitHub**: [https://github.com/google-ai-edge/LiteRT-LM](https://github.com/google-ai-edge/LiteRT-LM)
|
package/dist/conversation.d.ts
CHANGED
|
@@ -15,11 +15,12 @@
|
|
|
15
15
|
*/
|
|
16
16
|
import { Message, MessageLike } from './conversation_config.js';
|
|
17
17
|
import { Mutex } from './mutex.js';
|
|
18
|
-
import {
|
|
18
|
+
import { ChatInterface } from './orchestration/chat_interface.js';
|
|
19
|
+
import { BenchmarkInfo, Conversation as WasmConversation, Engine as WasmEngine } from './wasm_binding_types.js';
|
|
19
20
|
/**
|
|
20
21
|
* LiteRT-LM Conversation
|
|
21
22
|
*/
|
|
22
|
-
export declare class Conversation {
|
|
23
|
+
export declare class Conversation implements ChatInterface {
|
|
23
24
|
private readonly conversation;
|
|
24
25
|
private readonly engine;
|
|
25
26
|
private readonly mutexes;
|
|
@@ -37,7 +38,9 @@ export declare class Conversation {
|
|
|
37
38
|
* Sends a signal to cancel any current generation.
|
|
38
39
|
*/
|
|
39
40
|
cancel(): void;
|
|
40
|
-
getHistory(): Message[]
|
|
41
|
+
getHistory(): Promise<Message[]>;
|
|
42
|
+
getTokenCount(): Promise<number>;
|
|
43
|
+
getBenchmarkInfo(): Promise<BenchmarkInfo>;
|
|
41
44
|
delete(): Promise<void>;
|
|
42
45
|
}
|
|
43
46
|
//# sourceMappingURL=conversation.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"conversation.d.ts","sourceRoot":"","sources":["../src/conversation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAC,OAAO,EAAE,WAAW,EAAC,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAC,KAAK,EAAC,MAAM,YAAY,CAAC;AACjC,OAAO,EAAC,YAAY,IAAI,gBAAgB,EAAE,MAAM,IAAI,UAAU,EAAC,MAAM,yBAAyB,CAAC;
|
|
1
|
+
{"version":3,"file":"conversation.d.ts","sourceRoot":"","sources":["../src/conversation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAC,OAAO,EAAE,WAAW,EAAC,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAC,KAAK,EAAC,MAAM,YAAY,CAAC;AACjC,OAAO,EAAC,aAAa,EAAC,MAAM,mCAAmC,CAAC;AAChE,OAAO,EAAC,aAAa,EAAE,YAAY,IAAI,gBAAgB,EAAE,MAAM,IAAI,UAAU,EAAC,MAAM,yBAAyB,CAAC;AAK9G;;GAEG;AACH,qBAAa,YAAa,YAAW,aAAa;IAI5C,OAAO,CAAC,QAAQ,CAAC,YAAY;IAC7B,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,OAAO;IAL5B,OAAO,CAAC,MAAM,CAAS;gBAGF,YAAY,EAAE,gBAAgB,EAC9B,MAAM,EAAE,UAAU,EAClB,OAAO,EAAE;QAAC,QAAQ,EAAE,KAAK,CAAA;KAAC;IAEzC,WAAW,CAAC,OAAO,EAAE,WAAW,GAAC,WAAW,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;IAiBvE;;;OAGG;IACH,oBAAoB,CAAC,OAAO,EAAE,WAAW,GAAC,WAAW,EAAE,GACnD,cAAc,CAAC,OAAO,CAAC;IA6D3B;;OAEG;IACH,MAAM;IAIA,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAOhC,aAAa,IAAI,OAAO,CAAC,MAAM,CAAC;IAMhC,gBAAgB,IAAI,OAAO,CAAC,aAAa,CAAC;IAM1C,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC;CAK9B"}
|
package/dist/conversation.js
CHANGED
|
@@ -34,8 +34,8 @@ export class Conversation {
|
|
|
34
34
|
this.isBusy = true;
|
|
35
35
|
try {
|
|
36
36
|
return await this.mutexes.executor.acquireAndRun(async () => {
|
|
37
|
-
const
|
|
38
|
-
const resultStr = await this.conversation.sendMessage(
|
|
37
|
+
const currentMessageJson = messageToJsonString(message);
|
|
38
|
+
const resultStr = await this.conversation.sendMessage(currentMessageJson);
|
|
39
39
|
return JSON.parse(resultStr);
|
|
40
40
|
});
|
|
41
41
|
}
|
|
@@ -53,60 +53,50 @@ export class Conversation {
|
|
|
53
53
|
}
|
|
54
54
|
this.isBusy = true;
|
|
55
55
|
let isCancelled = false;
|
|
56
|
-
|
|
56
|
+
let hasErrored = false;
|
|
57
57
|
return new ReadableStream({
|
|
58
58
|
start: (controller) => {
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
return;
|
|
68
|
-
}
|
|
69
|
-
if (chunk) {
|
|
70
|
-
try {
|
|
71
|
-
controller.enqueue(JSON.parse(chunk));
|
|
72
|
-
}
|
|
73
|
-
catch (e) {
|
|
59
|
+
const currentMessageJson = messageToJsonString(message);
|
|
60
|
+
const executeGeneration = async () => {
|
|
61
|
+
await this.mutexes.executor.acquireAndRun(async () => {
|
|
62
|
+
await this.conversation.sendMessageAsync(currentMessageJson, (chunk, isDone, error) => {
|
|
63
|
+
if (isCancelled || hasErrored)
|
|
64
|
+
return;
|
|
65
|
+
if (error) {
|
|
66
|
+
hasErrored = true;
|
|
74
67
|
this.isBusy = false;
|
|
75
|
-
controller.error(
|
|
68
|
+
controller.error(new Error(error));
|
|
69
|
+
return;
|
|
70
|
+
}
|
|
71
|
+
if (chunk) {
|
|
72
|
+
try {
|
|
73
|
+
const msg = JSON.parse(chunk);
|
|
74
|
+
controller.enqueue(msg);
|
|
75
|
+
}
|
|
76
|
+
catch (e) {
|
|
77
|
+
hasErrored = true;
|
|
78
|
+
this.isBusy = false;
|
|
79
|
+
controller.error(e);
|
|
80
|
+
}
|
|
76
81
|
}
|
|
77
|
-
}
|
|
78
|
-
if (isDone) {
|
|
79
|
-
this.isBusy = false;
|
|
80
|
-
controller.close();
|
|
81
|
-
}
|
|
82
|
-
})
|
|
83
|
-
.catch((e) => {
|
|
84
|
-
if (isCancelled)
|
|
85
|
-
return;
|
|
86
|
-
this.isBusy = false;
|
|
87
|
-
controller.error(e);
|
|
88
|
-
});
|
|
89
|
-
// Since we're using the synchronus execution manager in C++, which
|
|
90
|
-
// lazily executes tasks, we must call waitUntilDone to start the
|
|
91
|
-
// execution.
|
|
92
|
-
const runWait = async () => {
|
|
93
|
-
await this.mutexes.executor.acquireAndRun(async () => {
|
|
94
|
-
await this.engine.waitUntilDone();
|
|
95
82
|
});
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
this.
|
|
101
|
-
controller.error(e);
|
|
83
|
+
// Since we're using the synchronus execution manager in C++, which
|
|
84
|
+
// lazily executes tasks, we must queue the message and call
|
|
85
|
+
// waitUntilDone to start the execution concurrently while holding
|
|
86
|
+
// the mutex lock.
|
|
87
|
+
await this.engine.waitUntilDone();
|
|
102
88
|
});
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
89
|
+
if (isCancelled || hasErrored)
|
|
90
|
+
return;
|
|
91
|
+
this.isBusy = false;
|
|
92
|
+
controller.close();
|
|
93
|
+
};
|
|
94
|
+
executeGeneration().catch((e) => {
|
|
95
|
+
if (isCancelled || hasErrored)
|
|
106
96
|
return;
|
|
107
97
|
this.isBusy = false;
|
|
108
98
|
controller.error(e);
|
|
109
|
-
}
|
|
99
|
+
});
|
|
110
100
|
},
|
|
111
101
|
cancel: () => {
|
|
112
102
|
isCancelled = true;
|
|
@@ -121,12 +111,21 @@ export class Conversation {
|
|
|
121
111
|
cancel() {
|
|
122
112
|
this.conversation.cancelProcess();
|
|
123
113
|
}
|
|
124
|
-
getHistory() {
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
114
|
+
async getHistory() {
|
|
115
|
+
return this.mutexes.executor.acquireAndRun(() => {
|
|
116
|
+
const historyStr = this.conversation.getHistory();
|
|
117
|
+
return JSON.parse(historyStr);
|
|
118
|
+
});
|
|
119
|
+
}
|
|
120
|
+
async getTokenCount() {
|
|
121
|
+
return this.mutexes.executor.acquireAndRun(() => {
|
|
122
|
+
return this.conversation.getTokenCount();
|
|
123
|
+
});
|
|
124
|
+
}
|
|
125
|
+
async getBenchmarkInfo() {
|
|
126
|
+
return this.mutexes.executor.acquireAndRun(() => {
|
|
127
|
+
return this.conversation.getBenchmarkInfo();
|
|
128
|
+
});
|
|
130
129
|
}
|
|
131
130
|
async delete() {
|
|
132
131
|
await this.mutexes.executor.acquireAndRun(() => {
|
package/dist/conversation.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"conversation.js","sourceRoot":"","sources":["../src/conversation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;
|
|
1
|
+
{"version":3,"file":"conversation.js","sourceRoot":"","sources":["../src/conversation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAOH,MAAM,YAAY,GACd,4DAA4D,CAAC;AAEjE;;GAEG;AACH,MAAM,OAAO,YAAY;IAIF;IACA;IACA;IALb,MAAM,GAAG,KAAK,CAAC;IAEvB,YACqB,YAA8B,EAC9B,MAAkB,EAClB,OAA0B;QAF1B,iBAAY,GAAZ,YAAY,CAAkB;QAC9B,WAAM,GAAN,MAAM,CAAY;QAClB,YAAO,GAAP,OAAO,CAAmB;IAAG,CAAC;IAEnD,KAAK,CAAC,WAAW,CAAC,OAAkC;QAClD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,IAAI,EAAE;gBAC1D,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;gBACxD,MAAM,SAAS,GACX,MAAM,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;gBAC5D,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAY,CAAC;YAC1C,CAAC,CAAC,CAAC;QACL,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,oBAAoB,CAAC,OAAkC;QAErD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QAEnB,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,OAAO,IAAI,cAAc,CAAU;YACjC,KAAK,EAAE,CAAC,UAAU,EAAE,EAAE;gBACpB,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;gBAExD,MAAM,iBAAiB,GAAG,KAAK,IAAkB,EAAE;oBACjD,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,IAAI,EAAE;wBACnD,MAAM,IAAI,CAAC,YAAY,CAAC,gBAAgB,CACpC,kBAAkB,EAClB,CAAC,KAAkB,EAAE,MAAe,EAAE,KAAkB,EAAE,EAAE;4BAC1D,IAAI,WAAW,IAAI,UAAU;gCAAE,OAAO;4BACtC,IAAI,KAAK,EAAE,CAAC;gCACV,UAAU,GAAG,IAAI,CAAC;gCAClB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;gCACpB,UAAU,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;gCACnC,OAAO;4BACT,CAAC;4BACD,IAAI,KAAK,EAAE,CAAC;gCACV,IAAI,CAAC;oCACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAY,CAAC;oCACzC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;gCAC1B,CAAC;gCAAC,OAAO,CAAC,EAAE,CAAC;oCACX,UAAU,GAAG,IAAI,CAAC;oCAClB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;oCACpB,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gCACtB,CAAC;4BACH,CAAC;wBACH,CAAC,CAAC,CAAC;wBACP,mEAAmE;wBACnE,4DAA4D;wBAC5D,kEAAkE;wBAClE,kBAAkB;wBAClB,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;oBACpC,CAAC,CAAC,CAAC;oBAEH,IAAI,WAAW,IAAI,UAAU;wBAAE,OAAO;oBACtC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;oBACpB,UAAU,CAAC,KAAK,EAAE,CAAC;gBACrB,CAAC,CAAC;gBAEF,iBAAiB,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC9B,IAAI,WAAW,IAAI,UAAU;wBAAE,OAAO;oBACtC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;oBACpB,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACtB,CAAC,CAAC,CAAC;YACL,CAAC;YACD,MAAM,EAAE,GAAG,EAAE;gBACX,WAAW,GAAG,IAAI,CAAC;gBACnB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;gBACpB,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE;YAC9C,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;YAClD,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAAc,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE;YAC9C,OAAO,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB;QACpB,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE;YAC9C,OAAO,IAAI,CAAC,YAAY,CAAC,gBAAgB,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM;QACV,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE;YAC7C,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,SAAS,mBAAmB,CAAC,WAAsC;IACjE,IAAI,OAA0B,CAAC;IAC/B,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;QAC/B,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,SAAS,CAAC,WAAW,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,SAAS,CAAC,WAAwB;IACzC,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;QACpC,OAAO,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAC,CAAC;IAC9C,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC"}
|
|
@@ -16,41 +16,86 @@
|
|
|
16
16
|
import { SessionConfig } from './session_config.js';
|
|
17
17
|
import { JsonValue } from './types.js';
|
|
18
18
|
/**
|
|
19
|
-
*
|
|
19
|
+
* A parameter schema definition, adhering to a subset of JSON Schema.
|
|
20
20
|
*/
|
|
21
|
-
export
|
|
22
|
-
type?: string;
|
|
23
|
-
|
|
21
|
+
export interface Schema {
|
|
22
|
+
type?: 'object' | 'array' | 'string' | 'number' | 'integer' | 'boolean' | 'null';
|
|
23
|
+
description?: string;
|
|
24
|
+
properties?: Record<string, Schema>;
|
|
24
25
|
required?: string[];
|
|
25
|
-
|
|
26
|
+
items?: Schema;
|
|
27
|
+
enum?: string[];
|
|
26
28
|
}
|
|
27
29
|
/**
|
|
28
|
-
*
|
|
30
|
+
* Declaration of a function that the model can call.
|
|
29
31
|
*/
|
|
30
|
-
export
|
|
32
|
+
export interface FunctionDeclaration {
|
|
31
33
|
name: string;
|
|
32
34
|
description?: string;
|
|
33
|
-
parameters?:
|
|
35
|
+
parameters?: Schema;
|
|
36
|
+
response?: Schema;
|
|
37
|
+
}
|
|
38
|
+
/**
|
|
39
|
+
* A wrapper for a function tool. This is the canonical format
|
|
40
|
+
* generated by platform SDKs.
|
|
41
|
+
*/
|
|
42
|
+
export declare interface FunctionTool {
|
|
43
|
+
type: string;
|
|
44
|
+
function: FunctionDeclaration;
|
|
45
|
+
[key: string]: JsonValue | FunctionDeclaration | undefined;
|
|
46
|
+
}
|
|
47
|
+
/**
|
|
48
|
+
* A tool that can be provided to the model.
|
|
49
|
+
* The C++ runtime is flexible and accepts both wrapped and unwrapped formats.
|
|
50
|
+
*/
|
|
51
|
+
export type Tool = FunctionDeclaration | FunctionTool;
|
|
52
|
+
/**
|
|
53
|
+
* Any support content part in a message.
|
|
54
|
+
*/
|
|
55
|
+
export type ContentPart = TextContentPart | ImageContentPart | AudioContentPart | VideoContentPart | ToolResponsePart;
|
|
56
|
+
/**
|
|
57
|
+
* Text content part of a message.
|
|
58
|
+
*/
|
|
59
|
+
export interface TextContentPart {
|
|
60
|
+
type: 'text';
|
|
61
|
+
text: string;
|
|
34
62
|
[key: string]: JsonValue | undefined;
|
|
35
63
|
}
|
|
36
64
|
/**
|
|
37
|
-
*
|
|
65
|
+
* Image content part of a message.
|
|
66
|
+
*
|
|
67
|
+
* This is a placeholder. Images are not supported in JS yet.
|
|
38
68
|
*/
|
|
39
|
-
export
|
|
40
|
-
|
|
41
|
-
tool_name?: string;
|
|
42
|
-
response?: Record<string, JsonValue>;
|
|
43
|
-
value?: JsonValue;
|
|
69
|
+
export interface ImageContentPart {
|
|
70
|
+
type: 'image';
|
|
44
71
|
[key: string]: JsonValue | undefined;
|
|
45
72
|
}
|
|
46
73
|
/**
|
|
47
|
-
*
|
|
74
|
+
* Audio content part of a message.
|
|
75
|
+
*
|
|
76
|
+
* This is a placeholder. Audio is not supported in JS yet.
|
|
48
77
|
*/
|
|
49
|
-
export
|
|
50
|
-
type:
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
78
|
+
export interface AudioContentPart {
|
|
79
|
+
type: 'audio';
|
|
80
|
+
[key: string]: JsonValue | undefined;
|
|
81
|
+
}
|
|
82
|
+
/**
|
|
83
|
+
* Video content part of a message.
|
|
84
|
+
*
|
|
85
|
+
* This is a placeholder. Video is not supported in JS yet.
|
|
86
|
+
*/
|
|
87
|
+
export interface VideoContentPart {
|
|
88
|
+
type: 'video';
|
|
89
|
+
[key: string]: JsonValue | undefined;
|
|
90
|
+
}
|
|
91
|
+
/**
|
|
92
|
+
* The response from a tool execution.
|
|
93
|
+
*/
|
|
94
|
+
export declare interface ToolResponsePart {
|
|
95
|
+
type: 'tool_response';
|
|
96
|
+
name?: string;
|
|
97
|
+
response?: JsonValue;
|
|
98
|
+
value?: JsonValue;
|
|
54
99
|
[key: string]: JsonValue | undefined;
|
|
55
100
|
}
|
|
56
101
|
/**
|
|
@@ -66,6 +111,7 @@ export declare interface ToolCallFunction {
|
|
|
66
111
|
*/
|
|
67
112
|
export declare interface ToolCall {
|
|
68
113
|
type: string;
|
|
114
|
+
id?: string;
|
|
69
115
|
function: ToolCallFunction;
|
|
70
116
|
[key: string]: JsonValue | undefined;
|
|
71
117
|
}
|
|
@@ -74,7 +120,7 @@ export declare interface ToolCall {
|
|
|
74
120
|
*/
|
|
75
121
|
export declare interface Message {
|
|
76
122
|
role: string;
|
|
77
|
-
content?: string |
|
|
123
|
+
content?: string | ContentPart[];
|
|
78
124
|
channels?: Record<string, string>;
|
|
79
125
|
tool_calls?: ToolCall[];
|
|
80
126
|
[key: string]: JsonValue | undefined;
|
|
@@ -92,7 +138,7 @@ export declare interface Preface {
|
|
|
92
138
|
messages?: Message[];
|
|
93
139
|
tools?: Tool[];
|
|
94
140
|
extra_context?: Record<string, JsonValue>;
|
|
95
|
-
[key: string]:
|
|
141
|
+
[key: string]: unknown;
|
|
96
142
|
}
|
|
97
143
|
/**
|
|
98
144
|
* Configuration for the Conversation instance.
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"conversation_config.d.ts","sourceRoot":"","sources":["../src/conversation_config.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAC,aAAa,EAAC,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAC,SAAS,EAAC,MAAM,YAAY,CAAC;AAErC;;GAEG;AACH,MAAM,CAAC,OAAO,
|
|
1
|
+
{"version":3,"file":"conversation_config.d.ts","sourceRoot":"","sources":["../src/conversation_config.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAC,aAAa,EAAC,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAC,SAAS,EAAC,MAAM,YAAY,CAAC;AAErC;;GAEG;AACH,MAAM,WAAW,MAAM;IACrB,IAAI,CAAC,EAAE,QAAQ,GAAC,OAAO,GAAC,QAAQ,GAAC,QAAQ,GAAC,SAAS,GAAC,SAAS,GAAC,MAAM,CAAC;IACrE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,MAAM,CAAC,OAAO,WAAW,YAAY;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,mBAAmB,CAAC;IAC9B,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAC,mBAAmB,GAAC,SAAS,CAAC;CACxD;AAED;;;GAGG;AACH,MAAM,MAAM,IAAI,GAAG,mBAAmB,GAAC,YAAY,CAAC;AAEpD;;GAEG;AACH,MAAM,MAAM,WAAW,GAAG,eAAe,GAAC,gBAAgB,GAAC,gBAAgB,GACvE,gBAAgB,GAAC,gBAAgB,CAAC;AAEtC;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAC,SAAS,CAAC;CACpC;AAED;;;;GAIG;AACH,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,OAAO,CAAC;IACd,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAC,SAAS,CAAC;CACpC;AAED;;;;GAIG;AACH,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,OAAO,CAAC;IACd,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAC,SAAS,CAAC;CACpC;AAED;;;;GAIG;AACH,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,OAAO,CAAC;IACd,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAC,SAAS,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,CAAC,OAAO,WAAW,gBAAgB;IACvC,IAAI,EAAE,eAAe,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,SAAS,CAAC;IACrB,KAAK,CAAC,EAAE,SAAS,CAAC;IAClB,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAC,SAAS,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,CAAC,OAAO,WAAW,gBAAgB;IACvC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACrC,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAC,SAAS,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,CAAC,OAAO,WAAW,QAAQ;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAC,SAAS,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,CAAC,OAAO,WAAW,OAAO;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,GAAC,WAAW,EAAE,CAAC;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClC,UAAU,CAAC,EAAE,QAAQ,EAAE,CAAC;IACxB,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAC,SAAS,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,MAAM,WAAW,GAAG,MAAM,GAAC,OAAO,CAAC;AAEzC;;;;GAIG;AACH,MAAM,CAAC,OAAO,WAAW,OAAO;IAI9B,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;IAErB,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IAIf,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC1C,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,CAAC,OAAO,WAAW,kBAAkB;IACzC,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,yBAAyB,CAAC,EAAE,OAAO,CAAC;IACpC,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,+BAA+B,CAAC,EAAE,OAAO,CAAC;CAE3C"}
|
package/dist/engine.js
CHANGED
|
@@ -130,7 +130,7 @@ export class Engine {
|
|
|
130
130
|
});
|
|
131
131
|
}
|
|
132
132
|
async createConversation(config) {
|
|
133
|
-
return this.mutexes.executor.acquireAndRun(() => {
|
|
133
|
+
return this.mutexes.executor.acquireAndRun(async () => {
|
|
134
134
|
let wasmConfig;
|
|
135
135
|
let wasmSessionConfig;
|
|
136
136
|
if (config) {
|
|
@@ -141,7 +141,7 @@ export class Engine {
|
|
|
141
141
|
else {
|
|
142
142
|
wasmConfig = this.wasm.ConversationConfig.createDefault(this.engine);
|
|
143
143
|
}
|
|
144
|
-
const wasmConversation = this.wasm.Conversation.create(this.engine, wasmConfig);
|
|
144
|
+
const wasmConversation = await this.wasm.Conversation.create(this.engine, wasmConfig);
|
|
145
145
|
wasmConfig.delete();
|
|
146
146
|
if (wasmSessionConfig) {
|
|
147
147
|
wasmSessionConfig.delete();
|
|
@@ -159,6 +159,9 @@ async function modelToStream(model) {
|
|
|
159
159
|
if (model instanceof ReadableStream) {
|
|
160
160
|
return model;
|
|
161
161
|
}
|
|
162
|
+
if (model instanceof Blob) {
|
|
163
|
+
return model.stream();
|
|
164
|
+
}
|
|
162
165
|
const modelUrl = model;
|
|
163
166
|
const response = await fetch(modelUrl, {
|
|
164
167
|
credentials: 'same-origin',
|
package/dist/engine.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"engine.js","sourceRoot":"","sources":["../src/engine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAC,OAAO,EAAC,MAAM,cAAc,CAAC;AACrC,OAAO,EAAC,YAAY,EAAC,MAAM,mBAAmB,CAAC;AAE/C,OAAO,EAAiB,wCAAwC,EAAE,kCAAkC,EAAC,MAAM,sBAAsB,CAAC;AAElI,OAAO,EAAC,uBAAuB,EAAC,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAC,KAAK,EAAC,MAAM,YAAY,CAAC;AACjC,OAAO,EAAC,+BAA+B,EAAC,MAAM,0CAA0C,CAAC;AACzF,OAAO,EAAC,OAAO,EAAC,MAAM,cAAc,CAAC;AACrC,OAAO,EAAC,gCAAgC,EAAC,MAAM,qBAAqB,CAAC;AAErE,OAAO,EAAC,OAAO,EAA+I,MAAM,yBAAyB,CAAC;AAE9L;;GAEG;AACH,gEAAgE;AAChE,IAAI,cAAc,GAAG,CAAC,CAAC;AAEvB;;GAEG;AACH,MAAM,OAAO,MAAM;IAOI;IACA;IAEA;IATZ,QAAQ,CAAoC;IAC7C,OAAO,GAAG;QAChB,QAAQ,EAAE,IAAI,KAAK,EAAE;KACtB,CAAC;IAEF,YACqB,IAAkB,EAClB,MAAkB,EACnC,WAAoC,EACnB,cAA0B;QAH1B,SAAI,GAAJ,IAAI,CAAc;QAClB,WAAM,GAAN,MAAM,CAAY;QAElB,mBAAc,GAAd,cAAc,CAAY;QAE7C,MAAM,kBAAkB,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC;QACtD,MAAM,oBAAoB,GACtB,kCAAkC,CAAC,kBAAkB,CAAC,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG;YACd,GAAG,oBAAoB;YACvB,KAAK,EAAE,WAAW;SACnB,CAAC;QACF,kBAAkB,CAAC,MAAM,EAAE,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,cAA8B,EAAE,iBAAiB,GAAG,EAAE;QAExE,MAAM,QAAQ,GAAG,MAAM,uBAAuB,EAAE,CAAC;QACjD,MAAM,IAAI,GAAG,QAAQ,CAAC,YAAY,CAAC;QACnC,wDAAwD;QACxD,MAAM,OAAO,GAAG,UAAU,cAAc,EAAE,WAAW,CAAC;QAEtD,2CAA2C;QAC3C,MAAM,OAAO,GAAG,cAAc,CAAC,OAAO,IAAI,OAAO,CAAC,WAAW,CAAC;QAC9D,cAAc,GAAG,EAAC,GAAG,cAAc,EAAE,OAAO,EAAC,CAAC;QAE9C,MAAM,cAAc,GAAG,cAAc,CAAC,oBAAoB,EAAE,cAAc,CAAC;QAC3E,IAAI,OAAO,KAAK,OAAO,CAAC,GAAG,IAAI,OAAO,KAAK,OAAO,CAAC,WAAW;YAC1D,cAAc,KAAK,OAAO,CAAC,GAAG;YAC9B,cAAc,KAAK,OAAO,CAAC,WAAW,EAAE,CAAC;YAC3C,MAAM,QAAQ,CAAC,wBAAwB,EAAE,CAAC;QAC5C,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;QAE9B,MAAM,WAAW,GAAG,MAAM,aAAa,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAC9D,IAAI,MAAkB,CAAC;QACvB,IAAI,CAAC;YACH,IAAI,WAAwB,CAAC;YAC7B,MAAM,WAAW,GAAG,OAAO,KAAK,OAAO,CAAC,WAAW,CAAC;YAEpD,IAAI,WAAW,EAAE,CAAC;gBAChB,yCAAyC;gBACzC,MAAM,aAAa,GACf,IAAI,+BAA+B,CAAC,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACxE,MAAM,UAAU,GAAG,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;gBACvE,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;oBACf,UAAU,CAAC,MAAM,EAAE,CAAC;gBACtB,CAAC,CAAC,CAAC;gBACH,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;YAC7D,CAAC;iBAAM,CAAC;gBACN,8DAA8D;gBAC9D,MAAM,cAAc,CAAC,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;gBACjD,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;oBACf,IAAI,CAAC;wBACH,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;oBAC1B,CAAC;oBAAC,OAAO,CAAC,EAAE,CAAC;wBACX,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;oBACpD,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;gBAC1C,WAAW,CAAC,MAAM,EAAE,CAAC;YACvB,CAAC,CAAC,CAAC;YAEH,MAAM,kBAAkB,GACpB,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,WAAW,EAAE,EAAC,KAAK,EAAE,OAAO,EAAC,CAAC,CAAC;YACrE,uEAAuE;YACvE,kBAAkB,EAAE,CAAC;YAErB,MAAM,yBAAyB,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;gBACjD,kBAAkB,CAAC,MAAM,EAAE,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,wCAAwC,CACpC,kBAAkB,EAAE,cAAc,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YACvD,kBAAkB,CAAC,6BAA6B,CAAC,KAAK,CAAC,CAAC;YACxD,kBAAkB,CAAC,0BAA0B,CAAC,IAAI,CAAC,CAAC;YAEpD,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CACtC,kBAAkB,EAAE,iBAAiB,CAAC,CAAC;YAC7C,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CACnC,kBAAkB,EAAE,iBAAiB,CAAC,CAAC;YAC7C,CAAC;YACD,yBAAyB,EAAE,CAAC;YAE5B,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;gBACf,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,CAAC,CAAC,CAAC;YACH,OAAO,IAAI,MAAM,CACb,IAAI,EAAE,MAAM,EAAE,cAAc,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,GAAG,EAAE,CAAC;YACd,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,aAAa,GAAG,EAAE;QACpC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE;YAC9C,MAAM,iBAAiB,GACnB,gCAAgC,CAAC,aAAa,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAE/D,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,iBAAiB,CAAC,CAAC;YACjE,iBAAiB,CAAC,MAAM,EAAE,CAAC;YAC3B,OAAO,IAAI,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,MAA2B;QAClD,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,
|
|
1
|
+
{"version":3,"file":"engine.js","sourceRoot":"","sources":["../src/engine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAC,OAAO,EAAC,MAAM,cAAc,CAAC;AACrC,OAAO,EAAC,YAAY,EAAC,MAAM,mBAAmB,CAAC;AAE/C,OAAO,EAAiB,wCAAwC,EAAE,kCAAkC,EAAC,MAAM,sBAAsB,CAAC;AAElI,OAAO,EAAC,uBAAuB,EAAC,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAC,KAAK,EAAC,MAAM,YAAY,CAAC;AACjC,OAAO,EAAC,+BAA+B,EAAC,MAAM,0CAA0C,CAAC;AACzF,OAAO,EAAC,OAAO,EAAC,MAAM,cAAc,CAAC;AACrC,OAAO,EAAC,gCAAgC,EAAC,MAAM,qBAAqB,CAAC;AAErE,OAAO,EAAC,OAAO,EAA+I,MAAM,yBAAyB,CAAC;AAE9L;;GAEG;AACH,gEAAgE;AAChE,IAAI,cAAc,GAAG,CAAC,CAAC;AAEvB;;GAEG;AACH,MAAM,OAAO,MAAM;IAOI;IACA;IAEA;IATZ,QAAQ,CAAoC;IAC7C,OAAO,GAAG;QAChB,QAAQ,EAAE,IAAI,KAAK,EAAE;KACtB,CAAC;IAEF,YACqB,IAAkB,EAClB,MAAkB,EACnC,WAAoC,EACnB,cAA0B;QAH1B,SAAI,GAAJ,IAAI,CAAc;QAClB,WAAM,GAAN,MAAM,CAAY;QAElB,mBAAc,GAAd,cAAc,CAAY;QAE7C,MAAM,kBAAkB,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC;QACtD,MAAM,oBAAoB,GACtB,kCAAkC,CAAC,kBAAkB,CAAC,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG;YACd,GAAG,oBAAoB;YACvB,KAAK,EAAE,WAAW;SACnB,CAAC;QACF,kBAAkB,CAAC,MAAM,EAAE,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,cAA8B,EAAE,iBAAiB,GAAG,EAAE;QAExE,MAAM,QAAQ,GAAG,MAAM,uBAAuB,EAAE,CAAC;QACjD,MAAM,IAAI,GAAG,QAAQ,CAAC,YAAY,CAAC;QACnC,wDAAwD;QACxD,MAAM,OAAO,GAAG,UAAU,cAAc,EAAE,WAAW,CAAC;QAEtD,2CAA2C;QAC3C,MAAM,OAAO,GAAG,cAAc,CAAC,OAAO,IAAI,OAAO,CAAC,WAAW,CAAC;QAC9D,cAAc,GAAG,EAAC,GAAG,cAAc,EAAE,OAAO,EAAC,CAAC;QAE9C,MAAM,cAAc,GAAG,cAAc,CAAC,oBAAoB,EAAE,cAAc,CAAC;QAC3E,IAAI,OAAO,KAAK,OAAO,CAAC,GAAG,IAAI,OAAO,KAAK,OAAO,CAAC,WAAW;YAC1D,cAAc,KAAK,OAAO,CAAC,GAAG;YAC9B,cAAc,KAAK,OAAO,CAAC,WAAW,EAAE,CAAC;YAC3C,MAAM,QAAQ,CAAC,wBAAwB,EAAE,CAAC;QAC5C,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;QAE9B,MAAM,WAAW,GAAG,MAAM,aAAa,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAC9D,IAAI,MAAkB,CAAC;QACvB,IAAI,CAAC;YACH,IAAI,WAAwB,CAAC;YAC7B,MAAM,WAAW,GAAG,OAAO,KAAK,OAAO,CAAC,WAAW,CAAC;YAEpD,IAAI,WAAW,EAAE,CAAC;gBAChB,yCAAyC;gBACzC,MAAM,aAAa,GACf,IAAI,+BAA+B,CAAC,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACxE,MAAM,UAAU,GAAG,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;gBACvE,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;oBACf,UAAU,CAAC,MAAM,EAAE,CAAC;gBACtB,CAAC,CAAC,CAAC;gBACH,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;YAC7D,CAAC;iBAAM,CAAC;gBACN,8DAA8D;gBAC9D,MAAM,cAAc,CAAC,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;gBACjD,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;oBACf,IAAI,CAAC;wBACH,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;oBAC1B,CAAC;oBAAC,OAAO,CAAC,EAAE,CAAC;wBACX,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;oBACpD,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;gBAC1C,WAAW,CAAC,MAAM,EAAE,CAAC;YACvB,CAAC,CAAC,CAAC;YAEH,MAAM,kBAAkB,GACpB,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,WAAW,EAAE,EAAC,KAAK,EAAE,OAAO,EAAC,CAAC,CAAC;YACrE,uEAAuE;YACvE,kBAAkB,EAAE,CAAC;YAErB,MAAM,yBAAyB,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;gBACjD,kBAAkB,CAAC,MAAM,EAAE,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,wCAAwC,CACpC,kBAAkB,EAAE,cAAc,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YACvD,kBAAkB,CAAC,6BAA6B,CAAC,KAAK,CAAC,CAAC;YACxD,kBAAkB,CAAC,0BAA0B,CAAC,IAAI,CAAC,CAAC;YAEpD,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CACtC,kBAAkB,EAAE,iBAAiB,CAAC,CAAC;YAC7C,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CACnC,kBAAkB,EAAE,iBAAiB,CAAC,CAAC;YAC7C,CAAC;YACD,yBAAyB,EAAE,CAAC;YAE5B,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;gBACf,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,CAAC,CAAC,CAAC;YACH,OAAO,IAAI,MAAM,CACb,IAAI,EAAE,MAAM,EAAE,cAAc,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,GAAG,EAAE,CAAC;YACd,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,aAAa,GAAG,EAAE;QACpC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE;YAC9C,MAAM,iBAAiB,GACnB,gCAAgC,CAAC,aAAa,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAE/D,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,iBAAiB,CAAC,CAAC;YACjE,iBAAiB,CAAC,MAAM,EAAE,CAAC;YAC3B,OAAO,IAAI,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,MAA2B;QAClD,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,IAAI,EAAE;YACpD,IAAI,UAAkC,CAAC;YACvC,IAAI,iBAA8C,CAAC;YAEnD,IAAI,MAAM,EAAE,CAAC;gBACX,iBAAiB,GAAG,gCAAgC,CAChD,MAAM,CAAC,aAAa,IAAI,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC3C,MAAM,WAAW,GACb,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzD,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAClD,IAAI,CAAC,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC,MAAM,CAAC,yBAAyB,EAClE,CAAC,CAAC,MAAM,CAAC,oBAAoB,EAC7B,CAAC,CAAC,MAAM,CAAC,+BAA+B,EAAE,WAAW,CAAC,CAAC;YAC7D,CAAC;iBAAM,CAAC;gBACN,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvE,CAAC;YAED,MAAM,gBAAgB,GAClB,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YAEjE,UAAU,CAAC,MAAM,EAAE,CAAC;YACpB,IAAI,iBAAiB,EAAE,CAAC;gBACtB,iBAAiB,CAAC,MAAM,EAAE,CAAC;YAC7B,CAAC;YAED,OAAO,IAAI,YAAY,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM;QACV,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,GAAG,EAAE;YAC7C,IAAI,CAAC,cAAc,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,KAAK,UAAU,aAAa,CAAC,KAA8B;IAEzD,IAAI,KAAK,YAAY,cAAc,EAAE,CAAC;QACpC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,KAAK,YAAY,IAAI,EAAE,CAAC;QAC1B,OAAO,KAAK,CAAC,MAAM,EAAE,CAAC;IACxB,CAAC;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC;IACvB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE;QACrC,WAAW,EAAE,aAAa;KAC3B,CAAC,CAAC;IACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,mCAAmC,QAAQ,EAAE,CAAC,CAAC;IACjE,CAAC;IACD,OAAO,QAAQ,CAAC,IAAK,CAAC;AACxB,CAAC;AAED,gEAAgE;AAChE,KAAK,UAAU,cAAc,CACzB,MAAoB,EACpB,WAAuC,EACvC,OAAe;IAEjB,IAAI,WAAuB,CAAC;IAE5B,MAAM,MAAM,GAAG,WAAW,CAAC,SAAS,EAAE,CAAC;IACvC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,OAAO,IAAI,EAAE,CAAC;QACZ,MAAM,EAAC,IAAI,EAAE,KAAK,EAAC,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,IAAI;YAAE,MAAM;QAChB,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnB,WAAW,IAAI,KAAK,CAAC,MAAM,CAAC;QAC9B,CAAC;IACH,CAAC;IACD,WAAW,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;IAC1C,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC/B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;IACzB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAC5C,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,OAAO,CAAC,KAAK,CAAC,4BAA4B,EAAE,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,CAAC;IACV,CAAC;AACH,CAAC"}
|
|
@@ -35,9 +35,10 @@ export interface LlmExecutorSettings {
|
|
|
35
35
|
* LiteRT-LM EngineSettings
|
|
36
36
|
*/
|
|
37
37
|
export interface EngineSettings {
|
|
38
|
-
model: string | ReadableStream<Uint8Array>;
|
|
38
|
+
model: string | Blob | ReadableStream<Uint8Array>;
|
|
39
39
|
backend?: Backend;
|
|
40
40
|
mainExecutorSettings?: LlmExecutorSettings;
|
|
41
|
+
benchmarkEnabled?: boolean;
|
|
41
42
|
}
|
|
42
43
|
/**
|
|
43
44
|
* Fills a WasmEngineSettings with the values from a EngineSettings.
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"engine_settings.d.ts","sourceRoot":"","sources":["../src/engine_settings.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAC,iBAAiB,EAAC,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAC,KAAK,gBAAgB,EAAE,OAAO,EAAE,KAAK,SAAS,EAAE,cAAc,IAAI,kBAAkB,EAAE,gBAAgB,IAAI,oBAAoB,EAAE,KAAK,SAAS,EAAC,MAAM,yBAAyB,CAAC;AAIvL,OAAO,EAAC,gBAAgB,EAAE,SAAS,EAAE,SAAS,EAAC,CAAC;AAEhD;;GAEG;AACH,MAAM,WAAW,gBAAiB,SAC9B,IAAI,CAAC,oBAAoB,EAAE,sBAAsB,CAAC;IAEpD,oBAAoB,EAAE,MAAM,EAAE,CAAC;CAChC;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,SAAS,GAAC,SAAS,GAAC,gBAAgB,CAAC;IACrD,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,gBAAgB,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;CAC9C;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,GAAC,cAAc,CAAC,UAAU,CAAC,CAAC;
|
|
1
|
+
{"version":3,"file":"engine_settings.d.ts","sourceRoot":"","sources":["../src/engine_settings.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAC,iBAAiB,EAAC,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAC,KAAK,gBAAgB,EAAE,OAAO,EAAE,KAAK,SAAS,EAAE,cAAc,IAAI,kBAAkB,EAAE,gBAAgB,IAAI,oBAAoB,EAAE,KAAK,SAAS,EAAC,MAAM,yBAAyB,CAAC;AAIvL,OAAO,EAAC,gBAAgB,EAAE,SAAS,EAAE,SAAS,EAAC,CAAC;AAEhD;;GAEG;AACH,MAAM,WAAW,gBAAiB,SAC9B,IAAI,CAAC,oBAAoB,EAAE,sBAAsB,CAAC;IAEpD,oBAAoB,EAAE,MAAM,EAAE,CAAC;CAChC;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,SAAS,GAAC,SAAS,GAAC,gBAAgB,CAAC;IACrD,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,gBAAgB,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;CAC9C;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,GAAC,IAAI,GAAC,cAAc,CAAC,UAAU,CAAC,CAAC;IAC9C,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,oBAAoB,CAAC,EAAE,mBAAmB,CAAC;IAC3C,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,wBAAgB,wCAAwC,CACpD,kBAAkB,EAAE,kBAAkB,EAAE,cAAc,EAAE,cAAc,EACtE,OAAO,EAAE,OAAO,EAAE,IAAI,iDAAmC,GAAG,IAAI,CAgDnE;AAED;;;;;;;;GAQG;AACH,wBAAgB,kCAAkC,CAC9C,kBAAkB,EAAE,kBAAkB,GACtC,iBAAiB,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAwCnD"}
|
package/dist/engine_settings.js
CHANGED
|
@@ -25,6 +25,9 @@ export function fillWasmEngineSettingsFromEngineSettings(wasmEngineSettings, eng
|
|
|
25
25
|
// and that is best handled in the Engine class.
|
|
26
26
|
const wasmExecutorSettings = wasmEngineSettings.getMutableMainExecutorSettings();
|
|
27
27
|
wasmExecutorSettings.setCacheDir(':nocache'); // Not supported in JS.
|
|
28
|
+
if (engineSettings.benchmarkEnabled) {
|
|
29
|
+
wasmEngineSettings.enableBenchmark();
|
|
30
|
+
}
|
|
28
31
|
if (engineSettings.mainExecutorSettings) {
|
|
29
32
|
const mainExecutorSettings = engineSettings.mainExecutorSettings;
|
|
30
33
|
if (mainExecutorSettings.maxNumTokens !== undefined) {
|
|
@@ -105,6 +108,7 @@ export function wasmEngineSettingsToEngineSettings(wasmEngineSettings) {
|
|
|
105
108
|
backendConfig,
|
|
106
109
|
advancedSettings: wasmExecutorSettings.getAdvancedSettings(),
|
|
107
110
|
},
|
|
111
|
+
benchmarkEnabled: false,
|
|
108
112
|
};
|
|
109
113
|
}
|
|
110
114
|
//# sourceMappingURL=engine_settings.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"engine_settings.js","sourceRoot":"","sources":["../src/engine_settings.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAC,iBAAiB,EAAC,MAAM,sBAAsB,CAAC;AAEvD,OAAO,EAAwB,OAAO,EAAiH,MAAM,yBAAyB,CAAC;AACvL,OAAO,EAAC,8BAA8B,EAAE,oBAAoB,EAAC,MAAM,iBAAiB,CAAC;
|
|
1
|
+
{"version":3,"file":"engine_settings.js","sourceRoot":"","sources":["../src/engine_settings.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAC,iBAAiB,EAAC,MAAM,sBAAsB,CAAC;AAEvD,OAAO,EAAwB,OAAO,EAAiH,MAAM,yBAAyB,CAAC;AACvL,OAAO,EAAC,8BAA8B,EAAE,oBAAoB,EAAC,MAAM,iBAAiB,CAAC;AAkCrF;;GAEG;AACH,MAAM,UAAU,wCAAwC,CACpD,kBAAsC,EAAE,cAA8B,EACtE,OAAgB,EAAE,IAAI,GAAG,iBAAiB,EAAE,CAAC,YAAY;IAC3D,yEAAyE;IACzE,8EAA8E;IAC9E,gDAAgD;IAEhD,MAAM,oBAAoB,GACtB,kBAAkB,CAAC,8BAA8B,EAAE,CAAC;IACxD,oBAAoB,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAE,uBAAuB;IAEtE,IAAI,cAAc,CAAC,gBAAgB,EAAE,CAAC;QACpC,kBAAkB,CAAC,eAAe,EAAE,CAAC;IACvC,CAAC;IAED,IAAI,cAAc,CAAC,oBAAoB,EAAE,CAAC;QACxC,MAAM,oBAAoB,GAAG,cAAc,CAAC,oBAAoB,CAAC;QAEjE,IAAI,oBAAoB,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;YACpD,oBAAoB,CAAC,eAAe,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;QAC1E,CAAC;QACD,IAAI,oBAAoB,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;YACtD,oBAAoB,CAAC,iBAAiB,CAClC,EAAC,KAAK,EAAE,oBAAoB,CAAC,cAAc,EAAC,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,oBAAoB,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;YACrD,MAAM,aAAa,GAAG,oBAAoB,CAAC,aAAa,CAAC;YACzD,IAAI,OAAO,KAAK,OAAO,CAAC,GAAG,EAAE,CAAC;gBAC5B,oBAAoB,CAAC,mBAAmB,CAAC,aAA0B,CAAC,CAAC;YACvE,CAAC;iBAAM,IAAI,OAAO,KAAK,OAAO,CAAC,GAAG,EAAE,CAAC;gBACnC,oBAAoB,CAAC,mBAAmB,CAAC,aAA0B,CAAC,CAAC;YACvE,CAAC;iBAAM,IAAI,OAAO,KAAK,OAAO,CAAC,WAAW,EAAE,CAAC;gBAC3C,MAAM,gBAAgB,GAAG,aAAiC,CAAC;gBAC3D,MAAM,YAAY,GAAG,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBAC7C,oBAAoB,CAChB,gBAAgB,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;gBACzD,MAAM,oBAAoB,GAAyB;oBACjD,GAAG,gBAAgB;oBACnB,oBAAoB,EAAE,YAAY;iBACnC,CAAC;gBACF,oBAAoB,CAAC,0BAA0B,CAAC,oBAAoB,CAAC,CAAC;gBACtE,YAAY,CAAC,MAAM,EAAE,CAAC;YACxB,CAAC;QACH,CAAC;QACD,IAAI,oBAAoB,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACxD,MAAM,UAAU,GAAG,oBAAoB,CAAC,mBAAmB,EAAE,CAAC;YAC9D,MAAM,MAAM,GAAG,EAAC,GAAG,UAAU,EAAE,GAAG,oBAAoB,CAAC,gBAAgB,EAAC,CAAC;YACzE,oBAAoB,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;AACH,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,kCAAkC,CAC9C,kBAAsC;IAExC,MAAM,oBAAoB,GACtB,kBAAkB,CAAC,8BAA8B,EAAE,CAAC;IAExD,IAAI,aAAmD,CAAC;IACxD,IAAI,OAAgB,CAAC;IACrB,IAAI,CAAC;QACH,0EAA0E;QAC1E,aAAa,GAAG,oBAAoB,CAAC,mBAAmB,EAAE,CAAC;QAC3D,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC;IACxB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,CAAC;YACH,aAAa,GAAG,oBAAoB,CAAC,mBAAmB,EAAE,CAAC;YAC3D,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC;QACxB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC;gBACH,MAAM,oBAAoB,GACtB,oBAAoB,CAAC,0BAA0B,EAAE,CAAC;gBACtD,aAAa,GAAG;oBACd,GAAG,oBAAoB;oBACvB,oBAAoB,EAAE,8BAA8B,CAChD,oBAAoB,CAAC,oBAAoB,CAAC;iBAC/C,CAAC;gBACF,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC;YAChC,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,OAAO;QACP,oBAAoB,EAAE;YACpB,YAAY,EAAE,oBAAoB,CAAC,eAAe,EAAE;YACpD,cAAc,EAAE,oBAAoB,CAAC,iBAAiB,EAAE,CAAC,KAAK;YAC9D,aAAa;YACb,gBAAgB,EAAE,oBAAoB,CAAC,mBAAmB,EAAE;SAC7D;QACD,gBAAgB,EAAE,KAAK;KACxB,CAAC;AACJ,CAAC"}
|
package/dist/index.d.ts
CHANGED
|
@@ -16,6 +16,9 @@
|
|
|
16
16
|
export * from './litertlm_web.js';
|
|
17
17
|
export * from './global_litertlm.js';
|
|
18
18
|
export * from './load_litertlm.js';
|
|
19
|
+
export * from './orchestration/chat_interface.js';
|
|
20
|
+
export * from './orchestration/auto_tool_chat.js';
|
|
21
|
+
export * from './orchestration/webmcp_tool.js';
|
|
19
22
|
export { Backend } from './wasm_binding_types.js';
|
|
20
23
|
export * from './engine_settings.js';
|
|
21
24
|
export * from './engine.js';
|
|
@@ -24,4 +27,5 @@ export * from './conversation.js';
|
|
|
24
27
|
export * from './session_config.js';
|
|
25
28
|
export * from './conversation_config.js';
|
|
26
29
|
export * as Wasm from './wasm_binding_types.js';
|
|
30
|
+
export * from './types.js';
|
|
27
31
|
//# sourceMappingURL=index.d.ts.map
|
package/dist/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,cAAc,mBAAmB,CAAC;AAClC,cAAc,sBAAsB,CAAC;AACrC,cAAc,oBAAoB,CAAC;AACnC,OAAO,EAAC,OAAO,EAAC,MAAM,yBAAyB,CAAC;AAChD,cAAc,sBAAsB,CAAC;AACrC,cAAc,aAAa,CAAC;AAC5B,cAAc,cAAc,CAAC;AAC7B,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,0BAA0B,CAAC;AAGzC,OAAO,KAAK,IAAI,MAAM,yBAAyB,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,cAAc,mBAAmB,CAAC;AAClC,cAAc,sBAAsB,CAAC;AACrC,cAAc,oBAAoB,CAAC;AACnC,cAAc,mCAAmC,CAAC;AAClD,cAAc,mCAAmC,CAAC;AAClD,cAAc,gCAAgC,CAAC;AAE/C,OAAO,EAAC,OAAO,EAAC,MAAM,yBAAyB,CAAC;AAChD,cAAc,sBAAsB,CAAC;AACrC,cAAc,aAAa,CAAC;AAC5B,cAAc,cAAc,CAAC;AAC7B,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,0BAA0B,CAAC;AAGzC,OAAO,KAAK,IAAI,MAAM,yBAAyB,CAAC;AAChD,cAAc,YAAY,CAAC"}
|
package/dist/index.js
CHANGED
|
@@ -16,6 +16,9 @@
|
|
|
16
16
|
export * from './litertlm_web.js';
|
|
17
17
|
export * from './global_litertlm.js';
|
|
18
18
|
export * from './load_litertlm.js';
|
|
19
|
+
export * from './orchestration/chat_interface.js';
|
|
20
|
+
export * from './orchestration/auto_tool_chat.js';
|
|
21
|
+
export * from './orchestration/webmcp_tool.js';
|
|
19
22
|
export { Backend } from './wasm_binding_types.js';
|
|
20
23
|
export * from './engine_settings.js';
|
|
21
24
|
export * from './engine.js';
|
|
@@ -24,4 +27,5 @@ export * from './conversation.js';
|
|
|
24
27
|
export * from './session_config.js';
|
|
25
28
|
export * from './conversation_config.js';
|
|
26
29
|
export * as Wasm from './wasm_binding_types.js';
|
|
30
|
+
export * from './types.js';
|
|
27
31
|
//# sourceMappingURL=index.js.map
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,cAAc,mBAAmB,CAAC;AAClC,cAAc,sBAAsB,CAAC;AACrC,cAAc,oBAAoB,CAAC;AACnC,OAAO,EAAC,OAAO,EAAC,MAAM,yBAAyB,CAAC;AAChD,cAAc,sBAAsB,CAAC;AACrC,cAAc,aAAa,CAAC;AAC5B,cAAc,cAAc,CAAC;AAC7B,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,0BAA0B,CAAC;AAGzC,OAAO,KAAK,IAAI,MAAM,yBAAyB,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,cAAc,mBAAmB,CAAC;AAClC,cAAc,sBAAsB,CAAC;AACrC,cAAc,oBAAoB,CAAC;AACnC,cAAc,mCAAmC,CAAC;AAClD,cAAc,mCAAmC,CAAC;AAClD,cAAc,gCAAgC,CAAC;AAE/C,OAAO,EAAC,OAAO,EAAC,MAAM,yBAAyB,CAAC;AAChD,cAAc,sBAAsB,CAAC;AACrC,cAAc,aAAa,CAAC;AAC5B,cAAc,cAAc,CAAC;AAC7B,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,0BAA0B,CAAC;AAGzC,OAAO,KAAK,IAAI,MAAM,yBAAyB,CAAC;AAChD,cAAc,YAAY,CAAC"}
|
package/dist/litertlm_web.js
CHANGED
|
@@ -26,7 +26,7 @@ export async function setupDefaultWebGpuDevice() {
|
|
|
26
26
|
*/
|
|
27
27
|
export class LiteRtLm {
|
|
28
28
|
liteRtLmWasm;
|
|
29
|
-
static DEFAULT_WASM_PATH = 'https://cdn.jsdelivr.net/npm/@litert-lm/core@0.
|
|
29
|
+
static DEFAULT_WASM_PATH = 'https://cdn.jsdelivr.net/npm/@litert-lm/core@0.14.0/wasm';
|
|
30
30
|
constructor(wasmModule) {
|
|
31
31
|
this.liteRtLmWasm = wasmModule;
|
|
32
32
|
this.liteRtLmWasm.setupLogging();
|
package/dist/load.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"load.d.ts","sourceRoot":"","sources":["../src/load.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAC,QAAQ,EAAC,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAkC,OAAO,EAAC,MAAM,qBAAqB,CAAC;
|
|
1
|
+
{"version":3,"file":"load.d.ts","sourceRoot":"","sources":["../src/load.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAC,QAAQ,EAAC,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAkC,OAAO,EAAC,MAAM,qBAAqB,CAAC;AAS7E;;IAEI;AACJ,MAAM,WAAW,WAAW;CAAG;AAE/B;;;GAGG;AACH,wBAAsB,IAAI,CACtB,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC,CAsB3D"}
|
package/dist/load.js
CHANGED
|
@@ -19,6 +19,8 @@ import { appendPathSegment, pathToString } from './url_path_utils.js';
|
|
|
19
19
|
import { supportsFeature } from './wasm_feature_detect.js';
|
|
20
20
|
const WASM_JS_FILE_NAME = 'litertlm_wasm_internal.js';
|
|
21
21
|
const WASM_JS_COMPAT_FILE_NAME = 'litertlm_wasm_compat_internal.js';
|
|
22
|
+
const WASM_JS_ASYNCIFY_FILE_NAME = 'litertlm_wasm_asyncify_internal.js';
|
|
23
|
+
const WASM_JS_COMPAT_ASYNCIFY_FILE_NAME = 'litertlm_wasm_compat_asyncify_internal.js';
|
|
22
24
|
/**
|
|
23
25
|
* Load the LiteRtLm library with WASM from the given URL. Does not set the
|
|
24
26
|
* global LiteRtLm instance.
|
|
@@ -26,9 +28,14 @@ const WASM_JS_COMPAT_FILE_NAME = 'litertlm_wasm_compat_internal.js';
|
|
|
26
28
|
export async function load(path, options) {
|
|
27
29
|
const pathString = pathToString(path);
|
|
28
30
|
const relaxedSimd = await supportsFeature('relaxedSimd');
|
|
29
|
-
|
|
31
|
+
const jspi = await supportsFeature('jspi');
|
|
32
|
+
let fileName = '';
|
|
30
33
|
if (relaxedSimd) {
|
|
31
|
-
fileName = WASM_JS_FILE_NAME;
|
|
34
|
+
fileName = jspi ? WASM_JS_FILE_NAME : WASM_JS_ASYNCIFY_FILE_NAME;
|
|
35
|
+
}
|
|
36
|
+
else {
|
|
37
|
+
fileName =
|
|
38
|
+
jspi ? WASM_JS_COMPAT_FILE_NAME : WASM_JS_COMPAT_ASYNCIFY_FILE_NAME;
|
|
32
39
|
}
|
|
33
40
|
let jsFilePath = path;
|
|
34
41
|
if (pathString.endsWith('.wasm')) {
|
package/dist/load.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"load.js","sourceRoot":"","sources":["../src/load.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAC,aAAa,EAAC,MAAM,sBAAsB,CAAC;AAEnD,OAAO,EAAC,QAAQ,EAAC,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAC,iBAAiB,EAAE,YAAY,EAAU,MAAM,qBAAqB,CAAC;AAC7E,OAAO,EAAC,eAAe,EAAC,MAAM,0BAA0B,CAAC;AAEzD,MAAM,iBAAiB,GAAG,2BAA2B,CAAC;AACtD,MAAM,wBAAwB,GAAG,kCAAkC,CAAC;
|
|
1
|
+
{"version":3,"file":"load.js","sourceRoot":"","sources":["../src/load.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAC,aAAa,EAAC,MAAM,sBAAsB,CAAC;AAEnD,OAAO,EAAC,QAAQ,EAAC,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAC,iBAAiB,EAAE,YAAY,EAAU,MAAM,qBAAqB,CAAC;AAC7E,OAAO,EAAC,eAAe,EAAC,MAAM,0BAA0B,CAAC;AAEzD,MAAM,iBAAiB,GAAG,2BAA2B,CAAC;AACtD,MAAM,wBAAwB,GAAG,kCAAkC,CAAC;AACpE,MAAM,0BAA0B,GAAG,oCAAoC,CAAC;AACxE,MAAM,iCAAiC,GACnC,2CAA2C,CAAC;AAOhD;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,IAAI,CACtB,IAAa,EAAE,OAAqB;IACtC,MAAM,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IACtC,MAAM,WAAW,GAAG,MAAM,eAAe,CAAC,aAAa,CAAC,CAAC;IACzD,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAC3C,IAAI,QAAQ,GAAG,EAAE,CAAC;IAClB,IAAI,WAAW,EAAE,CAAC;QAChB,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,0BAA0B,CAAC;IACnE,CAAC;SAAM,CAAC;QACN,QAAQ;YACJ,IAAI,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,iCAAiC,CAAC;IAC1E,CAAC;IAED,IAAI,UAAU,GAAG,IAAI,CAAC;IACtB,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACjC,MAAM,IAAI,KAAK,CACX,mEAAmE;YACnE,mCAAmC,CAAC,CAAC;IAC3C,CAAC;SAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACvC,UAAU,GAAG,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACjD,CAAC;IAED,OAAO,aAAa,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;AAC7C,CAAC"}
|
package/dist/load_litertlm.d.ts
CHANGED
|
@@ -53,6 +53,6 @@ export declare function unloadLiteRtLm(): void;
|
|
|
53
53
|
/**
|
|
54
54
|
* Get the global LiteRT-LM instance, or load it if it hasn't been loaded yet.
|
|
55
55
|
*/
|
|
56
|
-
export declare function getOrLoadGlobalLiteRtLm(): Promise<LiteRtLm>;
|
|
56
|
+
export declare function getOrLoadGlobalLiteRtLm(path?: UrlString): Promise<LiteRtLm>;
|
|
57
57
|
export {};
|
|
58
58
|
//# sourceMappingURL=load_litertlm.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"load_litertlm.d.ts","sourceRoot":"","sources":["../src/load_litertlm.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAKH,OAAO,EAAC,QAAQ,EAAC,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAO,WAAW,EAAC,MAAM,WAAW,CAAC;AAE5C,KAAK,SAAS,GAAG,MAAM,CAAC;
|
|
1
|
+
{"version":3,"file":"load_litertlm.d.ts","sourceRoot":"","sources":["../src/load_litertlm.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAKH,OAAO,EAAC,QAAQ,EAAC,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAO,WAAW,EAAC,MAAM,WAAW,CAAC;AAE5C,KAAK,SAAS,GAAG,MAAM,CAAC;AAGxB;;;;;;IAMI;AACJ,MAAM,WAAW,mBAAoB,SAAQ,WAAW;CAAG;AAE3D;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,YAAY,CACxB,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,mBAAmB,GAAG,OAAO,CAAC,QAAQ,CAAC,CAcrE;AAED;;;;;;GAMG;AACH,wBAAgB,cAAc,IAAI,IAAI,CAarC;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,IAAI,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,QAAQ,CAAC,CAa3E"}
|
package/dist/load_litertlm.js
CHANGED
|
@@ -17,6 +17,7 @@
|
|
|
17
17
|
import { getGlobalLiteRtLm, getGlobalLiteRtLmPromise, hasGlobalLiteRtLm, hasGlobalLiteRtLmPromise, setGlobalLiteRtLm, setGlobalLiteRtLmPromise } from './global_litertlm.js';
|
|
18
18
|
import { LiteRtLm } from './litertlm_web.js';
|
|
19
19
|
import { load } from './load.js';
|
|
20
|
+
let globalLiteRtLmPath = undefined;
|
|
20
21
|
/**
|
|
21
22
|
* Load LiteRT-LM Wasm files from the given URL. This needs to be called before
|
|
22
23
|
* calling any other LiteRT-LM functions.
|
|
@@ -63,13 +64,24 @@ export function unloadLiteRtLm() {
|
|
|
63
64
|
if (hasGlobalLiteRtLm()) {
|
|
64
65
|
getGlobalLiteRtLm().delete();
|
|
65
66
|
setGlobalLiteRtLm(undefined);
|
|
67
|
+
globalLiteRtLmPath = undefined;
|
|
66
68
|
}
|
|
67
69
|
setGlobalLiteRtLmPromise(undefined);
|
|
68
70
|
}
|
|
69
71
|
/**
|
|
70
72
|
* Get the global LiteRT-LM instance, or load it if it hasn't been loaded yet.
|
|
71
73
|
*/
|
|
72
|
-
export function getOrLoadGlobalLiteRtLm() {
|
|
73
|
-
|
|
74
|
+
export function getOrLoadGlobalLiteRtLm(path) {
|
|
75
|
+
const liteRtLmPromise = getGlobalLiteRtLmPromise();
|
|
76
|
+
if (liteRtLmPromise) {
|
|
77
|
+
if (path && globalLiteRtLmPath !== path) {
|
|
78
|
+
throw new Error(`LiteRT-LM is already loading / loaded with a different path ` +
|
|
79
|
+
`${globalLiteRtLmPath} but was requested to load ${path}. To ` +
|
|
80
|
+
`reload with a different path, call unloadLiteRtLm() first.`);
|
|
81
|
+
}
|
|
82
|
+
return liteRtLmPromise;
|
|
83
|
+
}
|
|
84
|
+
globalLiteRtLmPath = path ?? LiteRtLm.DEFAULT_WASM_PATH;
|
|
85
|
+
return liteRtLmPromise ?? loadLiteRtLm(globalLiteRtLmPath);
|
|
74
86
|
}
|
|
75
87
|
//# sourceMappingURL=load_litertlm.js.map
|