From f754eba6fb07bb6535d19b3a69f07d432ae6ade6 Mon Sep 17 00:00:00 2001 From: Bowen Liang Date: Wed, 27 Nov 2024 00:24:21 +0800 Subject: [PATCH] update (#1638) Co-authored-by: Brandon Hancock (bhancock_ai) <109994880+bhancockio@users.noreply.github.com> --- src/crewai/agent.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/src/crewai/agent.py b/src/crewai/agent.py index 386a7cc57..a3b40c2c9 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -160,28 +160,23 @@ class Agent(BaseAgent): for provider, env_vars in ENV_VARS.items(): if provider == set_provider: for env_var in env_vars: - if env_var["key_name"] in unnacepted_attributes: - continue # Check if the environment variable is set - if "key_name" in env_var: - env_value = os.environ.get(env_var["key_name"]) + key_name = env_var.get("key_name") + if key_name and key_name not in unnacepted_attributes: + env_value = os.environ.get(key_name) if env_value: # Map key names containing "API_KEY" to "api_key" key_name = ( - "api_key" - if "API_KEY" in env_var["key_name"] - else env_var["key_name"] + "api_key" if "API_KEY" in key_name else key_name ) # Map key names containing "API_BASE" to "api_base" key_name = ( - "api_base" - if "API_BASE" in env_var["key_name"] - else key_name + "api_base" if "API_BASE" in key_name else key_name ) # Map key names containing "API_VERSION" to "api_version" key_name = ( "api_version" - if "API_VERSION" in env_var["key_name"] + if "API_VERSION" in key_name else key_name ) llm_params[key_name] = env_value