diff --git a/tools/doc_generator/src/main.py b/tools/doc_generator/src/main.py index 40dd07c..31f1b92 100644 --- a/tools/doc_generator/src/main.py +++ b/tools/doc_generator/src/main.py @@ -170,4 +170,7 @@ def save_and_increment_index(state: DocumentProcessingState) -> DocumentProcessi # Execute if __name__ == "__main__": - document_processing_app.invoke({"document_index": 0, "llm_output_content": ""}) + document_processing_app.invoke( + {"document_index": 0, "llm_output_content": ""}, + {"recursion_limit": MAX_DOCUMENTS}, + )