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

feat: Add model name validation and expand test coverage

Browse Source 
- Add validation for Azure and Cerebras model names
- Add validation handling in create_crew.py
- Expand test coverage for model env var cases

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-15 16:12:22 +00:00
parent e1ed85d7bd
commit 583e6584eb
3 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/agent_test.py
View File

@@ -21,14 +21,16 @@ from crewai.utilities import RPMController
from crewai.utilities.events import Emitter
def test_agent_azure_model_env_var():
"""Test Azure MODEL environment variable handling with various cases."""
def test_agent_model_env_var():
"""Test MODEL environment variable handling with various cases."""
# Store original environment variables
original_model = os.environ.get("MODEL")
test_cases = [
("azure/test-model", "azure/test-model"), # Valid case
("azure/minimal", "azure/minimal"), # Another valid case
("azure/test-model", "azure/test-model"), # Valid Azure case
("azure/minimal", "azure/minimal"), # Another valid Azure case
("cerebras/test-model", "cerebras/test-model"), # Valid Cerebras case
("cerebras/minimal", "cerebras/minimal"), # Another valid Cerebras case
]
for input_model, expected_model in test_cases: