Update session_manager.py

session_manager.py

@@ -26,7 +26,11 @@ class SessionManager:
         """Load session data or create new if not found"""
         try:
             with open(self.sessions_dir / f"{session_id}.json") as f:
-                return json.load(f)
+                data = json.load(f)
+                # Ensure history exists for backward compatibility
+                if "history" not in data:
+                    data["history"] = []
+                return data
         except FileNotFoundError:
             return self.create_session()
     
@@ -34,5 +38,14 @@ class SessionManager:
         """Reset session while keeping ID"""
         self.save_session(session_id, {
             "created_at": datetime.now().isoformat(),
-            "history": []
+            "history": [],
+            "context": {}  # Add context storage
         })
+
+    def update_context(self, session_id: str, context: dict):
+        """Update session context with new data"""
+        session = self.load_session(session_id)
+        if "context" not in session:
+            session["context"] = {}
+        session["context"].update(context)
+        self.save_session(session_id, session)