feat: support env var LLM_DUMP_RESULTS (#144)

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() {