Spaces:
Sleeping
Sleeping
Update memory_manager.py
Browse files- memory_manager.py +3 -3
memory_manager.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
import os
|
| 2 |
import json
|
| 3 |
-
from datetime import datetime
|
| 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.
|
| 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.
|
| 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 |
})
|