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[list]=[]
    retrievals : Optional[list]=[]
    business_details : Optional[dict]={}
    latest_preferred_topics: Optional[list] = []
    images: Optional[list[str]] = []  # Base64-encoded strings of images
    image_captions: Optional[list] = []
    model_config = ConfigDict(arbitrary_types_allowed=True)