from langchain_core.pydantic_v1 import BaseModel, Field
from typing import List

class RequirementGathering(BaseModel):
    """Requirement Gathering Pydantic Schema"""

    overview: str = Field(description="Provide short overview of requirement")
    description: str = Field(description="Provide requirement description. Ensure each requirement is explicitly stated as per the document.")
    benefits: str = Field(description="The benefit or impact on the user or system as a result of the requirement. Ground it in the context of the document. Consider only for banking applications as product domain.")
    reason: str = Field(description="The reason for implementing the requirement.")
    priority: str = Field(description="The priority of the requirement. Return only one from list of values [High, Medium and Low]. Ground result based on the document.")
    tags: List[str] = Field(description="List of tags for the requirement to identify into relevant categories(e.g., Product Specifications, Validations and Quality Standards)")


class RequirementGatheringDetails(BaseModel):
    """Requirement Gathering Details Pydantic Schema"""

    header: str = Field(description="Provide the header of the document.")
    requirements: List[RequirementGathering] = Field(description="List of requirements gathered from the document.")