Circuit-Forge/avocet
3
0
Fork 0
Code Issues 10 Pull requests 2 Projects Releases 1 Packages Wiki Activity Actions

fix: use Literal type for SubmitRequest.action field

Browse source
This commit is contained in:
pyr0ball 2026-04-08 14:33:38 -07:00
parent f19cab60f7
commit 4ad2907ae8
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
app/sft.py
View file

@ -11,6 +11,7 @@ from __future__ import annotations
import logging import logging
from pathlib import Path from pathlib import Path
from typing import Literal
import yaml import yaml
from fastapi import APIRouter, HTTPException from fastapi import APIRouter, HTTPException
@ -140,15 +141,13 @@ def get_queue(page: int = 1, per_page: int = 20):
class SubmitRequest(BaseModel): class SubmitRequest(BaseModel):
id: str id: str
action: str # "correct" | "discard" | "flag" action: Literal["correct", "discard", "flag"]
corrected_response: str | None = None corrected_response: str | None = None
@router.post("/submit") @router.post("/submit")
def post_submit(req: SubmitRequest): def post_submit(req: SubmitRequest):
"""Record a reviewer decision for one SFT candidate.""" """Record a reviewer decision for one SFT candidate."""
if req.action not in ("correct", "discard", "flag"):
raise HTTPException(422, f"Unknown action {req.action!r}")
if req.action == "correct": if req.action == "correct":
if not req.corrected_response or not req.corrected_response.strip(): if not req.corrected_response or not req.corrected_response.strip():
raise HTTPException(422, "corrected_response must be non-empty when action is 'correct'") raise HTTPException(422, "corrected_response must be non-empty when action is 'correct'")