(no commit message)

main.py

@@ -37,8 +37,9 @@ class DocWriterProgram(PrecompiledProgram):
             verbose=True,
         )
 
-    def forward(self, source_root: str, source_tree: dict[str, Any]) -> dspy.Prediction:
+    def forward(self, source_root: str) -> dspy.Prediction:
         source_tree = load_source_tree(source_root)
+        print(f"Loaded source tree with {len(source_tree)} entries")
         return self.doc_writer(source_tree=source_tree)
 
 
@@ -62,7 +63,7 @@ SOURCE_ROOT = "."
 def main():
     """
     print("Starting documentation generation...")
-    result = doc_writer(source_root=SOURCE_ROOT, source_tree=None)
+    result = doc_writer(source_root=SOURCE_ROOT)
 
     with open("generated_documentation.md", "w", encoding="utf-8") as f:
         print("Writing documentation to file...")