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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
#!/usr/bin/env node
const path = require("path");
const fs = require("fs");
const os = require("os");
async function main() {
const [botName, botFunc, rawData] = parseArgv("run-bot.js");
const botData = parseRawData(rawData);
const rootDir = path.resolve(__dirname, "..");
setupEnv(rootDir, botName);
const botToolsPath = path.resolve(rootDir, `bots/${botName}/tools.js`);
await run(botToolsPath, botFunc, botData);
}
function parseArgv(thisFileName) {
let botName = process.argv[1];
let botFunc = "";
let botData = null;
if (botName.endsWith(thisFileName)) {
botName = process.argv[2];
botFunc = process.argv[3];
botData = process.argv[4];
} else {
botName = path.basename(botName);
botFunc = process.argv[2];
botData = process.argv[3];
}
if (botName.endsWith(".js")) {
botName = botName.slice(0, -3);
}
return [botName, botFunc, botData];
}
function parseRawData(data) {
if (!data) {
throw new Error("No JSON data");
}
try {
return JSON.parse(data);
} catch {
throw new Error("Invalid JSON data");
}
}
function setupEnv(rootDir, botName) {
process.env["LLM_ROOT_DIR"] = rootDir;
loadEnv(path.resolve(rootDir, ".env"));
process.env["LLM_BOT_NAME"] = botName;
process.env["LLM_BOT_ROOT_DIR"] = path.resolve(rootDir, "bots", botName);
process.env["LLM_BOT_CACHE_DIR"] = path.resolve(rootDir, "cache", botName);
}
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 {}
}
async function run(botPath, botFunc, botData) {
let mod;
if (os.platform() === "win32") {
botPath = `file://${botPath}`;
}
try {
mod = await import(botPath);
} catch {
throw new Error(`Unable to load bot tools at '${botPath}'`);
}
if (!mod || !mod[botFunc]) {
throw new Error(`Not module function '${botFunc}' at '${botPath}'`);
}
const value = await mod[botFunc](botData);
dumpValue(value);
}
function dumpValue(value) {
if (value === null || value === undefined) {
return;
}
const type = typeof value;
if (type === "string" || type === "number" || type === "boolean") {
console.log(value);
} else if (type === "object") {
const proto = Object.prototype.toString.call(value);
if (proto === "[object Object]" || proto === "[object Array]") {
const valueStr = JSON.stringify(value, null, 2);
require("assert").deepStrictEqual(value, JSON.parse(valueStr));
console.log(valueStr);
}
}
}
main();
|
