diff --git a/src/graph/nodes.py b/src/graph/nodes.py index ec2c31b..ec85929 100644 --- a/src/graph/nodes.py +++ b/src/graph/nodes.py @@ -104,7 +104,8 @@ def planner_node( if AGENT_LLM_MAP["planner"] == "basic": llm = get_llm_by_type(AGENT_LLM_MAP["planner"]).with_structured_output( Plan, - method="json_mode", + method="json_schema", + strict=True, ) else: llm = get_llm_by_type(AGENT_LLM_MAP["planner"])