"""
Created By: ishwor subedi
Date: 2024-08-29
"""
from pydantic import BaseModel, Field


class FollowUps(BaseModel):
    q1: str = Field(description="First Follow-up Question")
    q2: str = Field(description="Second Follow-up Question")
    q3: str = Field(description="Third Follow-up Question")


## langgraph models
class TopicSelectionParser(BaseModel):
    Topic: str = Field(description='Selected Topic')
    Reasoning: str = Field(description='Reasoning behind topic selection')