✨ Add logger message for completion of PDF processing

artefactory · Mar 26, 2024 · d956532 · d956532
1 parent ad4ad14
commit d956532
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/backend/rag_1/ingest.py b/backend/rag_1/ingest.py
@@ -129,6 +129,8 @@ def main(config: DictConfig) -> None:
     for file_path in tqdm(sorted(docs_folder.glob("**/*.pdf"))):
         ingest_pdf(file_path, config)
 
+    logger.info("Finished processing all PDF files")
+
 
 if __name__ == "__main__":
     main()
diff --git a/backend/rag_2/ingest.py b/backend/rag_2/ingest.py
@@ -155,6 +155,8 @@ def main(config: DictConfig) -> None:
         loop = asyncio.get_event_loop()
         loop.run_until_complete(ingest_pdf(file_path, config))
 
+    logger.info("Finished processing all PDF files")
+
 
 if __name__ == "__main__":
     main()
diff --git a/backend/rag_3/ingest.py b/backend/rag_3/ingest.py
@@ -155,6 +155,8 @@ def main(config: DictConfig) -> None:
         loop = asyncio.get_event_loop()
         loop.run_until_complete(ingest_pdf(file_path, config))
 
+    logger.info("Finished processing all PDF files")
+
 
 if __name__ == "__main__":
     main()