praisonaiagents 0.0.146__tar.gz → 0.0.148__tar.gz

{praisonaiagents-0.0.146 → praisonaiagents-0.0.148}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praisonaiagents
-Version: 0.0.146
+Version: 0.0.148
 Summary: Praison AI agents for completing complex tasks with Self Reflection Agents
 Author: Mervin Praison
 Requires-Python: >=3.10

praisonaiagents-0.0.148/praisonaiagents/__init__.py

@@ -0,0 +1,138 @@
+"""
+Praison AI Agents - A package for hierarchical AI agent task execution
+"""
+
+# Apply warning patch BEFORE any imports to intercept warnings at the source
+from . import _warning_patch
+
+# Import centralized logging configuration FIRST
+from . import _logging
+
+# 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
+from .agents.agents import PraisonAIAgents
+from .task.task import Task
+from .tools.tools import Tools
+from .agents.autoagents import AutoAgents
+from .knowledge.knowledge import Knowledge
+from .knowledge.chunking import Chunking
+# MCP support (optional)
+try:
+    from .mcp.mcp import MCP
+    _mcp_available = True
+except ImportError:
+    _mcp_available = False
+    MCP = None
+from .session import Session
+from .memory.memory import Memory
+from .guardrails import GuardrailResult, LLMGuardrail
+from .agent.handoff import Handoff, handoff, handoff_filters, RECOMMENDED_PROMPT_PREFIX, prompt_with_handoff_instructions
+from .main import (
+    TaskOutput,
+    ReflectionOutput,
+    display_interaction,
+    display_self_reflection,
+    display_instruction,
+    display_tool_call,
+    display_error,
+    display_generating,
+    clean_triple_backticks,
+    error_logs,
+    register_display_callback,
+    sync_display_callbacks,
+    async_display_callbacks,
+)
+
+# Telemetry support (lazy loaded)
+try:
+    from .telemetry import (
+        get_telemetry,
+        enable_telemetry,
+        disable_telemetry,
+        MinimalTelemetry,
+        TelemetryCollector
+    )
+    _telemetry_available = True
+except ImportError:
+    # Telemetry not available - provide stub functions
+    _telemetry_available = False
+    def get_telemetry():
+        return None
+    
+    def enable_telemetry(*args, **kwargs):
+        import logging
+        logging.warning(
+            "Telemetry not available. Install with: pip install praisonaiagents[telemetry]"
+        )
+        return None
+    
+    def disable_telemetry():
+        pass
+    
+    MinimalTelemetry = None
+    TelemetryCollector = None
+
+# Add Agents as an alias for PraisonAIAgents
+Agents = PraisonAIAgents
+
+# Apply telemetry auto-instrumentation after all imports
+if _telemetry_available:
+    try:
+        # Only instrument if telemetry is enabled
+        _telemetry = get_telemetry()
+        if _telemetry and _telemetry.enabled:
+            from .telemetry.integration import auto_instrument_all
+            auto_instrument_all(_telemetry)
+    except Exception:
+        # Silently fail if there are any issues
+        pass
+
+__all__ = [
+    'Agent',
+    'ImageAgent',
+    'ContextAgent',
+    'create_context_agent',
+    'PraisonAIAgents',
+    'Agents',
+    'Tools',
+    'Task',
+    'TaskOutput',
+    'ReflectionOutput',
+    'AutoAgents',
+    'Session',
+    'Memory',
+    'display_interaction',
+    'display_self_reflection',
+    'display_instruction',
+    'display_tool_call',
+    'display_error',
+    'display_generating',
+    'clean_triple_backticks',
+    'error_logs',
+    'register_display_callback',
+    'sync_display_callbacks',
+    'async_display_callbacks',
+    'Knowledge',
+    'Chunking',
+    'GuardrailResult',
+    'LLMGuardrail',
+    'Handoff',
+    'handoff',
+    'handoff_filters',
+    'RECOMMENDED_PROMPT_PREFIX',
+    'prompt_with_handoff_instructions',
+    'get_telemetry',
+    'enable_telemetry',
+    'disable_telemetry',
+    'MinimalTelemetry',
+    'TelemetryCollector'
+]
+
+# Add MCP to __all__ if available
+if _mcp_available:
+    __all__.append('MCP')

praisonaiagents-0.0.148/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", "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.148/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

{praisonaiagents-0.0.146 → praisonaiagents-0.0.148}/praisonaiagents/agent/agent.py

@@ -331,8 +331,15 @@ class Agent:
 
         # Configure logging to suppress unwanted outputs
         logging.getLogger("litellm").setLevel(logging.WARNING)
-        logging.getLogger("httpx").setLevel(logging.WARNING)
-        logging.getLogger("httpcore").setLevel(logging.WARNING)
+        
+        # Allow httpx logging when LOGLEVEL=debug, otherwise suppress it
+        loglevel = os.environ.get('LOGLEVEL', 'INFO').upper()
+        if loglevel == 'DEBUG':
+            logging.getLogger("httpx").setLevel(logging.INFO)
+            logging.getLogger("httpcore").setLevel(logging.INFO)
+        else:
+            logging.getLogger("httpx").setLevel(logging.WARNING)
+            logging.getLogger("httpcore").setLevel(logging.WARNING)
 
         # If instructions are provided, use them to set role, goal, and backstory
         if instructions:
@@ -1229,14 +1236,14 @@ Your Goal: {self.goal}"""
                 border_style="green",
                 expand=False
             )
-        else:
-            # No content yet: show generating message
-            return Panel(
-                f"[bold cyan]Generating response...[/bold cyan]",
-                title=f"[bold]{self.name}[/bold] - {elapsed:.1f}s",
-                border_style="cyan",
-                expand=False
-            )
+        # else:
+        #     # No content yet: show generating message
+        #     return Panel(
+        #         f"[bold cyan]Generating response...[/bold cyan]",
+        #         title=f"[bold]{self.name}[/bold] - {elapsed:.1f}s",
+        #         border_style="cyan",
+        #         expand=False
+        #     )
 
     def chat(self, prompt, temperature=0.2, tools=None, output_json=None, output_pydantic=None, reasoning_steps=False, stream=None, task_name=None, task_description=None, task_id=None):
         # Reset the final display flag for each new conversation

praisonaiagents-0.0.148/praisonaiagents/llm/__init__.py

@@ -0,0 +1,50 @@
+import os
+
+# Ensure litellm telemetry is disabled before imports
+os.environ["LITELLM_TELEMETRY"] = "False"
+
+# Import modules
+from .llm import LLM, LLMContextLengthExceededException
+from .openai_client import (
+    OpenAIClient, 
+    get_openai_client,
+    ChatCompletionMessage,
+    Choice,
+    CompletionTokensDetails,
+    PromptTokensDetails,
+    CompletionUsage,
+    ChatCompletion,
+    ToolCall,
+    process_stream_chunks
+)
+from .model_capabilities import (
+    supports_structured_outputs,
+    supports_streaming_with_tools
+)
+from .model_router import (
+    ModelRouter,
+    ModelProfile, 
+    TaskComplexity,
+    create_routing_agent
+)
+
+__all__ = [
+    "LLM", 
+    "LLMContextLengthExceededException", 
+    "OpenAIClient", 
+    "get_openai_client",
+    "ChatCompletionMessage",
+    "Choice",
+    "CompletionTokensDetails",
+    "PromptTokensDetails",
+    "CompletionUsage",
+    "ChatCompletion",
+    "ToolCall",
+    "process_stream_chunks",
+    "supports_structured_outputs",
+    "supports_streaming_with_tools",
+    "ModelRouter",
+    "ModelProfile",
+    "TaskComplexity",
+    "create_routing_agent"
+]