my_agent/utils/state.py

from pydantic import BaseModel, ConfigDict
from typing import Optional
import pandas as pd

class State(BaseModel):
    carry_on: Optional[bool]=False
    messages: Optional[str] = None
    topic: list
    brainstroming_topics: Optional[list] = []
    preferred_topics: Optional[list] = []
    stories : Optional[list]=[]
    final_story: Optional[str]=None
    retrievals : Optional[list]=[]
    model_config = ConfigDict(arbitrary_types_allowed=True)