mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-20 12:10:50 +08:00
Andreas KöpfandGitHub
8906854dbf
* Extract classes UserRepository and TaskRepository from PromptRepository * move close_task() to TaskRepository and get_user_leaderboard to UserRepository() * Use UserRepository in leaderboards endpoint, add type annotation to leaderboards endpoint
27 lines
857 B
Python
27 lines
857 B
Python
from fastapi import APIRouter, Depends
|
|
from oasst_backend.api import deps
|
|
from oasst_backend.models import ApiClient
|
|
from oasst_backend.user_repository import UserRepository
|
|
from oasst_shared.schemas.protocol import LeaderboardStats
|
|
from sqlmodel import Session
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/create/assistant")
|
|
def get_assistant_leaderboard(
|
|
db: Session = Depends(deps.get_db),
|
|
api_client: ApiClient = Depends(deps.get_trusted_api_client),
|
|
) -> LeaderboardStats:
|
|
ur = UserRepository(db, api_client)
|
|
return ur.get_user_leaderboard(role="assistant")
|
|
|
|
|
|
@router.get("/create/prompter")
|
|
def get_prompter_leaderboard(
|
|
db: Session = Depends(deps.get_db),
|
|
api_client: ApiClient = Depends(deps.get_trusted_api_client),
|
|
) -> LeaderboardStats:
|
|
ur = UserRepository(db, api_client)
|
|
return ur.get_user_leaderboard(role="prompter")
|