File size: 476 Bytes
9b14ff1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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}
|