feat: adjust the way of returning data to LLM (#69)

1 files changed, 10 insertions, 5 deletions

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;
 }