Update memory.py

memory.py

@@ -6,22 +6,25 @@ MEMORY_FILE = "memory.json"
 def init_memory():
     if not os.path.exists(MEMORY_FILE):
         with open(MEMORY_FILE, "w") as f:
-            json.dump({"history": []}, f)
+            json.dump({}, f)
     with open(MEMORY_FILE, "r") as f:
         return json.load(f)
 
-def save_memory(memory):
+def save_memory(data):
     with open(MEMORY_FILE, "w") as f:
-        json.dump(memory, f, indent=2)
+        json.dump(data, f, indent=2)
 
-def add_to_memory(prompt, response):
+def add_to_memory(session_id, prompt, response):
     memory = init_memory()
-    memory["history"].append({"prompt": prompt, "response": response})
-    memory["history"] = memory["history"][-50:]  # limit to last 50
+    if session_id not in memory:
+        memory[session_id] = []
+    memory[session_id].append({"prompt": prompt, "response": response})
+    memory[session_id] = memory[session_id][-50:]
     save_memory(memory)
 
-def get_memory():
-    return init_memory()["history"]
+def get_memory(session_id):
+    memory = init_memory()
+    return memory.get(session_id, [])
 
 def search_memory(query, memory):
     # Placeholder: Return last few memory entries related to query (simple all for now)