diff --git a/src/graph/nodes.py b/src/graph/nodes.py index 7c4a625..5fa574e 100644 --- a/src/graph/nodes.py +++ b/src/graph/nodes.py @@ -101,8 +101,10 @@ def planner_node( } ] - if AGENT_LLM_MAP["planner"] == "basic" and not configurable.enable_deep_thinking: - llm = get_llm_by_type(AGENT_LLM_MAP["planner"]).with_structured_output( + if configurable.enable_deep_thinking: + llm = get_llm_by_type("reasoning") + elif AGENT_LLM_MAP["planner"] == "basic": + llm = get_llm_by_type("basic").with_structured_output( Plan, method="json_mode", )