feat: add top-level reasoning toggle to ADE (#3813)

This commit is contained in:
cthomas
2025-08-07 17:21:12 -07:00
committed by GitHub
parent 6a713f9ae7
commit 2ee56f027f

View File

@@ -187,6 +187,8 @@ class LLMConfig(BaseModel):
@classmethod
def apply_reasoning_setting_to_config(cls, config: "LLMConfig", reasoning: bool):
if reasoning:
config.enable_reasoner = True
if (
config.model_endpoint_type == "anthropic"
and ("claude-opus-4" in config.model or "claude-sonnet-4" in config.model or "claude-3-7-sonnet" in config.model)
@@ -194,19 +196,19 @@ class LLMConfig(BaseModel):
config.model_endpoint_type == "google_vertex" and ("gemini-2.5-flash" in config.model or "gemini-2.0-pro" in config.model)
):
config.put_inner_thoughts_in_kwargs = False
config.enable_reasoner = True
if config.max_reasoning_tokens == 0:
config.max_reasoning_tokens = 1024
elif config.model_endpoint_type == "openai" and (
config.model.startswith("o1") or config.model.startswith("o3") or config.model.startswith("o4")
):
config.put_inner_thoughts_in_kwargs = True
config.enable_reasoner = True
config.put_inner_thoughts_in_kwargs = False
if config.reasoning_effort is None:
config.reasoning_effort = "medium"
else:
config.put_inner_thoughts_in_kwargs = True
config.enable_reasoner = False
else:
config.put_inner_thoughts_in_kwargs = False
config.enable_reasoner = False
config.put_inner_thoughts_in_kwargs = False
return config