Update src/memory.py

src/memory.py

@@ -7,7 +7,6 @@ from datetime import datetime
 class EnhancedInMemoryHistory(BaseMemory):
     """
     Custom memory class for storing chat history with timestamps.
-
     Attributes:
         messages (List[BaseMessage]): A list of messages exchanged between the user and the bot.
         timestamps (List[datetime]): A list of timestamps when the messages were exchanged.
@@ -24,7 +23,6 @@ class EnhancedInMemoryHistory(BaseMemory):
     def add_messages(self, messages: List[BaseMessage]):
         """
         Adds new messages to the memory and timestamps them.
-
         Parameters:
             messages (List[BaseMessage]): A list of messages to add to the memory.
         """
@@ -35,10 +33,8 @@ class EnhancedInMemoryHistory(BaseMemory):
     def get_recent_messages(self, limit: int = 5):
         """
         Retrieves the most recent messages.
-
         Parameters:
             limit (int): Number of recent messages to retrieve (default is 5).
-
         Returns:
             List[BaseMessage]: A list of the most recent messages.
         """
@@ -52,10 +48,8 @@ class EnhancedInMemoryHistory(BaseMemory):
     def load_memory_variables(self, inputs: dict):
         """
         Loads memory variables for the conversation chain.
-
         Parameters:
             inputs (dict): Input data for the conversation chain.
-
         Returns:
             dict: A dictionary with the conversation history.
         """
@@ -64,7 +58,6 @@ class EnhancedInMemoryHistory(BaseMemory):
     def save_context(self, inputs: dict, outputs: str):
         """
         Saves the context of the conversation by storing the user input and bot output.
-
         Parameters:
             inputs (dict): The user input.
             outputs (str): The bot's response.
@@ -73,5 +66,4 @@ class EnhancedInMemoryHistory(BaseMemory):
 
     def clear_memory(self):
         """Clears the memory (alias for the clear function)."""
-        self.clear()
-
+        self.clear()