File size: 1,143 Bytes
4ea6cf9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import sys
from pathlib import Path
sys.path.append(str(Path(__file__).resolve().parent.parent))
import sqlite3
from datetime import datetime
# Connect to SQLite database (will create if it doesn't exist)
conn = sqlite3.connect("memory.db")
cursor = conn.cursor()
# Create memory table if it doesn't exist
cursor.execute("""
CREATE TABLE IF NOT EXISTS memory (
id INTEGER PRIMARY KEY AUTOINCREMENT,
prompt TEXT NOT NULL,
response TEXT NOT NULL,
timestamp TEXT DEFAULT CURRENT_TIMESTAMP
)
""")
conn.commit()
def save_memory(prompt: str, response: str):
"""Save a prompt-response pair to the memory database."""
cursor.execute(
"INSERT INTO memory (prompt, response) VALUES (?, ?)",
(prompt, response)
)
conn.commit()
def recall_memories(limit: int = 5):
"""Retrieve the most recent prompt-response pairs."""
cursor.execute(
"SELECT prompt, response, timestamp FROM memory ORDER BY timestamp DESC LIMIT ?",
(limit,)
)
return cursor.fetchall()
def clear_memory():
"""Delete all memory records."""
cursor.execute("DELETE FROM memory")
conn.commit()
|