From bbc4bd91159502ff495f1790c992785043a8c98c Mon Sep 17 00:00:00 2001
From: Tanishq Dubey <dubey@dws.rip>
Date: Sun, 6 Oct 2024 21:32:16 -0400
Subject: [PATCH] sql gen

---
 main.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index 1328b5d..ef9c3ac 100644
--- a/main.py
+++ b/main.py
@@ -93,15 +93,22 @@ def init_db():
         db = get_db()
         db.execute(
             """
-            CREATE TABLE IF NOT EXISTS Queries (
-                id TEXT PRIMARY KEY,
+            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 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()