fix: don't include entities as not needed

planning_ai/graph.py

@@ -14,14 +14,15 @@ from planning_ai.states import OverallState
 
 def create_graph():
     graph = StateGraph(OverallState)
-    graph.add_node("add_entities", add_entities)
+    # graph.add_node("add_entities", add_entities)
     graph.add_node("generate_summary", generate_summary)
     graph.add_node("check_hallucination", check_hallucination)
     graph.add_node("fix_hallucination", fix_hallucination)
     graph.add_node("generate_final_report", generate_final_report)
 
-    graph.add_edge(START, "add_entities")
-    graph.add_conditional_edges("add_entities", map_documents, ["generate_summary"])
+    # graph.add_edge(START, "add_entities")
+    graph.add_conditional_edges(START, map_documents, ["generate_summary"])
+    # graph.add_conditional_edges("add_entities", map_documents, ["generate_summary"])
     graph.add_conditional_edges("generate_summary", map_check, ["check_hallucination"])
     graph.add_conditional_edges("check_hallucination", map_fix, ["fix_hallucination"])
     graph.add_conditional_edges("fix_hallucination", map_check, ["check_hallucination"])