Add Azure credential validation and error handling

Co-Authored-By: Joe Moura <joao@crewai.com>

tests/integration/test_azure_credentials.py

@@ -0,0 +1,38 @@
+import pytest
+from src.crewai.llm import LLM
+
+def test_azure_detection_with_credentials():
+    """Test that Azure is detected correctly when credentials are provided but model lacks azure/ prefix."""
+    # Create LLM instance with Azure parameters but without azure/ prefix
+    llm = LLM(
+        api_key='test_key',
+        api_base='test_base',
+        model='gpt-4o-mini-2024-07-18',  # Model from issue #2358
+        api_version='test_version'
+    )
+    
+    # Check if provider is detected correctly
+    provider = llm._get_custom_llm_provider()
+    assert provider == 'azure', "Azure provider should be detected based on credentials"
+    
+    # Prepare parameters that would be passed to LiteLLM
+    params = llm._prepare_completion_params(messages=[{"role": "user", "content": "test"}])
+    assert params.get('api_key') == 'test_key', "API key should be included in params"
+    assert params.get('api_base') == 'test_base', "API base should be included in params"
+    assert params.get('api_version') == 'test_version', "API version should be included in params"
+
+
+def test_azure_validation_error():
+    """Test that validation error is raised when Azure credentials are incomplete."""
+    # Create LLM instance with incomplete Azure parameters
+    llm = LLM(
+        model='azure/gpt-4',
+        api_key='test_key',
+        # Missing api_base and api_version
+    )
+    
+    # Validation should fail
+    with pytest.raises(ValueError) as excinfo:
+        llm._validate_call_params()
+    
+    assert "Incomplete Azure credentials" in str(excinfo.value)

tests/llm_test.py

@@ -267,6 +267,51 @@ def test_validate_call_params_no_response_format():
     llm._validate_call_params()
 
 
+def test_validate_azure_credentials_valid():
+    llm = LLM(
+        model="gpt-4",
+        api_key="test_key",
+        api_base="test_base",
+        api_version="test_version"
+    )
+    assert llm._validate_azure_credentials() == True
+
+
+def test_validate_azure_credentials_invalid():
+    # Missing api_version
+    llm = LLM(
+        model="gpt-4",
+        api_key="test_key",
+        api_base="test_base"
+    )
+    assert llm._validate_azure_credentials() == False
+    
+    # Non-string value
+    llm = LLM(
+        model="gpt-4",
+        api_key="test_key",
+        api_base="test_base",
+        api_version=123  # Not a string
+    )
+    assert llm._validate_azure_credentials() == False
+
+
+def test_validate_call_params_azure_invalid():
+    # Test with incomplete Azure credentials
+    llm = LLM(
+        model="azure/gpt-4",
+        api_key="test_key",
+        # Missing api_base and api_version
+    )
+    
+    # Should raise ValueError due to incomplete credentials
+    with pytest.raises(ValueError) as excinfo:
+        llm._validate_call_params()
+    
+    # Check error message
+    assert "Incomplete Azure credentials" in str(excinfo.value)
+
+
 @pytest.mark.vcr(filter_headers=["authorization"])
 def test_o3_mini_reasoning_effort_high():
     llm = LLM(