chuk-ai-session-manager 0.3__tar.gz → 0.4__tar.gz

{chuk_ai_session_manager-0.3 → chuk_ai_session_manager-0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chuk-ai-session-manager
-Version: 0.3
+Version: 0.4
 Summary: Session manager for AI applications
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
@@ -311,36 +311,6 @@ pip install chuk-ai-session-manager
 | **Production Ready** | Requires additional work | Built for production |
 | **Learning Curve** | Steep | 5 minutes to productivity |
 
-## 🚀 Migration Guides
-
-### From LangChain Memory
-```python
-# Old LangChain way
-from langchain.memory import ConversationBufferMemory
-memory = ConversationBufferMemory()
-memory.save_context({"input": "Hi"}, {"output": "Hello"})
-
-# New CHUK way (much simpler!)
-from chuk_ai_session_manager import track_conversation
-await track_conversation("Hi", "Hello")
-```
-
-### From Manual Session Management
-```python
-# Old manual way
-conversations = {}
-def save_conversation(user_id, message, response):
-    if user_id not in conversations:
-        conversations[user_id] = []
-    conversations[user_id].append({"user": message, "ai": response})
-
-# New CHUK way
-from chuk_ai_session_manager import SessionManager
-sm = SessionManager(session_id=user_id)
-await sm.user_says(message)
-await sm.ai_responds(response)
-```
-
 ## 📖 More Examples
 
 Check out the `/examples` directory for complete working examples:
@@ -362,16 +332,9 @@ Check out the `/examples` directory for complete working examples:
 - ✅ Complete conversation analytics and observability
 - ✅ Framework-agnostic solution that works with any LLM library
 
-**Consider alternatives if you:**
-- ❌ Only need basic in-memory conversation history
-- ❌ Are locked into a specific framework (LangChain, etc.)
-- ❌ Don't need cost tracking or analytics
-- ❌ Are building simple, stateless AI applications
-
 ## 🤝 Community & Support
 
 - 📖 **Documentation**: [Full docs with tutorials](link-to-docs)
-- 💬 **Discord**: Join our community for help and discussions
 - 🐛 **Issues**: Report bugs on GitHub
 - 💡 **Feature Requests**: Suggest new features
 - 📧 **Support**: enterprise@chuk.dev for production support

{chuk_ai_session_manager-0.3 → chuk_ai_session_manager-0.4}/README.md

@@ -289,36 +289,6 @@ pip install chuk-ai-session-manager
 | **Production Ready** | Requires additional work | Built for production |
 | **Learning Curve** | Steep | 5 minutes to productivity |
 
-## 🚀 Migration Guides
-
-### From LangChain Memory
-```python
-# Old LangChain way
-from langchain.memory import ConversationBufferMemory
-memory = ConversationBufferMemory()
-memory.save_context({"input": "Hi"}, {"output": "Hello"})
-
-# New CHUK way (much simpler!)
-from chuk_ai_session_manager import track_conversation
-await track_conversation("Hi", "Hello")
-```
-
-### From Manual Session Management
-```python
-# Old manual way
-conversations = {}
-def save_conversation(user_id, message, response):
-    if user_id not in conversations:
-        conversations[user_id] = []
-    conversations[user_id].append({"user": message, "ai": response})
-
-# New CHUK way
-from chuk_ai_session_manager import SessionManager
-sm = SessionManager(session_id=user_id)
-await sm.user_says(message)
-await sm.ai_responds(response)
-```
-
 ## 📖 More Examples
 
 Check out the `/examples` directory for complete working examples:
@@ -340,16 +310,9 @@ Check out the `/examples` directory for complete working examples:
 - ✅ Complete conversation analytics and observability
 - ✅ Framework-agnostic solution that works with any LLM library
 
-**Consider alternatives if you:**
-- ❌ Only need basic in-memory conversation history
-- ❌ Are locked into a specific framework (LangChain, etc.)
-- ❌ Don't need cost tracking or analytics
-- ❌ Are building simple, stateless AI applications
-
 ## 🤝 Community & Support
 
 - 📖 **Documentation**: [Full docs with tutorials](link-to-docs)
-- 💬 **Discord**: Join our community for help and discussions
 - 🐛 **Issues**: Report bugs on GitHub
 - 💡 **Feature Requests**: Suggest new features
 - 📧 **Support**: enterprise@chuk.dev for production support

{chuk_ai_session_manager-0.3 → chuk_ai_session_manager-0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "chuk-ai-session-manager"
-version = "0.3"
+version = "0.4"
 description = "Session manager for AI applications"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -36,10 +36,26 @@ include = ["chuk_ai_session_manager*"]
 # pytest settings so it finds your src/ layout automatically
 [tool.pytest.ini_options]
 testpaths = ["tests"]
-pythonpath = ["src"]
-addopts = "-v"
-asyncio_default_fixture_loop_scope = "function"
-asyncio_mode = "strict"
+asyncio_mode = "auto"
+python_files = ["test_*.py"]
+python_classes = ["Test*"]
+python_functions = ["test_*"]
+addopts = [
+    "--strict-markers",
+    "--strict-config",
+    "-ra",
+    "--tb=short",
+]
+markers = [
+    "unit: Unit tests",
+    "integration: Integration tests", 
+    "slow: Slow tests",
+    "asyncio: Async tests",
+]
+filterwarnings = [
+    "ignore::DeprecationWarning",
+    "ignore::PendingDeprecationWarning",
+]
 
 [tool.black]
 line-length = 88

{chuk_ai_session_manager-0.3 → chuk_ai_session_manager-0.4}/src/chuk_ai_session_manager/__init__.py

@@ -15,7 +15,18 @@ Quick Start:
     await sm.ai_responds("It's sunny and 72°F", model="gpt-4")
 """
 
-# Simple API exports - the main things developers need
+# Import core models first (these have no circular dependencies)
+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
+
+# Import other models (these might depend on storage being set up)
+from chuk_ai_session_manager.models.session import Session
+from chuk_ai_session_manager.models.session_event import SessionEvent
+
+# Import the simple API (this should work now that storage is fixed)
 from chuk_ai_session_manager.api.simple_api import (
     SessionManager,
     track_conversation,
@@ -24,15 +35,6 @@ from chuk_ai_session_manager.api.simple_api import (
     track_infinite_conversation
 )
 
-# Core models for advanced users
-from chuk_ai_session_manager.models.session import Session
-from chuk_ai_session_manager.models.session_event import SessionEvent
-from chuk_ai_session_manager.models.event_source import EventSource
-from chuk_ai_session_manager.models.event_type import EventType
-
-# Storage backend setup
-from chuk_ai_session_manager.session_storage import setup_chuk_sessions_storage
-
 __version__ = "0.1.0"
 
 # Main exports - keep it simple

{chuk_ai_session_manager-0.3 → chuk_ai_session_manager-0.4}/src/chuk_ai_session_manager/models/token_usage.py

@@ -175,7 +175,7 @@ class TokenUsage(BaseModel):
         )
     
     @staticmethod
-    def _count_tokens_sync(text: Optional[str], model: str = "gpt-3.5-turbo") -> int:
+    def _count_tokens_sync(text: Optional[Union[str, Any]], model: str = "gpt-3.5-turbo") -> int:
         """
         Synchronous implementation of count_tokens.
         
@@ -188,6 +188,17 @@ class TokenUsage(BaseModel):
         """
         if text is None:
             return 0
+        
+        # Convert to string if not already a string
+        if not isinstance(text, str):
+            try:
+                text = str(text)
+            except Exception:
+                return 0
+        
+        # Empty string has 0 tokens
+        if not text:
+            return 0
             
         if TIKTOKEN_AVAILABLE:
             try:
@@ -206,7 +217,7 @@ class TokenUsage(BaseModel):
         return int(len(text) / 4)
     
     @staticmethod
-    async def count_tokens(text: Optional[str], model: str = "gpt-3.5-turbo") -> int:
+    async def count_tokens(text: Optional[Union[str, Any]], model: str = "gpt-3.5-turbo") -> int:
         """
         Async version of count_tokens.
         

{chuk_ai_session_manager-0.3 → chuk_ai_session_manager-0.4}/src/chuk_ai_session_manager/sample_tools.py

@@ -1,4 +1,4 @@
-# sample_tools.py
+# chuk_ai_session_manager/sample_tools.py
 """
 Sample tools for chuk session manager demos - corrected version following registry example
 """

{chuk_ai_session_manager-0.3 → chuk_ai_session_manager-0.4}/src/chuk_ai_session_manager/session_prompt_builder.py

@@ -37,7 +37,8 @@ async def build_prompt_from_session(
     max_tokens: Optional[int] = None,
     model: str = "gpt-3.5-turbo",
     include_parent_context: bool = False,
-    current_query: Optional[str] = None
+    current_query: Optional[str] = None,
+    max_history: int = 5  # Add this parameter for conversation strategy
 ) -> List[Dict[str, str]]:
     """
     Build a prompt for the next LLM call from a Session asynchronously.
@@ -49,6 +50,7 @@ async def build_prompt_from_session(
         model: Model to use for token counting
         include_parent_context: Whether to include context from parent sessions
         current_query: Current user query for relevance-based context selection
+        max_history: Maximum number of messages to include for conversation strategy
         
     Returns:
         A list of message dictionaries suitable for LLM API calls
@@ -72,7 +74,7 @@ async def build_prompt_from_session(
     elif strategy == PromptStrategy.TOOL_FOCUSED:
         return await _build_tool_focused_prompt(session)
     elif strategy == PromptStrategy.CONVERSATION:
-        return await _build_conversation_prompt(session, max_history=5)
+        return await _build_conversation_prompt(session, max_history)
     elif strategy == PromptStrategy.HIERARCHICAL:
         return await _build_hierarchical_prompt(session, include_parent_context)
     else:
@@ -112,7 +114,7 @@ async def _build_minimal_prompt(session: Session) -> List[Dict[str, str]]:
     
     if assistant_msg is None:
         # Only the user message exists so far
-        return [{"role": "user", "content": first_user.message}] if first_user else []
+        return [{"role": "user", "content": _extract_content(first_user.message)}] if first_user else []
 
     # Children of that assistant
     children = [
@@ -126,11 +128,7 @@ async def _build_minimal_prompt(session: Session) -> List[Dict[str, str]]:
     # Assemble prompt
     prompt: List[Dict[str, str]] = []
     if first_user:
-        # Handle both string messages and dict messages
-        user_content = first_user.message
-        if isinstance(user_content, dict) and "content" in user_content:
-            user_content = user_content["content"]
-        prompt.append({"role": "user", "content": user_content})
+        prompt.append({"role": "user", "content": _extract_content(first_user.message)})
 
     # ALWAYS add the assistant marker - but strip its free text
     prompt.append({"role": "assistant", "content": None})
@@ -166,6 +164,24 @@ async def _build_minimal_prompt(session: Session) -> List[Dict[str, str]]:
     return prompt
 
 
+def _extract_content(message: Any) -> str:
+    """
+    Extract content string from a message that could be a string or dict.
+    
+    Args:
+        message: The message content (string, dict, or other)
+        
+    Returns:
+        The extracted content as a string
+    """
+    if isinstance(message, str):
+        return message
+    elif isinstance(message, dict) and "content" in message:
+        return message["content"]
+    else:
+        return str(message)
+
+
 async def _build_task_focused_prompt(session: Session) -> List[Dict[str, str]]:
     """
     Build a task-focused prompt.
@@ -201,17 +217,11 @@ async def _build_task_focused_prompt(session: Session) -> List[Dict[str, str]]:
     prompt = []
     
     # Always include the first user message (the main task)
-    first_content = first_user.message
-    if isinstance(first_content, dict) and "content" in first_content:
-        first_content = first_content["content"]
-    prompt.append({"role": "user", "content": first_content})
+    prompt.append({"role": "user", "content": _extract_content(first_user.message)})
     
     # Include the latest user message if different from the first
     if latest_user and latest_user.id != first_user.id:
-        latest_content = latest_user.message
-        if isinstance(latest_content, dict) and "content" in latest_content:
-            latest_content = latest_content["content"]
-        prompt.append({"role": "user", "content": latest_content})
+        prompt.append({"role": "user", "content": _extract_content(latest_user.message)})
     
     # Include assistant response placeholder
     if assistant_msg:
@@ -274,10 +284,7 @@ async def _build_tool_focused_prompt(session: Session) -> List[Dict[str, str]]:
     prompt = []
     
     # Include user message
-    user_content = latest_user.message
-    if isinstance(user_content, dict) and "content" in user_content:
-        user_content = user_content["content"]
-    prompt.append({"role": "user", "content": user_content})
+    prompt.append({"role": "user", "content": _extract_content(latest_user.message)})
     
     # Include assistant placeholder
     if assistant_msg:
@@ -334,17 +341,17 @@ async def _build_conversation_prompt(
     
     # Build the conversation history
     prompt = []
-    for msg in recent_messages:
+    for i, msg in enumerate(recent_messages):
         role = "user" if msg.source == EventSource.USER else "assistant"
-        content = msg.message
+        content = _extract_content(msg.message)
         
-        # Handle different message formats
-        if isinstance(content, dict) and "content" in content:
-            content = content["content"]
-        
-        # For the last assistant message, set content to None
-        if role == "assistant" and msg == recent_messages[-1] and msg.source != EventSource.USER:
-            content = None
+        # For the last assistant message, set content to None and add tool calls
+        if (role == "assistant" and 
+            msg == recent_messages[-1] and 
+            msg.source != EventSource.USER):
+            
+            # Add the message first with None content
+            prompt.append({"role": role, "content": None})
             
             # Add tool call results for this assistant message
             tool_calls = [
@@ -352,9 +359,6 @@ async def _build_conversation_prompt(
                 if e.type == EventType.TOOL_CALL and e.metadata.get("parent_event_id") == msg.id
             ]
             
-            # Add the message first, then tools
-            prompt.append({"role": role, "content": content})
-            
             # Add tool results
             for tc in tool_calls:
                 if isinstance(tc.message, dict):
@@ -366,11 +370,9 @@ async def _build_conversation_prompt(
                         "name": tool_name,
                         "content": json.dumps(tool_result, default=str),
                     })
-                    
-            # Skip adding this message again
-            continue
-            
-        prompt.append({"role": role, "content": content})
+        else:
+            # Regular message
+            prompt.append({"role": role, "content": content})
     
     return prompt
 
@@ -391,32 +393,38 @@ async def _build_hierarchical_prompt(
     
     # If parent context is enabled and session has a parent
     if include_parent_context and session.parent_id:
-        # Get the storage backend and create store
-        backend = get_backend()
-        store = ChukSessionsStore(backend)
-        parent = await store.get(session.parent_id)
-        
-        if parent:
-            # Find the most recent summary in parent
-            summary_event = next(
-                (e for e in reversed(parent.events) 
-                 if e.type == EventType.SUMMARY),
-                None
-            )
+        try:
+            # Get the storage backend and create store
+            backend = get_backend()
+            store = ChukSessionsStore(backend)
+            parent = await store.get(session.parent_id)
             
-            if summary_event:
-                # Extract summary content
-                summary_content = summary_event.message
-                if isinstance(summary_content, dict) and "note" in summary_content:
-                    summary_content = summary_content["note"]
-                elif isinstance(summary_content, dict) and "content" in summary_content:
-                    summary_content = summary_content["content"]
-                    
-                # Add parent context at the beginning
-                prompt.insert(0, {
-                    "role": "system",
-                    "content": f"Context from previous conversation: {summary_content}"
-                })
+            if parent:
+                # Find the most recent summary in parent
+                summary_event = next(
+                    (e for e in reversed(parent.events) 
+                     if e.type == EventType.SUMMARY),
+                    None
+                )
+                
+                if summary_event:
+                    # Extract summary content
+                    summary_content = summary_event.message
+                    if isinstance(summary_content, dict) and "note" in summary_content:
+                        summary_content = summary_content["note"]
+                    elif isinstance(summary_content, dict) and "content" in summary_content:
+                        summary_content = summary_content["content"]
+                    else:
+                        summary_content = str(summary_content)
+                        
+                    # Add parent context at the beginning
+                    prompt.insert(0, {
+                        "role": "system",
+                        "content": f"Context from previous conversation: {summary_content}"
+                    })
+        except Exception as e:
+            # If we can't load parent context, just continue with minimal prompt
+            logger.warning(f"Could not load parent context for session {session.parent_id}: {e}")
     
     return prompt
 

{chuk_ai_session_manager-0.3 → chuk_ai_session_manager-0.4}/src/chuk_ai_session_manager.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chuk-ai-session-manager
-Version: 0.3
+Version: 0.4
 Summary: Session manager for AI applications
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
@@ -311,36 +311,6 @@ pip install chuk-ai-session-manager
 | **Production Ready** | Requires additional work | Built for production |
 | **Learning Curve** | Steep | 5 minutes to productivity |
 
-## 🚀 Migration Guides
-
-### From LangChain Memory
-```python
-# Old LangChain way
-from langchain.memory import ConversationBufferMemory
-memory = ConversationBufferMemory()
-memory.save_context({"input": "Hi"}, {"output": "Hello"})
-
-# New CHUK way (much simpler!)
-from chuk_ai_session_manager import track_conversation
-await track_conversation("Hi", "Hello")
-```
-
-### From Manual Session Management
-```python
-# Old manual way
-conversations = {}
-def save_conversation(user_id, message, response):
-    if user_id not in conversations:
-        conversations[user_id] = []
-    conversations[user_id].append({"user": message, "ai": response})
-
-# New CHUK way
-from chuk_ai_session_manager import SessionManager
-sm = SessionManager(session_id=user_id)
-await sm.user_says(message)
-await sm.ai_responds(response)
-```
-
 ## 📖 More Examples
 
 Check out the `/examples` directory for complete working examples:
@@ -362,16 +332,9 @@ Check out the `/examples` directory for complete working examples:
 - ✅ Complete conversation analytics and observability
 - ✅ Framework-agnostic solution that works with any LLM library
 
-**Consider alternatives if you:**
-- ❌ Only need basic in-memory conversation history
-- ❌ Are locked into a specific framework (LangChain, etc.)
-- ❌ Don't need cost tracking or analytics
-- ❌ Are building simple, stateless AI applications
-
 ## 🤝 Community & Support
 
 - 📖 **Documentation**: [Full docs with tutorials](link-to-docs)
-- 💬 **Discord**: Join our community for help and discussions
 - 🐛 **Issues**: Report bugs on GitHub
 - 💡 **Feature Requests**: Suggest new features
 - 📧 **Support**: enterprise@chuk.dev for production support

{chuk_ai_session_manager-0.3 → chuk_ai_session_manager-0.4}/src/chuk_ai_session_manager.egg-info/SOURCES.txt

@@ -22,12 +22,12 @@ src/chuk_ai_session_manager/models/session_event.py
 src/chuk_ai_session_manager/models/session_metadata.py
 src/chuk_ai_session_manager/models/session_run.py
 src/chuk_ai_session_manager/models/token_usage.py
-src/chuk_ai_session_manager/utils/__init__.py
-src/chuk_ai_session_manager/utils/status_display_utils.py
 tests/test_basic_functionality.py
-tests/test_chuk_session_storage.py
+tests/test_exceptions.py
 tests/test_infinite_conversation.py
-tests/test_infinite_conversation_advanced.py
-tests/test_session.py
-tests/test_session_aware_tool_processor.py
-tests/test_session_prompt_builder.py
+tests/test_models.py
+tests/test_prompt_builder.py
+tests/test_simple.py
+tests/test_simple_api.py
+tests/test_storage.py
+tests/test_tools.py