Updated LangGraph reflection agent with max_rounds logic.: update main.py

2026-06-11 15:17:18 +00:00
parent 159baf16f9
commit a64cb75635
1 changed files with 6 additions and 1 deletions
@@ -67,7 +67,12 @@ builder.set_entry_point("draft_answer")
 builder.add_edge("draft_answer", "reflect")
 builder.add_conditional_edges(
    "reflect",
-    lambda state: "END" if state["verdict"] == "ok" else "rewrite",
+    lambda state: (
+        "END"
+        if state["verdict"] == "ok"
+        or state.get("round", 0) >= state.get("max_rounds", 2)
+        else "rewrite"
+    ),
 )
 builder.add_edge("rewrite", "reflect")