diff --git a/src/crewai/cli/constants.py b/src/crewai/cli/constants.py index b97b4f208..081ca74c6 100644 --- a/src/crewai/cli/constants.py +++ b/src/crewai/cli/constants.py @@ -66,7 +66,7 @@ ENV_VARS = { "azure": [ { "prompt": "Enter your Azure deployment name (must start with 'azure/')", - "key_name": "model", + "key_name": "MODEL", }, { "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", }, { "prompt": "Enter your Cerebras API version (press Enter to skip)", diff --git a/tests/agent_test.py b/tests/agent_test.py index e67a7454a..bc1c87bce 100644 --- a/tests/agent_test.py +++ b/tests/agent_test.py @@ -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")