From d9565329aaeb1fdba84faf738152ae059d02d6d2 Mon Sep 17 00:00:00 2001 From: Baptiste Pasquier Date: Tue, 26 Mar 2024 23:02:35 +0100 Subject: [PATCH] :sparkles: Add logger message for completion of PDF processing --- backend/rag_1/ingest.py | 2 ++ backend/rag_2/ingest.py | 2 ++ backend/rag_3/ingest.py | 2 ++ 3 files changed, 6 insertions(+) diff --git a/backend/rag_1/ingest.py b/backend/rag_1/ingest.py index 3e2ea75..a2b1b3c 100644 --- 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 index bda0397..fbe7fcc 100644 --- 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 index 38e484f..72bdd31 100644 --- 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()