Update memory_core.py

memory_core.py

@@ -1,26 +1,28 @@
-import json
-import os
+import json, os
 
-class MemoryCore:
-    def __init__(self, file_path="memory_store.json"):
-        self.file_path = file_path
-        self._init_store()
+class MemoryManager:
+    def __init__(self, agent_id, file_path=None):
+        self.agent_id = agent_id
+        self.file_path = file_path or f"memory_{agent_id}.json"
+        self._init_memory()
 
-    def _init_store(self):
+    def _init_memory(self):
+        # If no memory file exists, create it
         if not os.path.exists(self.file_path):
             with open(self.file_path, "w") as f:
-                json.dump({}, f)
+                json.dump({"history": []}, f)
 
-    def learn(self, agent_id, key, value):
+    def save(self, data):
+        # Append a new memory entry
         with open(self.file_path, "r") as f:
-            data = json.load(f)
-        if agent_id not in data:
-            data[agent_id] = []
-        data[agent_id].append({"key": key, "value": value})
+            memory = json.load(f)
+        memory["history"].append(data)
         with open(self.file_path, "w") as f:
-            json.dump(data, f, indent=2)
+            json.dump(memory, f, indent=2)
+        os.sync()  # ✅ ensures Hugging Face writes to disk immediately
 
-    def recall_context(self, agent_id):
+    def recall(self):
+        # Load all past memory for this agent
         with open(self.file_path, "r") as f:
-            data = json.load(f)
-        return data.get(agent_id, [])
+            memory = json.load(f)
+        return memory.get("history", [])