From 44fc664bde8e3b744788d827869ad9dba4563276 Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Tue, 3 Dec 2024 10:17:12 -0500 Subject: [PATCH] incorporate #1683 --- src/crewai/agent.py | 20 ++++++++------------ 1 file changed, 8 insertions(+), 12 deletions(-) diff --git a/src/crewai/agent.py b/src/crewai/agent.py index abe678db1..4277ec023 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -181,21 +181,17 @@ class Agent(BaseAgent): if key_name and key_name not in unaccepted_attributes: env_value = os.environ.get(key_name) if env_value: + param_name = env_var.lower() # Map key names containing "API_KEY" to "api_key" - key_name = ( - "api_key" if "API_KEY" in key_name else key_name - ) + if "api_key" in param_name: + param_name = "api_key" # Map key names containing "API_BASE" to "api_base" - key_name = ( - "api_base" if "API_BASE" in key_name else key_name - ) + elif "api_base" in param_name: + param_name = "api_base" # Map key names containing "API_VERSION" to "api_version" - key_name = ( - "api_version" - if "API_VERSION" in key_name - else key_name - ) - llm_params[key_name] = env_value + elif "api_version" in param_name: + param_name = "api_version" + llm_params[param_name] = env_value # Check for default values if the environment variable is not set elif env_var.get("default", False): for key, value in env_var.items():