andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-19 04:48:13 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix Azure OpenAI authentication for models without azure/ prefix (fixes #2358)

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-03-13 08:41:04 +00:00
parent 41a670166a
commit 1145900b39
3 changed files with 63 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
test_azure_integration.py Normal file
View File

@@ -0,0 +1,37 @@
# test_azure_integration.py
from src.crewai.llm import LLM
# Test with Azure parameters but without azure/ prefix
llm = LLM(
api_key='test_key',
api_base='test_base',
model='gpt-4o-mini-2024-07-18',
api_version='test_version'
)
# Print the detected provider
provider = llm._get_custom_llm_provider()
print(f"Detected provider: {provider}")
print(f"Is Azure detected correctly: {provider == 'azure'}")
# Prepare parameters that would be passed to LiteLLM
params = llm._prepare_completion_params(messages=[{"role": "user", "content": "test"}])
print(f"Parameters passed to LiteLLM: {params}")
# Test with Azure parameters and with azure/ prefix for comparison
llm_with_prefix = LLM(
api_key='test_key',
api_base='test_base',
model='azure/gpt-4o-mini-2024-07-18',
api_version='test_version'
)
# Print the detected provider
provider_with_prefix = llm_with_prefix._get_custom_llm_provider()
print(f"\nWith azure/ prefix:")
print(f"Detected provider: {provider_with_prefix}")
print(f"Is Azure detected correctly: {provider_with_prefix == 'azure'}")
# Prepare parameters that would be passed to LiteLLM
params_with_prefix = llm_with_prefix._prepare_completion_params(messages=[{"role": "user", "content": "test"}])
print(f"Parameters passed to LiteLLM: {params_with_prefix}")