From a64cb756353d75d9ae4a73fa383dc576ed4867a2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=A0=D0=B8=D0=BD=D0=B0=D1=80=20=D0=9C=D0=B8=D1=80=D0=B7?= =?UTF-8?q?=D0=B0=D0=B3=D0=B8=D1=82=D0=BE=D0=B2?= Date: Thu, 11 Jun 2026 15:17:18 +0000 Subject: [PATCH] Updated LangGraph reflection agent with max_rounds logic.: update main.py --- main.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 40d2f2e..672df5f 100644 --- a/main.py +++ b/main.py @@ -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")