mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-02 17:00:28 +08:00
Andreas Köpf
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
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
import datetime
|
|
from uuid import UUID
|
|
|
|
from fastapi import APIRouter, Depends, Query
|
|
from oasst_backend.api import deps
|
|
from oasst_backend.api.v1 import utils
|
|
from oasst_backend.models import ApiClient
|
|
from oasst_backend.prompt_repository import PromptRepository
|
|
from oasst_shared.schemas import protocol
|
|
from sqlmodel import Session
|
|
from starlette.status import HTTP_204_NO_CONTENT
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/{username}/messages", response_model=list[protocol.Message])
|
|
def query_frontend_user_messages(
|
|
username: str,
|
|
api_client_id: UUID = None,
|
|
max_count: int = Query(10, gt=0, le=1000),
|
|
start_date: datetime.datetime = None,
|
|
end_date: datetime.datetime = None,
|
|
only_roots: bool = False,
|
|
desc: bool = True,
|
|
include_deleted: bool = False,
|
|
api_client: ApiClient = Depends(deps.get_api_client),
|
|
db: Session = Depends(deps.get_db),
|
|
):
|
|
"""
|
|
Query frontend user messages.
|
|
"""
|
|
pr = PromptRepository(db, api_client)
|
|
messages = pr.query_messages(
|
|
username=username,
|
|
api_client_id=api_client_id,
|
|
desc=desc,
|
|
limit=max_count,
|
|
start_date=start_date,
|
|
end_date=end_date,
|
|
only_roots=only_roots,
|
|
deleted=None if include_deleted else False,
|
|
)
|
|
return utils.prepare_message_list(messages)
|
|
|
|
|
|
@router.delete("/{username}/messages", status_code=HTTP_204_NO_CONTENT)
|
|
def mark_frontend_user_messages_deleted(
|
|
username: str, api_client: ApiClient = Depends(deps.get_trusted_api_client), db: Session = Depends(deps.get_db)
|
|
):
|
|
pr = PromptRepository(db, api_client)
|
|
messages = pr.query_messages(username=username, api_client_id=api_client.id)
|
|
pr.mark_messages_deleted(messages)
|