| """ | |
| 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') |