Update agent.py

agent.py

@@ -88,10 +88,8 @@ workflow.set_entry_point("plan")
 workflow.add_edge("plan", "act")
 workflow.add_edge("act", "observe")
 workflow.add_edge("observe", "decide")
-workflow.add_conditional_edges("decide", {
-    "plan": "plan",
-    END: "summarize"
-})
+workflow.add_conditional_edge("decide", lambda state: "plan" if len(state["thoughts"]) < 3 else "summarize")
+
 workflow.add_edge("summarize", END)
 
 agent = workflow.compile()