mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-04 17:20:19 +08:00
James Melvin Ebenezer
063157355c
* fix: update user streaks * Moved streak_last_day_date & streak_days from UserStats to User * Updated Alembic version * update last_activity after tm.handle_interaction() * periodically executed function in main.py to update the user_streak every 4hrs * fix:removed log messages * fix: pre commit issues * refactor: incorporated review comments * fix: removed the managed -scope. * refactor: managed_tx to Outer REST Db Ops in /interaction API * fix: pre-commit changes * fix: added streak and last_activity_date to protocol.FrontEndUser * fix: pre-commit fixes after merge * fix: added streak info to user_stat_repository, leaderboard and review comments * fix: proper 4h delay and simpler startup_time initialisation --------- Co-authored-by: James Melvin <melvin@gameface.ai>
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
from datetime import datetime
|
|
from enum import Enum
|
|
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
import sqlalchemy as sa
|
|
import sqlalchemy.dialects.postgresql as pg
|
|
from sqlmodel import Field, Index, SQLModel
|
|
|
|
|
|
class UserStatsTimeFrame(str, Enum):
|
|
day = "day"
|
|
week = "week"
|
|
month = "month"
|
|
total = "total"
|
|
|
|
|
|
class UserStats(SQLModel, table=True):
|
|
__tablename__ = "user_stats"
|
|
__table_args__ = (
|
|
Index("ix_user_stats__timeframe__user_id", "time_frame", "user_id", unique=True),
|
|
Index("ix_user_stats__timeframe__rank__user_id", "time_frame", "rank", "user_id", unique=True),
|
|
)
|
|
|
|
time_frame: Optional[str] = Field(nullable=False, primary_key=True)
|
|
user_id: Optional[UUID] = Field(
|
|
sa_column=sa.Column(pg.UUID(as_uuid=True), sa.ForeignKey("user.id"), primary_key=True)
|
|
)
|
|
base_date: Optional[datetime] = Field(sa_column=sa.Column(sa.DateTime(timezone=True), nullable=True))
|
|
|
|
leader_score: int = 0
|
|
modified_date: Optional[datetime] = Field(
|
|
sa_column=sa.Column(sa.DateTime(timezone=True), nullable=False, server_default=sa.func.current_timestamp())
|
|
)
|
|
|
|
rank: int = Field(nullable=True)
|
|
|
|
prompts: int = 0
|
|
replies_assistant: int = 0
|
|
replies_prompter: int = 0
|
|
labels_simple: int = 0
|
|
labels_full: int = 0
|
|
rankings_total: int = 0
|
|
rankings_good: int = 0
|
|
|
|
accepted_prompts: int = 0
|
|
accepted_replies_assistant: int = 0
|
|
accepted_replies_prompter: int = 0
|
|
|
|
reply_ranked_1: int = 0
|
|
reply_ranked_2: int = 0
|
|
reply_ranked_3: int = 0
|
|
|
|
def compute_leader_score(self) -> int:
|
|
return (
|
|
self.prompts
|
|
+ self.replies_assistant * 4
|
|
+ self.replies_prompter
|
|
+ self.labels_simple
|
|
+ self.labels_full * 2
|
|
+ self.rankings_total
|
|
+ self.rankings_good
|
|
+ self.accepted_prompts
|
|
+ self.accepted_replies_assistant * 4
|
|
+ self.accepted_replies_prompter
|
|
+ self.reply_ranked_1 * 9
|
|
+ self.reply_ranked_2 * 3
|
|
+ self.reply_ranked_3
|
|
)
|