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

docs: Add comments and expand test coverage

Browse Source 
- 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
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/agent_test.py
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():