Added more memory management and fixed system prompt bug

1 files changed, 34 insertions, 6 deletions

diff --git a/personal_agent/memory/store.py b/personal_agent/memory/store.py
index e5fca96..164d11f 100644
--- a/personal_agent/memory/store.py
+++ b/personal_agent/memory/store.py
@@ -21,6 +21,16 @@ class LongTermMemoryStore:
                 (str(memory.id), memory.content, memory.category.value, created_at),
             )
 
+            connection.execute(
+                """
+                INSERT INTO memories_fts(rowid, content)
+                SELECT rowid, content
+                FROM memories
+                WHERE id = ?
+                """,
+                (str(memory.id),),
+            )
+
     def get(self, memory_id: UUID) -> LongTermMemory | None:
         with self.database.connect() as connection:
             row = connection.execute(
@@ -52,6 +62,18 @@ class LongTermMemoryStore:
     def delete(self, memory_id: UUID) -> None:
         with self.database.connect() as connection:
             connection.execute(
+                """
+                DELETE FROM memories_fts
+                WHERE rowid = (
+                    SELECT rowid
+                    FROM memories
+                    WHERE id = ?
+                )
+                """,
+                (str(memory_id),),
+            )
+
+            connection.execute(
                 "DELETE FROM memories WHERE id = ?",
                 (str(memory_id),),
             )
@@ -68,18 +90,24 @@ class LongTermMemoryStore:
         if not query.strip():
             raise ValueError("Search query must not be empty")
 
+        terms = query.split()
+        fts_query = " OR ".join(
+            f'"{term.replace(chr(34), chr(34) * 2)}"' for term in terms
+        )
+
         sql = """
-            SELECT id, content, category, created_at
-            FROM memories
-            WHERE content LIKE ?
+            SELECT m.id, m.content, m.category, m.created_at
+            FROM memories_fts as f
+            JOIN memories AS m ON m.rowid = f.rowid
+            WHERE memories_fts MATCH ?
         """
-        parameters: list[str | int] = [f"%{query}%"]
+        parameters: list[str | int] = [fts_query]
 
         if category is not None:
-            sql += " AND category = ?"
+            sql += " AND m.category = ?"
             parameters.append(category.value)
 
-        sql += " ORDER BY id LIMIT ?"
+        sql += " ORDER BY bm25(memories_fts) LIMIT ?"
         parameters.append(limit)
 
         with self.database.connect() as connection: