From 3584b5c31f4f1f1baa1c350fa46a18752f8bf7c4 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 12 Dec 2024 12:22:51 +0800
Subject: feat: support env var `LLM_DUMP_RESULTS` (#144)

---
 scripts/run-agent.sh | 34 ++++++++++------------------------
 1 file changed, 10 insertions(+), 24 deletions(-)

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

diff --git a/scripts/run-agent.sh b/scripts/run-agent.sh
index d7f9a20..b832577 100755
--- a/scripts/run-agent.sh
+++ b/scripts/run-agent.sh
@@ -1,4 +1,7 @@
 #!/usr/bin/env bash
+
+# Usage: ./run-agent.sh <agent-name> <agent-func> <agent-data>
+
 set -e
 
 main() {
@@ -88,46 +91,29 @@ EOF
         die "error: invalid JSON data"
     }
 
-    no_llm_output=0
     if [[ -z "$LLM_OUTPUT" ]]; then
-        no_llm_output=1
+        is_temp_llm_output=1
         export LLM_OUTPUT="$(mktemp)"
     fi
     eval "'$tools_path' '$agent_func' $args"
-    if [[ "$no_llm_output" -eq 1 ]]; then
+    if [[ "$is_temp_llm_output" -eq 1 ]]; then
         cat "$LLM_OUTPUT"
     else
-        dump_result
+        dump_result "${LLM_AGENT_NAME}:${LLM_AGENT_FUNC}"
     fi
 }
 
 dump_result() {
-    if [ ! -t 1 ]; then
+    if [[ "$LLM_OUTPUT" == "/dev/stdout" ]] || [[ -z "$LLM_DUMP_RESULTS" ]] ||  [[ ! -t 1 ]]; then
         return;
     fi
-    
-    local agent_env_name agent_env_value func_env_name func_env_value show_result=0
-    agent_env_name="LLM_AGENT_DUMP_RESULT_$(echo "$LLM_AGENT_NAME" | tr '[:lower:]' '[:upper:]' | tr '-' '_')"
-    agent_env_value="${!agent_env_name:-"$LLM_AGENT_DUMP_RESULT"}"
-    func_env_name="${agent_env_name}_$(echo "$LLM_AGENT_FUNC" | tr '[:lower:]' '[:upper:]' | tr '-' '_')"
-    func_env_value="${!func_env_name}"
-    if [[ "$agent_env_value" == "1" || "$agent_env_value" == "true" ]]; then
-        if [[ "$func_env_value" != "0" && "$func_env_value" != "false" ]]; then
-            show_result=1
-        fi
-    else
-        if [[ "$func_env_value" == "1" || "$func_env_value" == "true" ]]; then
-            show_result=1
-        fi
-    fi
-    if [[ "$show_result" -ne 1 ]]; then
-        return
-    fi
-    cat <<EOF
+    if grep -q -w -E "$LLM_DUMP_RESULTS" <<<"$1"; then
+        cat <<EOF
 $(echo -e "\e[2m")----------------------
 $(cat "$LLM_OUTPUT")
 ----------------------$(echo -e "\e[0m")
 EOF
+    fi
 }
 
 die() {
-- 
cgit v1.2.3