from pydantic import BaseModel, Field
from typing import List


class QAPair(BaseModel):
    question: str = Field(description="A question derived from the input")
    answer: str = Field(description="Answer to the question")


# Use a list of 3 QAPair objects as expected output
class QAList(BaseModel):
    qa_pairs: List[QAPair]


class FAQIndexRequest(BaseModel):
    agent_id: str