mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-24 12:50:50 +08:00
yeungadrianandGitHub
4715a53637
* Leaderboard endpoints proof of concept - New leaderboards router - Two endpoints /api/v1/leaderboards/assistant & /api/v1/leaderboards/prompter - Function in prompt_repository to calculate user scores * Separating create vs 'eventual' evaluate routes * Adding display_name to result * Missing / in routes * Updating route to be experimental * Fixing format for pre-commit
26 lines
778 B
Python
26 lines
778 B
Python
from fastapi import APIRouter, Depends
|
|
from oasst_backend.api import deps
|
|
from oasst_backend.models import ApiClient
|
|
from oasst_backend.prompt_repository import PromptRepository
|
|
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),
|
|
):
|
|
pr = PromptRepository(db, api_client, None)
|
|
return pr.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),
|
|
):
|
|
pr = PromptRepository(db, api_client, None)
|
|
return pr.get_user_leaderboard(role="prompter")
|