gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge branch 'main' into shared-tasks

Browse source
This commit is contained in:
Edwin Jose 2025-09-05 12:15:03 -04:00 • committed by GitHub
commit bb5e91969d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
warm_up_docling.py
View file

@ -1,16 +1,17 @@
from docling.document_converter import DocumentConverter from docling.document_converter import DocumentConverter
from src.utils.logging_config import get_logger import logging
logger = get_logger(__name__) logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
logger.info("Warming up docling models") logger.info("Warming up docling models")
try: try:
# Use the sample document to warm up docling # Use the sample document to warm up docling
test_file = "/app/warmup_ocr.pdf" test_file = "/app/warmup_ocr.pdf"
logger.info("Using test file to warm up docling", test_file=test_file) logger.info(f"Using test file to warm up docling: {test_file}")
DocumentConverter().convert(test_file) DocumentConverter().convert(test_file)
logger.info("Docling models warmed up successfully") logger.info("Docling models warmed up successfully")
except Exception as e: except Exception as e:
logger.info("Docling warm-up completed with exception", error=str(e)) logger.info(f"Docling warm-up completed with exception: {str(e)}")
# This is expected - we just want to trigger the model downloads # This is expected - we just want to trigger the model downloads