structures.py

from typing import TypedDict, List, Annotated
from pydantic import BaseModel, Field

class FactCheckInfo(BaseModel):
    """Details of individual fact-checking results"""
    matching_text: str = Field(description="Exact text excerpt from the document that supports the claim.")
    document_name: str = Field(description="Name of the document where the matching text was found.")

class FactCheck(BaseModel): # TODO add validation
    """Fact check details"""
    reasoning: str = Field(description="Explanation of the reasoning behind the fact-checking result.")
    details: List[FactCheckInfo] = Field(description="List of detailed fact-checking results, including matching texts and document names.")

class State(TypedDict):
    claim: str
    context: str
    answer: str