From cf6bf04c88035cf40afa1e8900cca4892b469f41 Mon Sep 17 00:00:00 2001 From: Amol Kapoor Date: Sat, 23 Aug 2025 23:48:46 -0400 Subject: [PATCH] Add support for gpt-oss-* models as reasoning models, and allow for reasoning_effort params --- openevolve/llm/openai.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/openevolve/llm/openai.py b/openevolve/llm/openai.py index bf5dde16..243ac02a 100644 --- a/openevolve/llm/openai.py +++ b/openevolve/llm/openai.py @@ -75,6 +75,8 @@ async def generate_with_context( "o4-", # o4-mini # GPT-5 series are also reasoning models "gpt-5-", "gpt-5" # gpt-5, gpt-5-mini, gpt-5-nano + # The GPT OSS series are also reasoning models + "gpt-oss-120b", "gpt-oss-20b" ) # Check if this is an OpenAI reasoning model @@ -106,6 +108,10 @@ async def generate_with_context( "max_tokens": kwargs.get("max_tokens", self.max_tokens), } + # Handle reasoning_effort for open source reasoning models. + if "reasoning_effort" in kwargs: + params["reasoning_effort"] = kwargs["reasoning_effort"] + # Add seed parameter for reproducibility if configured # Skip seed parameter for Google AI Studio endpoint as it doesn't support it seed = kwargs.get("seed", self.random_seed)