schema.py

from pydantic import BaseModel


class Question(BaseModel):
    transcript: str
    competency_id: int


class Competency(BaseModel):
    descriptions: list[str]
    competency_id: int


class PredictData(BaseModel):
    questions: list[Question]
    competency_sets: list[Competency]