Update modules/pmbl.py

modules/pmbl.py

@@ -31,22 +31,22 @@ class PMBL:
         c = conn.cursor()
 
         if mode == "full":
-            c.execute("SELECT timestamp, prompt, response FROM chats ORDER BY id")
+            c.execute("SELECT prompt, response FROM chats ORDER BY id")
             history = []
             for row in c.fetchall():
-                history.append({"role": "user", "content": row[1]})
-                history.append({"role": "PMB", "content": f"[{row[0]}] {row[2]}"})
+                history.append({"role": "user", "content": row[0]})
+                history.append({"role": "PMB", "content": row[1]})
         else:
             c.execute("SELECT id, prompt, response FROM chats WHERE topic != 'Untitled'")
             chats = c.fetchall()
             relevant_chat_id = self.find_relevant_chat(chats, user_message)
 
             if relevant_chat_id:
-                c.execute("SELECT timestamp, prompt, response FROM chats WHERE id = ?", (relevant_chat_id,))
+                c.execute("SELECT prompt, response FROM chats WHERE id = ?", (relevant_chat_id,))
                 row = c.fetchone()
                 history = [
-                    {"role": "user", "content": row[1]},
-                    {"role": "PMB", "content": f"[{row[0]}] {row[2]}"}
+                    {"role": "user", "content": row[0]},
+                    {"role": "PMB", "content": row[1]}
                 ]
             else:
                 history = []