import sqlite3
import os

DB_PATH = os.path.join("/mnt/data", "memory.db")

def init_db():
    conn = sqlite3.connect(DB_PATH)
    c = conn.cursor()
    c.execute("""
        CREATE TABLE IF NOT EXISTS memory_logs (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            agent TEXT,
            action TEXT,
            result TEXT,
            timestamp TEXT
        )
    """)
    conn.commit()
    conn.close()

def log_memory(agent: str, action: str, result: str):
    conn = sqlite3.connect(DB_PATH)
    c = conn.cursor()
    c.execute(
        "INSERT INTO memory_logs (agent, action, result, timestamp) VALUES (?,?,?,datetime('now'))",
        (agent, action, result)
    )
    conn.commit()
    conn.close()