File size: 605 Bytes
9b14ff1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from fastapi import APIRouter
from pydantic import BaseModel
from generator.prompt_builder import build_prompt
from generator.llm_inference import generate_answer
router = APIRouter()
# 요청 스키마
class GenerateRequest(BaseModel):
query: str
context_docs: list
# 응답 스키마
class GenerateResponse(BaseModel):
output: str
@router.post("/generate", response_model=GenerateResponse)
async def generate(request: GenerateRequest):
prompt = build_prompt(request.query, request.context_docs)
output = generate_answer(prompt)
return {"output": output}
|