From fb62dd5e18d77f465b1d27dad02a6eb1ab91185e Mon Sep 17 00:00:00 2001 From: sigoden Date: Fri, 18 Oct 2024 19:06:50 +0800 Subject: refactor: support LLM_AGENT_DUMP_RESULT (#112) --- scripts/run-agent.js | 2 +- scripts/run-agent.py | 2 +- scripts/run-agent.sh | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/scripts/run-agent.js b/scripts/run-agent.js index 3b176fe..7add1e2 100755 --- a/scripts/run-agent.js +++ b/scripts/run-agent.js @@ -129,7 +129,7 @@ async function dumpResult() { let showResult = false; const agentName = process.env["LLM_AGENT_NAME"].toUpperCase().replace(/-/g, '_'); const agentEnvName = `LLM_AGENT_DUMP_RESULT_${agentName}`; - const agentEnvValue = process.env[agentEnvName]; + const agentEnvValue = process.env[agentEnvName] || process.env["LLM_AGENT_DUMP_RESULT"]; const funcName = process.env["LLM_AGENT_FUNC"].toUpperCase().replace(/-/g, '_'); const funcEnvName = `${agentEnvName}_${funcName}`; diff --git a/scripts/run-agent.py b/scripts/run-agent.py index 582b792..b97dcd1 100755 --- a/scripts/run-agent.py +++ b/scripts/run-agent.py @@ -100,7 +100,7 @@ def dump_result(): show_result = False agent_name = os.environ["LLM_AGENT_NAME"].upper().replace("-", "_") agent_env_name = f"LLM_AGENT_DUMP_RESULT_{agent_name}" - agent_env_value = os.getenv(agent_env_name) + agent_env_value = os.getenv(agent_env_name, os.getenv("LLM_AGENT_DUMP_RESULT")) func_name = os.environ["LLM_AGENT_FUNC"].upper().replace("-", "_") func_env_name = f"{agent_env_name}_{func_name}" diff --git a/scripts/run-agent.sh b/scripts/run-agent.sh index 9ad0b49..0b6ed47 100755 --- a/scripts/run-agent.sh +++ b/scripts/run-agent.sh @@ -104,7 +104,7 @@ dump_result() { 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}" + 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 -- cgit v1.2.3