from fastapi import APIRouter

from axon.models import CitationValidationRequest, CitationValidationResponse
from axon.services.citations import validate_citations

router = APIRouter()


@router.post("/validate/citations", response_model=CitationValidationResponse)
async def validate_citation_report(
    request: CitationValidationRequest,
) -> CitationValidationResponse:
    count, missing, valid = validate_citations(request.report, request.sources)
    return CitationValidationResponse(
        citation_count=count,
        missing_citations=missing,
        valid=valid,
    )