fix : memory recovery bug

This commit is contained in:
martin legrand
2025-03-27 10:22:48 +01:00
parent 24329aa3d2
commit 697bc882c7
3 changed files with 7 additions and 4 deletions
+4
View File
@@ -25,8 +25,10 @@ class Memory():
self.session_time = datetime.datetime.now()
self.session_id = str(uuid.uuid4())
self.conversation_folder = f"conversations/"
self.session_recovered = False
if recover_last_session:
self.load_memory()
self.session_recovered = True
# memory compression system
self.model = "pszemraj/led-base-book-summary"
self.device = self.get_cuda_device()
@@ -65,6 +67,8 @@ class Memory():
def load_memory(self, agent_type: str = "casual_agent") -> None:
"""Load the memory from the last session."""
if self.session_recovered == True:
return
save_path = os.path.join(self.conversation_folder, agent_type)
if not os.path.exists(save_path):
return