andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-24 23:58:15 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

incorporate #1683

Browse Source
This commit is contained in:
Brandon Hancock
2024-12-03 10:17:12 -05:00
parent 7d9d0ff6f7
commit 44fc664bde
1 changed files with 8 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/crewai/agent.py
View File

@@ -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():