diff options
| author | sigoden <sigoden@gmail.com> | 2024-10-18 19:06:50 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-10-18 19:06:50 +0800 |
| commit | fb62dd5e18d77f465b1d27dad02a6eb1ab91185e (patch) | |
| tree | 7a34805c2673eb6a2a19d977c2df68dff6384da1 | |
| parent | 6eb391bdc76ed070299f1b70e96f021a03a6d97a (diff) | |
| download | llm-functions-docker-fb62dd5e18d77f465b1d27dad02a6eb1ab91185e.tar.gz | |
refactor: support LLM_AGENT_DUMP_RESULT (#112)
| -rwxr-xr-x | scripts/run-agent.js | 2 | ||||
| -rwxr-xr-x | scripts/run-agent.py | 2 | ||||
| -rwxr-xr-x | 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 |