scripts/run-tool.js


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

#!/usr/bin/env node

const path = require("path");
const fs = require("fs");

function parseArgv() {
  let toolName = process.argv[1];
  let toolData = null;

  if (toolName.endsWith("run-tool.js")) {
    toolName = process.argv[2];
    toolData = process.argv[3];
  } else {
    toolName = path.basename(toolName);
    toolData = process.argv[2];
  }

  if (toolName.endsWith(".js")) {
    toolName = toolName.slice(0, -3);
  }

  return [toolName, toolData];
}

function loadModule(toolName) {
  const toolFileName = `${toolName}.js`;
  const toolPath = path.resolve(
    process.env["LLM_ROOT_DIR"],
    `tools/${toolFileName}`,
  );
  try {
    return require(toolPath);
  } catch {
    console.log(`Invalid tooltion: ${toolFileName}`);
    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 {}
}

const LLM_ROOT_DIR = path.resolve(__dirname, "..");
process.env["LLM_ROOT_DIR"] = LLM_ROOT_DIR;

loadEnv(path.resolve(LLM_ROOT_DIR, ".env"));

const [toolName, toolData] = parseArgv();

process.env["LLM_TOOL_NAME"] = toolName;
process.env["LLM_TOOL_CACHE_DIR"] = path.resolve(
  LLM_ROOT_DIR,
  "cache",
  toolName,
);

if (!toolData) {
  console.log("No json data");
  process.exit(1);
}

let data = null;
try {
  data = JSON.parse(toolData);
} catch {
  console.log("Invalid json data");
  process.exit(1);
}

const { run } = loadModule(toolName);
run(data);