Ingestion Pipeline: Easy import Utils

Author: Kavyansh Chourasia

components/ingestion-pipeline/src/ingestion_pipeline/utils/__init__.py

@@ -0,0 +1,16 @@
+"""
+Utils package for ingestion pipeline.
+Provides easy access to all utility classes and functions.
+"""
+
+# Import all utility classes and functions
+from .pdf_splitter import PDFSplitter
+from .toc_page_finder import TOCPageFinder
+from .toc_extractor import TableOfContentsExtractor
+
+# Define what gets imported with "from utils import *"
+__all__ = [
+    "PDFSplitter",
+    "TOCPageFinder",
+    "TableOfContentsExtractor"
+]

components/ingestion-pipeline/tests/test_toc_page_finder.py

@@ -1,7 +1,7 @@
 import pytest
 import logging
 import os
-from ingestion_pipeline.utils.toc_page_finder import TOCPageFinder
+from ingestion_pipeline.utils import TOCPageFinder
 
 @pytest.fixture
 def azure_openai_credentials():