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-mcp-tool.sh | 34 ++++++++++------------------------
 1 file changed, 10 insertions(+), 24 deletions(-)

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

diff --git a/scripts/run-mcp-tool.sh b/scripts/run-mcp-tool.sh
index 11d3d0b..08beafd 100755
--- a/scripts/run-mcp-tool.sh
+++ b/scripts/run-mcp-tool.sh
@@ -1,4 +1,7 @@
 #!/usr/bin/env bash
+
+# Usage: ./run-mcp-tool.sh <tool-name> <tool-data>
+
 set -e
 
 main() {
@@ -53,9 +56,8 @@ run() {
         tool_data="$(echo "$tool_data" | sed 's/\\/\\\\/g')"
     fi
 
-    no_llm_output=0
     if [[ -z "$LLM_OUTPUT" ]]; then
-        no_llm_output=1
+        is_temp_llm_output=1
         export LLM_OUTPUT="$(mktemp)"
     fi
     curl -sS "http://localhost:${MCP_BRIDGE_PORT:-8808}/tools/$tool_name" \
@@ -63,40 +65,24 @@ run() {
         -H 'content-type: application/json' \
         -d "$tool_data" > "$LLM_OUTPUT"
 
-    if [[ "$no_llm_output" -eq 1 ]]; then
+    if [[ "$is_temp_llm_output" -eq 1 ]]; then
         cat "$LLM_OUTPUT"
     else
-        dump_result
+        dump_result "$tool_name" 
     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
 }
 
 main "$@"
-- 
cgit v1.2.3