praisonaiagents 0.0.145__tar.gz → 0.0.147__tar.gz

{praisonaiagents-0.0.145 → praisonaiagents-0.0.147}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praisonaiagents
-Version: 0.0.145
+Version: 0.0.147
 Summary: Praison AI agents for completing complex tasks with Self Reflection Agents
 Author: Mervin Praison
 Requires-Python: >=3.10
@@ -15,7 +15,7 @@ Requires-Dist: fastapi>=0.115.0; extra == "mcp"
 Requires-Dist: uvicorn>=0.34.0; extra == "mcp"
 Provides-Extra: memory
 Requires-Dist: chromadb>=1.0.0; extra == "memory"
-Requires-Dist: litellm>=1.72.0; extra == "memory"
+Requires-Dist: litellm>=1.72.6; extra == "memory"
 Provides-Extra: knowledge
 Requires-Dist: mem0ai>=0.1.0; extra == "knowledge"
 Requires-Dist: chromadb>=1.0.0; extra == "knowledge"
@@ -25,7 +25,7 @@ Provides-Extra: graph
 Requires-Dist: mem0ai[graph]>=0.1.0; extra == "graph"
 Requires-Dist: chromadb>=1.0.0; extra == "graph"
 Provides-Extra: llm
-Requires-Dist: litellm>=1.72.0; extra == "llm"
+Requires-Dist: litellm>=1.72.6; extra == "llm"
 Requires-Dist: pydantic>=2.4.2; extra == "llm"
 Provides-Extra: api
 Requires-Dist: fastapi>=0.115.0; extra == "api"
@@ -35,6 +35,11 @@ Requires-Dist: posthog>=3.0.0; extra == "telemetry"
 Provides-Extra: mongodb
 Requires-Dist: pymongo>=4.6.3; extra == "mongodb"
 Requires-Dist: motor>=3.4.0; extra == "mongodb"
+Provides-Extra: auth
+Requires-Dist: PyJWT>=2.8.0; extra == "auth"
+Requires-Dist: passlib[bcrypt]>=1.7.4; extra == "auth"
+Requires-Dist: python-jose[cryptography]>=3.3.0; extra == "auth"
+Requires-Dist: python-multipart>=0.0.6; extra == "auth"
 Provides-Extra: all
 Requires-Dist: praisonaiagents[memory]; extra == "all"
 Requires-Dist: praisonaiagents[knowledge]; extra == "all"
@@ -44,3 +49,4 @@ Requires-Dist: praisonaiagents[mcp]; extra == "all"
 Requires-Dist: praisonaiagents[api]; extra == "all"
 Requires-Dist: praisonaiagents[telemetry]; extra == "all"
 Requires-Dist: praisonaiagents[mongodb]; extra == "all"
+Requires-Dist: praisonaiagents[auth]; extra == "all"

{praisonaiagents-0.0.145 → praisonaiagents-0.0.147}/praisonaiagents/__init__.py

@@ -2,30 +2,16 @@
 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
+# Apply warning patch BEFORE any imports to intercept warnings at the source
+from . import _warning_patch
 
-# Get log level from environment variable
-LOGLEVEL = os.environ.get('LOGLEVEL', 'INFO').upper()
+# Import centralized logging configuration FIRST
+from . import _logging
 
-# 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)
+# Configure root logger after logging is initialized
+_logging.configure_root_logger()
 
+# Now import everything else
 from .agent.agent import Agent
 from .agent.image_agent import ImageAgent
 from .agent.context_agent import ContextAgent, create_context_agent
@@ -149,4 +135,4 @@ __all__ = [
 
 # Add MCP to __all__ if available
 if _mcp_available:
-    __all__.append('MCP') 
+    __all__.append('MCP')

praisonaiagents-0.0.147/praisonaiagents/_logging.py

@@ -0,0 +1,134 @@
+"""
+Centralized logging configuration for PraisonAI Agents.
+This module consolidates all logging configuration in one place to avoid duplication.
+"""
+
+import os
+import logging
+from typing import List
+
+# ========================================================================
+# ENVIRONMENT CONFIGURATION
+# ========================================================================
+def _configure_environment():
+    """Set environment variables to suppress debug messages at the source."""
+    env_vars = {
+        # LiteLLM configuration
+        "LITELLM_TELEMETRY": "False",
+        "LITELLM_DROP_PARAMS": "True",
+        "LITELLM_LOG": "ERROR",
+        "LITELLM_DEBUG": "False",
+        "LITELLM_SUPPRESS_DEBUG_INFO": "True",
+        "LITELLM_VERBOSE": "False",
+        "LITELLM_SET_VERBOSE": "False",
+        # HTTPX configuration
+        "HTTPX_DISABLE_WARNINGS": "True",
+        "HTTPX_LOG_LEVEL": "ERROR",
+        # Pydantic configuration
+        "PYDANTIC_WARNINGS_ENABLED": "False",
+    }
+    
+    for key, value in env_vars.items():
+        os.environ[key] = value
+
+
+# ========================================================================
+# LOGGER CONFIGURATION
+# ========================================================================
+def _get_all_noisy_loggers() -> List[str]:
+    """Get list of all loggers that should be suppressed."""
+    return [
+        # LiteLLM and variants
+        "litellm", "LiteLLM", "LiteLLM Router", "LiteLLM Proxy",
+        # HTTP libraries
+        "httpx", "httpx._trace", "httpx._client",
+        "httpcore", "httpcore._trace",
+        # OpenAI
+        "openai._base_client", "openai._client",
+        # Markdown
+        "markdown_it", "rich.markdown",
+        # System
+        "asyncio", "selector_events", "pydantic",
+        "praisonaiagents.telemetry.telemetry",
+    ]
+
+
+def _configure_loggers():
+    """Configure all loggers based on LOGLEVEL environment variable."""
+    loglevel = os.environ.get('LOGLEVEL', 'INFO').upper()
+    
+    # When DEBUG is set, allow some HTTP logging for API endpoints
+    if loglevel == 'DEBUG':
+        allowed_debug_loggers = {"httpx._client", "openai._client"}
+        
+        for logger_name in _get_all_noisy_loggers():
+            if logger_name not in allowed_debug_loggers:
+                logger = logging.getLogger(logger_name)
+                logger.setLevel(logging.CRITICAL)
+                logger.handlers = []
+                logger.propagate = False
+        
+        # Ensure allowed loggers are at INFO level to show API calls
+        for logger_name in allowed_debug_loggers:
+            logger = logging.getLogger(logger_name)
+            logger.setLevel(logging.INFO)
+    else:
+        # Suppress all noisy loggers when not in DEBUG mode
+        for logger_name in _get_all_noisy_loggers():
+            logger = logging.getLogger(logger_name)
+            logger.setLevel(logging.CRITICAL)
+            logger.handlers = []
+            logger.propagate = False
+
+
+# ========================================================================
+# LITELLM CONFIGURATION
+# ========================================================================
+def _configure_litellm():
+    """Configure litellm after it's imported."""
+    try:
+        import litellm
+        litellm.telemetry = False
+        litellm.drop_params = True
+        litellm.suppress_debug_info = True
+        
+        if hasattr(litellm, '_logging_obj'):
+            litellm._logging_obj.setLevel(logging.CRITICAL)
+        
+        if hasattr(litellm, 'set_verbose'):
+            litellm.set_verbose = False
+            
+    except (ImportError, AttributeError):
+        pass
+
+
+# ========================================================================
+# ROOT LOGGER CONFIGURATION
+# ========================================================================
+def configure_root_logger():
+    """Configure the root logger with RichHandler."""
+    from rich.logging import RichHandler
+    
+    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)],
+        force=True
+    )
+
+
+# ========================================================================
+# INITIALIZATION
+# ========================================================================
+def initialize_logging():
+    """Initialize all logging configuration."""
+    _configure_environment()
+    _configure_loggers()
+    _configure_litellm()
+
+
+# Auto-initialize on import
+initialize_logging()

praisonaiagents-0.0.147/praisonaiagents/_warning_patch.py

@@ -0,0 +1,73 @@
+"""
+Minimal warning patch to suppress specific third-party warnings.
+This module patches the warnings module to intercept specific messages.
+"""
+
+import warnings
+import functools
+import sys
+
+# Apply aggressive warning filters first
+warnings.filterwarnings("ignore", message=".*Pydantic serializer warnings.*")
+warnings.filterwarnings("ignore", message=".*PydanticSerializationUnexpectedValue.*")
+warnings.filterwarnings("ignore", message=".*Expected 9 fields but got.*")
+warnings.filterwarnings("ignore", message=".*Expected `StreamingChoices` but got.*")
+warnings.filterwarnings("ignore", message=".*serialized value may not be as expected.*")
+warnings.filterwarnings("ignore", message=".*Use 'content=<...>' to upload raw bytes/text content.*")
+warnings.filterwarnings("ignore", message=".*The `dict` method is deprecated.*")
+warnings.filterwarnings("ignore", category=UserWarning, module="pydantic.*")
+
+# Store the original warn function
+_original_warn = warnings.warn
+_original_warn_explicit = warnings.warn_explicit
+
+# Messages to suppress (partial matches)
+SUPPRESSED_PATTERNS = [
+    "Use 'content=<...>' to upload raw bytes/text content",
+    "The `dict` method is deprecated; use `model_dump` instead",
+    "Pydantic serializer warnings",
+    "PydanticSerializationUnexpectedValue",
+    "Expected 9 fields but got 5 for type `Message`",
+    "Expected `StreamingChoices` but got `Choices`",
+    "serialized value may not be as expected"
+]
+
+@functools.wraps(_original_warn)
+def _patched_warn(message, category=None, stacklevel=1, source=None):
+    """Patched warn function that suppresses specific messages."""
+    msg_str = str(message)
+    
+    for pattern in SUPPRESSED_PATTERNS:
+        if pattern in msg_str:
+            return
+    
+    if category == UserWarning and "pydantic" in msg_str.lower():
+        return
+    
+    _original_warn(message, category, stacklevel, source)
+
+@functools.wraps(_original_warn_explicit)
+def _patched_warn_explicit(message, category, filename, lineno, module=None, registry=None, module_globals=None, source=None):
+    """Patched warn_explicit function that suppresses specific messages."""
+    msg_str = str(message)
+    
+    for pattern in SUPPRESSED_PATTERNS:
+        if pattern in msg_str:
+            return
+    
+    if category == UserWarning and "pydantic" in msg_str.lower():
+        return
+    
+    if module and "pydantic" in str(module):
+        return
+        
+    _original_warn_explicit(message, category, filename, lineno, module, registry, module_globals, source)
+
+# Apply the patches
+warnings.warn = _patched_warn
+warnings.warn_explicit = _patched_warn_explicit
+
+# Also patch sys.modules warnings if it exists
+if 'warnings' in sys.modules:
+    sys.modules['warnings'].warn = _patched_warn
+    sys.modules['warnings'].warn_explicit = _patched_warn_explicit