scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(azure_ai): drop responses test __init__ to fix package name collision; cover api-version, missing api_base, AD fallback

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
shivam 2026-07-18 22:06:58 +00:00
parent 518c4b07a7
commit 34d32c04e5
2 changed files with 35 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

0
tests/test_litellm/llms/azure_ai/responses/__init__.py
View file

35
tests/test_litellm/llms/azure_ai/responses/test_azure_ai_responses_transformation.py
View file

@ -103,6 +103,25 @@ def test_get_complete_url(api_base, expected):
assert config.get_complete_url(api_base=api_base, litellm_params={}) == expected
def test_get_complete_url_adds_api_version_from_params():
config = AzureAIResponsesAPIConfig()
url = config.get_complete_url(
api_base="https://res.services.ai.azure.com/api/projects/proj",
litellm_params={"api_version": "2025-04-01-preview"},
)
assert url == (
"https://res.services.ai.azure.com/api/projects/proj/openai/v1/responses?api-version=2025-04-01-preview"
)
def test_get_complete_url_raises_without_api_base(monkeypatch):
monkeypatch.setattr(litellm, "api_base", None)
monkeypatch.delenv("AZURE_AI_API_BASE", raising=False)
config = AzureAIResponsesAPIConfig()
with pytest.raises(ValueError):
config.get_complete_url(api_base=None, litellm_params={})
def test_validate_environment_api_key_header_for_foundry_host():
config = AzureAIResponsesAPIConfig()
headers = config.validate_environment(
@ -129,6 +148,22 @@ def test_validate_environment_bearer_for_serverless_host():
assert "api-key" not in headers
def test_validate_environment_falls_back_to_base_azure_env_without_key(monkeypatch):
monkeypatch.setattr(litellm, "api_key", None)
monkeypatch.setattr(litellm, "openai_key", None)
monkeypatch.delenv("AZURE_AI_API_KEY", raising=False)
monkeypatch.delenv("AZURE_OPENAI_API_KEY", raising=False)
monkeypatch.delenv("AZURE_API_KEY", raising=False)
config = AzureAIResponsesAPIConfig()
headers = config.validate_environment(
headers={},
model="gpt-5.6-luna",
litellm_params=GenericLiteLLMParams(api_base="https://res.services.ai.azure.com/api/projects/proj"),
)
assert headers["Content-Type"] == "application/json"
assert "api-key" not in headers
@pytest.mark.asyncio
async def test_aresponses_routes_to_native_endpoint_with_reasoning_and_tools():
"""Core LIT-4427 regression: reasoning_effort + function tools must be sent to the