Update graph_builder.py

graph_builder.py

@@ -711,17 +711,9 @@ def build_document_extraction_graph():
     workflow.add_edge("extract_study_design", "store_in_knowledge_base")
     workflow.add_edge("store_in_knowledge_base", END)
     
-    # Handle errors - any node can output an error
-    for node in workflow.nodes:
-        # Check if status is error, if yes, go to END
-        workflow.add_conditional_edges(
-            node,
-            lambda state: "status" in state and state["status"] == "error",
-            {
-                True: END,
-                False: None  # Let the normal flow continue
-            }
-        )
+    # Instead of using conditional edges for all nodes, 
+    # let each function handle its own error status
+    # This simplifies the graph structure and avoids the conditional edge issue
     
     workflow.set_entry_point("parse_document")
     return workflow.compile()