Mireya Cueto Garrido
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
import uuid
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Depends
|
|
|
|
from app.api.dependencies import get_exam_service, get_llm_client
|
|
from app.schemas.exam import (
|
|
BuildPromptRequest,
|
|
GenerateExamRequest,
|
|
ParsedQuestionsResponse,
|
|
ParseRequest,
|
|
PromptResponse,
|
|
)
|
|
from app.services.exam_service import ExamService
|
|
from app.services.llm import LLMClient
|
|
|
|
router = APIRouter(tags=["generation"])
|
|
|
|
|
|
@router.post("/prompts/{template_id}", response_model=PromptResponse)
|
|
def build_prompt(
|
|
template_id: uuid.UUID,
|
|
payload: BuildPromptRequest,
|
|
service: Annotated[ExamService, Depends(get_exam_service)],
|
|
) -> PromptResponse:
|
|
return service.build_prompt(template_id, payload.topic_prompt)
|
|
|
|
|
|
@router.post("/generate", response_model=ParsedQuestionsResponse)
|
|
async def generate_exam(
|
|
payload: GenerateExamRequest,
|
|
service: Annotated[ExamService, Depends(get_exam_service)],
|
|
llm_client: Annotated[LLMClient, Depends(get_llm_client)],
|
|
) -> ParsedQuestionsResponse:
|
|
return await service.generate_with_llm(payload.template_id, payload.topic_prompt, llm_client)
|
|
|
|
|
|
@router.post("/parse", response_model=ParsedQuestionsResponse)
|
|
def parse_ai_output(
|
|
payload: ParseRequest,
|
|
service: Annotated[ExamService, Depends(get_exam_service)],
|
|
) -> ParsedQuestionsResponse:
|
|
return service.parse_and_persist(payload)
|