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)

---
 tools/demo_js.js | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

(limited to 'tools/demo_js.js')

diff --git a/tools/demo_js.js b/tools/demo_js.js
index c8b7237..5693d0e 100644
--- a/tools/demo_js.js
+++ b/tools/demo_js.js
@@ -12,13 +12,18 @@
  * @param {Args} args
  */
 exports.run = function run(args) {
-  for (const [key, value] of Object.entries(args)) {
-      console.log(`${key}: ${JSON.stringify(value)}`);
-  }
-
+  let output = `string: ${args.string}
+string_enum: ${args.string_enum}
+string_optional: ${args.string_optional}
+boolean: ${args.boolean}
+integer: ${args.integer}
+number: ${args.number}
+array: ${args.array}
+array_optional: ${args.array_optional}`;
   for (const [key, value] of Object.entries(process.env)) {
     if (key.startsWith("LLM_")) {
-      console.log(`${key}: ${value}`);
+      output = `${output}\n${key}: ${value}`;
     }
   }
+  return output;
 }
-- 
cgit v1.2.3