quivr/streamlit-demo/stats.py

from datetime import datetime, timedelta

# -- Create a table called "stats"
# create table
#   stats (
#     -- A column called "time" with data type "timestamp"
#     time timestamp,
#     -- A column called "details" with data type "text"
#     chat boolean,
#     embedding boolean,
#     details text,
#     metadata jsonb,
#     -- An "integer" primary key column called "id" that is generated always as identity
#     id integer primary key generated always as identity
#   );


def get_usage_today(supabase):
    # Returns the number of rows in the stats table for the last 24 hours
    response = supabase.table("stats").select("id", count="exact").gte("time", datetime.now() - timedelta(hours=24)).execute()
    return response.count

def add_usage(supabase, type, details, metadata):
    # Adds a row to the stats table
    supabase.table("stats").insert({
        "time": datetime.now().isoformat(),
        "chat": type == "chat",
        "embedding": type == "embedding",
        "details": details,
        "metadata": metadata
    }).execute()