cactus-react-native 0.1.3 → 0.2.1
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 +551 -720
- package/android/src/main/java/com/cactus/Cactus.java +41 -0
- package/android/src/main/java/com/cactus/LlamaContext.java +19 -0
- package/android/src/newarch/java/com/cactus/CactusModule.java +5 -0
- package/android/src/oldarch/java/com/cactus/CactusModule.java +5 -0
- package/ios/Cactus.mm +14 -0
- package/ios/CactusContext.h +1 -0
- package/ios/CactusContext.mm +18 -0
- package/ios/cactus.xcframework/ios-arm64_x86_64-simulator/cactus.framework/cactus +0 -0
- package/ios/cactus.xcframework/tvos-arm64_x86_64-simulator/cactus.framework/cactus +0 -0
- package/lib/commonjs/NativeCactus.js +10 -0
- package/lib/commonjs/NativeCactus.js.map +1 -1
- package/lib/commonjs/chat.js +37 -0
- package/lib/commonjs/grammar.js +560 -0
- package/lib/commonjs/index.js +545 -0
- package/lib/commonjs/index.js.map +1 -1
- package/lib/commonjs/lm.js +106 -0
- package/lib/commonjs/lm.js.map +1 -1
- package/lib/commonjs/projectId.js +8 -0
- package/lib/commonjs/projectId.js.map +1 -0
- package/lib/commonjs/remote.js +153 -0
- package/lib/commonjs/remote.js.map +1 -0
- package/lib/commonjs/telemetry.js +103 -0
- package/lib/commonjs/telemetry.js.map +1 -0
- package/lib/commonjs/tools.js +79 -0
- package/lib/commonjs/tools.js.map +1 -0
- package/lib/commonjs/tts.js +32 -0
- package/lib/commonjs/tts.js.map +1 -1
- package/lib/commonjs/vlm.js +150 -0
- package/lib/commonjs/vlm.js.map +1 -0
- package/lib/module/NativeCactus.js +8 -0
- package/lib/module/NativeCactus.js.map +1 -1
- package/lib/module/chat.js +33 -0
- package/lib/module/grammar.js +553 -0
- package/lib/module/index.js +435 -0
- package/lib/module/index.js.map +1 -1
- package/lib/module/lm.js +101 -0
- package/lib/module/lm.js.map +1 -0
- package/lib/module/projectId.js +4 -0
- package/lib/module/projectId.js.map +1 -0
- package/lib/module/remote.js +144 -0
- package/lib/module/remote.js.map +1 -0
- package/lib/module/telemetry.js +98 -0
- package/lib/module/telemetry.js.map +1 -0
- package/lib/module/tools.js +73 -0
- package/lib/module/tools.js.map +1 -0
- package/lib/module/tts.js +27 -0
- package/lib/module/tts.js.map +1 -1
- package/lib/module/vlm.js +145 -0
- package/lib/module/vlm.js.map +1 -1
- package/lib/typescript/NativeCactus.d.ts +7 -0
- package/lib/typescript/NativeCactus.d.ts.map +1 -1
- package/lib/typescript/index.d.ts +3 -1
- package/lib/typescript/index.d.ts.map +1 -1
- package/lib/typescript/lm.d.ts +11 -34
- package/lib/typescript/lm.d.ts.map +1 -1
- package/lib/typescript/projectId.d.ts +2 -0
- package/lib/typescript/projectId.d.ts.map +1 -0
- package/lib/typescript/remote.d.ts +7 -0
- package/lib/typescript/remote.d.ts.map +1 -0
- package/lib/typescript/telemetry.d.ts +25 -0
- package/lib/typescript/telemetry.d.ts.map +1 -0
- package/lib/typescript/tools.d.ts +0 -3
- package/lib/typescript/tools.d.ts.map +1 -1
- package/lib/typescript/tts.d.ts.map +1 -1
- package/lib/typescript/vlm.d.ts +14 -34
- package/lib/typescript/vlm.d.ts.map +1 -1
- package/package.json +4 -4
- package/scripts/postInstall.js +33 -0
- package/src/NativeCactus.ts +7 -0
- package/src/index.ts +122 -46
- package/src/lm.ts +80 -5
- package/src/projectId.ts +1 -0
- package/src/remote.ts +175 -0
- package/src/telemetry.ts +138 -0
- package/src/tools.ts +17 -58
- package/src/vlm.ts +129 -8
- package/android/src/main/jniLibs/x86_64/libcactus.so +0 -0
- package/android/src/main/jniLibs/x86_64/libcactus_x86_64.so +0 -0
|
@@ -0,0 +1,435 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
|
|
3
|
+
import { NativeEventEmitter, DeviceEventEmitter, Platform } from 'react-native';
|
|
4
|
+
import Cactus from './NativeCactus';
|
|
5
|
+
import { SchemaGrammarConverter, convertJsonSchemaToGrammar } from './grammar';
|
|
6
|
+
import { formatChat } from './chat';
|
|
7
|
+
import { Tools, parseAndExecuteTool } from './tools';
|
|
8
|
+
import { Telemetry } from './telemetry';
|
|
9
|
+
export { SchemaGrammarConverter, convertJsonSchemaToGrammar, Tools };
|
|
10
|
+
export * from './remote';
|
|
11
|
+
const EVENT_ON_INIT_CONTEXT_PROGRESS = '@Cactus_onInitContextProgress';
|
|
12
|
+
const EVENT_ON_TOKEN = '@Cactus_onToken';
|
|
13
|
+
const EVENT_ON_NATIVE_LOG = '@Cactus_onNativeLog';
|
|
14
|
+
let EventEmitter;
|
|
15
|
+
if (Platform.OS === 'ios') {
|
|
16
|
+
// @ts-ignore
|
|
17
|
+
EventEmitter = new NativeEventEmitter(Cactus);
|
|
18
|
+
}
|
|
19
|
+
if (Platform.OS === 'android') {
|
|
20
|
+
EventEmitter = DeviceEventEmitter;
|
|
21
|
+
}
|
|
22
|
+
const logListeners = [];
|
|
23
|
+
|
|
24
|
+
// @ts-ignore
|
|
25
|
+
if (EventEmitter) {
|
|
26
|
+
EventEmitter.addListener(EVENT_ON_NATIVE_LOG, evt => {
|
|
27
|
+
logListeners.forEach(listener => listener(evt.level, evt.text));
|
|
28
|
+
});
|
|
29
|
+
// Trigger unset to use default log callback
|
|
30
|
+
Cactus?.toggleNativeLog?.(false)?.catch?.(() => {});
|
|
31
|
+
}
|
|
32
|
+
const getJsonSchema = responseFormat => {
|
|
33
|
+
if (responseFormat?.type === 'json_schema') {
|
|
34
|
+
return responseFormat.json_schema?.schema;
|
|
35
|
+
}
|
|
36
|
+
if (responseFormat?.type === 'json_object') {
|
|
37
|
+
return responseFormat.schema || {};
|
|
38
|
+
}
|
|
39
|
+
return null;
|
|
40
|
+
};
|
|
41
|
+
const telemetryParams = {
|
|
42
|
+
n_gpu_layers: null,
|
|
43
|
+
n_ctx: null,
|
|
44
|
+
model: null
|
|
45
|
+
};
|
|
46
|
+
export class LlamaContext {
|
|
47
|
+
gpu = false;
|
|
48
|
+
reasonNoGPU = '';
|
|
49
|
+
constructor({
|
|
50
|
+
contextId,
|
|
51
|
+
gpu,
|
|
52
|
+
reasonNoGPU,
|
|
53
|
+
model
|
|
54
|
+
}) {
|
|
55
|
+
this.id = contextId;
|
|
56
|
+
this.gpu = gpu;
|
|
57
|
+
this.reasonNoGPU = reasonNoGPU;
|
|
58
|
+
this.model = model;
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
/**
|
|
62
|
+
* Load cached prompt & completion state from a file.
|
|
63
|
+
*/
|
|
64
|
+
async loadSession(filepath) {
|
|
65
|
+
let path = filepath;
|
|
66
|
+
if (path.startsWith('file://')) path = path.slice(7);
|
|
67
|
+
return Cactus.loadSession(this.id, path);
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
/**
|
|
71
|
+
* Save current cached prompt & completion state to a file.
|
|
72
|
+
*/
|
|
73
|
+
async saveSession(filepath, options) {
|
|
74
|
+
return Cactus.saveSession(this.id, filepath, options?.tokenSize || -1);
|
|
75
|
+
}
|
|
76
|
+
isLlamaChatSupported() {
|
|
77
|
+
return !!this.model.chatTemplates.llamaChat;
|
|
78
|
+
}
|
|
79
|
+
isJinjaSupported() {
|
|
80
|
+
const {
|
|
81
|
+
minja
|
|
82
|
+
} = this.model.chatTemplates;
|
|
83
|
+
return !!minja?.toolUse || !!minja?.default;
|
|
84
|
+
}
|
|
85
|
+
async getFormattedChat(messages, template, params) {
|
|
86
|
+
const chat = formatChat(messages);
|
|
87
|
+
const useJinja = this.isJinjaSupported() && params?.jinja;
|
|
88
|
+
let tmpl = this.isLlamaChatSupported() || useJinja ? undefined : 'chatml';
|
|
89
|
+
if (template) tmpl = template; // Force replace if provided
|
|
90
|
+
const jsonSchema = getJsonSchema(params?.response_format);
|
|
91
|
+
return Cactus.getFormattedChat(this.id, JSON.stringify(chat), tmpl, {
|
|
92
|
+
jinja: useJinja,
|
|
93
|
+
json_schema: jsonSchema ? JSON.stringify(jsonSchema) : undefined,
|
|
94
|
+
tools: params?.tools ? JSON.stringify(params.tools) : undefined,
|
|
95
|
+
parallel_tool_calls: params?.parallel_tool_calls ? JSON.stringify(params.parallel_tool_calls) : undefined,
|
|
96
|
+
tool_choice: params?.tool_choice
|
|
97
|
+
});
|
|
98
|
+
}
|
|
99
|
+
async completionWithTools(params, callback, recursionCount = 0, recursionLimit = 3) {
|
|
100
|
+
if (!params.messages) {
|
|
101
|
+
// tool calling only works with messages
|
|
102
|
+
return this.completion(params, callback);
|
|
103
|
+
}
|
|
104
|
+
if (!params.tools) {
|
|
105
|
+
// no tools => default completion
|
|
106
|
+
return this.completion(params, callback);
|
|
107
|
+
}
|
|
108
|
+
if (recursionCount >= recursionLimit) {
|
|
109
|
+
// console.log(`Recursion limit reached (${recursionCount}/${recursionLimit}), returning default completion`)
|
|
110
|
+
return this.completion({
|
|
111
|
+
...params,
|
|
112
|
+
jinja: true,
|
|
113
|
+
tools: params.tools.getSchemas()
|
|
114
|
+
}, callback);
|
|
115
|
+
}
|
|
116
|
+
const messages = [...params.messages]; // avoid mutating the original messages
|
|
117
|
+
|
|
118
|
+
// console.log('Calling completion...')
|
|
119
|
+
const result = await this.completion({
|
|
120
|
+
...params,
|
|
121
|
+
messages: messages,
|
|
122
|
+
jinja: true,
|
|
123
|
+
tools: params.tools.getSchemas()
|
|
124
|
+
}, callback);
|
|
125
|
+
// console.log('Completion result:', result);
|
|
126
|
+
|
|
127
|
+
const {
|
|
128
|
+
toolCalled,
|
|
129
|
+
toolName,
|
|
130
|
+
toolInput,
|
|
131
|
+
toolOutput
|
|
132
|
+
} = await parseAndExecuteTool(result, params.tools);
|
|
133
|
+
if (toolCalled && toolName && toolInput) {
|
|
134
|
+
const assistantMessage = {
|
|
135
|
+
role: 'assistant',
|
|
136
|
+
content: result.content,
|
|
137
|
+
tool_calls: result.tool_calls
|
|
138
|
+
};
|
|
139
|
+
messages.push(assistantMessage);
|
|
140
|
+
const toolCallId = result.tool_calls?.[0]?.id;
|
|
141
|
+
const toolMessage = {
|
|
142
|
+
role: 'tool',
|
|
143
|
+
content: JSON.stringify(toolOutput),
|
|
144
|
+
tool_call_id: toolCallId
|
|
145
|
+
};
|
|
146
|
+
messages.push(toolMessage);
|
|
147
|
+
|
|
148
|
+
// console.log('Messages being sent to next completion:', JSON.stringify(messages, null, 2));
|
|
149
|
+
|
|
150
|
+
return await this.completionWithTools({
|
|
151
|
+
...params,
|
|
152
|
+
messages: messages
|
|
153
|
+
}, callback, recursionCount + 1, recursionLimit);
|
|
154
|
+
}
|
|
155
|
+
return result;
|
|
156
|
+
}
|
|
157
|
+
async completion(params, callback) {
|
|
158
|
+
const nativeParams = {
|
|
159
|
+
...params,
|
|
160
|
+
prompt: params.prompt || '',
|
|
161
|
+
emit_partial_completion: !!callback
|
|
162
|
+
};
|
|
163
|
+
if (params.messages) {
|
|
164
|
+
// messages always win
|
|
165
|
+
const formattedResult = await this.getFormattedChat(params.messages, params.chat_template || params.chatTemplate, {
|
|
166
|
+
jinja: params.jinja,
|
|
167
|
+
tools: params.tools,
|
|
168
|
+
parallel_tool_calls: params.parallel_tool_calls,
|
|
169
|
+
tool_choice: params.tool_choice
|
|
170
|
+
});
|
|
171
|
+
if (typeof formattedResult === 'string') {
|
|
172
|
+
nativeParams.prompt = formattedResult || '';
|
|
173
|
+
} else {
|
|
174
|
+
nativeParams.prompt = formattedResult.prompt || '';
|
|
175
|
+
if (typeof formattedResult.chat_format === 'number') nativeParams.chat_format = formattedResult.chat_format;
|
|
176
|
+
if (formattedResult.grammar) nativeParams.grammar = formattedResult.grammar;
|
|
177
|
+
if (typeof formattedResult.grammar_lazy === 'boolean') nativeParams.grammar_lazy = formattedResult.grammar_lazy;
|
|
178
|
+
if (formattedResult.grammar_triggers) nativeParams.grammar_triggers = formattedResult.grammar_triggers;
|
|
179
|
+
if (formattedResult.preserved_tokens) nativeParams.preserved_tokens = formattedResult.preserved_tokens;
|
|
180
|
+
if (formattedResult.additional_stops) {
|
|
181
|
+
if (!nativeParams.stop) nativeParams.stop = [];
|
|
182
|
+
nativeParams.stop.push(...formattedResult.additional_stops);
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
} else {
|
|
186
|
+
nativeParams.prompt = params.prompt || '';
|
|
187
|
+
}
|
|
188
|
+
if (nativeParams.response_format && !nativeParams.grammar) {
|
|
189
|
+
const jsonSchema = getJsonSchema(params.response_format);
|
|
190
|
+
if (jsonSchema) nativeParams.json_schema = JSON.stringify(jsonSchema);
|
|
191
|
+
}
|
|
192
|
+
const startTime = Date.now();
|
|
193
|
+
let firstTokenTime = null;
|
|
194
|
+
const deviceInfo = await getDeviceInfo(this.id);
|
|
195
|
+
const wrappedCallback = callback ? data => {
|
|
196
|
+
if (firstTokenTime === null) firstTokenTime = Date.now();
|
|
197
|
+
callback(data);
|
|
198
|
+
} : undefined;
|
|
199
|
+
let tokenListener = wrappedCallback && EventEmitter.addListener(EVENT_ON_TOKEN, evt => {
|
|
200
|
+
const {
|
|
201
|
+
contextId,
|
|
202
|
+
tokenResult
|
|
203
|
+
} = evt;
|
|
204
|
+
if (contextId !== this.id) return;
|
|
205
|
+
wrappedCallback(tokenResult);
|
|
206
|
+
});
|
|
207
|
+
if (!nativeParams.prompt) throw new Error('Prompt is required');
|
|
208
|
+
const promise = Cactus.completion(this.id, nativeParams);
|
|
209
|
+
return promise.then(completionResult => {
|
|
210
|
+
Telemetry.track({
|
|
211
|
+
event: 'completion',
|
|
212
|
+
tok_per_sec: completionResult.timings?.predicted_per_second,
|
|
213
|
+
toks_generated: completionResult.timings?.predicted_n,
|
|
214
|
+
ttft: firstTokenTime ? firstTokenTime - startTime : null
|
|
215
|
+
}, telemetryParams, deviceInfo);
|
|
216
|
+
tokenListener?.remove();
|
|
217
|
+
tokenListener = null;
|
|
218
|
+
return completionResult;
|
|
219
|
+
}).catch(err => {
|
|
220
|
+
tokenListener?.remove();
|
|
221
|
+
tokenListener = null;
|
|
222
|
+
throw err;
|
|
223
|
+
});
|
|
224
|
+
}
|
|
225
|
+
stopCompletion() {
|
|
226
|
+
return Cactus.stopCompletion(this.id);
|
|
227
|
+
}
|
|
228
|
+
tokenize(text) {
|
|
229
|
+
return Cactus.tokenize(this.id, text);
|
|
230
|
+
}
|
|
231
|
+
detokenize(tokens) {
|
|
232
|
+
return Cactus.detokenize(this.id, tokens);
|
|
233
|
+
}
|
|
234
|
+
async embedding(text, params) {
|
|
235
|
+
const startTime = Date.now();
|
|
236
|
+
const embeddingResult = await Cactus.embedding(this.id, text, params || {});
|
|
237
|
+
const totalTime = Date.now() - startTime;
|
|
238
|
+
const deviceInfo = await getDeviceInfo(this.id);
|
|
239
|
+
Telemetry.track({
|
|
240
|
+
event: 'embedding',
|
|
241
|
+
mode: 'local',
|
|
242
|
+
embedding_time: totalTime
|
|
243
|
+
}, telemetryParams, deviceInfo);
|
|
244
|
+
return embeddingResult;
|
|
245
|
+
}
|
|
246
|
+
async bench(pp, tg, pl, nr) {
|
|
247
|
+
const result = await Cactus.bench(this.id, pp, tg, pl, nr);
|
|
248
|
+
const [modelDesc, modelSize, modelNParams, ppAvg, ppStd, tgAvg, tgStd] = JSON.parse(result);
|
|
249
|
+
return {
|
|
250
|
+
modelDesc,
|
|
251
|
+
modelSize,
|
|
252
|
+
modelNParams,
|
|
253
|
+
ppAvg,
|
|
254
|
+
ppStd,
|
|
255
|
+
tgAvg,
|
|
256
|
+
tgStd
|
|
257
|
+
};
|
|
258
|
+
}
|
|
259
|
+
async applyLoraAdapters(loraList) {
|
|
260
|
+
let loraAdapters = [];
|
|
261
|
+
if (loraList) loraAdapters = loraList.map(l => ({
|
|
262
|
+
path: l.path.replace(/file:\/\//, ''),
|
|
263
|
+
scaled: l.scaled
|
|
264
|
+
}));
|
|
265
|
+
return Cactus.applyLoraAdapters(this.id, loraAdapters);
|
|
266
|
+
}
|
|
267
|
+
async removeLoraAdapters() {
|
|
268
|
+
return Cactus.removeLoraAdapters(this.id);
|
|
269
|
+
}
|
|
270
|
+
async getLoadedLoraAdapters() {
|
|
271
|
+
return Cactus.getLoadedLoraAdapters(this.id);
|
|
272
|
+
}
|
|
273
|
+
async release() {
|
|
274
|
+
return Cactus.releaseContext(this.id);
|
|
275
|
+
}
|
|
276
|
+
async rewind() {
|
|
277
|
+
// @ts-ignore
|
|
278
|
+
return Cactus.rewind(this.id);
|
|
279
|
+
}
|
|
280
|
+
}
|
|
281
|
+
export async function toggleNativeLog(enabled) {
|
|
282
|
+
return Cactus.toggleNativeLog(enabled);
|
|
283
|
+
}
|
|
284
|
+
export function addNativeLogListener(listener) {
|
|
285
|
+
logListeners.push(listener);
|
|
286
|
+
return {
|
|
287
|
+
remove: () => {
|
|
288
|
+
logListeners.splice(logListeners.indexOf(listener), 1);
|
|
289
|
+
}
|
|
290
|
+
};
|
|
291
|
+
}
|
|
292
|
+
export async function setContextLimit(limit) {
|
|
293
|
+
return Cactus.setContextLimit(limit);
|
|
294
|
+
}
|
|
295
|
+
let contextIdCounter = 0;
|
|
296
|
+
const contextIdRandom = () => process.env.NODE_ENV === 'test' ? 0 : Math.floor(Math.random() * 100000);
|
|
297
|
+
const modelInfoSkip = [
|
|
298
|
+
// Large fields
|
|
299
|
+
'tokenizer.ggml.tokens', 'tokenizer.ggml.token_type', 'tokenizer.ggml.merges', 'tokenizer.ggml.scores'];
|
|
300
|
+
export async function loadLlamaModelInfo(model) {
|
|
301
|
+
let path = model;
|
|
302
|
+
if (path.startsWith('file://')) path = path.slice(7);
|
|
303
|
+
return Cactus.modelInfo(path, modelInfoSkip);
|
|
304
|
+
}
|
|
305
|
+
const poolTypeMap = {
|
|
306
|
+
// -1 is unspecified as undefined
|
|
307
|
+
none: 0,
|
|
308
|
+
mean: 1,
|
|
309
|
+
cls: 2,
|
|
310
|
+
last: 3,
|
|
311
|
+
rank: 4
|
|
312
|
+
};
|
|
313
|
+
export async function initLlama({
|
|
314
|
+
model,
|
|
315
|
+
is_model_asset: isModelAsset,
|
|
316
|
+
pooling_type: poolingType,
|
|
317
|
+
lora,
|
|
318
|
+
lora_list: loraList,
|
|
319
|
+
...rest
|
|
320
|
+
}, onProgress) {
|
|
321
|
+
let path = model;
|
|
322
|
+
if (path.startsWith('file://')) path = path.slice(7);
|
|
323
|
+
let loraPath = lora;
|
|
324
|
+
if (loraPath?.startsWith('file://')) loraPath = loraPath.slice(7);
|
|
325
|
+
let loraAdapters = [];
|
|
326
|
+
if (loraList) loraAdapters = loraList.map(l => ({
|
|
327
|
+
path: l.path.replace(/file:\/\//, ''),
|
|
328
|
+
scaled: l.scaled
|
|
329
|
+
}));
|
|
330
|
+
telemetryParams.n_gpu_layers = rest.n_gpu_layers || null;
|
|
331
|
+
telemetryParams.n_ctx = rest.n_ctx || null;
|
|
332
|
+
telemetryParams.model = model;
|
|
333
|
+
const contextId = contextIdCounter + contextIdRandom();
|
|
334
|
+
contextIdCounter += 1;
|
|
335
|
+
let removeProgressListener = null;
|
|
336
|
+
if (onProgress) {
|
|
337
|
+
removeProgressListener = EventEmitter.addListener(EVENT_ON_INIT_CONTEXT_PROGRESS, evt => {
|
|
338
|
+
if (evt.contextId !== contextId) return;
|
|
339
|
+
onProgress(evt.progress);
|
|
340
|
+
});
|
|
341
|
+
}
|
|
342
|
+
const poolType = poolTypeMap[poolingType];
|
|
343
|
+
const {
|
|
344
|
+
gpu,
|
|
345
|
+
reasonNoGPU,
|
|
346
|
+
model: modelDetails,
|
|
347
|
+
androidLib
|
|
348
|
+
} = await Cactus.initContext(contextId, {
|
|
349
|
+
model: path,
|
|
350
|
+
is_model_asset: !!isModelAsset,
|
|
351
|
+
use_progress_callback: !!onProgress,
|
|
352
|
+
pooling_type: poolType,
|
|
353
|
+
lora: loraPath,
|
|
354
|
+
lora_list: loraAdapters,
|
|
355
|
+
...rest
|
|
356
|
+
}).catch(err => {
|
|
357
|
+
removeProgressListener?.remove();
|
|
358
|
+
throw err;
|
|
359
|
+
});
|
|
360
|
+
removeProgressListener?.remove();
|
|
361
|
+
return new LlamaContext({
|
|
362
|
+
contextId,
|
|
363
|
+
gpu,
|
|
364
|
+
reasonNoGPU,
|
|
365
|
+
model: modelDetails,
|
|
366
|
+
androidLib
|
|
367
|
+
});
|
|
368
|
+
}
|
|
369
|
+
export async function releaseAllLlama() {
|
|
370
|
+
return Cactus.releaseAllContexts();
|
|
371
|
+
}
|
|
372
|
+
export const initContext = async params => {
|
|
373
|
+
return await Cactus.initContext(contextIdCounter++, params);
|
|
374
|
+
};
|
|
375
|
+
export const initMultimodal = async (contextId, mmprojPath, useGpu = false) => {
|
|
376
|
+
return await Cactus.initMultimodal(contextId, mmprojPath, useGpu);
|
|
377
|
+
};
|
|
378
|
+
export const isMultimodalEnabled = async contextId => {
|
|
379
|
+
return await Cactus.isMultimodalEnabled(contextId);
|
|
380
|
+
};
|
|
381
|
+
export const isMultimodalSupportVision = async contextId => {
|
|
382
|
+
return await Cactus.isMultimodalSupportVision(contextId);
|
|
383
|
+
};
|
|
384
|
+
export const isMultimodalSupportAudio = async contextId => {
|
|
385
|
+
return await Cactus.isMultimodalSupportAudio(contextId);
|
|
386
|
+
};
|
|
387
|
+
export const releaseMultimodal = async contextId => {
|
|
388
|
+
return await Cactus.releaseMultimodal(contextId);
|
|
389
|
+
};
|
|
390
|
+
export const multimodalCompletion = async (contextId, prompt, mediaPaths, params) => {
|
|
391
|
+
const result = await Cactus.multimodalCompletion(contextId, prompt, mediaPaths, params);
|
|
392
|
+
const deviceInfo = await getDeviceInfo(contextId);
|
|
393
|
+
Telemetry.track({
|
|
394
|
+
event: 'completion',
|
|
395
|
+
tok_per_sec: result.timings?.predicted_per_second,
|
|
396
|
+
toks_generated: result.timings?.predicted_n,
|
|
397
|
+
num_images: mediaPaths?.length
|
|
398
|
+
}, telemetryParams, deviceInfo);
|
|
399
|
+
return result;
|
|
400
|
+
};
|
|
401
|
+
export const initVocoder = async (contextId, vocoderModelPath) => {
|
|
402
|
+
return await Cactus.initVocoder(contextId, vocoderModelPath);
|
|
403
|
+
};
|
|
404
|
+
export const isVocoderEnabled = async contextId => {
|
|
405
|
+
return await Cactus.isVocoderEnabled(contextId);
|
|
406
|
+
};
|
|
407
|
+
export const getTTSType = async contextId => {
|
|
408
|
+
return await Cactus.getTTSType(contextId);
|
|
409
|
+
};
|
|
410
|
+
export const getFormattedAudioCompletion = async (contextId, speakerJsonStr, textToSpeak) => {
|
|
411
|
+
return await Cactus.getFormattedAudioCompletion(contextId, speakerJsonStr, textToSpeak);
|
|
412
|
+
};
|
|
413
|
+
export const getAudioCompletionGuideTokens = async (contextId, textToSpeak) => {
|
|
414
|
+
return await Cactus.getAudioCompletionGuideTokens(contextId, textToSpeak);
|
|
415
|
+
};
|
|
416
|
+
export const decodeAudioTokens = async (contextId, tokens) => {
|
|
417
|
+
return await Cactus.decodeAudioTokens(contextId, tokens);
|
|
418
|
+
};
|
|
419
|
+
export const releaseVocoder = async contextId => {
|
|
420
|
+
return await Cactus.releaseVocoder(contextId);
|
|
421
|
+
};
|
|
422
|
+
export const tokenize = async (contextId, text, mediaPaths) => {
|
|
423
|
+
if (mediaPaths && mediaPaths.length > 0) {
|
|
424
|
+
return await Cactus.tokenize(contextId, text, mediaPaths);
|
|
425
|
+
} else {
|
|
426
|
+
return await Cactus.tokenize(contextId, text);
|
|
427
|
+
}
|
|
428
|
+
};
|
|
429
|
+
export const getDeviceInfo = async contextId => {
|
|
430
|
+
return await Cactus.getDeviceInfo(contextId);
|
|
431
|
+
};
|
|
432
|
+
export { CactusLM } from './lm';
|
|
433
|
+
export { CactusVLM } from './vlm';
|
|
434
|
+
export { CactusTTS } from './tts';
|
|
435
|
+
//# sourceMappingURL=index.js.map
|
package/lib/module/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"names":["NativeEventEmitter","DeviceEventEmitter","Platform","Cactus","SchemaGrammarConverter","convertJsonSchemaToGrammar","formatChat","Tools","injectToolsIntoMessages","parseAndExecuteTool","updateMessagesWithToolCall","EVENT_ON_INIT_CONTEXT_PROGRESS","EVENT_ON_TOKEN","EVENT_ON_NATIVE_LOG","EventEmitter","OS","logListeners","addListener","evt","forEach","listener","level","text","toggleNativeLog","catch","getJsonSchema","responseFormat","type","json_schema","schema","LlamaContext","gpu","reasonNoGPU","constructor","contextId","model","id","loadSession","filepath","path","startsWith","slice","saveSession","options","tokenSize","isLlamaChatSupported","chatTemplates","llamaChat","isJinjaSupported","minja","toolUse","default","getFormattedChat","messages","template","params","chat","useJinja","jinja","tmpl","undefined","jsonSchema","response_format","JSON","stringify","tools","parallel_tool_calls","tool_choice","completionWithTools","callback","recursionCount","recursionLimit","completion","result","toolCalled","toolName","toolInput","toolOutput","newMessages","nativeParams","prompt","emit_partial_completion","formattedResult","chat_template","chatTemplate","chat_format","grammar","grammar_lazy","grammar_triggers","preserved_tokens","additional_stops","stop","push","tokenListener","tokenResult","Error","promise","then","completionResult","remove","err","stopCompletion","tokenize","detokenize","tokens","embedding","bench","pp","tg","pl","nr","modelDesc","modelSize","modelNParams","ppAvg","ppStd","tgAvg","tgStd","parse","applyLoraAdapters","loraList","loraAdapters","map","l","replace","scaled","removeLoraAdapters","getLoadedLoraAdapters","release","releaseContext","rewind","enabled","addNativeLogListener","splice","indexOf","setContextLimit","limit","contextIdCounter","contextIdRandom","process","env","NODE_ENV","Math","floor","random","modelInfoSkip","loadLlamaModelInfo","modelInfo","poolTypeMap","none","mean","cls","last","rank","initLlama","is_model_asset","isModelAsset","pooling_type","poolingType","lora","lora_list","rest","onProgress","loraPath","removeProgressListener","progress","poolType","modelDetails","androidLib","initContext","use_progress_callback","releaseAllLlama","releaseAllContexts","initMultimodal","mmprojPath","useGpu","isMultimodalEnabled","isMultimodalSupportVision","isMultimodalSupportAudio","releaseMultimodal","multimodalCompletion","mediaPaths","initVocoder","vocoderModelPath","isVocoderEnabled","getTTSType","getFormattedAudioCompletion","speakerJsonStr","textToSpeak","getAudioCompletionGuideTokens","decodeAudioTokens","releaseVocoder","length","CactusLM","CactusVLM","CactusTTS"],"sourceRoot":"../../src","sources":["index.ts"],"mappings":";;AAAA,SAASA,kBAAkB,EAAEC,kBAAkB,EAAEC,QAAQ,QAAQ,cAAc;AAE/E,OAAOC,MAAM,MAAM,gBAAgB;AAuBnC,SAASC,sBAAsB,EAAEC,0BAA0B,QAAQ,WAAW;AAE9E,SAASC,UAAU,QAAQ,QAAQ;AACnC,SAASC,KAAK,EAAEC,uBAAuB,EAAEC,mBAAmB,EAAEC,0BAA0B,QAAQ,SAAS;AAuBzG,SAASN,sBAAsB,EAAEC,0BAA0B,EAAEE,KAAK;AAElE,MAAMI,8BAA8B,GAAG,+BAA+B;AACtE,MAAMC,cAAc,GAAG,iBAAiB;AACxC,MAAMC,mBAAmB,GAAG,qBAAqB;AAEjD,IAAIC,YAA2D;AAC/D,IAAIZ,QAAQ,CAACa,EAAE,KAAK,KAAK,EAAE;EACzB;EACAD,YAAY,GAAG,IAAId,kBAAkB,CAACG,MAAM,CAAC;AAC/C;AACA,IAAID,QAAQ,CAACa,EAAE,KAAK,SAAS,EAAE;EAC7BD,YAAY,GAAGb,kBAAkB;AACnC;AAEA,MAAMe,YAA0D,GAAG,EAAE;;AAErE;AACA,IAAIF,YAAY,EAAE;EAChBA,YAAY,CAACG,WAAW,CACtBJ,mBAAmB,EAClBK,GAAoC,IAAK;IACxCF,YAAY,CAACG,OAAO,CAAEC,QAAQ,IAAKA,QAAQ,CAACF,GAAG,CAACG,KAAK,EAAEH,GAAG,CAACI,IAAI,CAAC,CAAC;EACnE,CACF,CAAC;EACD;EACAnB,MAAM,EAAEoB,eAAe,GAAG,KAAK,CAAC,EAAEC,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC;AACrD;AA2EA,MAAMC,aAAa,GAAIC,cAAyC,IAAK;EACnE,IAAIA,cAAc,EAAEC,IAAI,KAAK,aAAa,EAAE;IAC1C,OAAOD,cAAc,CAACE,WAAW,EAAEC,MAAM;EAC3C;EACA,IAAIH,cAAc,EAAEC,IAAI,KAAK,aAAa,EAAE;IAC1C,OAAOD,cAAc,CAACG,MAAM,IAAI,CAAC,CAAC;EACpC;EACA,OAAO,IAAI;AACb,CAAC;AAED,OAAO,MAAMC,YAAY,CAAC;EAGxBC,GAAG,GAAY,KAAK;EAEpBC,WAAW,GAAW,EAAE;EAIxBC,WAAWA,CAAC;IAAEC,SAAS;IAAEH,GAAG;IAAEC,WAAW;IAAEG;EAA0B,CAAC,EAAE;IACtE,IAAI,CAACC,EAAE,GAAGF,SAAS;IACnB,IAAI,CAACH,GAAG,GAAGA,GAAG;IACd,IAAI,CAACC,WAAW,GAAGA,WAAW;IAC9B,IAAI,CAACG,KAAK,GAAGA,KAAK;EACpB;;EAEA;AACF;AACA;EACE,MAAME,WAAWA,CAACC,QAAgB,EAAoC;IACpE,IAAIC,IAAI,GAAGD,QAAQ;IACnB,IAAIC,IAAI,CAACC,UAAU,CAAC,SAAS,CAAC,EAAED,IAAI,GAAGA,IAAI,CAACE,KAAK,CAAC,CAAC,CAAC;IACpD,OAAOtC,MAAM,CAACkC,WAAW,CAAC,IAAI,CAACD,EAAE,EAAEG,IAAI,CAAC;EAC1C;;EAEA;AACF;AACA;EACE,MAAMG,WAAWA,CACfJ,QAAgB,EAChBK,OAA+B,EACd;IACjB,OAAOxC,MAAM,CAACuC,WAAW,CAAC,IAAI,CAACN,EAAE,EAAEE,QAAQ,EAAEK,OAAO,EAAEC,SAAS,IAAI,CAAC,CAAC,CAAC;EACxE;EAEAC,oBAAoBA,CAAA,EAAY;IAC9B,OAAO,CAAC,CAAC,IAAI,CAACV,KAAK,CAACW,aAAa,CAACC,SAAS;EAC7C;EAEAC,gBAAgBA,CAAA,EAAY;IAC1B,MAAM;MAAEC;IAAM,CAAC,GAAG,IAAI,CAACd,KAAK,CAACW,aAAa;IAC1C,OAAO,CAAC,CAACG,KAAK,EAAEC,OAAO,IAAI,CAAC,CAACD,KAAK,EAAEE,OAAO;EAC7C;EAEA,MAAMC,gBAAgBA,CACpBC,QAAsC,EACtCC,QAAwB,EACxBC,MAMC,EAC2C;IAC5C,MAAMC,IAAI,GAAGlD,UAAU,CAAC+C,QAAQ,CAAC;IACjC,MAAMI,QAAQ,GAAG,IAAI,CAACT,gBAAgB,CAAC,CAAC,IAAIO,MAAM,EAAEG,KAAK;IACzD,IAAIC,IAAI,GAAG,IAAI,CAACd,oBAAoB,CAAC,CAAC,IAAIY,QAAQ,GAAGG,SAAS,GAAG,QAAQ;IACzE,IAAIN,QAAQ,EAAEK,IAAI,GAAGL,QAAQ,EAAC;IAC9B,MAAMO,UAAU,GAAGpC,aAAa,CAAC8B,MAAM,EAAEO,eAAe,CAAC;IACzD,OAAO3D,MAAM,CAACiD,gBAAgB,CAAC,IAAI,CAAChB,EAAE,EAAE2B,IAAI,CAACC,SAAS,CAACR,IAAI,CAAC,EAAEG,IAAI,EAAE;MAClED,KAAK,EAAED,QAAQ;MACf7B,WAAW,EAAEiC,UAAU,GAAGE,IAAI,CAACC,SAAS,CAACH,UAAU,CAAC,GAAGD,SAAS;MAChEK,KAAK,EAAEV,MAAM,EAAEU,KAAK,GAAGF,IAAI,CAACC,SAAS,CAACT,MAAM,CAACU,KAAK,CAAC,GAAGL,SAAS;MAC/DM,mBAAmB,EAAEX,MAAM,EAAEW,mBAAmB,GAC5CH,IAAI,CAACC,SAAS,CAACT,MAAM,CAACW,mBAAmB,CAAC,GAC1CN,SAAS;MACbO,WAAW,EAAEZ,MAAM,EAAEY;IACvB,CAAC,CAAC;EACJ;EAEA,MAAMC,mBAAmBA,CACrBb,MAAyC,EACzCc,QAAoC,EACpCC,cAAsB,GAAG,CAAC,EAC1BC,cAAsB,GAAG,CAAC,EACK;IAC/B,IAAI,CAAChB,MAAM,CAACF,QAAQ,EAAE;MAAE;MACpB,OAAO,IAAI,CAACmB,UAAU,CAACjB,MAAM,EAAEc,QAAQ,CAAC;IAC5C;IACA,IAAI,CAACd,MAAM,CAACU,KAAK,EAAE;MAAE;MACjB,OAAO,IAAI,CAACO,UAAU,CAACjB,MAAM,EAAEc,QAAQ,CAAC;IAC5C;IACA,IAAIC,cAAc,IAAIC,cAAc,EAAE;MAClC,OAAO,IAAI,CAACC,UAAU,CAACjB,MAAM,EAAEc,QAAQ,CAAC;IAC5C;IAEA,IAAIhB,QAAQ,GAAG,CAAC,IAAGE,MAAM,EAAEF,QAAQ,IAAI,EAAE,EAAC,CAAC,CAAC;;IAE5C,IAAIiB,cAAc,KAAK,CAAC,EAAE;MACtBjB,QAAQ,GAAG7C,uBAAuB,CAAC6C,QAAQ,EAAEE,MAAM,CAACU,KAAK,CAAC;IAC9D;IAEA,MAAMQ,MAAM,GAAG,MAAM,IAAI,CAACD,UAAU,CAAC;MAAC,GAAGjB,MAAM;MAAEF;IAAQ,CAAC,EAAEgB,QAAQ,CAAC;IAErE,MAAM;MAACK,UAAU;MAAEC,QAAQ;MAAEC,SAAS;MAAEC;IAAU,CAAC,GAC/C,MAAMpE,mBAAmB,CAACgE,MAAM,EAAElB,MAAM,CAACU,KAAK,CAAC;IAEnD,IAAIS,UAAU,IAAIC,QAAQ,IAAIC,SAAS,EAAE;MACrC,MAAME,WAAW,GAAGpE,0BAA0B,CAC1C2C,QAAQ,EAAEsB,QAAQ,EAAEC,SAAS,EAAEC,UACnC,CAAC;MAED,OAAO,MAAM,IAAI,CAACT,mBAAmB,CACjC;QAAC,GAAGb,MAAM;QAAEF,QAAQ,EAAEyB;MAAW,CAAC,EAClCT,QAAQ,EACRC,cAAc,GAAG,CAAC,EAClBC,cACJ,CAAC;IACL;IAEA,OAAOE,MAAM;EACnB;EAEE,MAAMD,UAAUA,CACdjB,MAAwB,EACxBc,QAAoC,EACH;IACjC,MAAMU,YAAY,GAAG;MACnB,GAAGxB,MAAM;MACTyB,MAAM,EAAEzB,MAAM,CAACyB,MAAM,IAAI,EAAE;MAC3BC,uBAAuB,EAAE,CAAC,CAACZ;IAC7B,CAAC;IACD,IAAId,MAAM,CAACF,QAAQ,EAAE;MACnB;MACA,MAAM6B,eAAe,GAAG,MAAM,IAAI,CAAC9B,gBAAgB,CACjDG,MAAM,CAACF,QAAQ,EACfE,MAAM,CAAC4B,aAAa,IAAI5B,MAAM,CAAC6B,YAAY,EAC3C;QACE1B,KAAK,EAAEH,MAAM,CAACG,KAAK;QACnBO,KAAK,EAAEV,MAAM,CAACU,KAAK;QACnBC,mBAAmB,EAAEX,MAAM,CAACW,mBAAmB;QAC/CC,WAAW,EAAEZ,MAAM,CAACY;MACtB,CACF,CAAC;MACD,IAAI,OAAOe,eAAe,KAAK,QAAQ,EAAE;QACvCH,YAAY,CAACC,MAAM,GAAGE,eAAe,IAAI,EAAE;MAC7C,CAAC,MAAM;QACLH,YAAY,CAACC,MAAM,GAAGE,eAAe,CAACF,MAAM,IAAI,EAAE;QAClD,IAAI,OAAOE,eAAe,CAACG,WAAW,KAAK,QAAQ,EACjDN,YAAY,CAACM,WAAW,GAAGH,eAAe,CAACG,WAAW;QACxD,IAAIH,eAAe,CAACI,OAAO,EACzBP,YAAY,CAACO,OAAO,GAAGJ,eAAe,CAACI,OAAO;QAChD,IAAI,OAAOJ,eAAe,CAACK,YAAY,KAAK,SAAS,EACnDR,YAAY,CAACQ,YAAY,GAAGL,eAAe,CAACK,YAAY;QAC1D,IAAIL,eAAe,CAACM,gBAAgB,EAClCT,YAAY,CAACS,gBAAgB,GAAGN,eAAe,CAACM,gBAAgB;QAClE,IAAIN,eAAe,CAACO,gBAAgB,EAClCV,YAAY,CAACU,gBAAgB,GAAGP,eAAe,CAACO,gBAAgB;QAClE,IAAIP,eAAe,CAACQ,gBAAgB,EAAE;UACpC,IAAI,CAACX,YAAY,CAACY,IAAI,EAAEZ,YAAY,CAACY,IAAI,GAAG,EAAE;UAC9CZ,YAAY,CAACY,IAAI,CAACC,IAAI,CAAC,GAAGV,eAAe,CAACQ,gBAAgB,CAAC;QAC7D;MACF;IACF,CAAC,MAAM;MACLX,YAAY,CAACC,MAAM,GAAGzB,MAAM,CAACyB,MAAM,IAAI,EAAE;IAC3C;IAEA,IAAID,YAAY,CAACjB,eAAe,IAAI,CAACiB,YAAY,CAACO,OAAO,EAAE;MACzD,MAAMzB,UAAU,GAAGpC,aAAa,CAAC8B,MAAM,CAACO,eAAe,CAAC;MACxD,IAAID,UAAU,EAAEkB,YAAY,CAACnD,WAAW,GAAGmC,IAAI,CAACC,SAAS,CAACH,UAAU,CAAC;IACvE;IAEA,IAAIgC,aAAkB,GACpBxB,QAAQ,IACRvD,YAAY,CAACG,WAAW,CAACL,cAAc,EAAGM,GAAqB,IAAK;MAClE,MAAM;QAAEgB,SAAS;QAAE4D;MAAY,CAAC,GAAG5E,GAAG;MACtC,IAAIgB,SAAS,KAAK,IAAI,CAACE,EAAE,EAAE;MAC3BiC,QAAQ,CAACyB,WAAW,CAAC;IACvB,CAAC,CAAC;IAEJ,IAAI,CAACf,YAAY,CAACC,MAAM,EAAE,MAAM,IAAIe,KAAK,CAAC,oBAAoB,CAAC;IAE/D,MAAMC,OAAO,GAAG7F,MAAM,CAACqE,UAAU,CAAC,IAAI,CAACpC,EAAE,EAAE2C,YAAY,CAAC;IACxD,OAAOiB,OAAO,CACXC,IAAI,CAAEC,gBAAgB,IAAK;MAC1BL,aAAa,EAAEM,MAAM,CAAC,CAAC;MACvBN,aAAa,GAAG,IAAI;MACpB,OAAOK,gBAAgB;IACzB,CAAC,CAAC,CACD1E,KAAK,CAAE4E,GAAQ,IAAK;MACnBP,aAAa,EAAEM,MAAM,CAAC,CAAC;MACvBN,aAAa,GAAG,IAAI;MACpB,MAAMO,GAAG;IACX,CAAC,CAAC;EACN;EAEAC,cAAcA,CAAA,EAAkB;IAC9B,OAAOlG,MAAM,CAACkG,cAAc,CAAC,IAAI,CAACjE,EAAE,CAAC;EACvC;EAEAkE,QAAQA,CAAChF,IAAY,EAAiC;IACpD,OAAOnB,MAAM,CAACmG,QAAQ,CAAC,IAAI,CAAClE,EAAE,EAAEd,IAAI,CAAC;EACvC;EAEAiF,UAAUA,CAACC,MAAgB,EAAmB;IAC5C,OAAOrG,MAAM,CAACoG,UAAU,CAAC,IAAI,CAACnE,EAAE,EAAEoE,MAAM,CAAC;EAC3C;EAEAC,SAASA,CACPnF,IAAY,EACZiC,MAAwB,EACQ;IAChC,OAAOpD,MAAM,CAACsG,SAAS,CAAC,IAAI,CAACrE,EAAE,EAAEd,IAAI,EAAEiC,MAAM,IAAI,CAAC,CAAC,CAAC;EACtD;EAEA,MAAMmD,KAAKA,CACTC,EAAU,EACVC,EAAU,EACVC,EAAU,EACVC,EAAU,EACY;IACtB,MAAMrC,MAAM,GAAG,MAAMtE,MAAM,CAACuG,KAAK,CAAC,IAAI,CAACtE,EAAE,EAAEuE,EAAE,EAAEC,EAAE,EAAEC,EAAE,EAAEC,EAAE,CAAC;IAC1D,MAAM,CAACC,SAAS,EAAEC,SAAS,EAAEC,YAAY,EAAEC,KAAK,EAAEC,KAAK,EAAEC,KAAK,EAAEC,KAAK,CAAC,GACpEtD,IAAI,CAACuD,KAAK,CAAC7C,MAAM,CAAC;IACpB,OAAO;MACLsC,SAAS;MACTC,SAAS;MACTC,YAAY;MACZC,KAAK;MACLC,KAAK;MACLC,KAAK;MACLC;IACF,CAAC;EACH;EAEA,MAAME,iBAAiBA,CACrBC,QAAkD,EACnC;IACf,IAAIC,YAAsD,GAAG,EAAE;IAC/D,IAAID,QAAQ,EACVC,YAAY,GAAGD,QAAQ,CAACE,GAAG,CAAEC,CAAC,KAAM;MAClCpF,IAAI,EAAEoF,CAAC,CAACpF,IAAI,CAACqF,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;MACrCC,MAAM,EAAEF,CAAC,CAACE;IACZ,CAAC,CAAC,CAAC;IACL,OAAO1H,MAAM,CAACoH,iBAAiB,CAAC,IAAI,CAACnF,EAAE,EAAEqF,YAAY,CAAC;EACxD;EAEA,MAAMK,kBAAkBA,CAAA,EAAkB;IACxC,OAAO3H,MAAM,CAAC2H,kBAAkB,CAAC,IAAI,CAAC1F,EAAE,CAAC;EAC3C;EAEA,MAAM2F,qBAAqBA,CAAA,EAEzB;IACA,OAAO5H,MAAM,CAAC4H,qBAAqB,CAAC,IAAI,CAAC3F,EAAE,CAAC;EAC9C;EAEA,MAAM4F,OAAOA,CAAA,EAAkB;IAC7B,OAAO7H,MAAM,CAAC8H,cAAc,CAAC,IAAI,CAAC7F,EAAE,CAAC;EACvC;EAEA,MAAM8F,MAAMA,CAAA,EAAkB;IAC5B;IACA,OAAQ/H,MAAM,CAAS+H,MAAM,CAAC,IAAI,CAAC9F,EAAE,CAAC;EACxC;AACF;AAEA,OAAO,eAAeb,eAAeA,CAAC4G,OAAgB,EAAiB;EACrE,OAAOhI,MAAM,CAACoB,eAAe,CAAC4G,OAAO,CAAC;AACxC;AAEA,OAAO,SAASC,oBAAoBA,CAClChH,QAA+C,EACvB;EACxBJ,YAAY,CAAC4E,IAAI,CAACxE,QAAQ,CAAC;EAC3B,OAAO;IACL+E,MAAM,EAAEA,CAAA,KAAM;MACZnF,YAAY,CAACqH,MAAM,CAACrH,YAAY,CAACsH,OAAO,CAAClH,QAAQ,CAAC,EAAE,CAAC,CAAC;IACxD;EACF,CAAC;AACH;AAEA,OAAO,eAAemH,eAAeA,CAACC,KAAa,EAAiB;EAClE,OAAOrI,MAAM,CAACoI,eAAe,CAACC,KAAK,CAAC;AACtC;AAEA,IAAIC,gBAAgB,GAAG,CAAC;AACxB,MAAMC,eAAe,GAAGA,CAAA,KACtBC,OAAO,CAACC,GAAG,CAACC,QAAQ,KAAK,MAAM,GAAG,CAAC,GAAGC,IAAI,CAACC,KAAK,CAACD,IAAI,CAACE,MAAM,CAAC,CAAC,GAAG,MAAM,CAAC;AAE1E,MAAMC,aAAa,GAAG;AACpB;AACA,uBAAuB,EACvB,2BAA2B,EAC3B,uBAAuB,EACvB,uBAAuB,CACxB;AACD,OAAO,eAAeC,kBAAkBA,CAAC/G,KAAa,EAAmB;EACvE,IAAII,IAAI,GAAGJ,KAAK;EAChB,IAAII,IAAI,CAACC,UAAU,CAAC,SAAS,CAAC,EAAED,IAAI,GAAGA,IAAI,CAACE,KAAK,CAAC,CAAC,CAAC;EACpD,OAAOtC,MAAM,CAACgJ,SAAS,CAAC5G,IAAI,EAAE0G,aAAa,CAAC;AAC9C;AAEA,MAAMG,WAAW,GAAG;EAClB;EACAC,IAAI,EAAE,CAAC;EACPC,IAAI,EAAE,CAAC;EACPC,GAAG,EAAE,CAAC;EACNC,IAAI,EAAE,CAAC;EACPC,IAAI,EAAE;AACR,CAAC;AAED,OAAO,eAAeC,SAASA,CAC7B;EACEvH,KAAK;EACLwH,cAAc,EAAEC,YAAY;EAC5BC,YAAY,EAAEC,WAAW;EACzBC,IAAI;EACJC,SAAS,EAAExC,QAAQ;EACnB,GAAGyC;AACU,CAAC,EAChBC,UAAuC,EAChB;EACvB,IAAI3H,IAAI,GAAGJ,KAAK;EAChB,IAAII,IAAI,CAACC,UAAU,CAAC,SAAS,CAAC,EAAED,IAAI,GAAGA,IAAI,CAACE,KAAK,CAAC,CAAC,CAAC;EAEpD,IAAI0H,QAAQ,GAAGJ,IAAI;EACnB,IAAII,QAAQ,EAAE3H,UAAU,CAAC,SAAS,CAAC,EAAE2H,QAAQ,GAAGA,QAAQ,CAAC1H,KAAK,CAAC,CAAC,CAAC;EAEjE,IAAIgF,YAAsD,GAAG,EAAE;EAC/D,IAAID,QAAQ,EACVC,YAAY,GAAGD,QAAQ,CAACE,GAAG,CAAEC,CAAC,KAAM;IAClCpF,IAAI,EAAEoF,CAAC,CAACpF,IAAI,CAACqF,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;IACrCC,MAAM,EAAEF,CAAC,CAACE;EACZ,CAAC,CAAC,CAAC;EAEL,MAAM3F,SAAS,GAAGuG,gBAAgB,GAAGC,eAAe,CAAC,CAAC;EACtDD,gBAAgB,IAAI,CAAC;EAErB,IAAI2B,sBAA2B,GAAG,IAAI;EACtC,IAAIF,UAAU,EAAE;IACdE,sBAAsB,GAAGtJ,YAAY,CAACG,WAAW,CAC/CN,8BAA8B,EAC7BO,GAA4C,IAAK;MAChD,IAAIA,GAAG,CAACgB,SAAS,KAAKA,SAAS,EAAE;MACjCgI,UAAU,CAAChJ,GAAG,CAACmJ,QAAQ,CAAC;IAC1B,CACF,CAAC;EACH;EAEA,MAAMC,QAAQ,GAAGlB,WAAW,CAACU,WAAW,CAA6B;EACrE,MAAM;IACJ/H,GAAG;IACHC,WAAW;IACXG,KAAK,EAAEoI,YAAY;IACnBC;EACF,CAAC,GAAG,MAAMrK,MAAM,CAACsK,WAAW,CAACvI,SAAS,EAAE;IACtCC,KAAK,EAAEI,IAAI;IACXoH,cAAc,EAAE,CAAC,CAACC,YAAY;IAC9Bc,qBAAqB,EAAE,CAAC,CAACR,UAAU;IACnCL,YAAY,EAAES,QAAQ;IACtBP,IAAI,EAAEI,QAAQ;IACdH,SAAS,EAAEvC,YAAY;IACvB,GAAGwC;EACL,CAAC,CAAC,CAACzI,KAAK,CAAE4E,GAAQ,IAAK;IACrBgE,sBAAsB,EAAEjE,MAAM,CAAC,CAAC;IAChC,MAAMC,GAAG;EACX,CAAC,CAAC;EACFgE,sBAAsB,EAAEjE,MAAM,CAAC,CAAC;EAChC,OAAO,IAAIrE,YAAY,CAAC;IACtBI,SAAS;IACTH,GAAG;IACHC,WAAW;IACXG,KAAK,EAAEoI,YAAY;IACnBC;EACF,CAAC,CAAC;AACJ;AAEA,OAAO,eAAeG,eAAeA,CAAA,EAAkB;EACrD,OAAOxK,MAAM,CAACyK,kBAAkB,CAAC,CAAC;AACpC;AAEA,OAAO,MAAMH,WAAW,GAAG,MAAOlH,MAA2B,IAAK;EAChE,OAAO,MAAMpD,MAAM,CAACsK,WAAW,CAAChC,gBAAgB,EAAE,EAAElF,MAAM,CAAC;AAC7D,CAAC;AAED,OAAO,MAAMsH,cAAc,GAAG,MAAAA,CAAO3I,SAAiB,EAAE4I,UAAkB,EAAEC,MAAe,GAAG,KAAK,KAAK;EACtG,OAAO,MAAM5K,MAAM,CAAC0K,cAAc,CAAC3I,SAAS,EAAE4I,UAAU,EAAEC,MAAM,CAAC;AACnE,CAAC;AAED,OAAO,MAAMC,mBAAmB,GAAG,MAAO9I,SAAiB,IAAK;EAC9D,OAAO,MAAM/B,MAAM,CAAC6K,mBAAmB,CAAC9I,SAAS,CAAC;AACpD,CAAC;AAED,OAAO,MAAM+I,yBAAyB,GAAG,MAAO/I,SAAiB,IAAK;EACpE,OAAO,MAAM/B,MAAM,CAAC8K,yBAAyB,CAAC/I,SAAS,CAAC;AAC1D,CAAC;AAED,OAAO,MAAMgJ,wBAAwB,GAAG,MAAOhJ,SAAiB,IAAK;EACnE,OAAO,MAAM/B,MAAM,CAAC+K,wBAAwB,CAAChJ,SAAS,CAAC;AACzD,CAAC;AAED,OAAO,MAAMiJ,iBAAiB,GAAG,MAAOjJ,SAAiB,IAAK;EAC5D,OAAO,MAAM/B,MAAM,CAACgL,iBAAiB,CAACjJ,SAAS,CAAC;AAClD,CAAC;AAED,OAAO,MAAMkJ,oBAAoB,GAAG,MAAAA,CAAOlJ,SAAiB,EAAE8C,MAAc,EAAEqG,UAAoB,EAAE9H,MAA8B,KAAsC;EACtK,OAAO,MAAMpD,MAAM,CAACiL,oBAAoB,CAAClJ,SAAS,EAAE8C,MAAM,EAAEqG,UAAU,EAAE9H,MAAM,CAAC;AACjF,CAAC;AAED,OAAO,MAAM+H,WAAW,GAAG,MAAAA,CAAOpJ,SAAiB,EAAEqJ,gBAAwB,KAAK;EAChF,OAAO,MAAMpL,MAAM,CAACmL,WAAW,CAACpJ,SAAS,EAAEqJ,gBAAgB,CAAC;AAC9D,CAAC;AAED,OAAO,MAAMC,gBAAgB,GAAG,MAAOtJ,SAAiB,IAAK;EAC3D,OAAO,MAAM/B,MAAM,CAACqL,gBAAgB,CAACtJ,SAAS,CAAC;AACjD,CAAC;AAED,OAAO,MAAMuJ,UAAU,GAAG,MAAOvJ,SAAiB,IAA6B;EAC7E,OAAO,MAAM/B,MAAM,CAACsL,UAAU,CAACvJ,SAAS,CAAC;AAC3C,CAAC;AAED,OAAO,MAAMwJ,2BAA2B,GAAG,MAAAA,CAAOxJ,SAAiB,EAAEyJ,cAAsB,EAAEC,WAAmB,KAA2C;EACzJ,OAAO,MAAMzL,MAAM,CAACuL,2BAA2B,CAACxJ,SAAS,EAAEyJ,cAAc,EAAEC,WAAW,CAAC;AACzF,CAAC;AAED,OAAO,MAAMC,6BAA6B,GAAG,MAAAA,CAAO3J,SAAiB,EAAE0J,WAAmB,KAAuC;EAC/H,OAAO,MAAMzL,MAAM,CAAC0L,6BAA6B,CAAC3J,SAAS,EAAE0J,WAAW,CAAC;AAC3E,CAAC;AAED,OAAO,MAAME,iBAAiB,GAAG,MAAAA,CAAO5J,SAAiB,EAAEsE,MAAgB,KAAuC;EAChH,OAAO,MAAMrG,MAAM,CAAC2L,iBAAiB,CAAC5J,SAAS,EAAEsE,MAAM,CAAC;AAC1D,CAAC;AAED,OAAO,MAAMuF,cAAc,GAAG,MAAO7J,SAAiB,IAAK;EACzD,OAAO,MAAM/B,MAAM,CAAC4L,cAAc,CAAC7J,SAAS,CAAC;AAC/C,CAAC;AAED,OAAO,MAAMoE,QAAQ,GAAG,MAAAA,CAAOpE,SAAiB,EAAEZ,IAAY,EAAE+J,UAAqB,KAAoC;EACvH,IAAIA,UAAU,IAAIA,UAAU,CAACW,MAAM,GAAG,CAAC,EAAE;IACvC,OAAO,MAAM7L,MAAM,CAACmG,QAAQ,CAACpE,SAAS,EAAEZ,IAAI,EAAE+J,UAAU,CAAC;EAC3D,CAAC,MAAM;IACL,OAAO,MAAMlL,MAAM,CAACmG,QAAQ,CAACpE,SAAS,EAAEZ,IAAI,CAAC;EAC/C;AACF,CAAC;AAED,SAAS2K,QAAQ,QAAQ,MAAM;AAC/B,SAASC,SAAS,QAAQ,OAAO;AACjC,SAASC,SAAS,QAAQ,OAAO","ignoreList":[]}
|
|
1
|
+
{"version":3,"names":["NativeEventEmitter","DeviceEventEmitter","Platform","Cactus","SchemaGrammarConverter","convertJsonSchemaToGrammar","formatChat","Tools","parseAndExecuteTool","Telemetry","EVENT_ON_INIT_CONTEXT_PROGRESS","EVENT_ON_TOKEN","EVENT_ON_NATIVE_LOG","EventEmitter","OS","logListeners","addListener","evt","forEach","listener","level","text","toggleNativeLog","catch","getJsonSchema","responseFormat","type","json_schema","schema","telemetryParams","n_gpu_layers","n_ctx","model","LlamaContext","gpu","reasonNoGPU","constructor","contextId","id","loadSession","filepath","path","startsWith","slice","saveSession","options","tokenSize","isLlamaChatSupported","chatTemplates","llamaChat","isJinjaSupported","minja","toolUse","default","getFormattedChat","messages","template","params","chat","useJinja","jinja","tmpl","undefined","jsonSchema","response_format","JSON","stringify","tools","parallel_tool_calls","tool_choice","completionWithTools","callback","recursionCount","recursionLimit","completion","getSchemas","result","toolCalled","toolName","toolInput","toolOutput","assistantMessage","role","content","tool_calls","push","toolCallId","toolMessage","tool_call_id","nativeParams","prompt","emit_partial_completion","formattedResult","chat_template","chatTemplate","chat_format","grammar","grammar_lazy","grammar_triggers","preserved_tokens","additional_stops","stop","startTime","Date","now","firstTokenTime","deviceInfo","getDeviceInfo","wrappedCallback","data","tokenListener","tokenResult","Error","promise","then","completionResult","track","event","tok_per_sec","timings","predicted_per_second","toks_generated","predicted_n","ttft","remove","err","stopCompletion","tokenize","detokenize","tokens","embedding","embeddingResult","totalTime","mode","embedding_time","bench","pp","tg","pl","nr","modelDesc","modelSize","modelNParams","ppAvg","ppStd","tgAvg","tgStd","parse","applyLoraAdapters","loraList","loraAdapters","map","l","replace","scaled","removeLoraAdapters","getLoadedLoraAdapters","release","releaseContext","rewind","enabled","addNativeLogListener","splice","indexOf","setContextLimit","limit","contextIdCounter","contextIdRandom","process","env","NODE_ENV","Math","floor","random","modelInfoSkip","loadLlamaModelInfo","modelInfo","poolTypeMap","none","mean","cls","last","rank","initLlama","is_model_asset","isModelAsset","pooling_type","poolingType","lora","lora_list","rest","onProgress","loraPath","removeProgressListener","progress","poolType","modelDetails","androidLib","initContext","use_progress_callback","releaseAllLlama","releaseAllContexts","initMultimodal","mmprojPath","useGpu","isMultimodalEnabled","isMultimodalSupportVision","isMultimodalSupportAudio","releaseMultimodal","multimodalCompletion","mediaPaths","num_images","length","initVocoder","vocoderModelPath","isVocoderEnabled","getTTSType","getFormattedAudioCompletion","speakerJsonStr","textToSpeak","getAudioCompletionGuideTokens","decodeAudioTokens","releaseVocoder","CactusLM","CactusVLM","CactusTTS"],"sourceRoot":"../../src","sources":["index.ts"],"mappings":";;AAAA,SAASA,kBAAkB,EAAEC,kBAAkB,EAAEC,QAAQ,QAAQ,cAAc;AAE/E,OAAOC,MAAM,MAAM,gBAAgB;AAwBnC,SAASC,sBAAsB,EAAEC,0BAA0B,QAAQ,WAAW;AAE9E,SAASC,UAAU,QAAQ,QAAQ;AACnC,SAASC,KAAK,EAAEC,mBAAmB,QAAQ,SAAS;AACpD,SAASC,SAAS,QAA8B,aAAa;AAuB7D,SAASL,sBAAsB,EAAEC,0BAA0B,EAAEE,KAAK;AAClE,cAAc,UAAU;AAExB,MAAMG,8BAA8B,GAAG,+BAA+B;AACtE,MAAMC,cAAc,GAAG,iBAAiB;AACxC,MAAMC,mBAAmB,GAAG,qBAAqB;AAEjD,IAAIC,YAA2D;AAC/D,IAAIX,QAAQ,CAACY,EAAE,KAAK,KAAK,EAAE;EACzB;EACAD,YAAY,GAAG,IAAIb,kBAAkB,CAACG,MAAM,CAAC;AAC/C;AACA,IAAID,QAAQ,CAACY,EAAE,KAAK,SAAS,EAAE;EAC7BD,YAAY,GAAGZ,kBAAkB;AACnC;AAEA,MAAMc,YAA0D,GAAG,EAAE;;AAErE;AACA,IAAIF,YAAY,EAAE;EAChBA,YAAY,CAACG,WAAW,CACtBJ,mBAAmB,EAClBK,GAAoC,IAAK;IACxCF,YAAY,CAACG,OAAO,CAAEC,QAAQ,IAAKA,QAAQ,CAACF,GAAG,CAACG,KAAK,EAAEH,GAAG,CAACI,IAAI,CAAC,CAAC;EACnE,CACF,CAAC;EACD;EACAlB,MAAM,EAAEmB,eAAe,GAAG,KAAK,CAAC,EAAEC,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC;AACrD;AA2EA,MAAMC,aAAa,GAAIC,cAAyC,IAAK;EACnE,IAAIA,cAAc,EAAEC,IAAI,KAAK,aAAa,EAAE;IAC1C,OAAOD,cAAc,CAACE,WAAW,EAAEC,MAAM;EAC3C;EACA,IAAIH,cAAc,EAAEC,IAAI,KAAK,aAAa,EAAE;IAC1C,OAAOD,cAAc,CAACG,MAAM,IAAI,CAAC,CAAC;EACpC;EACA,OAAO,IAAI;AACb,CAAC;AAED,MAAMC,eAAgC,GAAG;EACvCC,YAAY,EAAE,IAAI;EAClBC,KAAK,EAAE,IAAI;EACXC,KAAK,EAAE;AACT,CAAC;AAED,OAAO,MAAMC,YAAY,CAAC;EAGxBC,GAAG,GAAY,KAAK;EAEpBC,WAAW,GAAW,EAAE;EAIxBC,WAAWA,CAAC;IAAEC,SAAS;IAAEH,GAAG;IAAEC,WAAW;IAAEH;EAA0B,CAAC,EAAE;IACtE,IAAI,CAACM,EAAE,GAAGD,SAAS;IACnB,IAAI,CAACH,GAAG,GAAGA,GAAG;IACd,IAAI,CAACC,WAAW,GAAGA,WAAW;IAC9B,IAAI,CAACH,KAAK,GAAGA,KAAK;EACpB;;EAEA;AACF;AACA;EACE,MAAMO,WAAWA,CAACC,QAAgB,EAAoC;IACpE,IAAIC,IAAI,GAAGD,QAAQ;IACnB,IAAIC,IAAI,CAACC,UAAU,CAAC,SAAS,CAAC,EAAED,IAAI,GAAGA,IAAI,CAACE,KAAK,CAAC,CAAC,CAAC;IACpD,OAAOxC,MAAM,CAACoC,WAAW,CAAC,IAAI,CAACD,EAAE,EAAEG,IAAI,CAAC;EAC1C;;EAEA;AACF;AACA;EACE,MAAMG,WAAWA,CACfJ,QAAgB,EAChBK,OAA+B,EACd;IACjB,OAAO1C,MAAM,CAACyC,WAAW,CAAC,IAAI,CAACN,EAAE,EAAEE,QAAQ,EAAEK,OAAO,EAAEC,SAAS,IAAI,CAAC,CAAC,CAAC;EACxE;EAEAC,oBAAoBA,CAAA,EAAY;IAC9B,OAAO,CAAC,CAAC,IAAI,CAACf,KAAK,CAACgB,aAAa,CAACC,SAAS;EAC7C;EAEAC,gBAAgBA,CAAA,EAAY;IAC1B,MAAM;MAAEC;IAAM,CAAC,GAAG,IAAI,CAACnB,KAAK,CAACgB,aAAa;IAC1C,OAAO,CAAC,CAACG,KAAK,EAAEC,OAAO,IAAI,CAAC,CAACD,KAAK,EAAEE,OAAO;EAC7C;EAEA,MAAMC,gBAAgBA,CACpBC,QAAsC,EACtCC,QAAwB,EACxBC,MAMC,EAC2C;IAC5C,MAAMC,IAAI,GAAGpD,UAAU,CAACiD,QAAQ,CAAC;IACjC,MAAMI,QAAQ,GAAG,IAAI,CAACT,gBAAgB,CAAC,CAAC,IAAIO,MAAM,EAAEG,KAAK;IACzD,IAAIC,IAAI,GAAG,IAAI,CAACd,oBAAoB,CAAC,CAAC,IAAIY,QAAQ,GAAGG,SAAS,GAAG,QAAQ;IACzE,IAAIN,QAAQ,EAAEK,IAAI,GAAGL,QAAQ,EAAC;IAC9B,MAAMO,UAAU,GAAGvC,aAAa,CAACiC,MAAM,EAAEO,eAAe,CAAC;IACzD,OAAO7D,MAAM,CAACmD,gBAAgB,CAAC,IAAI,CAAChB,EAAE,EAAE2B,IAAI,CAACC,SAAS,CAACR,IAAI,CAAC,EAAEG,IAAI,EAAE;MAClED,KAAK,EAAED,QAAQ;MACfhC,WAAW,EAAEoC,UAAU,GAAGE,IAAI,CAACC,SAAS,CAACH,UAAU,CAAC,GAAGD,SAAS;MAChEK,KAAK,EAAEV,MAAM,EAAEU,KAAK,GAAGF,IAAI,CAACC,SAAS,CAACT,MAAM,CAACU,KAAK,CAAC,GAAGL,SAAS;MAC/DM,mBAAmB,EAAEX,MAAM,EAAEW,mBAAmB,GAC5CH,IAAI,CAACC,SAAS,CAACT,MAAM,CAACW,mBAAmB,CAAC,GAC1CN,SAAS;MACbO,WAAW,EAAEZ,MAAM,EAAEY;IACvB,CAAC,CAAC;EACJ;EAEA,MAAMC,mBAAmBA,CACvBb,MAAyC,EACzCc,QAAoC,EACpCC,cAAsB,GAAG,CAAC,EAC1BC,cAAsB,GAAG,CAAC,EACK;IAC/B,IAAI,CAAChB,MAAM,CAACF,QAAQ,EAAE;MAAE;MACpB,OAAO,IAAI,CAACmB,UAAU,CAACjB,MAAM,EAAEc,QAAQ,CAAC;IAC5C;IACA,IAAI,CAACd,MAAM,CAACU,KAAK,EAAE;MAAE;MACjB,OAAO,IAAI,CAACO,UAAU,CAACjB,MAAM,EAAEc,QAAQ,CAAC;IAC5C;IACA,IAAIC,cAAc,IAAIC,cAAc,EAAE;MAClC;MACA,OAAO,IAAI,CAACC,UAAU,CAAC;QACnB,GAAGjB,MAAM;QACTG,KAAK,EAAE,IAAI;QACXO,KAAK,EAAEV,MAAM,CAACU,KAAK,CAACQ,UAAU,CAAC;MACnC,CAAC,EAAEJ,QAAQ,CAAC;IAChB;IAEA,MAAMhB,QAAQ,GAAG,CAAC,GAAGE,MAAM,CAACF,QAAQ,CAAC,CAAC,CAAC;;IAEvC;IACA,MAAMqB,MAAM,GAAG,MAAM,IAAI,CAACF,UAAU,CAAC;MACjC,GAAGjB,MAAM;MACTF,QAAQ,EAAEA,QAAQ;MAClBK,KAAK,EAAE,IAAI;MACXO,KAAK,EAAEV,MAAM,CAACU,KAAK,CAACQ,UAAU,CAAC;IACnC,CAAC,EAAEJ,QAAQ,CAAC;IACZ;;IAEA,MAAM;MAACM,UAAU;MAAEC,QAAQ;MAAEC,SAAS;MAAEC;IAAU,CAAC,GAC/C,MAAMxE,mBAAmB,CAACoE,MAAM,EAAEnB,MAAM,CAACU,KAAK,CAAC;IAEnD,IAAIU,UAAU,IAAIC,QAAQ,IAAIC,SAAS,EAAE;MACrC,MAAME,gBAAgB,GAAG;QACrBC,IAAI,EAAE,WAAW;QACjBC,OAAO,EAAEP,MAAM,CAACO,OAAO;QACvBC,UAAU,EAAER,MAAM,CAACQ;MACvB,CAA+B;MAE/B7B,QAAQ,CAAC8B,IAAI,CAACJ,gBAAgB,CAAC;MAE/B,MAAMK,UAAU,GAAGV,MAAM,CAACQ,UAAU,GAAG,CAAC,CAAC,EAAE9C,EAAE;MAC7C,MAAMiD,WAAW,GAAG;QAChBL,IAAI,EAAE,MAAM;QACZC,OAAO,EAAElB,IAAI,CAACC,SAAS,CAACc,UAAU,CAAC;QACnCQ,YAAY,EAAEF;MAClB,CAA+B;MAE/B/B,QAAQ,CAAC8B,IAAI,CAACE,WAAW,CAAC;;MAE1B;;MAEA,OAAO,MAAM,IAAI,CAACjB,mBAAmB,CACjC;QAAC,GAAGb,MAAM;QAAEF,QAAQ,EAAEA;MAAQ,CAAC,EAC/BgB,QAAQ,EACRC,cAAc,GAAG,CAAC,EAClBC,cACJ,CAAC;IACL;IAEA,OAAOG,MAAM;EACf;EAEA,MAAMF,UAAUA,CACdjB,MAAwB,EACxBc,QAAoC,EACH;IACjC,MAAMkB,YAAY,GAAG;MACnB,GAAGhC,MAAM;MACTiC,MAAM,EAAEjC,MAAM,CAACiC,MAAM,IAAI,EAAE;MAC3BC,uBAAuB,EAAE,CAAC,CAACpB;IAC7B,CAAC;IACD,IAAId,MAAM,CAACF,QAAQ,EAAE;MACnB;MACA,MAAMqC,eAAe,GAAG,MAAM,IAAI,CAACtC,gBAAgB,CACjDG,MAAM,CAACF,QAAQ,EACfE,MAAM,CAACoC,aAAa,IAAIpC,MAAM,CAACqC,YAAY,EAC3C;QACElC,KAAK,EAAEH,MAAM,CAACG,KAAK;QACnBO,KAAK,EAAEV,MAAM,CAACU,KAAK;QACnBC,mBAAmB,EAAEX,MAAM,CAACW,mBAAmB;QAC/CC,WAAW,EAAEZ,MAAM,CAACY;MACtB,CACF,CAAC;MACD,IAAI,OAAOuB,eAAe,KAAK,QAAQ,EAAE;QACvCH,YAAY,CAACC,MAAM,GAAGE,eAAe,IAAI,EAAE;MAC7C,CAAC,MAAM;QACLH,YAAY,CAACC,MAAM,GAAGE,eAAe,CAACF,MAAM,IAAI,EAAE;QAClD,IAAI,OAAOE,eAAe,CAACG,WAAW,KAAK,QAAQ,EACjDN,YAAY,CAACM,WAAW,GAAGH,eAAe,CAACG,WAAW;QACxD,IAAIH,eAAe,CAACI,OAAO,EACzBP,YAAY,CAACO,OAAO,GAAGJ,eAAe,CAACI,OAAO;QAChD,IAAI,OAAOJ,eAAe,CAACK,YAAY,KAAK,SAAS,EACnDR,YAAY,CAACQ,YAAY,GAAGL,eAAe,CAACK,YAAY;QAC1D,IAAIL,eAAe,CAACM,gBAAgB,EAClCT,YAAY,CAACS,gBAAgB,GAAGN,eAAe,CAACM,gBAAgB;QAClE,IAAIN,eAAe,CAACO,gBAAgB,EAClCV,YAAY,CAACU,gBAAgB,GAAGP,eAAe,CAACO,gBAAgB;QAClE,IAAIP,eAAe,CAACQ,gBAAgB,EAAE;UACpC,IAAI,CAACX,YAAY,CAACY,IAAI,EAAEZ,YAAY,CAACY,IAAI,GAAG,EAAE;UAC9CZ,YAAY,CAACY,IAAI,CAAChB,IAAI,CAAC,GAAGO,eAAe,CAACQ,gBAAgB,CAAC;QAC7D;MACF;IACF,CAAC,MAAM;MACLX,YAAY,CAACC,MAAM,GAAGjC,MAAM,CAACiC,MAAM,IAAI,EAAE;IAC3C;IAEA,IAAID,YAAY,CAACzB,eAAe,IAAI,CAACyB,YAAY,CAACO,OAAO,EAAE;MACzD,MAAMjC,UAAU,GAAGvC,aAAa,CAACiC,MAAM,CAACO,eAAe,CAAC;MACxD,IAAID,UAAU,EAAE0B,YAAY,CAAC9D,WAAW,GAAGsC,IAAI,CAACC,SAAS,CAACH,UAAU,CAAC;IACvE;IAEA,MAAMuC,SAAS,GAAGC,IAAI,CAACC,GAAG,CAAC,CAAC;IAC5B,IAAIC,cAA6B,GAAG,IAAI;IACxC,MAAMC,UAAU,GAAG,MAAMC,aAAa,CAAC,IAAI,CAACrE,EAAE,CAAC;IAE/C,MAAMsE,eAAe,GAAGrC,QAAQ,GAAIsC,IAAS,IAAK;MAChD,IAAIJ,cAAc,KAAK,IAAI,EAAEA,cAAc,GAAGF,IAAI,CAACC,GAAG,CAAC,CAAC;MACxDjC,QAAQ,CAACsC,IAAI,CAAC;IAChB,CAAC,GAAG/C,SAAS;IAEb,IAAIgD,aAAkB,GACpBF,eAAe,IACf/F,YAAY,CAACG,WAAW,CAACL,cAAc,EAAGM,GAAqB,IAAK;MAClE,MAAM;QAAEoB,SAAS;QAAE0E;MAAY,CAAC,GAAG9F,GAAG;MACtC,IAAIoB,SAAS,KAAK,IAAI,CAACC,EAAE,EAAE;MAC3BsE,eAAe,CAACG,WAAW,CAAC;IAC9B,CAAC,CAAC;IAEJ,IAAI,CAACtB,YAAY,CAACC,MAAM,EAAE,MAAM,IAAIsB,KAAK,CAAC,oBAAoB,CAAC;IAE/D,MAAMC,OAAO,GAAG9G,MAAM,CAACuE,UAAU,CAAC,IAAI,CAACpC,EAAE,EAAEmD,YAAY,CAAC;IACxD,OAAOwB,OAAO,CACXC,IAAI,CAAEC,gBAAgB,IAAK;MAC1B1G,SAAS,CAAC2G,KAAK,CAAC;QACdC,KAAK,EAAE,YAAY;QACnBC,WAAW,EAAGH,gBAAgB,CAASI,OAAO,EAAEC,oBAAoB;QACpEC,cAAc,EAAGN,gBAAgB,CAASI,OAAO,EAAEG,WAAW;QAC9DC,IAAI,EAAElB,cAAc,GAAGA,cAAc,GAAGH,SAAS,GAAG;MACtD,CAAC,EAAEzE,eAAe,EAAE6E,UAAU,CAAC;MAC/BI,aAAa,EAAEc,MAAM,CAAC,CAAC;MACvBd,aAAa,GAAG,IAAI;MACpB,OAAOK,gBAAgB;IACzB,CAAC,CAAC,CACD5F,KAAK,CAAEsG,GAAQ,IAAK;MACnBf,aAAa,EAAEc,MAAM,CAAC,CAAC;MACvBd,aAAa,GAAG,IAAI;MACpB,MAAMe,GAAG;IACX,CAAC,CAAC;EACN;EAEAC,cAAcA,CAAA,EAAkB;IAC9B,OAAO3H,MAAM,CAAC2H,cAAc,CAAC,IAAI,CAACxF,EAAE,CAAC;EACvC;EAEAyF,QAAQA,CAAC1G,IAAY,EAAiC;IACpD,OAAOlB,MAAM,CAAC4H,QAAQ,CAAC,IAAI,CAACzF,EAAE,EAAEjB,IAAI,CAAC;EACvC;EAEA2G,UAAUA,CAACC,MAAgB,EAAmB;IAC5C,OAAO9H,MAAM,CAAC6H,UAAU,CAAC,IAAI,CAAC1F,EAAE,EAAE2F,MAAM,CAAC;EAC3C;EAEA,MAAMC,SAASA,CACb7G,IAAY,EACZoC,MAAwB,EACQ;IAChC,MAAM6C,SAAS,GAAGC,IAAI,CAACC,GAAG,CAAC,CAAC;IAC5B,MAAM2B,eAAe,GAAG,MAAMhI,MAAM,CAAC+H,SAAS,CAAC,IAAI,CAAC5F,EAAE,EAAEjB,IAAI,EAAEoC,MAAM,IAAI,CAAC,CAAC,CAAC;IAC3E,MAAM2E,SAAS,GAAG7B,IAAI,CAACC,GAAG,CAAC,CAAC,GAAGF,SAAS;IACxC,MAAMI,UAAU,GAAG,MAAMC,aAAa,CAAC,IAAI,CAACrE,EAAE,CAAC;IAC/C7B,SAAS,CAAC2G,KAAK,CAAC;MACdC,KAAK,EAAE,WAAW;MAClBgB,IAAI,EAAE,OAAO;MACbC,cAAc,EAAEF;IAClB,CAAC,EAAEvG,eAAe,EAAE6E,UAAU,CAAC;IAC/B,OAAOyB,eAAe;EAExB;EAEA,MAAMI,KAAKA,CACTC,EAAU,EACVC,EAAU,EACVC,EAAU,EACVC,EAAU,EACY;IACtB,MAAM/D,MAAM,GAAG,MAAMzE,MAAM,CAACoI,KAAK,CAAC,IAAI,CAACjG,EAAE,EAAEkG,EAAE,EAAEC,EAAE,EAAEC,EAAE,EAAEC,EAAE,CAAC;IAC1D,MAAM,CAACC,SAAS,EAAEC,SAAS,EAAEC,YAAY,EAAEC,KAAK,EAAEC,KAAK,EAAEC,KAAK,EAAEC,KAAK,CAAC,GACpEjF,IAAI,CAACkF,KAAK,CAACvE,MAAM,CAAC;IACpB,OAAO;MACLgE,SAAS;MACTC,SAAS;MACTC,YAAY;MACZC,KAAK;MACLC,KAAK;MACLC,KAAK;MACLC;IACF,CAAC;EACH;EAEA,MAAME,iBAAiBA,CACrBC,QAAkD,EACnC;IACf,IAAIC,YAAsD,GAAG,EAAE;IAC/D,IAAID,QAAQ,EACVC,YAAY,GAAGD,QAAQ,CAACE,GAAG,CAAEC,CAAC,KAAM;MAClC/G,IAAI,EAAE+G,CAAC,CAAC/G,IAAI,CAACgH,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;MACrCC,MAAM,EAAEF,CAAC,CAACE;IACZ,CAAC,CAAC,CAAC;IACL,OAAOvJ,MAAM,CAACiJ,iBAAiB,CAAC,IAAI,CAAC9G,EAAE,EAAEgH,YAAY,CAAC;EACxD;EAEA,MAAMK,kBAAkBA,CAAA,EAAkB;IACxC,OAAOxJ,MAAM,CAACwJ,kBAAkB,CAAC,IAAI,CAACrH,EAAE,CAAC;EAC3C;EAEA,MAAMsH,qBAAqBA,CAAA,EAEzB;IACA,OAAOzJ,MAAM,CAACyJ,qBAAqB,CAAC,IAAI,CAACtH,EAAE,CAAC;EAC9C;EAEA,MAAMuH,OAAOA,CAAA,EAAkB;IAC7B,OAAO1J,MAAM,CAAC2J,cAAc,CAAC,IAAI,CAACxH,EAAE,CAAC;EACvC;EAEA,MAAMyH,MAAMA,CAAA,EAAkB;IAC5B;IACA,OAAQ5J,MAAM,CAAS4J,MAAM,CAAC,IAAI,CAACzH,EAAE,CAAC;EACxC;AACF;AAEA,OAAO,eAAehB,eAAeA,CAAC0I,OAAgB,EAAiB;EACrE,OAAO7J,MAAM,CAACmB,eAAe,CAAC0I,OAAO,CAAC;AACxC;AAEA,OAAO,SAASC,oBAAoBA,CAClC9I,QAA+C,EACvB;EACxBJ,YAAY,CAACsE,IAAI,CAAClE,QAAQ,CAAC;EAC3B,OAAO;IACLyG,MAAM,EAAEA,CAAA,KAAM;MACZ7G,YAAY,CAACmJ,MAAM,CAACnJ,YAAY,CAACoJ,OAAO,CAAChJ,QAAQ,CAAC,EAAE,CAAC,CAAC;IACxD;EACF,CAAC;AACH;AAEA,OAAO,eAAeiJ,eAAeA,CAACC,KAAa,EAAiB;EAClE,OAAOlK,MAAM,CAACiK,eAAe,CAACC,KAAK,CAAC;AACtC;AAEA,IAAIC,gBAAgB,GAAG,CAAC;AACxB,MAAMC,eAAe,GAAGA,CAAA,KACtBC,OAAO,CAACC,GAAG,CAACC,QAAQ,KAAK,MAAM,GAAG,CAAC,GAAGC,IAAI,CAACC,KAAK,CAACD,IAAI,CAACE,MAAM,CAAC,CAAC,GAAG,MAAM,CAAC;AAE1E,MAAMC,aAAa,GAAG;AACpB;AACA,uBAAuB,EACvB,2BAA2B,EAC3B,uBAAuB,EACvB,uBAAuB,CACxB;AACD,OAAO,eAAeC,kBAAkBA,CAAC/I,KAAa,EAAmB;EACvE,IAAIS,IAAI,GAAGT,KAAK;EAChB,IAAIS,IAAI,CAACC,UAAU,CAAC,SAAS,CAAC,EAAED,IAAI,GAAGA,IAAI,CAACE,KAAK,CAAC,CAAC,CAAC;EACpD,OAAOxC,MAAM,CAAC6K,SAAS,CAACvI,IAAI,EAAEqI,aAAa,CAAC;AAC9C;AAEA,MAAMG,WAAW,GAAG;EAClB;EACAC,IAAI,EAAE,CAAC;EACPC,IAAI,EAAE,CAAC;EACPC,GAAG,EAAE,CAAC;EACNC,IAAI,EAAE,CAAC;EACPC,IAAI,EAAE;AACR,CAAC;AAED,OAAO,eAAeC,SAASA,CAC7B;EACEvJ,KAAK;EACLwJ,cAAc,EAAEC,YAAY;EAC5BC,YAAY,EAAEC,WAAW;EACzBC,IAAI;EACJC,SAAS,EAAExC,QAAQ;EACnB,GAAGyC;AACU,CAAC,EAChBC,UAAuC,EAChB;EACvB,IAAItJ,IAAI,GAAGT,KAAK;EAChB,IAAIS,IAAI,CAACC,UAAU,CAAC,SAAS,CAAC,EAAED,IAAI,GAAGA,IAAI,CAACE,KAAK,CAAC,CAAC,CAAC;EAEpD,IAAIqJ,QAAQ,GAAGJ,IAAI;EACnB,IAAII,QAAQ,EAAEtJ,UAAU,CAAC,SAAS,CAAC,EAAEsJ,QAAQ,GAAGA,QAAQ,CAACrJ,KAAK,CAAC,CAAC,CAAC;EAEjE,IAAI2G,YAAsD,GAAG,EAAE;EAC/D,IAAID,QAAQ,EACVC,YAAY,GAAGD,QAAQ,CAACE,GAAG,CAAEC,CAAC,KAAM;IAClC/G,IAAI,EAAE+G,CAAC,CAAC/G,IAAI,CAACgH,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;IACrCC,MAAM,EAAEF,CAAC,CAACE;EACZ,CAAC,CAAC,CAAC;EAEL7H,eAAe,CAACC,YAAY,GAAGgK,IAAI,CAAChK,YAAY,IAAI,IAAI;EACxDD,eAAe,CAACE,KAAK,GAAG+J,IAAI,CAAC/J,KAAK,IAAI,IAAI;EAC1CF,eAAe,CAACG,KAAK,GAAGA,KAAK;EAE7B,MAAMK,SAAS,GAAGiI,gBAAgB,GAAGC,eAAe,CAAC,CAAC;EACtDD,gBAAgB,IAAI,CAAC;EAErB,IAAI2B,sBAA2B,GAAG,IAAI;EACtC,IAAIF,UAAU,EAAE;IACdE,sBAAsB,GAAGpL,YAAY,CAACG,WAAW,CAC/CN,8BAA8B,EAC7BO,GAA4C,IAAK;MAChD,IAAIA,GAAG,CAACoB,SAAS,KAAKA,SAAS,EAAE;MACjC0J,UAAU,CAAC9K,GAAG,CAACiL,QAAQ,CAAC;IAC1B,CACF,CAAC;EACH;EAEA,MAAMC,QAAQ,GAAGlB,WAAW,CAACU,WAAW,CAA6B;EACrE,MAAM;IACJzJ,GAAG;IACHC,WAAW;IACXH,KAAK,EAAEoK,YAAY;IACnBC;EACF,CAAC,GAAG,MAAMlM,MAAM,CAACmM,WAAW,CAACjK,SAAS,EAAE;IACtCL,KAAK,EAAES,IAAI;IACX+I,cAAc,EAAE,CAAC,CAACC,YAAY;IAC9Bc,qBAAqB,EAAE,CAAC,CAACR,UAAU;IACnCL,YAAY,EAAES,QAAQ;IACtBP,IAAI,EAAEI,QAAQ;IACdH,SAAS,EAAEvC,YAAY;IACvB,GAAGwC;EACL,CAAC,CAAC,CAACvK,KAAK,CAAEsG,GAAQ,IAAK;IACrBoE,sBAAsB,EAAErE,MAAM,CAAC,CAAC;IAChC,MAAMC,GAAG;EACX,CAAC,CAAC;EACFoE,sBAAsB,EAAErE,MAAM,CAAC,CAAC;EAChC,OAAO,IAAI3F,YAAY,CAAC;IACtBI,SAAS;IACTH,GAAG;IACHC,WAAW;IACXH,KAAK,EAAEoK,YAAY;IACnBC;EACF,CAAC,CAAC;AACJ;AAEA,OAAO,eAAeG,eAAeA,CAAA,EAAkB;EACrD,OAAOrM,MAAM,CAACsM,kBAAkB,CAAC,CAAC;AACpC;AAEA,OAAO,MAAMH,WAAW,GAAG,MAAO7I,MAA2B,IAAK;EAChE,OAAO,MAAMtD,MAAM,CAACmM,WAAW,CAAChC,gBAAgB,EAAE,EAAE7G,MAAM,CAAC;AAC7D,CAAC;AAED,OAAO,MAAMiJ,cAAc,GAAG,MAAAA,CAAOrK,SAAiB,EAAEsK,UAAkB,EAAEC,MAAe,GAAG,KAAK,KAAK;EACtG,OAAO,MAAMzM,MAAM,CAACuM,cAAc,CAACrK,SAAS,EAAEsK,UAAU,EAAEC,MAAM,CAAC;AACnE,CAAC;AAED,OAAO,MAAMC,mBAAmB,GAAG,MAAOxK,SAAiB,IAAK;EAC9D,OAAO,MAAMlC,MAAM,CAAC0M,mBAAmB,CAACxK,SAAS,CAAC;AACpD,CAAC;AAED,OAAO,MAAMyK,yBAAyB,GAAG,MAAOzK,SAAiB,IAAK;EACpE,OAAO,MAAMlC,MAAM,CAAC2M,yBAAyB,CAACzK,SAAS,CAAC;AAC1D,CAAC;AAED,OAAO,MAAM0K,wBAAwB,GAAG,MAAO1K,SAAiB,IAAK;EACnE,OAAO,MAAMlC,MAAM,CAAC4M,wBAAwB,CAAC1K,SAAS,CAAC;AACzD,CAAC;AAED,OAAO,MAAM2K,iBAAiB,GAAG,MAAO3K,SAAiB,IAAK;EAC5D,OAAO,MAAMlC,MAAM,CAAC6M,iBAAiB,CAAC3K,SAAS,CAAC;AAClD,CAAC;AAED,OAAO,MAAM4K,oBAAoB,GAAG,MAAAA,CAAO5K,SAAiB,EAAEqD,MAAc,EAAEwH,UAAoB,EAAEzJ,MAA8B,KAAsC;EACtK,MAAMmB,MAAM,GAAG,MAAMzE,MAAM,CAAC8M,oBAAoB,CAAC5K,SAAS,EAAEqD,MAAM,EAAEwH,UAAU,EAAEzJ,MAAM,CAAC;EAEvF,MAAMiD,UAAU,GAAG,MAAMC,aAAa,CAACtE,SAAS,CAAC;EAEjD5B,SAAS,CAAC2G,KAAK,CAAC;IACdC,KAAK,EAAE,YAAY;IACnBC,WAAW,EAAG1C,MAAM,CAAS2C,OAAO,EAAEC,oBAAoB;IAC1DC,cAAc,EAAG7C,MAAM,CAAS2C,OAAO,EAAEG,WAAW;IACpDyF,UAAU,EAAED,UAAU,EAAEE;EAC1B,CAAC,EAAEvL,eAAe,EAAE6E,UAAU,CAAC;EAE/B,OAAO9B,MAAM;AACf,CAAC;AAED,OAAO,MAAMyI,WAAW,GAAG,MAAAA,CAAOhL,SAAiB,EAAEiL,gBAAwB,KAAK;EAChF,OAAO,MAAMnN,MAAM,CAACkN,WAAW,CAAChL,SAAS,EAAEiL,gBAAgB,CAAC;AAC9D,CAAC;AAED,OAAO,MAAMC,gBAAgB,GAAG,MAAOlL,SAAiB,IAAK;EAC3D,OAAO,MAAMlC,MAAM,CAACoN,gBAAgB,CAAClL,SAAS,CAAC;AACjD,CAAC;AAED,OAAO,MAAMmL,UAAU,GAAG,MAAOnL,SAAiB,IAA6B;EAC7E,OAAO,MAAMlC,MAAM,CAACqN,UAAU,CAACnL,SAAS,CAAC;AAC3C,CAAC;AAED,OAAO,MAAMoL,2BAA2B,GAAG,MAAAA,CAAOpL,SAAiB,EAAEqL,cAAsB,EAAEC,WAAmB,KAA2C;EACzJ,OAAO,MAAMxN,MAAM,CAACsN,2BAA2B,CAACpL,SAAS,EAAEqL,cAAc,EAAEC,WAAW,CAAC;AACzF,CAAC;AAED,OAAO,MAAMC,6BAA6B,GAAG,MAAAA,CAAOvL,SAAiB,EAAEsL,WAAmB,KAAuC;EAC/H,OAAO,MAAMxN,MAAM,CAACyN,6BAA6B,CAACvL,SAAS,EAAEsL,WAAW,CAAC;AAC3E,CAAC;AAED,OAAO,MAAME,iBAAiB,GAAG,MAAAA,CAAOxL,SAAiB,EAAE4F,MAAgB,KAAuC;EAChH,OAAO,MAAM9H,MAAM,CAAC0N,iBAAiB,CAACxL,SAAS,EAAE4F,MAAM,CAAC;AAC1D,CAAC;AAED,OAAO,MAAM6F,cAAc,GAAG,MAAOzL,SAAiB,IAAK;EACzD,OAAO,MAAMlC,MAAM,CAAC2N,cAAc,CAACzL,SAAS,CAAC;AAC/C,CAAC;AAED,OAAO,MAAM0F,QAAQ,GAAG,MAAAA,CAAO1F,SAAiB,EAAEhB,IAAY,EAAE6L,UAAqB,KAAoC;EACvH,IAAIA,UAAU,IAAIA,UAAU,CAACE,MAAM,GAAG,CAAC,EAAE;IACvC,OAAO,MAAMjN,MAAM,CAAC4H,QAAQ,CAAC1F,SAAS,EAAEhB,IAAI,EAAE6L,UAAU,CAAC;EAC3D,CAAC,MAAM;IACL,OAAO,MAAM/M,MAAM,CAAC4H,QAAQ,CAAC1F,SAAS,EAAEhB,IAAI,CAAC;EAC/C;AACF,CAAC;AAED,OAAO,MAAMsF,aAAa,GAAG,MAAOtE,SAAiB,IAAgC;EACnF,OAAO,MAAMlC,MAAM,CAACwG,aAAa,CAACtE,SAAS,CAAC;AAC9C,CAAC;AAED,SAAS0L,QAAQ,QAAQ,MAAM;AAC/B,SAASC,SAAS,QAAQ,OAAO;AACjC,SAASC,SAAS,QAAQ,OAAO","ignoreList":[]}
|
package/lib/module/lm.js
ADDED
|
@@ -0,0 +1,101 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
|
|
3
|
+
import { initLlama } from './index';
|
|
4
|
+
import { Telemetry } from './telemetry';
|
|
5
|
+
import { setCactusToken, getVertexAIEmbedding } from './remote';
|
|
6
|
+
export class CactusLM {
|
|
7
|
+
constructor(context) {
|
|
8
|
+
this.context = context;
|
|
9
|
+
}
|
|
10
|
+
static async init(params, onProgress, cactusToken) {
|
|
11
|
+
if (cactusToken) {
|
|
12
|
+
setCactusToken(cactusToken);
|
|
13
|
+
}
|
|
14
|
+
const configs = [params, {
|
|
15
|
+
...params,
|
|
16
|
+
n_gpu_layers: 0
|
|
17
|
+
}];
|
|
18
|
+
for (const config of configs) {
|
|
19
|
+
try {
|
|
20
|
+
const context = await initLlama(config, onProgress);
|
|
21
|
+
return {
|
|
22
|
+
lm: new CactusLM(context),
|
|
23
|
+
error: null
|
|
24
|
+
};
|
|
25
|
+
} catch (e) {
|
|
26
|
+
Telemetry.error(e, {
|
|
27
|
+
n_gpu_layers: config.n_gpu_layers ?? null,
|
|
28
|
+
n_ctx: config.n_ctx ?? null,
|
|
29
|
+
model: config.model ?? null
|
|
30
|
+
});
|
|
31
|
+
if (configs.indexOf(config) === configs.length - 1) {
|
|
32
|
+
return {
|
|
33
|
+
lm: null,
|
|
34
|
+
error: e
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
return {
|
|
40
|
+
lm: null,
|
|
41
|
+
error: new Error('Failed to initialize CactusLM')
|
|
42
|
+
};
|
|
43
|
+
}
|
|
44
|
+
async completion(messages, params = {}, callback) {
|
|
45
|
+
return await this.context.completion({
|
|
46
|
+
messages,
|
|
47
|
+
...params
|
|
48
|
+
}, callback);
|
|
49
|
+
}
|
|
50
|
+
async embedding(text, params, mode = 'local') {
|
|
51
|
+
let result;
|
|
52
|
+
let lastError = null;
|
|
53
|
+
if (mode === 'remote') {
|
|
54
|
+
result = await this._handleRemoteEmbedding(text);
|
|
55
|
+
} else if (mode === 'local') {
|
|
56
|
+
result = await this._handleLocalEmbedding(text, params);
|
|
57
|
+
} else if (mode === 'localfirst') {
|
|
58
|
+
try {
|
|
59
|
+
result = await this._handleLocalEmbedding(text, params);
|
|
60
|
+
} catch (e) {
|
|
61
|
+
lastError = e;
|
|
62
|
+
try {
|
|
63
|
+
result = await this._handleRemoteEmbedding(text);
|
|
64
|
+
} catch (remoteError) {
|
|
65
|
+
throw lastError;
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
} else if (mode === 'remotefirst') {
|
|
69
|
+
try {
|
|
70
|
+
result = await this._handleRemoteEmbedding(text);
|
|
71
|
+
} catch (e) {
|
|
72
|
+
lastError = e;
|
|
73
|
+
try {
|
|
74
|
+
result = await this._handleLocalEmbedding(text, params);
|
|
75
|
+
} catch (localError) {
|
|
76
|
+
throw lastError;
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
} else {
|
|
80
|
+
throw new Error('Invalid mode: ' + mode + '. Must be "local", "remote", "localfirst", or "remotefirst"');
|
|
81
|
+
}
|
|
82
|
+
return result;
|
|
83
|
+
}
|
|
84
|
+
async _handleLocalEmbedding(text, params) {
|
|
85
|
+
return this.context.embedding(text, params);
|
|
86
|
+
}
|
|
87
|
+
async _handleRemoteEmbedding(text) {
|
|
88
|
+
const embeddingValues = await getVertexAIEmbedding(text);
|
|
89
|
+
return {
|
|
90
|
+
embedding: embeddingValues
|
|
91
|
+
};
|
|
92
|
+
}
|
|
93
|
+
async rewind() {
|
|
94
|
+
// @ts-ignore
|
|
95
|
+
return this.context?.rewind();
|
|
96
|
+
}
|
|
97
|
+
async release() {
|
|
98
|
+
return this.context.release();
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
//# sourceMappingURL=lm.js.map
|
package/lib/module/lm.js.map
CHANGED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"names":["initLlama","Telemetry","setCactusToken","getVertexAIEmbedding","CactusLM","constructor","context","init","params","onProgress","cactusToken","configs","n_gpu_layers","config","lm","error","e","n_ctx","model","indexOf","length","Error","completion","messages","callback","embedding","text","mode","result","lastError","_handleRemoteEmbedding","_handleLocalEmbedding","remoteError","localError","embeddingValues","rewind","release"],"sourceRoot":"../../src","sources":["lm.ts"],"mappings":";;AAAA,SAASA,SAAS,QAAsB,SAAS;AASjD,SAASC,SAAS,QAAQ,aAAa;AACvC,SAASC,cAAc,EAAEC,oBAAoB,QAAQ,UAAU;AAO/D,OAAO,MAAMC,QAAQ,CAAC;EAGZC,WAAWA,CAACC,OAAqB,EAAE;IACzC,IAAI,CAACA,OAAO,GAAGA,OAAO;EACxB;EAEA,aAAaC,IAAIA,CACfC,MAAqB,EACrBC,UAAuC,EACvCC,WAAoB,EACK;IACzB,IAAIA,WAAW,EAAE;MACfR,cAAc,CAACQ,WAAW,CAAC;IAC7B;IAEA,MAAMC,OAAO,GAAG,CACdH,MAAM,EACN;MAAE,GAAGA,MAAM;MAAEI,YAAY,EAAE;IAAE,CAAC,CAC/B;IAED,KAAK,MAAMC,MAAM,IAAIF,OAAO,EAAE;MAC5B,IAAI;QACF,MAAML,OAAO,GAAG,MAAMN,SAAS,CAACa,MAAM,EAAEJ,UAAU,CAAC;QACnD,OAAO;UAAEK,EAAE,EAAE,IAAIV,QAAQ,CAACE,OAAO,CAAC;UAAES,KAAK,EAAE;QAAK,CAAC;MACnD,CAAC,CAAC,OAAOC,CAAC,EAAE;QACVf,SAAS,CAACc,KAAK,CAACC,CAAC,EAAW;UAC1BJ,YAAY,EAAEC,MAAM,CAACD,YAAY,IAAI,IAAI;UACzCK,KAAK,EAAEJ,MAAM,CAACI,KAAK,IAAI,IAAI;UAC3BC,KAAK,EAAEL,MAAM,CAACK,KAAK,IAAI;QACzB,CAAC,CAAC;QACF,IAAIP,OAAO,CAACQ,OAAO,CAACN,MAAM,CAAC,KAAKF,OAAO,CAACS,MAAM,GAAG,CAAC,EAAE;UAClD,OAAO;YAAEN,EAAE,EAAE,IAAI;YAAEC,KAAK,EAAEC;UAAW,CAAC;QACxC;MACF;IACF;IACA,OAAO;MAAEF,EAAE,EAAE,IAAI;MAAEC,KAAK,EAAE,IAAIM,KAAK,CAAC,+BAA+B;IAAE,CAAC;EACxE;EAEA,MAAMC,UAAUA,CACdC,QAAsC,EACtCf,MAAwB,GAAG,CAAC,CAAC,EAC7BgB,QAA8B,EACG;IACjC,OAAO,MAAM,IAAI,CAAClB,OAAO,CAACgB,UAAU,CAAC;MAAEC,QAAQ;MAAE,GAAGf;IAAO,CAAC,EAAEgB,QAAQ,CAAC;EACzE;EAEA,MAAMC,SAASA,CACbC,IAAY,EACZlB,MAAwB,EACxBmB,IAAY,GAAG,OAAO,EACU;IAChC,IAAIC,MAA6B;IACjC,IAAIC,SAAuB,GAAG,IAAI;IAElC,IAAIF,IAAI,KAAK,QAAQ,EAAE;MACrBC,MAAM,GAAG,MAAM,IAAI,CAACE,sBAAsB,CAACJ,IAAI,CAAC;IAClD,CAAC,MAAM,IAAIC,IAAI,KAAK,OAAO,EAAE;MAC3BC,MAAM,GAAG,MAAM,IAAI,CAACG,qBAAqB,CAACL,IAAI,EAAElB,MAAM,CAAC;IACzD,CAAC,MAAM,IAAImB,IAAI,KAAK,YAAY,EAAE;MAChC,IAAI;QACFC,MAAM,GAAG,MAAM,IAAI,CAACG,qBAAqB,CAACL,IAAI,EAAElB,MAAM,CAAC;MACzD,CAAC,CAAC,OAAOQ,CAAC,EAAE;QACVa,SAAS,GAAGb,CAAU;QACtB,IAAI;UACFY,MAAM,GAAG,MAAM,IAAI,CAACE,sBAAsB,CAACJ,IAAI,CAAC;QAClD,CAAC,CAAC,OAAOM,WAAW,EAAE;UACpB,MAAMH,SAAS;QACjB;MACF;IACF,CAAC,MAAM,IAAIF,IAAI,KAAK,aAAa,EAAE;MACjC,IAAI;QACFC,MAAM,GAAG,MAAM,IAAI,CAACE,sBAAsB,CAACJ,IAAI,CAAC;MAClD,CAAC,CAAC,OAAOV,CAAC,EAAE;QACVa,SAAS,GAAGb,CAAU;QACtB,IAAI;UACFY,MAAM,GAAG,MAAM,IAAI,CAACG,qBAAqB,CAACL,IAAI,EAAElB,MAAM,CAAC;QACzD,CAAC,CAAC,OAAOyB,UAAU,EAAE;UACnB,MAAMJ,SAAS;QACjB;MACF;IACF,CAAC,MAAM;MACL,MAAM,IAAIR,KAAK,CAAC,gBAAgB,GAAGM,IAAI,GAAG,6DAA6D,CAAC;IAC1G;IACA,OAAOC,MAAM;EACf;EAEA,MAAcG,qBAAqBA,CAACL,IAAY,EAAElB,MAAwB,EAAkC;IAC1G,OAAO,IAAI,CAACF,OAAO,CAACmB,SAAS,CAACC,IAAI,EAAElB,MAAM,CAAC;EAC7C;EAEA,MAAcsB,sBAAsBA,CAACJ,IAAY,EAAkC;IACjF,MAAMQ,eAAe,GAAG,MAAM/B,oBAAoB,CAACuB,IAAI,CAAC;IACxD,OAAO;MACLD,SAAS,EAAES;IACb,CAAC;EACH;EAEA,MAAMC,MAAMA,CAAA,EAAkB;IAC5B;IACA,OAAO,IAAI,CAAC7B,OAAO,EAAE6B,MAAM,CAAC,CAAC;EAC/B;EAEA,MAAMC,OAAOA,CAAA,EAAkB;IAC7B,OAAO,IAAI,CAAC9B,OAAO,CAAC8B,OAAO,CAAC,CAAC;EAC/B;AACF","ignoreList":[]}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"names":["PROJECT_ID"],"sourceRoot":"../../src","sources":["projectId.ts"],"mappings":";;AAAA,OAAO,MAAMA,UAAU,GAAG,SAAS","ignoreList":[]}
|