Skip to content

Commit

Permalink
feat(agents): json-viewer use jq cli other than node-jq module (#163)
Browse files Browse the repository at this point in the history
  • Loading branch information
sigoden authored Feb 12, 2025
1 parent 4ac72b5 commit 8e25114
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 7 deletions.
3 changes: 1 addition & 2 deletions agents/json-viewer/package.json
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
{
"dependencies": {
"@inquirer/input": "^4.0.2",
"json-schema-generator": "^2.0.6",
"node-jq": "^6.0.1"
"json-schema-generator": "^2.0.6"
}
}
22 changes: 17 additions & 5 deletions agents/json-viewer/tools.js
Original file line number Diff line number Diff line change
@@ -1,10 +1,9 @@
const fs = require("node:fs/promises");
const { exec } = require("node:child_process");
const { exec, spawn } = require("node:child_process");
const { promisify } = require("node:util");
const path = require("node:path");
const { tmpdir } = require("node:os");

const jq = require("node-jq");
const jsonSchemaGenerator = require("json-schema-generator");
const input = require("@inquirer/input").default;

Expand All @@ -17,7 +16,7 @@ exports._instructions = async function () {
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});
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}`);
Expand All @@ -44,6 +43,19 @@ json_schema: ${JSON.stringify(json_schema, null, 2)}
*/
exports.print_json = async function (args) {
const { json_file_path, jq_expr } = args;
const result = await jq.run(jq_expr, json_file_path, { raw: true });
console.log(result);
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);
});
});
}

0 comments on commit 8e25114

Please sign in to comment.