Update src/memory.py

src/memory.py

@@ -66,4 +66,19 @@ class EnhancedInMemoryHistory(BaseMemory):
 
     def clear_memory(self):
         """Clears the memory (alias for the clear function)."""
-        self.clear()
+        self.clear()
+
+# Store sessions in a dictionary
+store = {}
+
+def get_by_session_id(session_id: str) -> EnhancedInMemoryHistory:
+    """
+    Retrieves or creates a new memory session by session ID.
+    Parameters:
+        session_id (str): The session ID for the chat session.
+    Returns:
+        EnhancedInMemoryHistory: The memory object for the session.
+    """
+    if session_id not in store:
+        store[session_id] = EnhancedInMemoryHistory()
+    return store[session_id]