scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-24 00:52:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 91

fix(xai): accept max_completion_tokens as a supported param

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-09-21 22:37:13 +00:00
parent 5c0589207c
commit 4ff251e6b1
1 changed files with 13 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/test_litellm/llms/xai/test_xai_chat_transformation.py
View file

@ -13,6 +13,7 @@ from litellm.types.utils import (
ModelResponse,
Usage,
)
from litellm.utils import get_optional_params
class TestXAIReasoningTokenFolding:
@ -309,3 +310,15 @@ class TestXAIReportedCost:
0.0,
0.0037756,
)
class TestXAIMaxCompletionTokens:
def test_max_completion_tokens_is_accepted_and_mapped_to_max_tokens(self):
result = get_optional_params(
model="grok-4.20-beta",
custom_llm_provider="xai",
max_completion_tokens=64,
)
assert result["max_tokens"] == 64
assert "max_completion_tokens" not in result