File size: 488 Bytes
0870bc8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
"""
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')