fix: cover o1 max-completion token fallback

Maintainer edit: keep the GPT-5/o-series fallback on slug-boundary matching so unrelated model names are not caught by substring checks, and include o1 alongside o3/o4 because it is also an o-series chat model.
This commit is contained in:
chengyongru
2026-06-10 14:47:10 +08:00
committed by Xubin Ren
parent a779e7c29e
commit 99f7f371fa
2 changed files with 6 additions and 2 deletions
+4
View File
@@ -933,8 +933,12 @@ def test_openai_compat_build_kwargs_uses_gpt5_safe_parameters() -> None:
("model_name", "expected_key"),
[
("gpt-5.4", "max_completion_tokens"),
("o1-mini", "max_completion_tokens"),
("o3-mini", "max_completion_tokens"),
("o4-mini", "max_completion_tokens"),
("gpt-4", "max_tokens"),
("foo3-mini", "max_tokens"),
("foo4-mini", "max_tokens"),
],
)
def test_openai_compat_build_kwargs_max_completion_tokens_by_model_name(