Update agent.py

agent.py

@@ -88,7 +88,12 @@ workflow.set_entry_point("plan")
 workflow.add_edge("plan", "act")
 workflow.add_edge("act", "observe")
 workflow.add_edge("observe", "decide")
-workflow.add_conditional_edge("decide", lambda state: "plan" if len(state["thoughts"]) < 3 else "summarize")
+def route_decision(state: AgentState) -> str:
+    return "plan" if len(state["thoughts"]) < 3 else "summarize"
+
+workflow.add_conditional_edges("decide", {
+    "plan": route_decision
+})
 
 workflow.add_edge("summarize", END)