Add user level to user stats (LAION-AI#2083)

The backend part of LAION-AI#1957

Author: AbdBarho

.devcontainer/backend-dev/post_create_command.sh

@@ -1,5 +1,9 @@
 #!/bin/bash
 
+
+# allow git usage
+git config --global --add safe.directory "*"
+
 # ensure pre-commit is installed
 pre-commit install
 

backend/oasst_backend/user_stats_repository.py

@@ -2,6 +2,7 @@
 from typing import Optional
 from uuid import UUID
 
+import numpy as np
 import sqlalchemy as sa
 from loguru import logger
 from oasst_backend.config import settings
@@ -38,6 +39,15 @@
 from sqlmodel import Session, delete, func, text
 
 
+def get_thresholds(baseline: int = 3, alpha: float = 1.1521, max_level: int = 100) -> np.ndarray:
+    level = np.round(np.cumsum(np.arange(1, max_level) * alpha + baseline))
+    return np.array([0] + level.astype(int).tolist())
+
+
+# lookup table, never changes
+THRESHOLDS = get_thresholds()
+
+
 def _create_user_score(r, highlighted_user_id: UUID | None) -> UserScore:
     if r["UserStats"]:
         d = r["UserStats"].dict()
@@ -55,6 +65,7 @@ def _create_user_score(r, highlighted_user_id: UUID | None) -> UserScore:
         d[k] = r[k]
     if highlighted_user_id:
         d["highlighted"] = r["user_id"] == highlighted_user_id
+    d["level"] = (THRESHOLDS <= d["leader_score"]).sum()
     return UserScore(**d)
 
 

oasst-shared/oasst_shared/schemas/protocol.py

@@ -460,6 +460,7 @@ class UserScore(BaseModel):
     display_name: str
 
     leader_score: int = 0
+    level: int = 0  # between 0 and 100
 
     base_date: Optional[datetime]
     modified_date: Optional[datetime]