From 6eb391bdc76ed070299f1b70e96f021a03a6d97a Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 18 Oct 2024 18:28:03 +0800
Subject: feat: controls displaying the results from function call (#111)

---
 scripts/run-tool.sh | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

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

diff --git a/scripts/run-tool.sh b/scripts/run-tool.sh
index bdc5547..4ede4a1 100755
--- a/scripts/run-tool.sh
+++ b/scripts/run-tool.sh
@@ -88,7 +88,35 @@ EOF
     eval "'$tool_path' $args"
     if [[ "$no_llm_output" -eq 1 ]]; then
         cat "$LLM_OUTPUT"
+    else
+        dump_result
+    fi
+}
+
+dump_result() {
+    if [ ! -t 1 ]; then
+        return;
+    fi
+    local env_name env_value show_result=0
+    env_name="LLM_TOOL_DUMP_RESULT_$(echo "$LLM_TOOL_NAME" | tr '[:lower:]' '[:upper:]' | tr '-' '_')"
+    env_value="${!env_name}"
+    if [[ "$LLM_TOOL_DUMP_RESULT" == "1" || "$LLM_TOOL_DUMP_RESULT" == "true" ]]; then
+        if [[ "$env_value" != "0" && "$env_value" != "false" ]]; then
+            show_result=1
+        fi
+    else
+        if [[ "$env_value" == "1" || "$env_value" == "true" ]]; then
+            show_result=1
+        fi
     fi
+    if [[ "$show_result" -ne 1 ]]; then
+        return
+    fi
+    cat <<EOF
+$(echo -e "\e[2m")----------------------
+$(cat "$LLM_OUTPUT")
+----------------------$(echo -e "\e[0m")
+EOF
 }
 
 die() {
-- 
cgit v1.2.3