This commit is contained in:
Tanishq Dubey 2024-10-06 21:32:16 -04:00
parent 5111ae6a7b
commit bbc4bd9115

11
main.py
View File

@ -93,15 +93,22 @@ def init_db():
db = get_db()
db.execute(
"""
CREATE TABLE IF NOT EXISTS Keys (
id INTEGER PRIMARY KEY AUTOINCREMENT,
username TEXT NOT NULL UNIQUE,
api_key TEXT NOT NULL UNIQUE
);
CREATE TABLE IF NOT EXISTS Queries (
id TEXT PRIMARY KEY,
id INTEGER PRIMARY KEY AUTOINCREMENT,
request_id TEXT NOT NULL UNIQUE,
ip TEXT NOT NULL,
timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
query TEXT NOT NULL,
api_key_id INTEGER,
conversation_history TEXT,
FOREIGN KEY (api_key_id) REFERENCES Keys (id)
)
);
"""
)
db.commit()