mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-05-01 23:32:39 +00:00
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:
@@ -66,7 +66,7 @@ ENV_VARS = {
|
|||||||
"azure": [
|
"azure": [
|
||||||
{
|
{
|
||||||
"prompt": "Enter your Azure deployment name (must start with '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)",
|
"prompt": "Enter your AZURE API key (press Enter to skip)",
|
||||||
@@ -84,7 +84,7 @@ ENV_VARS = {
|
|||||||
"cerebras": [
|
"cerebras": [
|
||||||
{
|
{
|
||||||
"prompt": "Enter your Cerebras model name (must start with '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)",
|
"prompt": "Enter your Cerebras API version (press Enter to skip)",
|
||||||
|
|||||||
@@ -21,6 +21,22 @@ from crewai.utilities import RPMController
|
|||||||
from crewai.utilities.events import Emitter
|
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():
|
def test_agent_llm_creation_with_env_vars():
|
||||||
# Store original environment variables
|
# Store original environment variables
|
||||||
original_api_key = os.environ.get("OPENAI_API_KEY")
|
original_api_key = os.environ.get("OPENAI_API_KEY")
|
||||||
|
|||||||
Reference in New Issue
Block a user