File size: 763 Bytes
d101df6
 
c9c8879
95dba9d
 
 
c9c8879
 
95dba9d
d101df6
 
c28b1fc
c9c8879
 
 
 
 
c28b1fc
 
c9c8879
 
 
c28b1fc
95dba9d
d101df6
 
c9c8879
 
 
 
 
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
# memory/database.py

import sqlite3
import os

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

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

def log_action(agent: str, action: str, result: str):
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
    cursor.execute(
        "INSERT INTO agent_logs (agent, action, result) VALUES (?, ?, ?)",
        (agent, action, result)
    )
    conn.commit()
    conn.close()