chuk-ai-session-manager 0.4__py3-none-any.whl → 0.5__py3-none-any.whl

chuk_ai_session_manager/__init__.py

@@ -1,7 +1,15 @@
-# chuk_ai_session_manager/__init__.py
+# src/chuk_ai_session_manager/__init__.py
 """
 CHUK AI Session Manager - Simple Developer API
 
+A powerful session management system for AI applications that provides:
+- Automatic conversation tracking
+- Token usage monitoring  
+- Tool call logging
+- Infinite context support with automatic summarization
+- Hierarchical session relationships
+- CHUK Sessions backend integration
+
 Quick Start:
     from chuk_ai_session_manager import track_conversation
 
@@ -13,61 +21,185 @@ Quick Start:
     sm = SessionManager()
     await sm.user_says("What's the weather?")
     await sm.ai_responds("It's sunny and 72°F", model="gpt-4")
+
+Infinite Context Example:
+    # Automatically handles long conversations with summarization
+    sm = SessionManager(infinite_context=True, token_threshold=4000)
+    await sm.user_says("Tell me about machine learning")
+    await sm.ai_responds("Machine learning is...", model="gpt-4")
+    # Session will auto-segment when limits are reached
 """
 
-# Import core models first (these have no circular dependencies)
+import logging
+
+# Package version
+__version__ = "0.4"
+
+# Set up package-level logger
+logger = logging.getLogger(__name__)
+
+# Core enums and constants
 from chuk_ai_session_manager.models.event_source import EventSource
 from chuk_ai_session_manager.models.event_type import EventType
 
-# Import storage setup (this should work now with the fixed session_storage.py)
-from chuk_ai_session_manager.session_storage import setup_chuk_sessions_storage
+# Exception classes
+from chuk_ai_session_manager.exceptions import (
+    SessionManagerError,
+    SessionNotFound,
+    SessionAlreadyExists,
+    InvalidSessionOperation,
+    TokenLimitExceeded,
+    StorageError,
+    ToolProcessingError
+)
 
-# Import other models (these might depend on storage being set up)
-from chuk_ai_session_manager.models.session import Session
+# Core models
+from chuk_ai_session_manager.models.session_metadata import SessionMetadata
+from chuk_ai_session_manager.models.session_run import SessionRun, RunStatus
+from chuk_ai_session_manager.models.token_usage import TokenUsage, TokenSummary
 from chuk_ai_session_manager.models.session_event import SessionEvent
+from chuk_ai_session_manager.models.session import Session
+
+# Storage backend
+from chuk_ai_session_manager.session_storage import setup_chuk_sessions_storage
 
-# Import the simple API (this should work now that storage is fixed)
+# High-level session manager
+from chuk_ai_session_manager.session_manager import SessionManager
+
+# Simple API - The main interface most users will use
 from chuk_ai_session_manager.api.simple_api import (
-    SessionManager,
     track_conversation,
     track_llm_call,
     quick_conversation,
-    track_infinite_conversation
+    track_infinite_conversation,
+    track_tool_use,
+    get_session_stats,
+    get_conversation_history
+)
+
+# Advanced components
+from chuk_ai_session_manager.infinite_conversation import (
+    InfiniteConversationManager,
+    SummarizationStrategy
+)
+
+from chuk_ai_session_manager.session_aware_tool_processor import SessionAwareToolProcessor
+
+from chuk_ai_session_manager.session_prompt_builder import (
+    build_prompt_from_session,
+    PromptStrategy,
+    truncate_prompt_to_token_limit
 )
 
-__version__ = "0.1.0"
+# Configuration functions
+def configure_storage(sandbox_id: str = "chuk-ai-session-manager", 
+                     default_ttl_hours: int = 24) -> bool:
+    """
+    Configure the storage backend.
+    
+    Args:
+        sandbox_id: CHUK Sessions sandbox ID to use
+        default_ttl_hours: Default TTL for sessions
+        
+    Returns:
+        True if configuration was successful, False otherwise
+    """
+    try:
+        setup_chuk_sessions_storage(
+            sandbox_id=sandbox_id,
+            default_ttl_hours=default_ttl_hours
+        )
+        logger.info(f"Storage configured with sandbox_id='{sandbox_id}'")
+        return True
+    except Exception as e:
+        logger.error(f"Failed to configure storage: {e}")
+        return False
+
 
-# Main exports - keep it simple
+def get_version() -> str:
+    """Get the package version."""
+    return __version__
+
+
+def is_available() -> dict:
+    """
+    Check which components are available.
+    
+    Returns:
+        Dictionary showing availability of each component
+    """
+    return {
+        "core_enums": True,
+        "core_models": True,
+        "simple_api": True,
+        "storage": True,
+        "infinite_context": True,
+        "tool_processor": True,
+        "prompt_builder": True,
+        "token_tracking": True,
+        "exceptions": True,
+        "session_manager": True,
+    }
+
+
+# Main exports - everything should be available
 __all__ = [
-    # Simple API - most developers start here
+    # Version and utilities
+    "__version__",
+    "get_version", 
+    "is_available",
+    "configure_storage",
+    
+    # Core enums
+    "EventSource",
+    "EventType",
+    
+    # Exception classes
+    "SessionManagerError",
+    "SessionNotFound", 
+    "SessionAlreadyExists",
+    "InvalidSessionOperation",
+    "TokenLimitExceeded",
+    "StorageError",
+    "ToolProcessingError",
+    
+    # Core models
+    "Session",
+    "SessionEvent",
+    "SessionMetadata",
+    "SessionRun",
+    "RunStatus",
+    "TokenUsage",
+    "TokenSummary",
+    
+    # Storage
+    "setup_chuk_sessions_storage",
+    
+    # Primary interfaces - what most users will use
     "SessionManager",
     "track_conversation", 
     "track_llm_call",
     "quick_conversation",
     "track_infinite_conversation",
+    "track_tool_use",
+    "get_session_stats",
+    "get_conversation_history",
     
-    # Core models for advanced usage
-    "Session",
-    "SessionEvent", 
-    "EventSource",
-    "EventType",
-    
-    # Setup
-    "setup_chuk_sessions_storage",
+    # Advanced components
+    "InfiniteConversationManager",
+    "SummarizationStrategy",
+    "SessionAwareToolProcessor",
+    "build_prompt_from_session",
+    "PromptStrategy",
+    "truncate_prompt_to_token_limit",
 ]
 
-# Auto-setup with sensible defaults
-def _auto_setup():
-    """Auto-setup with good defaults if not already configured."""
-    try:
-        from chuk_ai_session_manager.session_storage import get_backend
-        get_backend()  # This will trigger setup if needed
-    except Exception:
-        # Silently setup with defaults
-        setup_chuk_sessions_storage(
-            sandbox_id="chuk-ai-session-manager",
-            default_ttl_hours=24
-        )
+# Auto-setup storage on import
+try:
+    configure_storage()
+    logger.debug("Auto-configured storage backend")
+except Exception as e:
+    logger.debug(f"Auto-setup skipped: {e}")
 
-# Run auto-setup on import
-_auto_setup()
+# Log successful import
+logger.debug(f"CHUK AI Session Manager v{__version__} imported successfully")