refactor: support LLM_AGENT_DUMP_RESULT (#112)
This commit is contained in:
@@ -129,7 +129,7 @@ async function dumpResult() {
|
|||||||
let showResult = false;
|
let showResult = false;
|
||||||
const agentName = process.env["LLM_AGENT_NAME"].toUpperCase().replace(/-/g, '_');
|
const agentName = process.env["LLM_AGENT_NAME"].toUpperCase().replace(/-/g, '_');
|
||||||
const agentEnvName = `LLM_AGENT_DUMP_RESULT_${agentName}`;
|
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 funcName = process.env["LLM_AGENT_FUNC"].toUpperCase().replace(/-/g, '_');
|
||||||
const funcEnvName = `${agentEnvName}_${funcName}`;
|
const funcEnvName = `${agentEnvName}_${funcName}`;
|
||||||
|
|||||||
@@ -100,7 +100,7 @@ def dump_result():
|
|||||||
show_result = False
|
show_result = False
|
||||||
agent_name = os.environ["LLM_AGENT_NAME"].upper().replace("-", "_")
|
agent_name = os.environ["LLM_AGENT_NAME"].upper().replace("-", "_")
|
||||||
agent_env_name = f"LLM_AGENT_DUMP_RESULT_{agent_name}"
|
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_name = os.environ["LLM_AGENT_FUNC"].upper().replace("-", "_")
|
||||||
func_env_name = f"{agent_env_name}_{func_name}"
|
func_env_name = f"{agent_env_name}_{func_name}"
|
||||||
|
|||||||
@@ -104,7 +104,7 @@ dump_result() {
|
|||||||
|
|
||||||
local agent_env_name agent_env_value func_env_name func_env_value show_result=0
|
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_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_name="${agent_env_name}_$(echo "$LLM_AGENT_FUNC" | tr '[:lower:]' '[:upper:]' | tr '-' '_')"
|
||||||
func_env_value="${!func_env_name}"
|
func_env_value="${!func_env_name}"
|
||||||
if [[ "$agent_env_value" == "1" || "$agent_env_value" == "true" ]]; then
|
if [[ "$agent_env_value" == "1" || "$agent_env_value" == "true" ]]; then
|
||||||
|
|||||||
Reference in New Issue
Block a user