schemas.py

from pydantic import BaseModel, HttpUrl, Field
from typing import Annotated

class OutputFormat(BaseModel):
    answer: Annotated[str, Field(description="The answer to the user's question.")]
    sources: Annotated[list[HttpUrl], Field(description="The non-redundant links of source videos used to generate the answer.")]

class InputModel(BaseModel):
    question: str
    chat_history: list[dict]