endpoints/groundedness.py

# endpoints/groundedness.py
from fastapi import APIRouter
from pydantic import BaseModel
from model_inference.groundedness_checker import GroundednessChecker
from endpoints.api_models import LLMResponse

router = APIRouter(prefix="/groundedness", tags=["Groundedness"])


checker = GroundednessChecker(model_path="./grounding_detector")

@router.post("/check")
async def check_groundedness(req: LLMResponse):
    result = checker.check(req.question, req.answer, req.context)
    return result