feat: setting CHATLAS_LOG now enables other relevant, provider specific, logs (#97)

* feat: setting CHATLAS_LOG will enable other relevant, provider specific, logs

* Update changelog

Author: cpsievert

CHANGELOG.md

@@ -15,6 +15,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Improvements
 
+* The `CHATLAS_LOG` environment variable will now enable logs for the relevant model provider. It now also supports a lovel of `debug` in addition to `info`. (#97)
 * `Chat` instances can now be deep copied, which is useful for forking the chat session. (#96)
 
 ### Bug fixes

chatlas/_logging.py

@@ -1,6 +1,7 @@
 import logging
 import os
 import warnings
+from typing import Literal
 
 from rich.logging import RichHandler
 
@@ -12,15 +13,38 @@ def _rich_handler() -> RichHandler:
     return handler
 
 
-logger = logging.getLogger("chatlas")
-
-if os.environ.get("CHATLAS_LOG") == "info":
+def setup_logger(x: str, level: Literal["debug", "info"]) -> logging.Logger:
+    logger = logging.getLogger(x)
+    if level == "debug":
+        logger.setLevel(logging.DEBUG)
+    elif level == "info":
+        logger.setLevel(logging.INFO)
     # By adding a RichHandler to chatlas' logger, we can guarantee that they
     # never get dropped, even if the root logger's handlers are not
     # RichHandlers.
-    logger.setLevel(logging.INFO)
-    logger.addHandler(_rich_handler())
+    if not any(isinstance(h, RichHandler) for h in logger.handlers):
+        logger.addHandler(_rich_handler())
     logger.propagate = False
+    return logger
+
+
+logger = logging.getLogger("chatlas")
+log_level = os.environ.get("CHATLAS_LOG")
+if log_level:
+    if log_level != "debug" and log_level != "info":
+        warnings.warn(
+            f"CHATLAS_LOG is set to '{log_level}', but the log level must "
+            "be one of 'debug' or 'info'. Defaulting to 'info'.",
+        )
+        log_level = "info"
+
+    # Manually setup the logger for each dependency we care about. This way, we
+    # can ensure that the logs won't get dropped when a rich display is activate
+    logger = setup_logger("chatlas", log_level)
+    openai_logger = setup_logger("openai", log_level)
+    anthropic_logger = setup_logger("anthropic", log_level)
+    google_logger = setup_logger("google_genai.models", log_level)
+    httpx_logger = setup_logger("httpx", log_level)
 
     # Add a RichHandler to the root logger if there are no handlers. Note that
     # if chatlas is imported before other libraries that set up logging, (like