diff --git a/python/sglang/lang/backend/openai.py b/python/sglang/lang/backend/openai.py index 502d2f3e4..a2d006bb7 100644 --- a/python/sglang/lang/backend/openai.py +++ b/python/sglang/lang/backend/openai.py @@ -161,7 +161,11 @@ class OpenAI(BaseBackend): prompt = s.text_ kwargs = sampling_params.to_openai_kwargs() - if self.model_name.startswith("o1") or self.model_name.startswith("o3"): + if ( + self.model_name.startswith("o1") + or self.model_name.startswith("o3") + or "o1" in self.model_name + ): kwargs.pop("max_tokens", None) else: kwargs.pop("max_completion_tokens", None)