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
+2 -2
View File
@@ -94,10 +94,10 @@ def _model_slug(model_name: str) -> str:
def _requires_max_completion_tokens(model_name: str) -> bool:
"""Return True for models that reject ``max_tokens`` (GPT-5 family, o3/o4)."""
"""Return True for models that reject ``max_tokens`` (GPT-5 family, o-series)."""
slug = _model_slug(model_name)
return "gpt-5" in slug or any(
slug == p or slug.startswith((p + "-", p + ".")) for p in ("o3", "o4")
slug == p or slug.startswith((p + "-", p + ".")) for p in ("o1", "o3", "o4")
)