praisonaiagents 0.0.101__py3-none-any.whl → 0.0.103__py3-none-any.whl

praisonaiagents/__init__.py

@@ -12,6 +12,7 @@ from .knowledge.knowledge import Knowledge
 from .knowledge.chunking import Chunking
 from .mcp.mcp import MCP
 from .session import Session
+from .memory.memory import Memory
 from .guardrails import GuardrailResult, LLMGuardrail
 from .main import (
     TaskOutput,
@@ -29,9 +30,50 @@ from .main import (
     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',
@@ -43,6 +85,7 @@ __all__ = [
     'ReflectionOutput',
     'AutoAgents',
     'Session',
+    'Memory',
     'display_interaction',
     'display_self_reflection',
     'display_instruction',
@@ -58,5 +101,10 @@ __all__ = [
     'Chunking',
     'MCP',
     'GuardrailResult',
-    'LLMGuardrail'
+    'LLMGuardrail',
+    'get_telemetry',
+    'enable_telemetry',
+    'disable_telemetry',
+    'MinimalTelemetry',
+    'TelemetryCollector'
 ] 

praisonaiagents/llm/__init__.py

@@ -1,5 +1,9 @@
 import logging
 import warnings
+import os
+
+# Disable litellm telemetry before any imports
+os.environ["LITELLM_TELEMETRY"] = "False"
 
 # Suppress all relevant logs at module level
 logging.getLogger("litellm").setLevel(logging.ERROR)
@@ -17,4 +21,11 @@ logging.basicConfig(level=logging.WARNING)
 # Import after suppressing warnings
 from .llm import LLM, LLMContextLengthExceededException
 
+# Ensure telemetry is disabled after import as well
+try:
+    import litellm
+    litellm.telemetry = False
+except ImportError:
+    pass
+
 __all__ = ["LLM", "LLMContextLengthExceededException"]

praisonaiagents/llm/llm.py

@@ -17,6 +17,9 @@ from ..main import (
 from rich.console import Console
 from rich.live import Live
 
+# Disable litellm telemetry before any imports
+os.environ["LITELLM_TELEMETRY"] = "False"
+
 # TODO: Include in-build tool calling in LLM class
 # TODO: Restructure so that duplicate calls are not made (Sync with agent.py)
 class LLMContextLengthExceededException(Exception):
@@ -108,6 +111,9 @@ class LLM:
     ):
         try:
             import litellm
+            # Disable telemetry
+            litellm.telemetry = False
+            
             # Set litellm options globally
             litellm.set_verbose = False
             litellm.success_callback = []

praisonaiagents/memory/memory.py

@@ -6,6 +6,9 @@ import shutil
 from typing import Any, Dict, List, Optional, Union, Literal
 import logging
 
+# Disable litellm telemetry before any imports
+os.environ["LITELLM_TELEMETRY"] = "False"
+
 # Set up logger
 logger = logging.getLogger(__name__)
 
@@ -29,6 +32,13 @@ try:
 except ImportError:
     OPENAI_AVAILABLE = False
 
+try:
+    import litellm
+    litellm.telemetry = False  # Disable telemetry
+    LITELLM_AVAILABLE = True
+except ImportError:
+    LITELLM_AVAILABLE = False
+
 
 
 
@@ -340,14 +350,28 @@ class Memory:
             
         elif self.use_rag and hasattr(self, "chroma_col"):
             try:
-                from openai import OpenAI
-                client = OpenAI()
-                
-                response = client.embeddings.create(
-                    input=query,
-                    model="text-embedding-3-small"
-                )
-                query_embedding = response.data[0].embedding
+                if LITELLM_AVAILABLE:
+                    # Use LiteLLM for consistency with the rest of the codebase
+                    import litellm
+                    
+                    response = litellm.embedding(
+                        model="text-embedding-3-small",
+                        input=query
+                    )
+                    query_embedding = response.data[0]["embedding"]
+                elif OPENAI_AVAILABLE:
+                    # Fallback to OpenAI client
+                    from openai import OpenAI
+                    client = OpenAI()
+                    
+                    response = client.embeddings.create(
+                        input=query,
+                        model="text-embedding-3-small"
+                    )
+                    query_embedding = response.data[0].embedding
+                else:
+                    self._log_verbose("Neither litellm nor openai available for embeddings", logging.WARNING)
+                    return []
                 
                 resp = self.chroma_col.query(
                     query_embeddings=[query_embedding],
@@ -464,19 +488,39 @@ class Memory:
         # Store in vector database if enabled
         if self.use_rag and hasattr(self, "chroma_col"):
             try:
-                from openai import OpenAI
-                client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))  # Ensure API key is correctly set
-                
-                logger.info("Getting embeddings from OpenAI...")
-                logger.debug(f"Embedding input text: {text}")  # Log the input text
-                
-                response = client.embeddings.create(
-                    input=text,
-                    model="text-embedding-3-small"
-                )
-                embedding = response.data[0].embedding
-                logger.info("Successfully got embeddings")
-                logger.debug(f"Received embedding of length: {len(embedding)}")  # Log embedding details
+                if LITELLM_AVAILABLE:
+                    # Use LiteLLM for consistency with the rest of the codebase
+                    import litellm
+                    
+                    logger.info("Getting embeddings from LiteLLM...")
+                    logger.debug(f"Embedding input text: {text}")
+                    
+                    response = litellm.embedding(
+                        model="text-embedding-3-small",
+                        input=text
+                    )
+                    embedding = response.data[0]["embedding"]
+                    logger.info("Successfully got embeddings from LiteLLM")
+                    logger.debug(f"Received embedding of length: {len(embedding)}")
+                    
+                elif OPENAI_AVAILABLE:
+                    # Fallback to OpenAI client
+                    from openai import OpenAI
+                    client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+                    
+                    logger.info("Getting embeddings from OpenAI...")
+                    logger.debug(f"Embedding input text: {text}")
+                    
+                    response = client.embeddings.create(
+                        input=text,
+                        model="text-embedding-3-small"
+                    )
+                    embedding = response.data[0].embedding
+                    logger.info("Successfully got embeddings from OpenAI")
+                    logger.debug(f"Received embedding of length: {len(embedding)}")
+                else:
+                    logger.warning("Neither litellm nor openai available for embeddings")
+                    return
                 
                 # Sanitize metadata for ChromaDB
                 sanitized_metadata = self._sanitize_metadata(metadata)
@@ -527,15 +571,28 @@ class Memory:
 
         elif self.use_rag and hasattr(self, "chroma_col"):
             try:
-                from openai import OpenAI
-                client = OpenAI()
-                
-                # Get query embedding
-                response = client.embeddings.create(
-                    input=query,
-                    model="text-embedding-3-small"  # Using consistent model
-                )
-                query_embedding = response.data[0].embedding
+                if LITELLM_AVAILABLE:
+                    # Use LiteLLM for consistency with the rest of the codebase
+                    import litellm
+                    
+                    response = litellm.embedding(
+                        model="text-embedding-3-small",
+                        input=query
+                    )
+                    query_embedding = response.data[0]["embedding"]
+                elif OPENAI_AVAILABLE:
+                    # Fallback to OpenAI client
+                    from openai import OpenAI
+                    client = OpenAI()
+                    
+                    response = client.embeddings.create(
+                        input=query,
+                        model="text-embedding-3-small"
+                    )
+                    query_embedding = response.data[0].embedding
+                else:
+                    self._log_verbose("Neither litellm nor openai available for embeddings", logging.WARNING)
+                    return []
                 
                 # Search ChromaDB with embedding
                 resp = self.chroma_col.query(
@@ -910,21 +967,44 @@ class Memory:
         """
 
         try:
-            # Use LiteLLM for consistency with the rest of the codebase
-            import litellm
-            
-            # Convert model name if it's in litellm format
-            model_name = llm or "gpt-4o-mini"
-            
-            response = litellm.completion(
-                model=model_name,
-                messages=[{
-                    "role": "user", 
-                    "content": custom_prompt or default_prompt
-                }],
-                response_format={"type": "json_object"},
-                temperature=0.3
-            )
+            if LITELLM_AVAILABLE:
+                # Use LiteLLM for consistency with the rest of the codebase
+                import litellm
+                
+                # Convert model name if it's in litellm format
+                model_name = llm or "gpt-4o-mini"
+                
+                response = litellm.completion(
+                    model=model_name,
+                    messages=[{
+                        "role": "user", 
+                        "content": custom_prompt or default_prompt
+                    }],
+                    response_format={"type": "json_object"},
+                    temperature=0.3
+                )
+            elif OPENAI_AVAILABLE:
+                # Fallback to OpenAI client
+                from openai import OpenAI
+                client = OpenAI()
+                
+                response = client.chat.completions.create(
+                    model=llm or "gpt-4o-mini",
+                    messages=[{
+                        "role": "user", 
+                        "content": custom_prompt or default_prompt
+                    }],
+                    response_format={"type": "json_object"},
+                    temperature=0.3
+                )
+            else:
+                logger.error("Neither litellm nor openai available for quality calculation")
+                return {
+                    "completeness": 0.0,
+                    "relevance": 0.0,
+                    "clarity": 0.0,
+                    "accuracy": 0.0
+                }
             
             metrics = json.loads(response.choices[0].message.content)
             

praisonaiagents/telemetry/__init__.py

@@ -0,0 +1,102 @@
+"""
+PraisonAI Agents Minimal Telemetry Module
+
+This module provides anonymous usage tracking with privacy-first design.
+Telemetry is opt-out and can be disabled via environment variables:
+- PRAISONAI_TELEMETRY_DISABLED=true
+- PRAISONAI_DISABLE_TELEMETRY=true  
+- DO_NOT_TRACK=true
+
+No personal data, prompts, or responses are collected.
+"""
+
+import os
+import atexit
+from typing import Optional, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .telemetry import MinimalTelemetry, TelemetryCollector
+
+# Import the classes for real (not just type checking)
+from .telemetry import MinimalTelemetry, TelemetryCollector
+
+__all__ = [
+    'get_telemetry',
+    'enable_telemetry',
+    'disable_telemetry',
+    'MinimalTelemetry',
+    'TelemetryCollector',  # For backward compatibility
+]
+
+
+def get_telemetry() -> 'MinimalTelemetry':
+    """Get the global telemetry instance."""
+    from .telemetry import get_telemetry as _get_telemetry
+    return _get_telemetry()
+
+
+def enable_telemetry():
+    """Enable telemetry (if not disabled by environment)."""
+    from .telemetry import enable_telemetry as _enable_telemetry
+    _enable_telemetry()
+
+
+def disable_telemetry():
+    """Disable telemetry."""
+    from .telemetry import disable_telemetry as _disable_telemetry
+    _disable_telemetry()
+
+
+# Auto-instrumentation and cleanup setup
+_initialized = False
+_atexit_registered = False
+
+def _ensure_atexit():
+    """Ensure atexit handler is registered."""
+    global _atexit_registered
+    if _atexit_registered:
+        return
+    
+    # Check if telemetry should be disabled
+    telemetry_disabled = any([
+        os.environ.get('PRAISONAI_TELEMETRY_DISABLED', '').lower() in ('true', '1', 'yes'),
+        os.environ.get('PRAISONAI_DISABLE_TELEMETRY', '').lower() in ('true', '1', 'yes'),
+        os.environ.get('DO_NOT_TRACK', '').lower() in ('true', '1', 'yes'),
+    ])
+    
+    if not telemetry_disabled:
+        # Register atexit handler to flush telemetry on exit
+        atexit.register(lambda: get_telemetry().flush())
+        _atexit_registered = True
+
+def _initialize_telemetry():
+    """Initialize telemetry with auto-instrumentation and cleanup."""
+    global _initialized
+    if _initialized:
+        return
+    
+    # Ensure atexit is registered
+    _ensure_atexit()
+    
+    # Check if telemetry should be disabled
+    telemetry_disabled = any([
+        os.environ.get('PRAISONAI_TELEMETRY_DISABLED', '').lower() in ('true', '1', 'yes'),
+        os.environ.get('PRAISONAI_DISABLE_TELEMETRY', '').lower() in ('true', '1', 'yes'),
+        os.environ.get('DO_NOT_TRACK', '').lower() in ('true', '1', 'yes'),
+    ])
+    
+    if not telemetry_disabled:
+        try:
+            # Defer the actual instrumentation to avoid circular imports
+            # This will be called when get_telemetry() is first accessed
+            _initialized = True
+        except Exception:
+            # Silently fail if there are any issues
+            pass
+
+
+# No need for lazy auto-instrumentation here since main __init__.py handles it
+
+
+# Initialize atexit handler early
+_ensure_atexit()

praisonaiagents/telemetry/integration.py

@@ -0,0 +1,242 @@
+"""
+Simplified integration module for adding telemetry to core PraisonAI components.
+"""
+
+from typing import Any, Optional, TYPE_CHECKING
+from functools import wraps
+import time
+
+if TYPE_CHECKING:
+    from .telemetry import MinimalTelemetry
+    from ..agent.agent import Agent
+    from ..task.task import Task
+    from ..agents.agents import PraisonAIAgents
+
+
+def instrument_agent(agent: 'Agent', telemetry: Optional['MinimalTelemetry'] = None):
+    """
+    Instrument an Agent instance with minimal telemetry.
+    
+    Args:
+        agent: The Agent instance to instrument
+        telemetry: Optional telemetry instance (uses global if not provided)
+    """
+    if not telemetry:
+        from .telemetry import get_telemetry
+        telemetry = get_telemetry()
+    
+    if not telemetry.enabled:
+        return agent
+    
+    # Check if agent is already instrumented to avoid double-counting
+    if hasattr(agent, '_telemetry_instrumented'):
+        return agent
+    
+    # Store original methods
+    original_chat = agent.chat if hasattr(agent, 'chat') else None
+    original_start = agent.start if hasattr(agent, 'start') else None
+    original_run = agent.run if hasattr(agent, 'run') else None
+    original_execute_tool = agent.execute_tool if hasattr(agent, 'execute_tool') else None
+    
+    # Wrap chat method if it exists (this is the main method called by workflow)
+    if original_chat:
+        @wraps(original_chat)
+        def instrumented_chat(*args, **kwargs):
+            try:
+                result = original_chat(*args, **kwargs)
+                telemetry.track_agent_execution(agent.name, success=True)
+                return result
+            except Exception as e:
+                telemetry.track_agent_execution(agent.name, success=False)
+                telemetry.track_error(type(e).__name__)
+                raise
+        
+        agent.chat = instrumented_chat
+    
+    # Wrap start method if it exists
+    if original_start:
+        @wraps(original_start)
+        def instrumented_start(*args, **kwargs):
+            try:
+                result = original_start(*args, **kwargs)
+                telemetry.track_agent_execution(agent.name, success=True)
+                return result
+            except Exception as e:
+                telemetry.track_agent_execution(agent.name, success=False)
+                telemetry.track_error(type(e).__name__)
+                raise
+        
+        agent.start = instrumented_start
+    
+    # Wrap run method if it exists
+    if original_run:
+        @wraps(original_run)
+        def instrumented_run(*args, **kwargs):
+            try:
+                result = original_run(*args, **kwargs)
+                telemetry.track_agent_execution(agent.name, success=True)
+                return result
+            except Exception as e:
+                telemetry.track_agent_execution(agent.name, success=False)
+                telemetry.track_error(type(e).__name__)
+                raise
+        
+        agent.run = instrumented_run
+    
+    # Wrap execute_tool method
+    if original_execute_tool:
+        @wraps(original_execute_tool)
+        def instrumented_execute_tool(tool_name: str, *args, **kwargs):
+            try:
+                result = original_execute_tool(tool_name, *args, **kwargs)
+                telemetry.track_tool_usage(tool_name, success=True)
+                return result
+            except Exception as e:
+                telemetry.track_tool_usage(tool_name, success=False)
+                telemetry.track_error(type(e).__name__)
+                raise
+        
+        agent.execute_tool = instrumented_execute_tool
+    
+    # Mark agent as instrumented to avoid double instrumentation
+    agent._telemetry_instrumented = True
+    
+    return agent
+
+
+def instrument_workflow(workflow: 'PraisonAIAgents', telemetry: Optional['MinimalTelemetry'] = None):
+    """
+    Instrument a PraisonAIAgents workflow with minimal telemetry.
+    
+    Args:
+        workflow: The PraisonAIAgents instance to instrument
+        telemetry: Optional telemetry instance (uses global if not provided)
+    """
+    if not telemetry:
+        from .telemetry import get_telemetry
+        telemetry = get_telemetry()
+    
+    if not telemetry.enabled:
+        return workflow
+    
+    # Check if workflow is already instrumented to avoid double-counting
+    if hasattr(workflow, '_telemetry_instrumented'):
+        return workflow
+    
+    # Track feature usage
+    telemetry.track_feature_usage(f"workflow_{workflow.process}" if hasattr(workflow, 'process') else "workflow")
+    
+    # Instrument all agents in the workflow
+    if hasattr(workflow, 'agents') and workflow.agents:
+        for agent in workflow.agents:
+            instrument_agent(agent, telemetry)
+    
+    # Wrap the execute_task method to track task completions
+    if hasattr(workflow, 'execute_task'):
+        original_execute_task = workflow.execute_task
+        
+        @wraps(original_execute_task)
+        def instrumented_execute_task(task_id, *args, **kwargs):
+            task = None
+            try:
+                # Get task info
+                if hasattr(workflow, 'tasks') and task_id < len(workflow.tasks):
+                    task = workflow.tasks[task_id]
+                
+                result = original_execute_task(task_id, *args, **kwargs)
+                
+                # Track task completion
+                task_name = task.name if task and hasattr(task, 'name') else f"task_{task_id}"
+                telemetry.track_task_completion(task_name, success=True)
+                
+                return result
+            except Exception as e:
+                telemetry.track_error(type(e).__name__)
+                if task:
+                    task_name = task.name if hasattr(task, 'name') else f"task_{task_id}"
+                    telemetry.track_task_completion(task_name, success=False)
+                raise
+        
+        workflow.execute_task = instrumented_execute_task
+    
+    # Wrap the start method
+    original_start = workflow.start
+    
+    @wraps(original_start)
+    def instrumented_start(*args, **kwargs):
+        try:
+            result = original_start(*args, **kwargs)
+            # Don't double-track here since agent.chat already tracks execution
+            return result
+        except Exception as e:
+            telemetry.track_error(type(e).__name__)
+            raise
+    
+    workflow.start = instrumented_start
+    
+    # Also wrap astart if it exists (async version)
+    if hasattr(workflow, 'astart'):
+        original_astart = workflow.astart
+        
+        @wraps(original_astart)
+        async def instrumented_astart(*args, **kwargs):
+            try:
+                result = await original_astart(*args, **kwargs)
+                # Don't double-track here since agent.chat already tracks execution
+                return result
+            except Exception as e:
+                telemetry.track_error(type(e).__name__)
+                raise
+        
+        workflow.astart = instrumented_astart
+    
+    # Mark workflow as instrumented to avoid double instrumentation
+    workflow._telemetry_instrumented = True
+    
+    return workflow
+
+
+# Auto-instrumentation helper
+def auto_instrument_all(telemetry: Optional['MinimalTelemetry'] = None):
+    """
+    Automatically instrument all new instances of Agent and PraisonAIAgents.
+    This should be called after enabling telemetry.
+    
+    Args:
+        telemetry: Optional telemetry instance (uses global if not provided)
+    """
+    if not telemetry:
+        from .telemetry import get_telemetry
+        telemetry = get_telemetry()
+    
+    if not telemetry.enabled:
+        return
+    
+    try:
+        # Import the classes
+        from ..agent.agent import Agent
+        from ..agents.agents import PraisonAIAgents
+        
+        # Store original __init__ methods
+        original_agent_init = Agent.__init__
+        original_workflow_init = PraisonAIAgents.__init__
+        
+        # Wrap Agent.__init__
+        @wraps(original_agent_init)
+        def agent_init_wrapper(self, *args, **kwargs):
+            original_agent_init(self, *args, **kwargs)
+            instrument_agent(self, telemetry)
+        
+        # Wrap PraisonAIAgents.__init__
+        @wraps(original_workflow_init)
+        def workflow_init_wrapper(self, *args, **kwargs):
+            original_workflow_init(self, *args, **kwargs)
+            instrument_workflow(self, telemetry)
+        
+        # Apply wrapped constructors
+        Agent.__init__ = agent_init_wrapper
+        PraisonAIAgents.__init__ = workflow_init_wrapper
+        
+    except ImportError:
+        # Classes not available, skip auto-instrumentation
+        pass

praisonaiagents/telemetry/telemetry.py

@@ -0,0 +1,350 @@
+"""
+Minimal telemetry implementation for PraisonAI Agents.
+
+This module provides anonymous usage tracking with privacy-first design.
+All telemetry is opt-out via environment variables.
+"""
+
+import os
+import time
+import platform
+import hashlib
+from typing import Dict, Any, Optional
+from datetime import datetime
+import logging
+
+# Try to import PostHog
+try:
+    from posthog import Posthog
+    POSTHOG_AVAILABLE = True
+except ImportError:
+    POSTHOG_AVAILABLE = False
+
+# Check for opt-out environment variables
+_TELEMETRY_DISABLED = any([
+    os.environ.get('PRAISONAI_TELEMETRY_DISABLED', '').lower() in ('true', '1', 'yes'),
+    os.environ.get('PRAISONAI_DISABLE_TELEMETRY', '').lower() in ('true', '1', 'yes'),
+    os.environ.get('DO_NOT_TRACK', '').lower() in ('true', '1', 'yes'),
+])
+
+
+class MinimalTelemetry:
+    """
+    Minimal telemetry collector for anonymous usage tracking.
+    
+    Privacy guarantees:
+    - No personal data is collected
+    - No prompts, responses, or user content is tracked
+    - Only anonymous metrics about feature usage
+    - Respects DO_NOT_TRACK standard
+    - Can be disabled via environment variables
+    """
+    
+    def __init__(self, enabled: bool = None):
+        """
+        Initialize the minimal telemetry collector.
+        
+        Args:
+            enabled: Override the environment-based enable/disable setting
+        """
+        # Respect explicit enabled parameter, otherwise check environment
+        if enabled is not None:
+            self.enabled = enabled
+        else:
+            self.enabled = not _TELEMETRY_DISABLED
+            
+        self.logger = logging.getLogger(__name__)
+        
+        if not self.enabled:
+            self.logger.debug("Telemetry is disabled")
+            return
+            
+        # Generate anonymous session ID (not user ID)
+        session_data = f"{datetime.now().isoformat()}-{os.getpid()}-{time.time()}"
+        self.session_id = hashlib.sha256(session_data.encode()).hexdigest()[:16]
+        
+        # Basic metrics storage
+        self._metrics = {
+            "agent_executions": 0,
+            "task_completions": 0,
+            "tool_calls": 0,
+            "errors": 0,
+        }
+        
+        # Collect basic environment info (anonymous)
+        self._environment = {
+            "python_version": platform.python_version(),
+            "os_type": platform.system(),
+            "framework_version": self._get_framework_version(),
+        }
+        
+        self.logger.debug(f"Telemetry enabled with session {self.session_id}")
+        
+        # Initialize PostHog if available
+        if POSTHOG_AVAILABLE:
+            try:
+                self._posthog = Posthog(
+                    project_api_key='phc_skZpl3eFLQJ4iYjsERNMbCO6jfeSJi2vyZlPahKgxZ7',
+                    host='https://eu.i.posthog.com',
+                    disable_geoip=True
+                )
+            except:
+                self._posthog = None
+        else:
+            self._posthog = None
+    
+    def _get_framework_version(self) -> str:
+        """Get the PraisonAI Agents version."""
+        try:
+            from .. import __version__
+            return __version__
+        except ImportError:
+            return "unknown"
+    
+    def track_agent_execution(self, agent_name: str = None, success: bool = True):
+        """
+        Track an agent execution event.
+        
+        Args:
+            agent_name: Name of the agent (not logged, just for counting)
+            success: Whether the execution was successful
+        """
+        if not self.enabled:
+            return
+            
+        self._metrics["agent_executions"] += 1
+        
+        # In a real implementation, this would send to a backend
+        # For now, just log at debug level
+        self.logger.debug(f"Agent execution tracked: success={success}")
+    
+    def track_task_completion(self, task_name: str = None, success: bool = True):
+        """
+        Track a task completion event.
+        
+        Args:
+            task_name: Name of the task (not logged, just for counting)
+            success: Whether the task completed successfully
+        """
+        if not self.enabled:
+            return
+            
+        self._metrics["task_completions"] += 1
+        
+        self.logger.debug(f"Task completion tracked: success={success}")
+    
+    def track_tool_usage(self, tool_name: str, success: bool = True):
+        """
+        Track tool usage event.
+        
+        Args:
+            tool_name: Name of the tool being used
+            success: Whether the tool call was successful
+        """
+        if not self.enabled:
+            return
+            
+        self._metrics["tool_calls"] += 1
+        
+        # Only track tool name, not arguments or results
+        self.logger.debug(f"Tool usage tracked: {tool_name}, success={success}")
+    
+    def track_error(self, error_type: str = None):
+        """
+        Track an error event.
+        
+        Args:
+            error_type: Type of error (not the full message)
+        """
+        if not self.enabled:
+            return
+            
+        self._metrics["errors"] += 1
+        
+        # Only track error type, not full error messages
+        self.logger.debug(f"Error tracked: type={error_type or 'unknown'}")
+    
+    def track_feature_usage(self, feature_name: str):
+        """
+        Track usage of a specific feature.
+        
+        Args:
+            feature_name: Name of the feature being used
+        """
+        if not self.enabled:
+            return
+            
+        # Track which features are being used
+        self.logger.debug(f"Feature usage tracked: {feature_name}")
+    
+    def get_metrics(self) -> Dict[str, Any]:
+        """
+        Get current metrics summary.
+        
+        Returns:
+            Dictionary of current metrics
+        """
+        if not self.enabled:
+            return {"enabled": False}
+            
+        return {
+            "enabled": True,
+            "session_id": self.session_id,
+            "metrics": self._metrics.copy(),
+            "environment": self._environment.copy(),
+        }
+    
+    def flush(self):
+        """
+        Flush any pending telemetry data.
+        
+        In a real implementation, this would send data to a backend.
+        """
+        if not self.enabled:
+            return
+            
+        metrics = self.get_metrics()
+        self.logger.debug(f"Telemetry flush: {metrics}")
+        
+        # Send to PostHog if available
+        if hasattr(self, '_posthog') and self._posthog:
+            
+            try:
+                self._posthog.capture(
+                    distinct_id='anonymous',
+                    event='sdk_used',
+                    properties={
+                        'version': self._environment['framework_version'],
+                        'os': platform.system(),
+                        '$process_person_profile': False,
+                        '$geoip_disable': True
+                    }
+                )
+            except:
+                pass
+        
+        # Reset counters
+        for key in self._metrics:
+            if isinstance(self._metrics[key], int):
+                self._metrics[key] = 0
+
+
+# Global telemetry instance
+_telemetry_instance = None
+
+
+def get_telemetry() -> MinimalTelemetry:
+    """
+    Get the global telemetry instance.
+    
+    Returns:
+        The global MinimalTelemetry instance
+    """
+    global _telemetry_instance
+    if _telemetry_instance is None:
+        _telemetry_instance = MinimalTelemetry()
+    return _telemetry_instance
+
+
+def disable_telemetry():
+    """Programmatically disable telemetry."""
+    global _telemetry_instance
+    if _telemetry_instance:
+        _telemetry_instance.enabled = False
+    else:
+        _telemetry_instance = MinimalTelemetry(enabled=False)
+
+
+def enable_telemetry():
+    """Programmatically enable telemetry (if not disabled by environment)."""
+    global _telemetry_instance
+    if not _TELEMETRY_DISABLED:
+        if _telemetry_instance:
+            _telemetry_instance.enabled = True
+        else:
+            _telemetry_instance = MinimalTelemetry(enabled=True)
+
+
+# For backward compatibility with existing code
+class TelemetryCollector:
+    """Backward compatibility wrapper for the old TelemetryCollector interface."""
+    
+    def __init__(self, backend: str = "minimal", service_name: str = "praisonai-agents", **kwargs):
+        self.telemetry = get_telemetry()
+        
+    def start(self):
+        """Start telemetry collection."""
+        # No-op for minimal implementation
+        pass
+        
+    def stop(self):
+        """Stop telemetry collection and flush data."""
+        self.telemetry.flush()
+    
+    def trace_agent_execution(self, agent_name: str, **attributes):
+        """Compatibility method for agent execution tracking."""
+        from contextlib import contextmanager
+        
+        @contextmanager
+        def _trace():
+            try:
+                yield None
+                self.telemetry.track_agent_execution(agent_name, success=True)
+            except Exception:
+                self.telemetry.track_agent_execution(agent_name, success=False)
+                raise
+                
+        return _trace()
+    
+    def trace_task_execution(self, task_name: str, agent_name: str = None, **attributes):
+        """Compatibility method for task execution tracking."""
+        from contextlib import contextmanager
+        
+        @contextmanager
+        def _trace():
+            try:
+                yield None
+                self.telemetry.track_task_completion(task_name, success=True)
+            except Exception:
+                self.telemetry.track_task_completion(task_name, success=False)
+                raise
+                
+        return _trace()
+    
+    def trace_tool_call(self, tool_name: str, **attributes):
+        """Compatibility method for tool call tracking."""
+        from contextlib import contextmanager
+        
+        @contextmanager
+        def _trace():
+            try:
+                yield None
+                self.telemetry.track_tool_usage(tool_name, success=True)
+            except Exception:
+                self.telemetry.track_tool_usage(tool_name, success=False)
+                raise
+                
+        return _trace()
+    
+    def trace_llm_call(self, model: str = None, **attributes):
+        """Compatibility method for LLM call tracking."""
+        from contextlib import contextmanager
+        
+        @contextmanager
+        def _trace():
+            # We don't track LLM calls in minimal telemetry
+            yield None
+            
+        return _trace()
+    
+    def record_tokens(self, prompt_tokens: int, completion_tokens: int, model: str = None):
+        """Compatibility method - we don't track token usage."""
+        pass
+        
+    def record_cost(self, cost: float, model: str = None):
+        """Compatibility method - we don't track costs."""
+        pass
+        
+    def get_metrics(self) -> Dict[str, Any]:
+        """Get current metrics."""
+        return self.telemetry.get_metrics()

{praisonaiagents-0.0.101.dist-info → praisonaiagents-0.0.103.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praisonaiagents
-Version: 0.0.101
+Version: 0.0.103
 Summary: Praison AI agents for completing complex tasks with Self Reflection Agents
 Author: Mervin Praison
 Requires-Python: >=3.10
@@ -8,13 +8,14 @@ Requires-Dist: pydantic
 Requires-Dist: rich
 Requires-Dist: openai
 Requires-Dist: mcp>=1.6.0
+Requires-Dist: posthog>=4.0.0
 Provides-Extra: mcp
 Requires-Dist: mcp>=1.6.0; extra == "mcp"
 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.50.0; extra == "memory"
+Requires-Dist: litellm>=1.72.0; extra == "memory"
 Provides-Extra: knowledge
 Requires-Dist: mem0ai>=0.1.0; extra == "knowledge"
 Requires-Dist: chromadb>=1.0.0; extra == "knowledge"
@@ -24,11 +25,13 @@ 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.50.0; extra == "llm"
+Requires-Dist: litellm>=1.72.0; extra == "llm"
 Requires-Dist: pydantic>=2.4.2; extra == "llm"
 Provides-Extra: api
 Requires-Dist: fastapi>=0.115.0; extra == "api"
 Requires-Dist: uvicorn>=0.34.0; extra == "api"
+Provides-Extra: telemetry
+Requires-Dist: posthog>=4.0.0; extra == "telemetry"
 Provides-Extra: all
 Requires-Dist: praisonaiagents[memory]; extra == "all"
 Requires-Dist: praisonaiagents[knowledge]; extra == "all"
@@ -36,3 +39,4 @@ Requires-Dist: praisonaiagents[graph]; extra == "all"
 Requires-Dist: praisonaiagents[llm]; extra == "all"
 Requires-Dist: praisonaiagents[mcp]; extra == "all"
 Requires-Dist: praisonaiagents[api]; extra == "all"
+Requires-Dist: praisonaiagents[telemetry]; extra == "all"

{praisonaiagents-0.0.101.dist-info → praisonaiagents-0.0.103.dist-info}/RECORD

@@ -1,4 +1,4 @@
-praisonaiagents/__init__.py,sha256=GmTiMNta4iwmfarh_6cTUpry50hpqFE8YqolrYfZ_7U,1465
+praisonaiagents/__init__.py,sha256=10wExtoVkZ31OwxoKjj9gtWq2uvC05dYBommV2Eii4M,2769
 praisonaiagents/approval.py,sha256=UJ4OhfihpFGR5CAaMphqpSvqdZCHi5w2MGw1MByZ1FQ,9813
 praisonaiagents/main.py,sha256=_-XE7_Y7ChvtLQMivfNFrrnAhv4wSSDhH9WJMWlkS0w,16315
 praisonaiagents/session.py,sha256=CI-ffCiOfmgB-1zFFik9daKCB5Sm41Q9ZOaq1-oSLW8,9250
@@ -14,17 +14,20 @@ praisonaiagents/guardrails/llm_guardrail.py,sha256=MTTqmYDdZX-18QN9T17T5P_6H2qnV
 praisonaiagents/knowledge/__init__.py,sha256=xL1Eh-a3xsHyIcU4foOWF-JdWYIYBALJH9bge0Ujuto,246
 praisonaiagents/knowledge/chunking.py,sha256=G6wyHa7_8V0_7VpnrrUXbEmUmptlT16ISJYaxmkSgmU,7678
 praisonaiagents/knowledge/knowledge.py,sha256=OKPar-XGyAp1ndmbOOdCgqFnTCqpOThYVSIZRxZyP58,15683
-praisonaiagents/llm/__init__.py,sha256=ttPQQJQq6Tah-0updoEXDZFKWtJAM93rBWRoIgxRWO8,689
-praisonaiagents/llm/llm.py,sha256=hoIxHzo9aNygeOiw9RtoPhpuSCVTUrKPe3OPvsT5qLc,98212
+praisonaiagents/llm/__init__.py,sha256=bSywIHBHH0YUf4hSx-FmFXkRv2g1Rlhuk-gjoImE8j8,925
+praisonaiagents/llm/llm.py,sha256=4AyXTgcolemKf4kHQOwQAIE4MJ0z-CuGs9F7UkHbyfE,98385
 praisonaiagents/mcp/__init__.py,sha256=ibbqe3_7XB7VrIcUcetkZiUZS1fTVvyMy_AqCSFG8qc,240
 praisonaiagents/mcp/mcp.py,sha256=_gfp8hrSVT9aPqEDDfU8MiCdg0-3dVQpEQUE6AbrJlo,17243
 praisonaiagents/mcp/mcp_sse.py,sha256=DLh3F_aoVRM1X-7hgIOWOw4FQ1nGmn9YNbQTesykzn4,6792
 praisonaiagents/memory/__init__.py,sha256=aEFdhgtTqDdMhc_JCWM-f4XI9cZIj7Wz5g_MUa-0amg,397
-praisonaiagents/memory/memory.py,sha256=x6CEMYhgzvlJH6SGKHPLRDt6kF0DVFFSUQbgr1OK3JM,38729
+praisonaiagents/memory/memory.py,sha256=eYXVvuXrvt4LaEJ-AAbAiwpFUCuS5LH5F7Z0cBW5_gQ,42186
 praisonaiagents/process/__init__.py,sha256=lkYbL7Hn5a0ldvJtkdH23vfIIZLIcanK-65C0MwaorY,52
 praisonaiagents/process/process.py,sha256=gxhMXG3s4CzaREyuwE5zxCMx2Wp_b_Wd53tDfkj8Qk8,66567
 praisonaiagents/task/__init__.py,sha256=VL5hXVmyGjINb34AalxpBMl-YW9m5EDcRkMTKkSSl7c,80
 praisonaiagents/task/task.py,sha256=imqJ8wzZzVyUSym2EyF2tC-vAsV1UdfI_P3YM5mqAiw,20786
+praisonaiagents/telemetry/__init__.py,sha256=5iAOrj_N_cKMmh2ltWGYs3PfOYt_jcwUoElW8fTAIsc,3062
+praisonaiagents/telemetry/integration.py,sha256=36vvYac8tW92YzQYbBeKWKM8JC9IiizlxhUy3AFqPlA,8667
+praisonaiagents/telemetry/telemetry.py,sha256=T2Mv_iOXYbL-C3CZW5EEEs7N0dUk1S2xrD1FQjVaxmc,11064
 praisonaiagents/tools/README.md,sha256=bIQGTSqQbC8l_UvTAnKbnh1TxrybSFGbCqxnhvDwkE4,4450
 praisonaiagents/tools/__init__.py,sha256=Rrgi7_3-yLHpfBB81WUi0-wD_wb_BsukwHVdjDYAF-0,9316
 praisonaiagents/tools/arxiv_tools.py,sha256=1stb31zTjLTon4jCnpZG5de9rKc9QWgC0leLegvPXWo,10528
@@ -48,7 +51,7 @@ praisonaiagents/tools/xml_tools.py,sha256=iYTMBEk5l3L3ryQ1fkUnNVYK-Nnua2Kx2S0dxN
 praisonaiagents/tools/yaml_tools.py,sha256=uogAZrhXV9O7xvspAtcTfpKSQYL2nlOTvCQXN94-G9A,14215
 praisonaiagents/tools/yfinance_tools.py,sha256=s2PBj_1v7oQnOobo2fDbQBACEHl61ftG4beG6Z979ZE,8529
 praisonaiagents/tools/train/data/generatecot.py,sha256=H6bNh-E2hqL5MW6kX3hqZ05g9ETKN2-kudSjiuU_SD8,19403
-praisonaiagents-0.0.101.dist-info/METADATA,sha256=E7LFE0liQmrpsCnURK80ynkYPxMKuNTTSJM4YKUS57o,1503
-praisonaiagents-0.0.101.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-praisonaiagents-0.0.101.dist-info/top_level.txt,sha256=_HsRddrJ23iDx5TTqVUVvXG2HeHBL5voshncAMDGjtA,16
-praisonaiagents-0.0.101.dist-info/RECORD,,
+praisonaiagents-0.0.103.dist-info/METADATA,sha256=QPwe9pWnpP519hGH179MF5Vt_Jt0uD-9F-O5D5IhK84,1669
+praisonaiagents-0.0.103.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+praisonaiagents-0.0.103.dist-info/top_level.txt,sha256=_HsRddrJ23iDx5TTqVUVvXG2HeHBL5voshncAMDGjtA,16
+praisonaiagents-0.0.103.dist-info/RECORD,,