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)

---
 agents/demo/tools.js | 3 +--
 agents/demo/tools.py | 1 -
 agents/demo/tools.sh | 7 ++++---
 3 files changed, 5 insertions(+), 6 deletions(-)

(limited to 'agents/demo')

diff --git a/agents/demo/tools.js b/agents/demo/tools.js
index 37f8937..6a5c071 100644
--- a/agents/demo/tools.js
+++ b/agents/demo/tools.js
@@ -4,6 +4,5 @@ const os = require("node:os");
  */
 exports.get_sysinfo = function getSysinfo() {
     return `OS: ${os.type()}
-Arch: ${os.arch()}
-User: ${process.env["USER"]}`
+Arch: ${os.arch()}`
 }
diff --git a/agents/demo/tools.py b/agents/demo/tools.py
index 5f1a4fc..ec7ca2b 100644
--- a/agents/demo/tools.py
+++ b/agents/demo/tools.py
@@ -8,5 +8,4 @@ def get_sysinfo():
   return "\n".join([
     f"OS: {platform.system()}",
     f"Arch: {platform.machine()}",
-    f"User: {os.environ.get('USER')}"
   ])
\ No newline at end of file
diff --git a/agents/demo/tools.sh b/agents/demo/tools.sh
index 4b5849b..98f3d47 100755
--- a/agents/demo/tools.sh
+++ b/agents/demo/tools.sh
@@ -3,9 +3,10 @@ set -e
 
 # @cmd Get the system info
 get_sysinfo() {
-    echo "OS: $(uname)"
-    echo "Arch: $(arch)"
-    echo "User: $USER"
+    cat <<EOF >> "$LLM_OUTPUT"
+OS: $(uname)
+Arch: $(arch)
+EOF
 }
 
 # See more details at https://github.com/sigoden/argc
-- 
cgit v1.2.3