Create memory.py

memory.py

@@ -0,0 +1,21 @@
+import json
+import os
+
+class MemoryManager:
+    def __init__(self, agent_id):
+        self.agent_id = agent_id
+        self.file_path = f"memory_{agent_id}.json"
+        self._init_memory()
+
+    def _init_memory(self):
+        if not os.path.exists(self.file_path):
+            with open(self.file_path, "w") as f:
+                json.dump({"history": []}, f)
+
+    def save(self, data):
+        with open(self.file_path, "r") as f:
+            memory = json.load(f)
+        memory["history"].append(data)
+        with open(self.file_path, "w") as f:
+            json.dump(memory, f, indent=2)
+        os.sync()  # Ensures the file actually saves on Hugging Face