Update agents/graph_builder_langgraph.py

agents/graph_builder_langgraph.py

@@ -32,11 +32,22 @@ async def node_extract(state: State, config=None, runtime=None) -> State:
     return {"requirements": reqs}
 
 async def node_generate(state: State, config=None, runtime=None) -> State:
-    # Safely read reference HTML from runtime or env
+    """
+    Generate recruitment-style UI HTML using requirements and reference template.
+    Works with both old and new LangGraph runtime objects.
+    """
     reference_html = None
-    if runtime and getattr(runtime.graph, "config", None):
-        reference_html = runtime.graph.config.get("reference_html")
 
+    # ✅ Handle all known runtime configurations
+    if runtime:
+        # Newer LangGraph (>=0.2)
+        if hasattr(runtime, "graph") and getattr(runtime.graph, "config", None):
+            reference_html = runtime.graph.config.get("reference_html")
+        # Older LangGraph (<0.2)
+        elif hasattr(runtime, "config"):
+            reference_html = runtime.config.get("reference_html")
+
+    # Fallback to local file if none supplied
     if not reference_html and os.path.exists("templates/demo_qms_design.html"):
         with open("templates/demo_qms_design.html", "r", encoding="utf-8") as f:
             reference_html = f.read()