from langchain_core.pydantic_v1 import BaseModel
from langchain_core.pydantic_v1 import Field


"""
This module contains helper pydantic models to validate the inputs
and outputs of LLM calls used for Ragas
"""


class AnswerRelevanceClassification(BaseModel):
    question: str
    noncommittal: int


class ContextPrecisionVerification(BaseModel):
    """Answer for the verification task whether the context was useful."""

    reason: str = Field(..., description="Reason for verification")
    verdict: int = Field(..., description="Binary (0/1) verdict of verification")


class StatementFaithfulnessAnswer(BaseModel):
    statement: str = Field(..., description="the original statement, word-by-word")
    reason: str = Field(..., description="the reason of the verdict")
    verdict: int = Field(..., description="the verdict(0/1) of the faithfulness.")


class StatementFaithfulnessAnswers(BaseModel):
    __root__: list[StatementFaithfulnessAnswer]

    def dicts(self) -> list[dict]:
        return self.dict()["__root__"]


class Statements(BaseModel):
    sentence_index: int = Field(..., description="Index of the sentence from the statement list")
    simpler_statements: list[str] = Field(..., description="the simpler statements")


class StatementsAnswers(BaseModel):
    __root__: list[Statements]

    def dicts(self) -> list[dict]:
        return self.dict()["__root__"]