Upload stats.py

stats.py

@@ -0,0 +1,31 @@
+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()