Update memory.py

memory.py

@@ -12,6 +12,7 @@ CREATE TABLE IF NOT EXISTS conversations (
 """)
 conn.commit()
 
+
 def save_message(user_id, role, content):
     cursor.execute(
         "INSERT INTO conversations VALUES (?, ?, ?)",
@@ -19,6 +20,7 @@ def save_message(user_id, role, content):
     )
     conn.commit()
 
+
 def load_memory(user_id, limit=6):
     cursor.execute("""
     SELECT role, content FROM conversations
@@ -28,5 +30,11 @@ def load_memory(user_id, limit=6):
     """, (user_id, limit))
 
     rows = cursor.fetchall()
-    return list(reversed(rows))
-    
+
+    # Convert into LLM message format
+    formatted = [
+        {"role": role, "content": content}
+        for role, content in reversed(rows)
+    ]
+
+    return formatted