mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-18 12:10:34 +08:00
Oliver StanleyandGitHub
69bc799cd9
* Implement label task for initial prompts and replies * Resolve formatting * Include missing argument * Modify text_labels API to match new model, update DB schema accordingly * Send valid labels as part of label tasks * Send correctly formatted valid_labels list * Fix request format * Fix request details for text-frontend reply label task * Include message_id in tasks * Address review comments * Fix alembic tree
35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from fastapi.security.api_key import APIKey
|
|
from loguru import logger
|
|
from oasst_backend.api import deps
|
|
from oasst_backend.prompt_repository import PromptRepository
|
|
from oasst_shared.schemas import protocol as protocol_schema
|
|
from sqlmodel import Session
|
|
from starlette.status import HTTP_204_NO_CONTENT, HTTP_400_BAD_REQUEST
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/", status_code=HTTP_204_NO_CONTENT)
|
|
def label_text(
|
|
*,
|
|
db: Session = Depends(deps.get_db),
|
|
api_key: APIKey = Depends(deps.get_api_key),
|
|
text_labels: protocol_schema.TextLabels,
|
|
) -> None:
|
|
"""
|
|
Label a piece of text.
|
|
"""
|
|
api_client = deps.api_auth(api_key, db)
|
|
|
|
try:
|
|
logger.info(f"Labeling text {text_labels=}.")
|
|
pr = PromptRepository(db, api_client, user=text_labels.user)
|
|
pr.store_text_labels(text_labels)
|
|
|
|
except Exception:
|
|
logger.exception("Failed to store label.")
|
|
raise HTTPException(
|
|
status_code=HTTP_400_BAD_REQUEST,
|
|
)
|