add documentation

planning_ai/graph.py

@@ -13,11 +13,22 @@ from planning_ai.states import OverallState
 
 
 def create_graph():
+    """Creates and compiles a state graph for document processing.
+
+    This function sets up a state graph using the `StateGraph` class, defining nodes
+    and edges for processing documents. It includes nodes for generating summaries,
+    checking and fixing hallucinations, and generating a final summary. Conditional
+    edges are added to manage the flow of data between nodes based on the processing
+    state.
+
+    Returns:
+        StateGraph: The compiled state graph ready for execution.
+    """
     graph = StateGraph(OverallState)
     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_summary", generate_final_summary)
+    # graph.add_node("generate_final_summary", generate_final_summary)
 
     graph.add_conditional_edges(
         START,
@@ -40,7 +51,7 @@ def create_graph():
         ["check_hallucination"],
     )
 
-    graph.add_edge("check_hallucination", "generate_final_summary")
-    graph.add_edge("generate_final_summary", END)
+    # graph.add_edge("check_hallucination", "generate_final_summary")
+    # graph.add_edge("generate_final_summary", END)
 
     return graph.compile()

planning_ai/main.py

@@ -174,7 +174,7 @@ def build_quarto_doc(doc_title, out):
 
 def main():
     loader = DirectoryLoader(
-        path=str(Paths.STAGING),
+        path=str(Paths.STAGING / "pdfs"),
         show_progress=True,
         use_multithreading=True,
         loader_cls=TextLoader,
@@ -208,7 +208,7 @@ if __name__ == "__main__":
 
     tic = time.time()
     out = main()
-    build_quarto_doc(doc_title, out)
+    # build_quarto_doc(doc_title, out)
     toc = time.time()
 
     print(f"Time taken: {(toc - tic) / 60:.2f} minutes.")