From 396c59b3e2ab96956994fef41c21eced3d2a0977 Mon Sep 17 00:00:00 2001
From: James Braza <jamesbraza@gmail.com>
Date: Wed, 9 Oct 2024 16:07:19 -0700
Subject: [PATCH] Cleaned up log verbosity code (#554)

---
 paperqa/agents/__init__.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/paperqa/agents/__init__.py b/paperqa/agents/__init__.py
index 579afea43..3da5414bb 100644
--- a/paperqa/agents/__init__.py
+++ b/paperqa/agents/__init__.py
@@ -18,7 +18,7 @@
 
 logger = logging.getLogger(__name__)
 
-LOG_VERBOSITY_MAP = {
+LOG_VERBOSITY_MAP: dict[int, dict[str, int]] = {
     0: {
         "paperqa.agents": logging.INFO,
         "paperqa.agents.helpers": logging.WARNING,
@@ -28,6 +28,7 @@
         "paperqa.agents.search": logging.INFO,
         "anthropic": logging.WARNING,
         "openai": logging.WARNING,
+        "httpcore": logging.WARNING,
         "httpx": logging.WARNING,
         "LiteLLM": logging.WARNING,
         "LiteLLM Router": logging.WARNING,
@@ -51,6 +52,7 @@
 LOG_VERBOSITY_MAP[3] = LOG_VERBOSITY_MAP[2] | {
     "LiteLLM": logging.DEBUG,  # <-- every single LLM call
 }
+_MAX_PRESET_VERBOSITY: int = max(k for k in LOG_VERBOSITY_MAP)
 
 _PAPERQA_PKG_ROOT_LOGGER = logging.getLogger(__name__.split(".", maxsplit=1)[0])
 _INITIATED_FROM_CLI = False
@@ -75,12 +77,10 @@ def set_up_rich_handler(install: bool = True) -> RichHandler:
 
 
 def configure_log_verbosity(verbosity: int = 0) -> None:
-    max_preset_verbosity: int = max(list(LOG_VERBOSITY_MAP.keys()))
+    key = min(verbosity, _MAX_PRESET_VERBOSITY)
     for logger_name, logger_ in logging.Logger.manager.loggerDict.items():
         if isinstance(logger_, logging.Logger) and (
-            log_level := LOG_VERBOSITY_MAP.get(
-                min(verbosity, max_preset_verbosity), {}
-            ).get(logger_name)
+            log_level := LOG_VERBOSITY_MAP.get(key, {}).get(logger_name)
         ):
             logger_.setLevel(log_level)