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
|
const fs = require("node:fs/promises");
const { exec, spawn } = require("node:child_process");
const { promisify } = require("node:util");
const path = require("node:path");
const { tmpdir } = require("node:os");
const toJsonSchema = require('to-json-schema');
const input = require("@inquirer/input").default;
exports._instructions = async function () {
const value = await input({ message: "Enter the json file path or command to generate json", required: true });
let json_file_path;
let generate_json_command_context = "";
try {
await fs.access(value);
json_file_path = value;
} catch {
generate_json_command_context = `command_to_generate_json: \`${value}\`\n`;
const { stdout } = await promisify(exec)(value, { maxBuffer: 100 * 1024 * 1024 });
json_file_path = path.join(tmpdir(), `${process.env.LLM_AGENT_NAME}-${process.pid}.data.json`);
await fs.writeFile(json_file_path, stdout);
console.log(`ⓘ Generated json data saved to: ${json_file_path}`);
}
const json_data = await fs.readFile(json_file_path, "utf8");
const json_schema = toJsonSchema(JSON.parse(json_data));
return `You are a AI agent that can view and filter json data with jq.
## Context
${generate_json_command_context}json_file_path: ${json_file_path}
json_schema: ${JSON.stringify(json_schema, null, 2)}
`
}
/**
* Print the json data.
*
* @typedef {Object} Args
* @property {string} json_file_path The json file path
* @property {string} jq_expr The jq expression
* @param {Args} args
*/
exports.print_json = async function (args) {
const { json_file_path, jq_expr } = args;
return new Promise((resolve, reject) => {
const child = spawn("jq", ["-r", jq_expr, json_file_path], { stdio: "inherit" });
child.on('close', code => {
if (code === 0) {
resolve();
} else {
reject(new Error(`jq exited with code ${code}`));
}
});
child.on('error', err => {
reject(err);
});
});
}
|
