Update agents/graph_builder_langgraph.py

agents/graph_builder_langgraph.py

@@ -55,15 +55,23 @@ async def node_generate(state: State, config=None, runtime=None) -> State:
     res = await generate_ui_html(state["requirements"], reference_html)
     return {"html": res["html"], "attempts": state.get("attempts", 0) + 1}
 
+a_guard functions (replace both completely)
 async def node_qa(state: State, config=None, runtime=None) -> State:
-    """Runs QA and guarantees dict output."""
+    """Runs QA and guarantees normalized output."""
     qa_result = await run_ui_qa(state["html"])
     if not isinstance(qa_result, dict):
         qa_result = {"score": 0, "comment": str(qa_result)}
-    if "score" not in qa_result:
+
+    # Normalize values
+    try:
+        qa_result["score"] = int(qa_result.get("score", 0))
+    except Exception:
         qa_result["score"] = 0
-    if "comment" not in qa_result:
-        qa_result["comment"] = "No comment"
+
+    if not isinstance(qa_result.get("comment"), str):
+        qa_result["comment"] = str(qa_result.get("comment", "No comment"))
+
+    print(f"[QA NODE OUTPUT] => {qa_result}")  # debug log
     return {"qa": qa_result}
 
 async def node_repair(state: State, config=None, runtime=None) -> State:
@@ -91,8 +99,8 @@ async def node_report(state: State, config=None, runtime=None) -> State:
 
 # ----- conditional guard -----
 def qa_guard(state: State) -> str:
-    """Decide whether to proceed to report or repair."""
-    qa_data = state.get("qa", {})
+    """Route flow safely based on QA result."""
+    qa_data = state.get("qa") or {}
     if not isinstance(qa_data, dict):
         qa_data = {"score": 0, "comment": "Invalid QA format"}
 
@@ -105,8 +113,7 @@ def qa_guard(state: State) -> str:
     max_retries = int(os.environ.get("MAX_RETRIES", "2"))
     threshold = int(os.environ.get("QA_THRESHOLD", "85"))
 
-    if os.environ.get("DEBUG_QA_GUARD", "0") == "1":
-        print(f"[QA GUARD] score={score}, attempts={attempts}")
+    print(f"[QA GUARD] score={score}, attempts={attempts}, threshold={threshold}")
 
     if score >= threshold or attempts >= max_retries:
         return "report"