jblast94 commited on
Commit
5e6055c
·
verified ·
1 Parent(s): 1ce8863

Update memory_manager.py

Browse files
Files changed (1) hide show
  1. memory_manager.py +3 -3
memory_manager.py CHANGED
@@ -1,6 +1,6 @@
1
  import os
2
  import json
3
- from datetime import datetime # <--- THIS IS THE FIX
4
  import time
5
 
6
  class MemoryManager:
@@ -12,7 +12,7 @@ class MemoryManager:
12
  def initialize_user_profile(self, user_id: str):
13
  if user_id not in self.memory:
14
  self.memory[user_id] = {
15
- "created_at": datetime.datetime.now().isoformat(),
16
  "conversations": []
17
  }
18
  self._save_memory()
@@ -21,7 +21,7 @@ class MemoryManager:
21
  def store_conversation(self, user_id: str, message: str, role: str):
22
  if user_id in self.memory:
23
  self.memory[user_id]["conversations"].append({
24
- "timestamp": datetime.datetime.now().isoformat(),
25
  "role": role,
26
  "message": message
27
  })
 
1
  import os
2
  import json
3
+ from datetime import datetime
4
  import time
5
 
6
  class MemoryManager:
 
12
  def initialize_user_profile(self, user_id: str):
13
  if user_id not in self.memory:
14
  self.memory[user_id] = {
15
+ "created_at": datetime.now().isoformat(),
16
  "conversations": []
17
  }
18
  self._save_memory()
 
21
  def store_conversation(self, user_id: str, message: str, role: str):
22
  if user_id in self.memory:
23
  self.memory[user_id]["conversations"].append({
24
+ "timestamp": datetime.now().isoformat(),
25
  "role": role,
26
  "message": message
27
  })