docs: Add comments and expand test coverage

- Add documentation comments for MODEL env var
- Expand test coverage for Azure MODEL env var
- Add test for missing MODEL env var case

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-15 16:09:03 +00:00
parent 96b6d91084
commit e1ed85d7bd
2 changed files with 20 additions and 6 deletions

View File

@@ -66,7 +66,7 @@ ENV_VARS = {
"azure": [
{
"prompt": "Enter your Azure deployment name (must start with 'azure/')",
"key_name": "MODEL",
"key_name": "MODEL", # Uppercase MODEL used for consistency across environment variables
},
{
"prompt": "Enter your AZURE API key (press Enter to skip)",
@@ -84,7 +84,7 @@ ENV_VARS = {
"cerebras": [
{
"prompt": "Enter your Cerebras model name (must start with 'cerebras/')",
"key_name": "MODEL",
"key_name": "MODEL", # Uppercase MODEL used for consistency across environment variables
},
{
"prompt": "Enter your Cerebras API version (press Enter to skip)",

View File

@@ -22,19 +22,33 @@ from crewai.utilities.events import Emitter
def test_agent_azure_model_env_var():
"""Test Azure MODEL environment variable handling with various cases."""
# Store original environment variables
original_model = os.environ.get("MODEL")
# Test with uppercase MODEL
os.environ["MODEL"] = "azure/test-model"
test_cases = [
("azure/test-model", "azure/test-model"), # Valid case
("azure/minimal", "azure/minimal"), # Another valid case
]
for input_model, expected_model in test_cases:
# Set test MODEL value
os.environ["MODEL"] = input_model
agent = Agent(role="test role", goal="test goal", backstory="test backstory")
assert agent.llm.model == expected_model
# Test missing MODEL env var
if "MODEL" in os.environ:
del os.environ["MODEL"]
agent = Agent(role="test role", goal="test goal", backstory="test backstory")
assert agent.llm.model == "azure/test-model"
assert agent.llm.model == "gpt-4o-mini" # Default model
# Clean up environment variables
if original_model:
os.environ["MODEL"] = original_model
else:
del os.environ["MODEL"]
if "MODEL" in os.environ:
del os.environ["MODEL"]
def test_agent_llm_creation_with_env_vars():