mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
Add Azure credential validation and error handling
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
38
tests/integration/test_azure_credentials.py
Normal file
38
tests/integration/test_azure_credentials.py
Normal file
@@ -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)
|
||||
Reference in New Issue
Block a user