fix: centralize logging configuration to respect LOGLEVEL environment variable

- Move logging configuration to top of __init__.py to ensure it runs before any imports
- Remove duplicate logging.basicConfig from llm/__init__.py that was overriding settings
- Clean up main.py to remove redundant logging configuration
- Add test_logging.py script to verify the fix works correctly

This ensures that the LOGLEVEL environment variable (DEBUG, INFO, WARNING, ERROR) properly controls logging verbosity throughout the package.

Fixes #831

Co-authored-by: Mervin Praison <[email protected]>

Author: claude[bot]

src/praisonai-agents/praisonaiagents/__init__.py

@@ -2,6 +2,30 @@
 Praison AI Agents - A package for hierarchical AI agent task execution
 """
 
+# Configure logging before any other imports
+import os
+import logging
+from rich.logging import RichHandler
+
+# Get log level from environment variable
+LOGLEVEL = os.environ.get('LOGLEVEL', 'INFO').upper()
+
+# Configure root logger
+logging.basicConfig(
+    level=getattr(logging, LOGLEVEL, logging.INFO),
+    format="%(asctime)s %(filename)s:%(lineno)d %(levelname)s %(message)s",
+    datefmt="[%X]",
+    handlers=[RichHandler(rich_tracebacks=True)]
+)
+
+# Suppress specific noisy loggers
+logging.getLogger("litellm").setLevel(logging.WARNING)
+logging.getLogger("litellm.utils").setLevel(logging.WARNING)
+logging.getLogger("markdown_it").setLevel(logging.WARNING)
+logging.getLogger("rich.markdown").setLevel(logging.WARNING)
+logging.getLogger("httpx").setLevel(logging.WARNING)
+logging.getLogger("httpcore").setLevel(logging.WARNING)
+
 from .agent.agent import Agent
 from .agent.image_agent import ImageAgent
 from .agents.agents import PraisonAIAgents

src/praisonai-agents/praisonaiagents/llm/__init__.py

@@ -15,9 +15,6 @@
 # Suppress pydantic warnings
 warnings.filterwarnings("ignore", category=UserWarning, module="pydantic")
 
-# Configure logging to suppress all INFO messages
-logging.basicConfig(level=logging.WARNING)
-
 # Import after suppressing warnings
 from .llm import LLM, LLMContextLengthExceededException
 from .openai_client import (

src/praisonai-agents/praisonaiagents/main.py

@@ -9,32 +9,15 @@
 from rich.panel import Panel
 from rich.text import Text
 from rich.markdown import Markdown
-from rich.logging import RichHandler
 from rich.live import Live
 import asyncio
 
-# # Configure root logger
-# logging.basicConfig(level=logging.WARNING)
-
-# Suppress litellm logs
+# Logging is already configured in __init__.py, just clean up handlers for litellm
 logging.getLogger("litellm").handlers = []
 logging.getLogger("litellm.utils").handlers = []
 logging.getLogger("litellm").propagate = False
 logging.getLogger("litellm.utils").propagate = False
 
-LOGLEVEL = os.environ.get('LOGLEVEL', 'INFO').upper()
-
-logging.basicConfig(
-    level=getattr(logging, LOGLEVEL, logging.INFO),
-    format="%(asctime)s %(filename)s:%(lineno)d %(levelname)s %(message)s",
-    datefmt="[%X]",
-    handlers=[RichHandler(rich_tracebacks=True)]
-)
-
-# Add these lines to suppress markdown parser debug logs
-logging.getLogger('markdown_it').setLevel(logging.WARNING)
-logging.getLogger('rich.markdown').setLevel(logging.WARNING)
-
 # Global list to store error logs
 error_logs = []
 

src/praisonai-agents/test_logging.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python3
+"""Test script to verify logging is working correctly with LOGLEVEL environment variable"""
+
+import os
+import sys
+import logging
+
+# Test 1: Show current environment variable
+print(f"Current LOGLEVEL environment variable: {os.environ.get('LOGLEVEL', 'Not set')}")
+
+# Test 2: Import praisonaiagents and check logging level
+print("\nImporting praisonaiagents...")
+import praisonaiagents
+
+print(f"Root logger level after import: {logging.getLevelName(logging.getLogger().getEffectiveLevel())}")
+
+# Test 3: Create a test agent and check its logging
+print("\nCreating test agent...")
+from praisonaiagents import Agent
+
+agent = Agent(
+    name="TestAgent",
+    role="Logger Tester",
+    goal="Test if logging works",
+    backstory="I am a test agent created to verify logging functionality"
+)
+
+# Test 4: Test different log levels
+print("\nTesting different log levels:")
+test_logger = logging.getLogger(__name__)
+
+test_logger.debug("This is a DEBUG message - should show only when LOGLEVEL=DEBUG")
+test_logger.info("This is an INFO message - should show when LOGLEVEL=INFO or DEBUG")
+test_logger.warning("This is a WARNING message - should always show")
+test_logger.error("This is an ERROR message - should always show")
+
+print("\nTest complete! Check above for log messages based on your LOGLEVEL setting.")
+print("\nTo test different levels, run:")
+print("  export LOGLEVEL=DEBUG && python test_logging.py")
+print("  export LOGLEVEL=INFO && python test_logging.py")
+print("  export LOGLEVEL=WARNING && python test_logging.py")