cactus-react-native 0.1.3 → 0.1.4
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 +5 -3
- 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/chat.js +37 -0
- package/lib/commonjs/grammar.js +560 -0
- package/lib/commonjs/index.js +459 -0
- package/lib/commonjs/index.js.map +1 -1
- package/lib/commonjs/lm.js +72 -0
- package/lib/commonjs/lm.js.map +1 -1
- package/lib/commonjs/telemetry.js +97 -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 +83 -0
- package/lib/commonjs/vlm.js.map +1 -0
- package/lib/module/NativeCactus.js +8 -0
- package/lib/module/chat.js +33 -0
- package/lib/module/grammar.js +553 -0
- package/lib/module/index.js +392 -0
- package/lib/module/index.js.map +1 -1
- package/lib/module/lm.js +67 -0
- package/lib/module/lm.js.map +1 -0
- package/lib/module/telemetry.js +92 -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 +78 -0
- package/lib/module/vlm.js.map +1 -1
- package/lib/typescript/index.d.ts.map +1 -1
- package/lib/typescript/lm.d.ts +9 -33
- package/lib/typescript/lm.d.ts.map +1 -1
- package/lib/typescript/telemetry.d.ts +21 -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 +12 -34
- package/lib/typescript/vlm.d.ts.map +1 -1
- package/package.json +1 -1
- package/src/index.ts +64 -41
- package/src/lm.ts +45 -5
- package/src/telemetry.ts +123 -0
- package/src/tools.ts +17 -58
- package/src/vlm.ts +50 -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,97 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
|
|
3
|
+
Object.defineProperty(exports, "__esModule", {
|
|
4
|
+
value: true
|
|
5
|
+
});
|
|
6
|
+
exports.Telemetry = void 0;
|
|
7
|
+
var _reactNative = require("react-native");
|
|
8
|
+
// Import package.json to get version
|
|
9
|
+
const packageJson = require('../package.json');
|
|
10
|
+
class Telemetry {
|
|
11
|
+
static instance = null;
|
|
12
|
+
constructor(config) {
|
|
13
|
+
this.config = {
|
|
14
|
+
table: 'telemetry',
|
|
15
|
+
...config
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
static getFilename(path) {
|
|
19
|
+
try {
|
|
20
|
+
return path.split('/').pop() || path.split('\\').pop() || 'unknown';
|
|
21
|
+
} catch {
|
|
22
|
+
return 'unknown';
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
static autoInit() {
|
|
26
|
+
if (!Telemetry.instance) {
|
|
27
|
+
Telemetry.instance = new Telemetry({
|
|
28
|
+
supabaseUrl: 'https://vlqqczxwyaodtcdmdmlw.supabase.co',
|
|
29
|
+
supabaseKey: 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6InZscXFjenh3eWFvZHRjZG1kbWx3Iiwicm9sZSI6ImFub24iLCJpYXQiOjE3NTE1MTg2MzIsImV4cCI6MjA2NzA5NDYzMn0.nBzqGuK9j6RZ6mOPWU2boAC_5H9XDs-fPpo5P3WZYbI' // Anon!
|
|
30
|
+
});
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
static init(config) {
|
|
34
|
+
if (!Telemetry.instance) {
|
|
35
|
+
Telemetry.instance = new Telemetry(config);
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
static track(payload, options) {
|
|
39
|
+
if (!Telemetry.instance) {
|
|
40
|
+
Telemetry.autoInit();
|
|
41
|
+
}
|
|
42
|
+
Telemetry.instance.trackInternal(payload, options);
|
|
43
|
+
}
|
|
44
|
+
static error(error, options) {
|
|
45
|
+
if (!Telemetry.instance) {
|
|
46
|
+
Telemetry.autoInit();
|
|
47
|
+
}
|
|
48
|
+
Telemetry.instance.errorInternal(error, options);
|
|
49
|
+
}
|
|
50
|
+
trackInternal(payload, options) {
|
|
51
|
+
const record = {
|
|
52
|
+
os: _reactNative.Platform.OS === 'ios' ? 'iOS' : 'Android',
|
|
53
|
+
os_version: _reactNative.Platform.Version.toString(),
|
|
54
|
+
framework: 'react-native',
|
|
55
|
+
framework_version: packageJson.version,
|
|
56
|
+
telemetry_payload: payload,
|
|
57
|
+
timestamp: new Date().toISOString(),
|
|
58
|
+
model_filename: Telemetry.getFilename(options.model),
|
|
59
|
+
n_ctx: options.n_ctx,
|
|
60
|
+
n_gpu_layers: options.n_gpu_layers
|
|
61
|
+
};
|
|
62
|
+
this.sendRecord(record).catch(() => {});
|
|
63
|
+
}
|
|
64
|
+
errorInternal(error, options) {
|
|
65
|
+
const errorPayload = {
|
|
66
|
+
message: error.message,
|
|
67
|
+
stack: error.stack,
|
|
68
|
+
name: error.name
|
|
69
|
+
};
|
|
70
|
+
const record = {
|
|
71
|
+
os: _reactNative.Platform.OS === 'ios' ? 'iOS' : 'Android',
|
|
72
|
+
os_version: _reactNative.Platform.Version.toString(),
|
|
73
|
+
framework: 'react-native',
|
|
74
|
+
framework_version: packageJson.version,
|
|
75
|
+
error_payload: errorPayload,
|
|
76
|
+
timestamp: new Date().toISOString(),
|
|
77
|
+
model_filename: Telemetry.getFilename(options.model),
|
|
78
|
+
n_ctx: options.n_ctx,
|
|
79
|
+
n_gpu_layers: options.n_gpu_layers
|
|
80
|
+
};
|
|
81
|
+
this.sendRecord(record).catch(() => {});
|
|
82
|
+
}
|
|
83
|
+
async sendRecord(record) {
|
|
84
|
+
await globalThis.fetch(`${this.config.supabaseUrl}/rest/v1/${this.config.table}`, {
|
|
85
|
+
method: 'POST',
|
|
86
|
+
headers: {
|
|
87
|
+
'apikey': this.config.supabaseKey,
|
|
88
|
+
'Authorization': `Bearer ${this.config.supabaseKey}`,
|
|
89
|
+
'Content-Type': 'application/json',
|
|
90
|
+
'Prefer': 'return=minimal'
|
|
91
|
+
},
|
|
92
|
+
body: JSON.stringify([record])
|
|
93
|
+
});
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
exports.Telemetry = Telemetry;
|
|
97
|
+
//# sourceMappingURL=telemetry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"names":["_reactNative","require","packageJson","Telemetry","instance","constructor","config","table","getFilename","path","split","pop","autoInit","supabaseUrl","supabaseKey","init","track","payload","options","trackInternal","error","errorInternal","record","os","Platform","OS","os_version","Version","toString","framework","framework_version","version","telemetry_payload","timestamp","Date","toISOString","model_filename","model","n_ctx","n_gpu_layers","sendRecord","catch","errorPayload","message","stack","name","error_payload","globalThis","fetch","method","headers","body","JSON","stringify","exports"],"sourceRoot":"../../src","sources":["telemetry.ts"],"mappings":";;;;;;AAAA,IAAAA,YAAA,GAAAC,OAAA;AAEA;AACA,MAAMC,WAAW,GAAGD,OAAO,CAAC,iBAAiB,CAAC;AAqBvC,MAAME,SAAS,CAAC;EACrB,OAAeC,QAAQ,GAAqB,IAAI;EAGxCC,WAAWA,CAACC,MAAuB,EAAE;IAC3C,IAAI,CAACA,MAAM,GAAG;MACZC,KAAK,EAAE,WAAW;MAClB,GAAGD;IACL,CAAC;EACH;EAEA,OAAeE,WAAWA,CAACC,IAAY,EAAU;IAC/C,IAAI;MACF,OAAOA,IAAI,CAACC,KAAK,CAAC,GAAG,CAAC,CAACC,GAAG,CAAC,CAAC,IAAIF,IAAI,CAACC,KAAK,CAAC,IAAI,CAAC,CAACC,GAAG,CAAC,CAAC,IAAI,SAAS;IACrE,CAAC,CAAC,MAAM;MACN,OAAO,SAAS;IAClB;EACF;EAEA,OAAOC,QAAQA,CAAA,EAAS;IACtB,IAAI,CAACT,SAAS,CAACC,QAAQ,EAAE;MACvBD,SAAS,CAACC,QAAQ,GAAG,IAAID,SAAS,CAAC;QACjCU,WAAW,EAAE,0CAA0C;QACvDC,WAAW,EAAE,kNAAkN,CAAE;MACnO,CAAC,CAAC;IACJ;EACF;EAEA,OAAOC,IAAIA,CAACT,MAAuB,EAAQ;IACzC,IAAI,CAACH,SAAS,CAACC,QAAQ,EAAE;MACvBD,SAAS,CAACC,QAAQ,GAAG,IAAID,SAAS,CAACG,MAAM,CAAC;IAC5C;EACF;EAEA,OAAOU,KAAKA,CAACC,OAA4B,EAAEC,OAAsB,EAAQ;IACvE,IAAI,CAACf,SAAS,CAACC,QAAQ,EAAE;MACvBD,SAAS,CAACS,QAAQ,CAAC,CAAC;IACtB;IACAT,SAAS,CAACC,QAAQ,CAAEe,aAAa,CAACF,OAAO,EAAEC,OAAO,CAAC;EACrD;EAEA,OAAOE,KAAKA,CAACA,KAAY,EAAEF,OAAsB,EAAQ;IACvD,IAAI,CAACf,SAAS,CAACC,QAAQ,EAAE;MACvBD,SAAS,CAACS,QAAQ,CAAC,CAAC;IACtB;IACAT,SAAS,CAACC,QAAQ,CAAEiB,aAAa,CAACD,KAAK,EAAEF,OAAO,CAAC;EACnD;EAEQC,aAAaA,CAACF,OAA4B,EAAEC,OAAsB,EAAQ;IAChF,MAAMI,MAAuB,GAAG;MAC9BC,EAAE,EAAEC,qBAAQ,CAACC,EAAE,KAAK,KAAK,GAAG,KAAK,GAAG,SAAS;MAC7CC,UAAU,EAAEF,qBAAQ,CAACG,OAAO,CAACC,QAAQ,CAAC,CAAC;MACvCC,SAAS,EAAE,cAAc;MACzBC,iBAAiB,EAAE5B,WAAW,CAAC6B,OAAO;MACtCC,iBAAiB,EAAEf,OAAO;MAC1BgB,SAAS,EAAE,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC,CAAC;MACnCC,cAAc,EAAEjC,SAAS,CAACK,WAAW,CAACU,OAAO,CAACmB,KAAK,CAAC;MACpDC,KAAK,EAAEpB,OAAO,CAACoB,KAAK;MACpBC,YAAY,EAAErB,OAAO,CAACqB;IACxB,CAAC;IAED,IAAI,CAACC,UAAU,CAAClB,MAAM,CAAC,CAACmB,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;EACzC;EAEQpB,aAAaA,CAACD,KAAY,EAAEF,OAAsB,EAAQ;IAChE,MAAMwB,YAAY,GAAG;MACnBC,OAAO,EAAEvB,KAAK,CAACuB,OAAO;MACtBC,KAAK,EAAExB,KAAK,CAACwB,KAAK;MAClBC,IAAI,EAAEzB,KAAK,CAACyB;IACd,CAAC;IAED,MAAMvB,MAAuB,GAAG;MAC9BC,EAAE,EAAEC,qBAAQ,CAACC,EAAE,KAAK,KAAK,GAAG,KAAK,GAAG,SAAS;MAC7CC,UAAU,EAAEF,qBAAQ,CAACG,OAAO,CAACC,QAAQ,CAAC,CAAC;MACvCC,SAAS,EAAE,cAAc;MACzBC,iBAAiB,EAAE5B,WAAW,CAAC6B,OAAO;MACtCe,aAAa,EAAEJ,YAAY;MAC3BT,SAAS,EAAE,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC,CAAC;MACnCC,cAAc,EAAEjC,SAAS,CAACK,WAAW,CAACU,OAAO,CAACmB,KAAK,CAAC;MACpDC,KAAK,EAAEpB,OAAO,CAACoB,KAAK;MACpBC,YAAY,EAAErB,OAAO,CAACqB;IACxB,CAAC;IAED,IAAI,CAACC,UAAU,CAAClB,MAAM,CAAC,CAACmB,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;EACzC;EAEA,MAAcD,UAAUA,CAAClB,MAAuB,EAAiB;IAC/D,MAAOyB,UAAU,CAASC,KAAK,CAAC,GAAG,IAAI,CAAC1C,MAAM,CAACO,WAAW,YAAY,IAAI,CAACP,MAAM,CAACC,KAAK,EAAE,EAAE;MACzF0C,MAAM,EAAE,MAAM;MACdC,OAAO,EAAE;QACP,QAAQ,EAAE,IAAI,CAAC5C,MAAM,CAACQ,WAAW;QACjC,eAAe,EAAE,UAAU,IAAI,CAACR,MAAM,CAACQ,WAAW,EAAE;QACpD,cAAc,EAAE,kBAAkB;QAClC,QAAQ,EAAE;MACZ,CAAC;MACDqC,IAAI,EAAEC,IAAI,CAACC,SAAS,CAAC,CAAC/B,MAAM,CAAC;IAC/B,CAAC,CAAC;EACJ;AACF;AAACgC,OAAA,CAAAnD,SAAA,GAAAA,SAAA","ignoreList":[]}
|
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
|
|
3
|
+
Object.defineProperty(exports, "__esModule", {
|
|
4
|
+
value: true
|
|
5
|
+
});
|
|
6
|
+
exports.Tools = void 0;
|
|
7
|
+
exports.parseAndExecuteTool = parseAndExecuteTool;
|
|
8
|
+
class Tools {
|
|
9
|
+
tools = new Map();
|
|
10
|
+
add(func, description, parameters) {
|
|
11
|
+
this.tools.set(func.name, {
|
|
12
|
+
func,
|
|
13
|
+
description,
|
|
14
|
+
parameters,
|
|
15
|
+
required: Object.entries(parameters).filter(([_, param]) => param.required).map(([key, _]) => key)
|
|
16
|
+
});
|
|
17
|
+
return func;
|
|
18
|
+
}
|
|
19
|
+
getSchemas() {
|
|
20
|
+
return Array.from(this.tools.entries()).map(([name, {
|
|
21
|
+
description,
|
|
22
|
+
parameters,
|
|
23
|
+
required
|
|
24
|
+
}]) => ({
|
|
25
|
+
type: "function",
|
|
26
|
+
function: {
|
|
27
|
+
name,
|
|
28
|
+
description,
|
|
29
|
+
parameters: {
|
|
30
|
+
type: "object",
|
|
31
|
+
properties: parameters,
|
|
32
|
+
required
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
}));
|
|
36
|
+
}
|
|
37
|
+
async execute(name, args) {
|
|
38
|
+
const tool = this.tools.get(name);
|
|
39
|
+
if (!tool) throw new Error(`Tool ${name} not found`);
|
|
40
|
+
return await tool.func(...Object.values(args));
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
exports.Tools = Tools;
|
|
44
|
+
async function parseAndExecuteTool(result, tools) {
|
|
45
|
+
if (!result.tool_calls || result.tool_calls.length === 0) {
|
|
46
|
+
// console.log('No tool calls found');
|
|
47
|
+
return {
|
|
48
|
+
toolCalled: false
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
try {
|
|
52
|
+
const toolCall = result.tool_calls[0];
|
|
53
|
+
if (!toolCall) {
|
|
54
|
+
// console.log('No tool call found');
|
|
55
|
+
return {
|
|
56
|
+
toolCalled: false
|
|
57
|
+
};
|
|
58
|
+
}
|
|
59
|
+
const toolName = toolCall.function.name;
|
|
60
|
+
const toolInput = JSON.parse(toolCall.function.arguments);
|
|
61
|
+
|
|
62
|
+
// console.log('Calling tool:', toolName, toolInput);
|
|
63
|
+
const toolOutput = await tools.execute(toolName, toolInput);
|
|
64
|
+
// console.log('Tool called result:', toolOutput);
|
|
65
|
+
|
|
66
|
+
return {
|
|
67
|
+
toolCalled: true,
|
|
68
|
+
toolName,
|
|
69
|
+
toolInput,
|
|
70
|
+
toolOutput
|
|
71
|
+
};
|
|
72
|
+
} catch (error) {
|
|
73
|
+
// console.error('Error parsing tool call:', error);
|
|
74
|
+
return {
|
|
75
|
+
toolCalled: false
|
|
76
|
+
};
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
//# sourceMappingURL=tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"names":["Tools","tools","Map","add","func","description","parameters","set","name","required","Object","entries","filter","_","param","map","key","getSchemas","Array","from","type","function","properties","execute","args","tool","get","Error","values","exports","parseAndExecuteTool","result","tool_calls","length","toolCalled","toolCall","toolName","toolInput","JSON","parse","arguments","toolOutput","error"],"sourceRoot":"../../src","sources":["tools.ts"],"mappings":";;;;;;;AAeO,MAAMA,KAAK,CAAC;EACTC,KAAK,GAAG,IAAIC,GAAG,CAAe,CAAC;EAEvCC,GAAGA,CACCC,IAAc,EACdC,WAAmB,EACnBC,UAAsC,EACtC;IACA,IAAI,CAACL,KAAK,CAACM,GAAG,CAACH,IAAI,CAACI,IAAI,EAAE;MACxBJ,IAAI;MACJC,WAAW;MACXC,UAAU;MACVG,QAAQ,EAAEC,MAAM,CAACC,OAAO,CAACL,UAAU,CAAC,CACjCM,MAAM,CAAC,CAAC,CAACC,CAAC,EAAEC,KAAK,CAAC,KAAKA,KAAK,CAACL,QAAQ,CAAC,CACtCM,GAAG,CAAC,CAAC,CAACC,GAAG,EAAEH,CAAC,CAAC,KAAKG,GAAG;IAC1B,CAAC,CAAC;IACF,OAAOZ,IAAI;EACb;EAEFa,UAAUA,CAAA,EAAG;IACT,OAAOC,KAAK,CAACC,IAAI,CAAC,IAAI,CAAClB,KAAK,CAACU,OAAO,CAAC,CAAC,CAAC,CAACI,GAAG,CAAC,CAAC,CAACP,IAAI,EAAE;MAAEH,WAAW;MAAEC,UAAU;MAAEG;IAAS,CAAC,CAAC,MAAM;MAC9FW,IAAI,EAAE,UAAU;MAChBC,QAAQ,EAAE;QACRb,IAAI;QACJH,WAAW;QACXC,UAAU,EAAE;UACVc,IAAI,EAAE,QAAQ;UACdE,UAAU,EAAEhB,UAAU;UACtBG;QACF;MACF;IACF,CAAC,CAAC,CAAC;EACL;EAEF,MAAMc,OAAOA,CAACf,IAAY,EAAEgB,IAAS,EAAE;IACnC,MAAMC,IAAI,GAAG,IAAI,CAACxB,KAAK,CAACyB,GAAG,CAAClB,IAAI,CAAC;IACjC,IAAI,CAACiB,IAAI,EAAE,MAAM,IAAIE,KAAK,CAAC,QAAQnB,IAAI,YAAY,CAAC;IACpD,OAAO,MAAMiB,IAAI,CAACrB,IAAI,CAAC,GAAGM,MAAM,CAACkB,MAAM,CAACJ,IAAI,CAAC,CAAC;EAClD;AACF;AAACK,OAAA,CAAA7B,KAAA,GAAAA,KAAA;AAEM,eAAe8B,mBAAmBA,CAACC,MAA8B,EAAE9B,KAAY,EAAwF;EAC5K,IAAI,CAAC8B,MAAM,CAACC,UAAU,IAAID,MAAM,CAACC,UAAU,CAACC,MAAM,KAAK,CAAC,EAAE;IACtD;IACA,OAAO;MAACC,UAAU,EAAE;IAAK,CAAC;EAC9B;EAEA,IAAI;IACA,MAAMC,QAAQ,GAAGJ,MAAM,CAACC,UAAU,CAAC,CAAC,CAAC;IACrC,IAAI,CAACG,QAAQ,EAAE;MACb;MACA,OAAO;QAACD,UAAU,EAAE;MAAK,CAAC;IAC5B;IACA,MAAME,QAAQ,GAAGD,QAAQ,CAACd,QAAQ,CAACb,IAAI;IACvC,MAAM6B,SAAS,GAAGC,IAAI,CAACC,KAAK,CAACJ,QAAQ,CAACd,QAAQ,CAACmB,SAAS,CAAC;;IAEzD;IACA,MAAMC,UAAU,GAAG,MAAMxC,KAAK,CAACsB,OAAO,CAACa,QAAQ,EAAEC,SAAS,CAAC;IAC3D;;IAEA,OAAO;MACHH,UAAU,EAAE,IAAI;MAChBE,QAAQ;MACRC,SAAS;MACTI;IACJ,CAAC;EACL,CAAC,CAAC,OAAOC,KAAK,EAAE;IACZ;IACA,OAAO;MAACR,UAAU,EAAE;IAAK,CAAC;EAC9B;AACF","ignoreList":[]}
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
|
|
3
|
+
Object.defineProperty(exports, "__esModule", {
|
|
4
|
+
value: true
|
|
5
|
+
});
|
|
6
|
+
exports.CactusTTS = void 0;
|
|
7
|
+
var _index = require("./index");
|
|
8
|
+
class CactusTTS {
|
|
9
|
+
constructor(context) {
|
|
10
|
+
this.context = context;
|
|
11
|
+
}
|
|
12
|
+
static async init(context, vocoderModelPath) {
|
|
13
|
+
await (0, _index.initVocoder)(context.id, vocoderModelPath);
|
|
14
|
+
return new CactusTTS(context);
|
|
15
|
+
}
|
|
16
|
+
async generate(textToSpeak, speakerJsonStr) {
|
|
17
|
+
const {
|
|
18
|
+
formatted_prompt
|
|
19
|
+
} = await (0, _index.getFormattedAudioCompletion)(this.context.id, speakerJsonStr, textToSpeak);
|
|
20
|
+
// This part is simplified. In a real scenario, the tokens from
|
|
21
|
+
// the main model would be generated and passed to decodeAudioTokens.
|
|
22
|
+
// For now, we are assuming a direct path which may not be fully functional
|
|
23
|
+
// without the main model's token output for TTS.
|
|
24
|
+
const tokens = (await this.context.tokenize(formatted_prompt)).tokens;
|
|
25
|
+
return (0, _index.decodeAudioTokens)(this.context.id, tokens);
|
|
26
|
+
}
|
|
27
|
+
async release() {
|
|
28
|
+
return (0, _index.releaseVocoder)(this.context.id);
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
exports.CactusTTS = CactusTTS;
|
|
32
|
+
//# sourceMappingURL=tts.js.map
|
package/lib/commonjs/tts.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"names":["_index","require","CactusTTS","constructor","context","init","vocoderModelPath","initVocoder","id","generate","textToSpeak","speakerJsonStr","formatted_prompt","getFormattedAudioCompletion","tokens","tokenize","decodeAudioTokens","release","releaseVocoder","exports"],"sourceRoot":"../../src","sources":["tts.ts"],"mappings":";;;;;;AAAA,IAAAA,MAAA,GAAAC,OAAA;AASO,MAAMC,SAAS,CAAC;EAGbC,WAAWA,CAACC,OAAqB,EAAE;IACzC,IAAI,CAACA,OAAO,GAAGA,OAAO;EACxB;EAEA,aAAaC,IAAIA,CACfD,OAAqB,EACrBE,gBAAwB,EACJ;IACpB,MAAM,IAAAC,kBAAW,EAACH,OAAO,CAACI,EAAE,EAAEF,gBAAgB,CAAC;IAC/C,OAAO,IAAIJ,SAAS,CAACE,OAAO,CAAC;EAC/B;EAEA,MAAMK,QAAQA,CACZC,WAAmB,EACnBC,cAAsB,EACY;IAClC,MAAM;MAAEC;IAAiB,CAAC,GAAG,MAAM,IAAAC,kCAA2B,EAC5D,IAAI,CAACT,OAAO,CAACI,EAAE,EACfG,cAAc,EACdD,WACF,CAAC;IACD,MAAMI,MAAM,GAAG,CAAC,MAAM,IAAI,CAACV,OAAO,CAACW,QAAQ,CAACH,gBAAgB,CAAC,EAAEE,MAAM;IACrE,OAAO,IAAAE,wBAAiB,EAAC,IAAI,CAACZ,OAAO,CAACI,EAAE,EAAEM,MAAM,CAAC;EACnD;EAEA,MAAMG,OAAOA,CAAA,EAAkB;IAC7B,OAAO,IAAAC,qBAAc,EAAC,IAAI,CAACd,OAAO,CAACI,EAAE,CAAC;EACxC;AACF;AAACW,OAAA,CAAAjB,SAAA,GAAAA,SAAA","ignoreList":[]}
|
|
1
|
+
{"version":3,"names":["_index","require","CactusTTS","constructor","context","init","vocoderModelPath","initVocoder","id","generate","textToSpeak","speakerJsonStr","formatted_prompt","getFormattedAudioCompletion","tokens","tokenize","decodeAudioTokens","release","releaseVocoder","exports"],"sourceRoot":"../../src","sources":["tts.ts"],"mappings":";;;;;;AAAA,IAAAA,MAAA,GAAAC,OAAA;AASO,MAAMC,SAAS,CAAC;EAGbC,WAAWA,CAACC,OAAqB,EAAE;IACzC,IAAI,CAACA,OAAO,GAAGA,OAAO;EACxB;EAEA,aAAaC,IAAIA,CACfD,OAAqB,EACrBE,gBAAwB,EACJ;IACpB,MAAM,IAAAC,kBAAW,EAACH,OAAO,CAACI,EAAE,EAAEF,gBAAgB,CAAC;IAC/C,OAAO,IAAIJ,SAAS,CAACE,OAAO,CAAC;EAC/B;EAEA,MAAMK,QAAQA,CACZC,WAAmB,EACnBC,cAAsB,EACY;IAClC,MAAM;MAAEC;IAAiB,CAAC,GAAG,MAAM,IAAAC,kCAA2B,EAC5D,IAAI,CAACT,OAAO,CAACI,EAAE,EACfG,cAAc,EACdD,WACF,CAAC;IACD;IACA;IACA;IACA;IACA,MAAMI,MAAM,GAAG,CAAC,MAAM,IAAI,CAACV,OAAO,CAACW,QAAQ,CAACH,gBAAgB,CAAC,EAAEE,MAAM;IACrE,OAAO,IAAAE,wBAAiB,EAAC,IAAI,CAACZ,OAAO,CAACI,EAAE,EAAEM,MAAM,CAAC;EACnD;EAEA,MAAMG,OAAOA,CAAA,EAAkB;IAC7B,OAAO,IAAAC,qBAAc,EAAC,IAAI,CAACd,OAAO,CAACI,EAAE,CAAC;EACxC;AACF;AAACW,OAAA,CAAAjB,SAAA,GAAAA,SAAA","ignoreList":[]}
|
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
|
|
3
|
+
Object.defineProperty(exports, "__esModule", {
|
|
4
|
+
value: true
|
|
5
|
+
});
|
|
6
|
+
exports.CactusVLM = void 0;
|
|
7
|
+
var _index = require("./index");
|
|
8
|
+
var _telemetry = require("./telemetry");
|
|
9
|
+
class CactusVLM {
|
|
10
|
+
constructor(context, initParams) {
|
|
11
|
+
this.context = context;
|
|
12
|
+
this.initParams = initParams;
|
|
13
|
+
}
|
|
14
|
+
static async init(params, onProgress) {
|
|
15
|
+
const configs = [params, {
|
|
16
|
+
...params,
|
|
17
|
+
n_gpu_layers: 0
|
|
18
|
+
}];
|
|
19
|
+
for (const config of configs) {
|
|
20
|
+
try {
|
|
21
|
+
const context = await (0, _index.initLlama)(config, onProgress);
|
|
22
|
+
// Explicitly disable GPU for the multimodal projector for stability.
|
|
23
|
+
await (0, _index.initMultimodal)(context.id, params.mmproj, false);
|
|
24
|
+
return {
|
|
25
|
+
vlm: new CactusVLM(context, params),
|
|
26
|
+
error: null
|
|
27
|
+
};
|
|
28
|
+
} catch (e) {
|
|
29
|
+
_telemetry.Telemetry.error(e, config);
|
|
30
|
+
if (configs.indexOf(config) === configs.length - 1) {
|
|
31
|
+
return {
|
|
32
|
+
vlm: null,
|
|
33
|
+
error: e
|
|
34
|
+
};
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
return {
|
|
39
|
+
vlm: null,
|
|
40
|
+
error: new Error('Failed to initialize CactusVLM')
|
|
41
|
+
};
|
|
42
|
+
}
|
|
43
|
+
async completion(messages, params = {}, callback) {
|
|
44
|
+
const startTime = Date.now();
|
|
45
|
+
let firstTokenTime = null;
|
|
46
|
+
const wrappedCallback = callback ? data => {
|
|
47
|
+
if (firstTokenTime === null) firstTokenTime = Date.now();
|
|
48
|
+
callback(data);
|
|
49
|
+
} : undefined;
|
|
50
|
+
let result;
|
|
51
|
+
if (params.images && params.images.length > 0) {
|
|
52
|
+
const formattedPrompt = await this.context.getFormattedChat(messages);
|
|
53
|
+
const prompt = typeof formattedPrompt === 'string' ? formattedPrompt : formattedPrompt.prompt;
|
|
54
|
+
result = await (0, _index.multimodalCompletion)(this.context.id, prompt, params.images, {
|
|
55
|
+
...params,
|
|
56
|
+
prompt,
|
|
57
|
+
emit_partial_completion: !!callback
|
|
58
|
+
});
|
|
59
|
+
} else {
|
|
60
|
+
result = await this.context.completion({
|
|
61
|
+
messages,
|
|
62
|
+
...params
|
|
63
|
+
}, wrappedCallback);
|
|
64
|
+
}
|
|
65
|
+
_telemetry.Telemetry.track({
|
|
66
|
+
event: 'completion',
|
|
67
|
+
tok_per_sec: result.timings?.predicted_per_second,
|
|
68
|
+
toks_generated: result.timings?.predicted_n,
|
|
69
|
+
ttft: firstTokenTime ? firstTokenTime - startTime : null,
|
|
70
|
+
num_images: params.images?.length
|
|
71
|
+
}, this.initParams);
|
|
72
|
+
return result;
|
|
73
|
+
}
|
|
74
|
+
async rewind() {
|
|
75
|
+
// @ts-ignore
|
|
76
|
+
return this.context?.rewind();
|
|
77
|
+
}
|
|
78
|
+
async release() {
|
|
79
|
+
return this.context.release();
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
exports.CactusVLM = CactusVLM;
|
|
83
|
+
//# sourceMappingURL=vlm.js.map
|
package/lib/commonjs/vlm.js.map
CHANGED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"names":["_index","require","_telemetry","CactusVLM","constructor","context","initParams","init","params","onProgress","configs","n_gpu_layers","config","initLlama","initMultimodal","id","mmproj","vlm","error","e","Telemetry","indexOf","length","Error","completion","messages","callback","startTime","Date","now","firstTokenTime","wrappedCallback","data","undefined","result","images","formattedPrompt","getFormattedChat","prompt","multimodalCompletion","emit_partial_completion","track","event","tok_per_sec","timings","predicted_per_second","toks_generated","predicted_n","ttft","num_images","rewind","release","exports"],"sourceRoot":"../../src","sources":["vlm.ts"],"mappings":";;;;;;AAAA,IAAAA,MAAA,GAAAC,OAAA;AAYA,IAAAC,UAAA,GAAAD,OAAA;AAeO,MAAME,SAAS,CAAC;EAIbC,WAAWA,CAACC,OAAqB,EAAEC,UAA4B,EAAE;IACvE,IAAI,CAACD,OAAO,GAAGA,OAAO;IACtB,IAAI,CAACC,UAAU,GAAGA,UAAU;EAC9B;EAEA,aAAaC,IAAIA,CACfC,MAAwB,EACxBC,UAAuC,EACb;IAC1B,MAAMC,OAAO,GAAG,CACdF,MAAM,EACN;MAAE,GAAGA,MAAM;MAAEG,YAAY,EAAE;IAAE,CAAC,CAC/B;IAED,KAAK,MAAMC,MAAM,IAAIF,OAAO,EAAE;MAC5B,IAAI;QACF,MAAML,OAAO,GAAG,MAAM,IAAAQ,gBAAS,EAACD,MAAM,EAAEH,UAAU,CAAC;QACnD;QACA,MAAM,IAAAK,qBAAc,EAACT,OAAO,CAACU,EAAE,EAAEP,MAAM,CAACQ,MAAM,EAAE,KAAK,CAAC;QACtD,OAAO;UAACC,GAAG,EAAE,IAAId,SAAS,CAACE,OAAO,EAAEG,MAAM,CAAC;UAAEU,KAAK,EAAE;QAAI,CAAC;MAC3D,CAAC,CAAC,OAAOC,CAAC,EAAE;QACVC,oBAAS,CAACF,KAAK,CAACC,CAAC,EAAWP,MAAM,CAAC;QACnC,IAAIF,OAAO,CAACW,OAAO,CAACT,MAAM,CAAC,KAAKF,OAAO,CAACY,MAAM,GAAG,CAAC,EAAE;UAClD,OAAO;YAACL,GAAG,EAAE,IAAI;YAAEC,KAAK,EAAEC;UAAU,CAAC;QACvC;MACF;IACF;IAEA,OAAO;MAACF,GAAG,EAAE,IAAI;MAAEC,KAAK,EAAE,IAAIK,KAAK,CAAC,gCAAgC;IAAC,CAAC;EACxE;EAEA,MAAMC,UAAUA,CACdC,QAAsC,EACtCjB,MAA2B,GAAG,CAAC,CAAC,EAChCkB,QAA8B,EACG;IACjC,MAAMC,SAAS,GAAGC,IAAI,CAACC,GAAG,CAAC,CAAC;IAC5B,IAAIC,cAA6B,GAAG,IAAI;IAExC,MAAMC,eAAe,GAAGL,QAAQ,GAAIM,IAAS,IAAK;MAChD,IAAIF,cAAc,KAAK,IAAI,EAAEA,cAAc,GAAGF,IAAI,CAACC,GAAG,CAAC,CAAC;MACxDH,QAAQ,CAACM,IAAI,CAAC;IAChB,CAAC,GAAGC,SAAS;IAEb,IAAIC,MAA8B;IAClC,IAAI1B,MAAM,CAAC2B,MAAM,IAAI3B,MAAM,CAAC2B,MAAM,CAACb,MAAM,GAAG,CAAC,EAAE;MAC7C,MAAMc,eAAe,GAAG,MAAM,IAAI,CAAC/B,OAAO,CAACgC,gBAAgB,CAACZ,QAAQ,CAAC;MACrE,MAAMa,MAAM,GACV,OAAOF,eAAe,KAAK,QAAQ,GAC/BA,eAAe,GACfA,eAAe,CAACE,MAAM;MAC5BJ,MAAM,GAAG,MAAM,IAAAK,2BAAoB,EACjC,IAAI,CAAClC,OAAO,CAACU,EAAE,EACfuB,MAAM,EACN9B,MAAM,CAAC2B,MAAM,EACb;QAAE,GAAG3B,MAAM;QAAE8B,MAAM;QAAEE,uBAAuB,EAAE,CAAC,CAACd;MAAS,CAC3D,CAAC;IACH,CAAC,MAAM;MACLQ,MAAM,GAAG,MAAM,IAAI,CAAC7B,OAAO,CAACmB,UAAU,CAAC;QAAEC,QAAQ;QAAE,GAAGjB;MAAO,CAAC,EAAEuB,eAAe,CAAC;IAClF;IAEAX,oBAAS,CAACqB,KAAK,CAAC;MACdC,KAAK,EAAE,YAAY;MACnBC,WAAW,EAAGT,MAAM,CAASU,OAAO,EAAEC,oBAAoB;MAC1DC,cAAc,EAAGZ,MAAM,CAASU,OAAO,EAAEG,WAAW;MACpDC,IAAI,EAAElB,cAAc,GAAGA,cAAc,GAAGH,SAAS,GAAG,IAAI;MACxDsB,UAAU,EAAEzC,MAAM,CAAC2B,MAAM,EAAEb;IAC7B,CAAC,EAAE,IAAI,CAAChB,UAAU,CAAC;IAEnB,OAAO4B,MAAM;EACf;EAEA,MAAMgB,MAAMA,CAAA,EAAkB;IAC5B;IACA,OAAO,IAAI,CAAC7C,OAAO,EAAE6C,MAAM,CAAC,CAAC;EAC/B;EAEA,MAAMC,OAAOA,CAAA,EAAkB;IAC7B,OAAO,IAAI,CAAC9C,OAAO,CAAC8C,OAAO,CAAC,CAAC;EAC/B;AACF;AAACC,OAAA,CAAAjD,SAAA,GAAAA,SAAA","ignoreList":[]}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
|
|
3
|
+
export function formatChat(messages) {
|
|
4
|
+
const chat = [];
|
|
5
|
+
messages.forEach(currMsg => {
|
|
6
|
+
const role = currMsg.role || '';
|
|
7
|
+
let content = '';
|
|
8
|
+
if ('content' in currMsg) {
|
|
9
|
+
if (typeof currMsg.content === 'string') {
|
|
10
|
+
;
|
|
11
|
+
({
|
|
12
|
+
content
|
|
13
|
+
} = currMsg);
|
|
14
|
+
} else if (Array.isArray(currMsg.content)) {
|
|
15
|
+
currMsg.content.forEach(part => {
|
|
16
|
+
if ('text' in part) {
|
|
17
|
+
content += `${content ? '\n' : ''}${part.text}`;
|
|
18
|
+
}
|
|
19
|
+
});
|
|
20
|
+
} else {
|
|
21
|
+
throw new TypeError("Invalid 'content' type (ref: https://github.com/ggerganov/llama.cpp/issues/8367)");
|
|
22
|
+
}
|
|
23
|
+
} else {
|
|
24
|
+
throw new Error("Missing 'content' (ref: https://github.com/ggerganov/llama.cpp/issues/8367)");
|
|
25
|
+
}
|
|
26
|
+
chat.push({
|
|
27
|
+
role,
|
|
28
|
+
content
|
|
29
|
+
});
|
|
30
|
+
});
|
|
31
|
+
return chat;
|
|
32
|
+
}
|
|
33
|
+
//# sourceMappingURL=chat.js.map
|