diff --git a/src/haiku/rag/logging.py b/src/haiku/rag/logging.py index 3627a5e9..34aeaf5a 100644 --- a/src/haiku/rag/logging.py +++ b/src/haiku/rag/logging.py @@ -6,6 +6,7 @@ from rich.logging import RichHandler logging.basicConfig(level=logging.DEBUG) logging.getLogger("httpx").setLevel(logging.WARNING) logging.getLogger("httpcore").setLevel(logging.WARNING) +logging.getLogger("docling").setLevel(logging.WARNING) def get_logger() -> logging.Logger: diff --git a/tests/generate_benchmark_db.py b/tests/generate_benchmark_db.py index 7318a27b..47baabd9 100644 --- a/tests/generate_benchmark_db.py +++ b/tests/generate_benchmark_db.py @@ -6,6 +6,7 @@ from llm_judge import LLMJudge from rich.console import Console from rich.progress import Progress +from haiku.rag import logging # noqa from haiku.rag.client import HaikuRAG from haiku.rag.qa import get_qa_agent