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.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'scripts/run-agent.py') 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}" -- cgit v1.2.3