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

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)