from fastapi import APIRouter, HTTPException

from backend.database.schemas import QueryRequest
from backend.services.query_service import query_service

router = APIRouter(prefix="/repos", tags=["query"])


@router.post("/{repo_id}/query")
def query_repo(repo_id: str, request: QueryRequest):
    try:
        return query_service.answer(repo_id, request.question, request.top_k)
    except Exception as exc:
        raise HTTPException(status_code=400, detail=f"Could not answer query: {exc}") from exc