basic credit usage tracking for workspaces

daras_ai_v2/icons.py

@@ -36,6 +36,7 @@
 chevron_right = '<i class="fa-sharp-duotone fa-solid fa-sm fa-chevron-right"></i>'
 check = '<i class="fa-solid fa-check"></i>'
 octopus = '<i class="fa-regular fa-octopus"></i>'
+analytics = '<i class="fa-regular fa-chart-pie-simple-circle-dollar"></i>'
 
 # brands
 github = '<i class="fa-brands fa-github"></i>'

routers/account.py

@@ -110,6 +110,12 @@ def account_route(request: Request):
     )
 
 
+@gui.route(app, "/account/analytics")
+def analytics_route(request: Request):
+    with account_page_wrapper(request, AccountTabs.analytics) as current_workspace:
+        analytics_tab(request)
+
+
 @gui.route(app, "/account/profile/")
 def profile_route(request: Request):
     is_switching_workspace = gui.session_state.get(SWITCH_WORKSPACE_KEY)
@@ -238,6 +244,7 @@ class AccountTabs(TabData, Enum):
     saved = TabData(title=f"{icons.save} Saved", route=saved_route)
     api_keys = TabData(title=f"{icons.api} API Keys", route=api_keys_route)
     billing = TabData(title=f"{icons.billing} Billing", route=account_route)
+    analytics = TabData(title=f"{icons.analytics} Analytics", route=analytics_route)
 
     @property
     def url_path(self) -> str:
@@ -252,10 +259,12 @@ def get_tabs_for_user(
 
         if workspace.is_personal:
             ret.remove(cls.members)
+            ret.remove(cls.analytics)
         else:
             ret.remove(cls.profile)
             if not workspace.memberships.get(user=user).can_edit_workspace():
                 ret.remove(cls.billing)
+                ret.remove(cls.analytics)
 
         return ret
 
@@ -356,6 +365,61 @@ def _render_run(pr: PublishedRun):
     paginate_button(url=request.url, cursor=cursor)
 
 
+def analytics_tab(request: Request):
+    workspace = get_current_workspace(request.user, request.session)
+    gui.write("# Usage & Limits")
+    gui.caption(
+        f"Member, API & Integration usage for **{workspace.display_name(request.user)}**."
+    )
+
+    with gui.div(className="table-responsive"), gui.tag("table", className="table"):
+        with gui.tag("thead"), gui.tag("tr"):
+            with gui.tag("th", scope="col"):
+                gui.html("Name")
+            with gui.tag("th", scope="col"):
+                gui.html("Type")
+            with gui.tag("th", scope="col"):
+                gui.html("Runs")
+            with gui.tag("th", scope="col"):
+                gui.html("Credits Used")
+            with gui.tag("th", scope="col"):
+                gui.html("")
+
+        with gui.tag("tbody"):
+            for m in workspace.memberships.all():
+                with gui.tag("tr", className="no-margin"):
+                    with gui.tag("td"):
+                        gui.write(m.user.full_name())
+                    with gui.tag("td"):
+                        gui.html("User")
+                    with gui.tag("td"):
+                        gui.html(f"{m.get_run_count()}")
+                    with gui.tag("td"):
+                        gui.html(f"{m.get_credit_usage()} Cr")
+                    with gui.tag("td"):
+                        gui.html("")
+
+            with gui.tag("tr", className="no-margin"):
+                with gui.tag("td"):
+                    gui.write(f"[API Keys]({get_route_path(api_keys_route)})")
+                with gui.tag("td"):
+                    gui.html("API Key")
+                with gui.tag("td"):
+                    gui.html(f"{workspace.get_api_key_run_count()}")
+                with gui.tag("td"):
+                    gui.html(f"{workspace.get_api_key_credit_usage()} Cr")
+
+            with gui.tag("tr", className="no-margin"):
+                with gui.tag("td"):
+                    gui.write("Bot Integrations")
+                with gui.tag("td"):
+                    gui.html("Bot")
+                with gui.tag("td"):
+                    gui.html(f"{workspace.get_bot_run_count()}")
+                with gui.tag("td"):
+                    gui.html(f"{workspace.get_bot_credit_usage()} Cr")
+
+
 def api_keys_tab(request: Request):
     workspace = get_current_workspace(request.user, request.session)
 

workspaces/models.py

@@ -12,6 +12,7 @@
 from django.db import models, transaction, IntegrityError
 from django.db.backends.base.schema import logger
 from django.db.models.aggregates import Sum
+from django.db.models.functions import Abs
 from django.db.models.query_utils import Q
 from django.utils import timezone
 from django.utils.text import slugify
@@ -294,6 +295,35 @@ def add_balance(
                 pass
             raise
 
+    def get_api_key_run_count(self) -> int:
+        return (
+            self.saved_runs.filter(is_api_call=True, price__gt=0)
+            .exclude(messages__isnull=False)
+            .count()
+        )
+
+    def get_api_key_credit_usage(self) -> int:
+        return (
+            self.saved_runs.filter(is_api_call=True)
+            .exclude(messages__isnull=False)  # exclude bot-integration runs
+            .aggregate(total=Sum("price"))
+            .get("total")
+            or 0
+        )
+
+    def get_bot_run_count(self) -> int:
+        return self.saved_runs.filter(
+            is_api_call=True, price__gt=0, messages__isnull=False
+        ).count()
+
+    def get_bot_credit_usage(self) -> int:
+        return (
+            self.saved_runs.filter(is_api_call=True, messages__isnull=False)
+            .aggregate(total=Sum("price"))
+            .get("total")
+            or 0
+        )
+
     def get_or_create_stripe_customer(self) -> stripe.Customer:
         customer = None
 
@@ -509,6 +539,21 @@ def can_invite(self):
             and not self.workspace.is_personal
         )
 
+    def get_run_count(self) -> int:
+        return self.workspace.saved_runs.filter(
+            uid=self.user.uid,
+            price__gt=0,  # proxy for successful runs
+            is_api_call=False,
+        ).count()
+
+    def get_credit_usage(self) -> int:
+        return (
+            self.workspace.saved_runs.filter(
+                uid=self.user.uid, price__gt=0, is_api_call=False
+            ).aggregate(total=Sum("price"))["total"]
+            or 0
+        )
+
 
 class WorkspaceInviteQuerySet(models.QuerySet):
     def create_and_send_invite(