chuk-ai-session-manager 0.3__py3-none-any.whl → 0.4.1__py3-none-any.whl

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/session_storage.py

@@ -1,4 +1,4 @@
-# chuk_ai_session_manager/session_storage.py
+# src/chuk_ai_session_manager/session_storage.py
 """
 CHUK Sessions storage backend for AI Session Manager.
 
@@ -11,7 +11,18 @@ import json
 import logging
 from typing import Any, Dict, List, Optional
 
-from chuk_sessions import SessionManager as ChukSessionManager
+# Import the correct class from chuk_sessions
+# We need to determine the correct import based on what's actually available
+try:
+    from chuk_sessions import SessionManager as ChukSessionManager
+except ImportError as e:
+    raise ImportError(
+        f"Cannot import SessionManager from chuk_sessions: {e}\n"
+        "Please ensure chuk_sessions is properly installed with: uv add chuk-sessions\n"
+        "Or check the available classes in chuk_sessions by running:\n"
+        "python -c \"import chuk_sessions; print(dir(chuk_sessions))\""
+    ) from e
+
 from chuk_ai_session_manager.models.session import Session
 
 logger = logging.getLogger(__name__)
@@ -45,19 +56,21 @@ class SessionStorage:
             return self._cache[session_id]
         
         try:
-            if not await self.chuk.validate_session(session_id):
-                return None
-            
+            # Get session info from CHUK Sessions
             info = await self.chuk.get_session_info(session_id)
             if not info:
                 return None
             
+            # Check if it's an AI session manager session
             custom_metadata = info.get('custom_metadata', {})
-            ai_session_json = custom_metadata.get('ai_session_data')
+            if custom_metadata.get('session_type') != 'ai_session_manager':
+                return None
             
+            ai_session_json = custom_metadata.get('ai_session_data')
             if not ai_session_json:
                 return None
             
+            # Parse the JSON data
             session_data = json.loads(ai_session_json)
             ai_session = Session.model_validate(session_data)
             

{chuk_ai_session_manager-0.3.dist-info → chuk_ai_session_manager-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chuk-ai-session-manager
-Version: 0.3
+Version: 0.4.1
 Summary: Session manager for AI applications
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
@@ -14,6 +14,7 @@ Requires-Dist: redis>=4.0.0; extra == "redis"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
 Requires-Dist: redis>=4.0.0; extra == "dev"
 Requires-Dist: black>=23.0.0; extra == "dev"
 Requires-Dist: isort>=5.12.0; extra == "dev"
@@ -311,36 +312,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 +333,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.4.1.dist-info/RECORD

@@ -0,0 +1,22 @@
+chuk_ai_session_manager/__init__.py,sha256=Kg8_fqwkV88b0f9H8Q45MDqaZdHHTo3Wf0Oko89DhHg,10217
+chuk_ai_session_manager/exceptions.py,sha256=WqrrUZuOAiUmz7tKnSnk0y222U_nV9a8LyaXLayn2fg,4420
+chuk_ai_session_manager/infinite_conversation.py,sha256=7j3caMnsX27M5rjj4oOkqiy_2AfcupWwsAWRflnKiSo,12092
+chuk_ai_session_manager/sample_tools.py,sha256=U-jTGveTJ95uSnA4jB30fJQJG3K-TGxN9jcOY6qVHZQ,8179
+chuk_ai_session_manager/session_aware_tool_processor.py,sha256=iVe3d-qfp5QGkdNrgfZeRYoOjd8nLZ0g6K7HW1thFE8,7274
+chuk_ai_session_manager/session_manager.py,sha256=AauyE2XTqAM53fGxpJD73nVwxB3cWH0jMlBXakYTAD0,29195
+chuk_ai_session_manager/session_prompt_builder.py,sha256=Jeg_MWses_hFtHtDL7ZQl6EdSNVmVIIrLDrWEoPumfM,17613
+chuk_ai_session_manager/session_storage.py,sha256=wb6SPwOKVoov2cGmK6DuorpPM7xbjnCRHrA6VTvG7GE,6427
+chuk_ai_session_manager/api/__init__.py,sha256=Lo_BoDW2rSn0Zw-CbjahOxc6ykjjTpucxHZo5FA2Gnc,41
+chuk_ai_session_manager/api/simple_api.py,sha256=aUDH2KzkI_M2eUiW9DAQo2KmUYaVa5u25-gSIF3PzGM,10504
+chuk_ai_session_manager/models/__init__.py,sha256=H1rRuDQDRf821JPUWUn5Zgwvc5BAqcEGekkHEmX-IgE,1167
+chuk_ai_session_manager/models/event_source.py,sha256=mn_D16sXMa6nAX-5BzssygJPz6VF24GRe-3IaH7bTnI,196
+chuk_ai_session_manager/models/event_type.py,sha256=TPPvAz-PlXVtrwXDNVFVnhdt1yEfgDGmKDGt8ArYcTk,275
+chuk_ai_session_manager/models/session.py,sha256=yUPuu6yt8HbBebkIejpgSDGTLhZjtLOxFRkWG2C6XKM,12193
+chuk_ai_session_manager/models/session_event.py,sha256=RTghC9_sDHzD8qdgEYCoclJzpVo8cE6B2f7jlsDOVr0,9006
+chuk_ai_session_manager/models/session_metadata.py,sha256=KFG7lc_E0BQTP2OD9Y529elVGJXppDUMqz8vVONW0rw,1510
+chuk_ai_session_manager/models/session_run.py,sha256=uhMM4-WSrqOUsiWQPnyakInd-foZhxI-YnSHSWiZZwE,4369
+chuk_ai_session_manager/models/token_usage.py,sha256=M9Qwmeb2woILaSRwA2SIAiG-sIwC3cL_1H-y3NjW5Ik,11436
+chuk_ai_session_manager-0.4.1.dist-info/METADATA,sha256=0PvxzwOw2HG8XujmXkMInNVAlrhF149rxyLzXTUROAg,11136
+chuk_ai_session_manager-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chuk_ai_session_manager-0.4.1.dist-info/top_level.txt,sha256=5RinqD0v-niHuLYePUREX4gEWTlrpgtUg0RfexVRBMk,24
+chuk_ai_session_manager-0.4.1.dist-info/RECORD,,