scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test(azure_ai): cover max token mapping branches

Browse source
This commit is contained in:
Vic Wen (Manpower Services Taiwan Co Ltd) 2026-09-20 23:15:49 +08:00
parent 9a1aff1bf5
commit 154aa39e94
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
tests/unit/llms/azure_ai/chat/test_azure_ai_transformation.py
View file

@ -158,6 +158,29 @@ def test_azure_ai_maps_max_completion_tokens_to_max_tokens():
assert "max_completion_tokens" not in mapped_params
def test_azure_ai_keeps_max_completion_tokens_for_gpt_5():
mapped_params: Final = litellm.get_optional_params(
model="gpt-5",
custom_llm_provider="azure_ai",
max_completion_tokens=256,
)
assert mapped_params["max_completion_tokens"] == 256
assert "max_tokens" not in mapped_params
def test_azure_ai_keeps_params_without_max_completion_tokens():
mapped_params: Final = litellm.get_optional_params(
model="mistral-large-3",
custom_llm_provider="azure_ai",
temperature=0.2,
)
assert mapped_params["temperature"] == 0.2
assert "max_tokens" not in mapped_params
assert "max_completion_tokens" not in mapped_params
def test_foundry_gpt_6_astra_keeps_sampling_params_when_reasoning_effort_is_none(_local_model_cost_map):
optional_params = AzureAIStudioConfig().map_openai_params(
non_default_params={"reasoning_effort": "none", "temperature": 0.2, "top_p": 0.9},