from llm_server.database.conn import db_pool


def create_db():
    conn = db_pool.connection()
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS prompts (
        ip TEXT,
        token TEXT DEFAULT NULL,
        model TEXT,
        backend_mode TEXT,
        backend_url TEXT,
        request_url TEXT,
        generation_time FLOAT,
        prompt LONGTEXT,
        prompt_tokens INTEGER,
        response LONGTEXT,
        response_tokens INTEGER,
        response_status INTEGER,
        parameters TEXT,
        # CHECK (parameters IS NULL OR JSON_VALID(parameters)),
        headers TEXT,
        # CHECK (headers IS NULL OR JSON_VALID(headers)),
        timestamp INTEGER
    )
    ''')
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS token_auth (
        token TEXT,
        UNIQUE (token),
        type TEXT NOT NULL,
        priority INTEGER DEFAULT 9999,
        simultaneous_ip INTEGER DEFAULT NULL,
        uses INTEGER DEFAULT 0,
        max_uses INTEGER,
        expire INTEGER,
        disabled BOOLEAN DEFAULT 0
    )
    ''')
    conn.commit()
    cursor.close()