refactor: support LLM_AGENT_DUMP_RESULT (#112)

Author: sigoden

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}`;

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}"

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