File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -75,6 +75,8 @@ async def generate_with_context(
7575 "o4-" , # o4-mini
7676 # GPT-5 series are also reasoning models
7777 "gpt-5-" , "gpt-5" # gpt-5, gpt-5-mini, gpt-5-nano
78+ # The GPT OSS series are also reasoning models
79+ "gpt-oss-120b" , "gpt-oss-20b"
7880 )
7981
8082 # Check if this is an OpenAI reasoning model
@@ -106,6 +108,10 @@ async def generate_with_context(
106108 "max_tokens" : kwargs .get ("max_tokens" , self .max_tokens ),
107109 }
108110
111+ # Handle reasoning_effort for open source reasoning models.
112+ if "reasoning_effort" in kwargs :
113+ params ["reasoning_effort" ] = kwargs ["reasoning_effort" ]
114+
109115 # Add seed parameter for reproducibility if configured
110116 # Skip seed parameter for Google AI Studio endpoint as it doesn't support it
111117 seed = kwargs .get ("seed" , self .random_seed )
You can’t perform that action at this time.
0 commit comments