blob: 3c8e966c2201b7aafefb7a72be409ec49326a7b6 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
#!/usr/bin/env node
const path = require("path");
const fs = require('fs');
function parseArgv() {
let func_file = process.argv[1];
let func_data = null;
if (func_file.endsWith("tool.js")) {
func_file = process.argv[2]
func_data = process.argv[3]
} else {
func_file = path.basename(func_file)
func_data = process.argv[2];
}
if (!func_file.endsWith(".js")) {
func_file += '.js'
}
return [func_file, func_data]
}
function loadFunc(func_file) {
const func_path = path.resolve(process.env["LLM_FUNCTIONS_DIR"], `tools/${func_file}`)
try {
return require(func_path);
} catch {
console.log(`Invalid function: ${func_file}`)
process.exit(1)
}
}
function loadEnv(filePath) {
try {
const data = fs.readFileSync(filePath, 'utf-8');
const lines = data.split('\n');
lines.forEach(line => {
if (line.trim().startsWith('#') || line.trim() === '') return;
const [key, ...value] = line.split('=');
process.env[key.trim()] = value.join('=').trim();
});
} catch {}
}
process.env["LLM_FUNCTIONS_DIR"] = path.resolve(__dirname, "..");
loadEnv(path.resolve(process.env["LLM_FUNCTIONS_DIR"], ".env"));
const [func_file, func_data] = parseArgv();
if (process.env["LLM_FUNCTION_ACTION"] == "declarate") {
const { declarate } = loadFunc(func_file);
console.log(JSON.stringify(declarate(), null, 2))
} else {
if (!func_data) {
console.log("No json data");
process.exit(1)
}
let args;
try {
args = JSON.parse(func_data)
} catch {
console.log("Invalid json data")
process.exit(1)
}
const { execute } = loadFunc(func_file);
execute(args)
}
|
