fix: Use uppercase MODEL env var for Azure provider

- Update Azure provider config to use uppercase MODEL env var
- Add test case to verify MODEL env var handling
- Fixes #2139

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

View File

@@ -21,6 +21,22 @@ from crewai.utilities import RPMController
from crewai.utilities.events import Emitter
def test_agent_azure_model_env_var():
# Store original environment variables
original_model = os.environ.get("MODEL")
# Test with uppercase MODEL
os.environ["MODEL"] = "azure/test-model"
agent = Agent(role="test role", goal="test goal", backstory="test backstory")
assert agent.llm.model == "azure/test-model"
# Clean up environment variables
if original_model:
os.environ["MODEL"] = original_model
else:
del os.environ["MODEL"]
def test_agent_llm_creation_with_env_vars():
# Store original environment variables
original_api_key = os.environ.get("OPENAI_API_KEY")