api/ask.py

from fastapi import APIRouter
from pydantic import BaseModel
from services.rag_pipeline import rag_pipeline

router = APIRouter()

# 요청 스키마
class AskRequest(BaseModel):
    query: str
    top_k: int = 5

# 응답 스키마
class AskResponse(BaseModel):
    output: str

@router.post("/ask", response_model=AskResponse)
async def ask(request: AskRequest):
    output = rag_pipeline(request.query, top_k=request.top_k)
    return {"output": output}