From fd6887c274857dd788f045c3412dca84823e2920 Mon Sep 17 00:00:00 2001 From: Xubin Ren Date: Tue, 12 May 2026 03:33:04 +0000 Subject: [PATCH] test(providers): cover VolcEngine token parameter Co-authored-by: Cursor --- tests/providers/test_litellm_kwargs.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/providers/test_litellm_kwargs.py b/tests/providers/test_litellm_kwargs.py index 94455fd40..c2e9efeba 100644 --- a/tests/providers/test_litellm_kwargs.py +++ b/tests/providers/test_litellm_kwargs.py @@ -847,6 +847,18 @@ def test_volcengine_thinking_enabled() -> None: assert kw["extra_body"] == {"thinking": {"type": "enabled"}} +def test_volcengine_uses_max_completion_tokens() -> None: + kw = _build_kwargs_for("volcengine", "doubao-seed-2-0-pro") + assert kw["max_completion_tokens"] == 1024 + assert "max_tokens" not in kw + + +def test_volcengine_coding_plan_uses_max_completion_tokens() -> None: + kw = _build_kwargs_for("volcengine_coding_plan", "doubao-seed-2-0-pro") + assert kw["max_completion_tokens"] == 1024 + assert "max_tokens" not in kw + + def test_byteplus_thinking_disabled_for_minimal() -> None: kw = _build_kwargs_for("byteplus", "doubao-seed-2-0-pro", reasoning_effort="minimal") assert kw["extra_body"] == {"thinking": {"type": "disabled"}}