Match output format with APIs (#1595)

mem0/memory/storage.py

@@ -6,7 +6,45 @@ from datetime import datetime
 class SQLiteManager:
     def __init__(self, db_path=":memory:"):
         self.connection = sqlite3.connect(db_path, check_same_thread=False)
+        self._migrate_history_table()
         self._create_history_table()
+    
+    def _migrate_history_table(self):
+        with self.connection:
+            cursor = self.connection.cursor()
+
+            cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='history'")
+            table_exists = cursor.fetchone() is not None
+
+            if table_exists:
+                # Rename the old table
+                cursor.execute("ALTER TABLE history RENAME TO old_history")
+
+                cursor.execute("""
+                    CREATE TABLE IF NOT EXISTS history (
+                        id TEXT PRIMARY KEY,
+                        memory_id TEXT,
+                        old_memory TEXT,
+                        new_memory TEXT,
+                        new_value TEXT,
+                        event TEXT,
+                        created_at DATETIME,
+                        updated_at DATETIME,
+                        is_deleted INTEGER
+                    )
+                """)
+
+                # Copy data from the old table to the new table
+                cursor.execute("""
+                    INSERT INTO history (id, memory_id, old_memory, new_memory, new_value, event, created_at, updated_at, is_deleted)
+                    SELECT id, memory_id, prev_value, new_value, new_value, event, timestamp, timestamp, is_deleted
+                    FROM old_history
+                """)
+
+                cursor.execute("DROP TABLE old_history")
+
+                self.connection.commit()
+
 
     def _create_history_table(self):
         with self.connection:
@@ -15,29 +53,32 @@ class SQLiteManager:
                 CREATE TABLE IF NOT EXISTS history (
                     id TEXT PRIMARY KEY,
                     memory_id TEXT,
-                    prev_value TEXT,
+                    old_memory TEXT,
+                    new_memory TEXT,
                     new_value TEXT,
                     event TEXT,
-                    timestamp DATETIME,
+                    created_at DATETIME,
+                    updated_at DATETIME,
                     is_deleted INTEGER
                 )
             """
             )
 
-    def add_history(self, memory_id, prev_value, new_value, event, is_deleted=0):
+    def add_history(self, memory_id, old_memory, new_memory, event, created_at = None, updated_at = None, is_deleted=0):
         with self.connection:
             self.connection.execute(
                 """
-                INSERT INTO history (id, memory_id, prev_value, new_value, event, timestamp, is_deleted)
-                VALUES (?, ?, ?, ?, ?, ?, ?)
+                INSERT INTO history (id, memory_id, old_memory, new_memory, event, created_at, updated_at, is_deleted)
+                VALUES (?, ?, ?, ?, ?, ?, ?, ?)
             """,
                 (
                     str(uuid.uuid4()),
                     memory_id,
-                    prev_value,
-                    new_value,
+                    old_memory,
+                    new_memory,
                     event,
-                    datetime.utcnow(),
+                    created_at,
+                    updated_at,
                     is_deleted,
                 ),
             )
@@ -45,10 +86,10 @@ class SQLiteManager:
     def get_history(self, memory_id):
         cursor = self.connection.execute(
             """
-            SELECT id, memory_id, prev_value, new_value, event, timestamp, is_deleted
+            SELECT id, memory_id, old_memory, new_memory, event, created_at, updated_at
             FROM history
             WHERE memory_id = ?
-            ORDER BY timestamp ASC
+            ORDER BY updated_at ASC
         """,
             (memory_id,),
         )
@@ -57,11 +98,11 @@ class SQLiteManager:
             {
                 "id": row[0],
                 "memory_id": row[1],
-                "prev_value": row[2],
-                "new_value": row[3],
+                "old_memory": row[2],
+                "new_memory": row[3],
                 "event": row[4],
-                "timestamp": row[5],
-                "is_deleted": row[6],
+                "created_at": row[5],
+                "updated_at": row[6],
             }
             for row in rows
         ]