File size: 402 Bytes
c01955c
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
from pydantic import BaseModel
from typing import List, Annotated, Optional
from langgraph.graph.message import add_messages
from langchain_core.messages import BaseMessage

class ChatState(BaseModel):
    messages: Annotated[List[BaseMessage], add_messages]
    topic: Optional[str] = None
    time_remaining: int  # in seconds
    questions_generated: bool = False
    questions: Optional[str] = None