From 732eae532c8e8632db95ab80e0dde5071e744386 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 10 Jul 2024 18:53:32 +0800
Subject: feat: adjust the way of returning data to LLM (#69)

---
 scripts/run-tool.js | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

(limited to 'scripts/run-tool.js')

diff --git a/scripts/run-tool.js b/scripts/run-tool.js
index bc09b5e..9728f7d 100755
--- a/scripts/run-tool.js
+++ b/scripts/run-tool.js
@@ -46,8 +46,8 @@ function parseRawData(data) {
 }
 
 function setupEnv(rootDir, toolName) {
-  process.env["LLM_ROOT_DIR"] = rootDir;
   loadEnv(path.resolve(rootDir, ".env"));
+  process.env["LLM_ROOT_DIR"] = rootDir;
   process.env["LLM_TOOL_NAME"] = toolName;
   process.env["LLM_TOOL_CACHE_DIR"] = path.resolve(rootDir, "cache", toolName);
 }
@@ -80,22 +80,26 @@ async function run(toolPath, toolFunc, toolData) {
     throw new Error(`Not module function '${toolFunc}' at '${toolPath}'`);
   }
   const value = await mod[toolFunc](toolData);
-  dumpValue(value);
+  returnToLLM(value);
 }
 
-function dumpValue(value) {
+function returnToLLM(value) {
   if (value === null || value === undefined) {
     return;
   }
+  let writer = process.stdout;
+  if (process.env["LLM_OUTPUT"]) {
+    writer = fs.createWriteStream(process.env["LLM_OUTPUT"]);
+  }
   const type = typeof value;
   if (type === "string" || type === "number" || type === "boolean") {
-    console.log(value);
+    writer.write(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);
+      writer.write(value);
     }
   }
 }
-- 
cgit v1.2.3