Update memory.py

memory.py

@@ -1,19 +1,15 @@
-from sentence_transformers import SentenceTransformer
-import numpy as np
-import faiss
-
-model = SentenceTransformer("all-MiniLM-L6-v2")
+# memory.py
 
 def init_memory():
-    dim = 384
-    return {"index": faiss.IndexFlatL2(dim), "texts": []}
+    # Initialize memory as a simple dict with a 'texts' list
+    return {"texts": []}
 
 def add_to_memory(text, memory):
-    vec = model.encode([text])[0]
-    memory["index"].add(np.array([vec]))
+    # Append text to memory, keep last 100 entries
     memory["texts"].append(text)
+    if len(memory["texts"]) > 100:
+        memory["texts"] = memory["texts"][-100:]
 
-def search_memory(query, memory, k=5):
-    vec = model.encode([query])[0]
-    D, I = memory["index"].search(np.array([vec]), k)
-    return [memory["texts"][i] for i in I[0] if i < len(memory["texts"])]
+def search_memory(query, memory):
+    # Placeholder: Return last few memory entries related to query (simple all for now)
+    return memory["texts"][-5:]