feat: enhanced memory flows for better handle loaded conversation

Author: daltonnyx

AgentCrew/modules/chat/message/conversation.py

@@ -111,6 +111,9 @@ def load_conversation(self, conversation_id: str) -> Optional[List[Dict[str, Any
                         self.message_handler.current_conversation_id
                     )
                     self.message_handler.memory_service.loaded_conversation = True
+                    self.message_handler.memory_service.load_conversation_context(
+                        self.message_handler.current_conversation_id
+                    )
                 last_agent_name = history[-1].get("agent", "")
                 if last_agent_name and self.message_handler.agent_manager.select_agent(
                     last_agent_name

AgentCrew/modules/memory/base_service.py

@@ -49,6 +49,10 @@ async def need_generate_user_context(self, user_input) -> bool:
     def clear_conversation_context(self):
         pass
 
+    @abstractmethod
+    def load_conversation_context(self, session_id: str):
+        pass
+
     @abstractmethod
     def generate_user_context(self, user_input: str, agent_name: str = "None") -> str:
         """

AgentCrew/modules/memory/chroma_service.py

@@ -378,8 +378,21 @@ async def need_generate_user_context(self, user_input: str) -> bool:
 
     def clear_conversation_context(self):
         self.current_embedding_context = None
+        self.current_conversation_context = {}
         self.context_embedding = []
 
+    def load_conversation_context(self, session_id: str):
+        collection = self._initialize_collection()
+        latest_memory = collection.get(
+            where={
+                "session_id": session_id,
+            },
+        )
+        if latest_memory["documents"]:
+            self.current_conversation_context[session_id] = latest_memory["documents"][
+                -1
+            ]
+
     def generate_user_context(self, user_input: str, agent_name: str = "None") -> str:
         """
         Generate context based on user input by retrieving relevant memories.