from typing import List, Optional

from pydantic import BaseModel, Field


class RAGOutput(BaseModel):
    answer: Optional[str] = Field(None, description="Answer to the questions asked")
    citations: Optional[List[str]] = Field(None, description="A list of sources used in generating the answer")
    insufficient_info: Optional[bool] = Field(None, description="An indication if there's insufficient information to answer")