From fb62dd5e18d77f465b1d27dad02a6eb1ab91185e Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 18 Oct 2024 19:06:50 +0800
Subject: refactor: support LLM_AGENT_DUMP_RESULT (#112)

---
 scripts/run-agent.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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

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}`;
-- 
cgit v1.2.3